mirrors/lemmy
1
0
Fork 0
mirror of https://github.com/LemmyNet/lemmy.git synced 2025-02-28 00:27:10 +00:00
Code Issues Projects Releases Wiki Activity

simplify test context

Browse source
This commit is contained in:
Felix Ableitner 2024-03-06 15:56:49 +01:00
parent cca2a123d0
commit d202838670
2 changed files with 4 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
crates/api_common/src/context.rs
View file

@ -55,32 +55,16 @@ impl LemmyContext {
&self.rate_limit_cell &self.rate_limit_cell
} }
/// Initialize a context for use in tests, optionally blocks network requests. /// Initialize a context for use in tests which blocks federation network calls.
/// ///
/// Do not use this in production code. /// Do not use this in production code.
pub async fn init_test_context() -> Data<LemmyContext> { pub async fn init_test_context() -> Data<LemmyContext> {
Self::build_test_context(true).await
}
/// Initialize a context for use in tests, with network requests allowed.
/// TODO: get rid of this if possible.
///
/// Do not use this in production code.
pub async fn init_test_context_with_networking() -> Data<LemmyContext> {
Self::build_test_context(false).await
}
async fn build_test_context(block_networking: bool) -> Data<LemmyContext> {
// call this to run migrations // call this to run migrations
let pool = build_db_pool_for_tests().await; let pool = build_db_pool_for_tests().await;
let client = client_builder(&SETTINGS).build().expect("build client"); let client = client_builder(&SETTINGS).build().expect("build client");
let mut client = ClientBuilder::new(client); let client = ClientBuilder::new(client).build();
if block_networking {
client = client.with(BlockedMiddleware);
}
let client = client.build();
let secret = Secret { let secret = Secret {
id: 0, id: 0,
jwt_secret: String::new(), jwt_secret: String::new(),
@ -92,6 +76,7 @@ impl LemmyContext {
let config = FederationConfig::builder() let config = FederationConfig::builder()
.domain(context.settings().hostname.clone()) .domain(context.settings().hostname.clone())
.app_data(context) .app_data(context)
// Dont allow any network fetches
.http_fetch_limit(0) .http_fetch_limit(0)
.build() .build()
.await .await
@ -99,18 +84,3 @@ impl LemmyContext {
config.to_request_data() config.to_request_data()
} }
} }
struct BlockedMiddleware;
/// A reqwest middleware which blocks all requests
#[async_trait::async_trait]
impl Middleware for BlockedMiddleware {
async fn handle(
&self,
_req: Request,
_extensions: &mut Extensions,
_next: Next<'_>,
) -> reqwest_middleware::Result<Response> {
Err(anyhow!("Network requests not allowed").into())
}
}

2
crates/api_common/src/request.rs
View file

@ -327,7 +327,7 @@ mod tests {
#[tokio::test] #[tokio::test]
#[serial] #[serial]
async fn test_link_metadata() { async fn test_link_metadata() {
let context = LemmyContext::init_test_context_with_networking().await; let context = LemmyContext::init_test_context().await;
let sample_url = Url::parse("https://gitlab.com/IzzyOnDroid/repo/-/wikis/FAQ").unwrap(); let sample_url = Url::parse("https://gitlab.com/IzzyOnDroid/repo/-/wikis/FAQ").unwrap();
let sample_res = fetch_link_metadata(&sample_url, false, &context) let sample_res = fetch_link_metadata(&sample_url, false, &context)
.await .await