proxy downloads

backend/tests/common/mod.rs

@@ -136,10 +136,17 @@ pub struct TestApp {
 
 impl TestApp {
     pub async fn new() -> Result<Self> {
+        Self::with_config(|_| {}).await
+    }
+
+    pub async fn with_config<F>(configure: F) -> Result<Self>
+    where
+        F: FnOnce(&mut AppConfig),
+    {
         let database_url = env::var("TEST_DATABASE_URL")
             .context("TEST_DATABASE_URL must be set for integration tests")?;
 
-        let config = AppConfig {
+        let mut config = AppConfig {
             database_url: database_url.clone(),
             database_max_pool_size: db::DEFAULT_MAX_POOL_SIZE,
             server_host: "127.0.0.1".to_string(),
@@ -156,6 +163,7 @@ impl TestApp {
             refresh_cookie_secure: false,
             refresh_cookie_domain: None,
             cors_allowed_origin: None,
+            proxy_downloads: false,
             aws_endpoint_url: None,
             aws_access_key_id: None,
             aws_secret_access_key: None,
@@ -170,6 +178,8 @@ impl TestApp {
             webauthn_rp_name: "Papercrate".to_string(),
         };
 
+        configure(&mut config);
+
         let pool = db::init_pool_with_size(&config.database_url, config.database_max_pool_size)?;
         prepare_database(&pool).await?;
 
@@ -206,6 +216,11 @@ impl TestApp {
         Ok(())
     }
 
+    #[allow(dead_code)]
+    pub async fn tenant_id(&self) -> Result<Uuid> {
+        self.ensure_default_tenant().await
+    }
+
     #[allow(dead_code)]
     pub fn storage(&self) -> Arc<FakeStorage> {
         self.storage.clone()