tests/TestAppBuilder: Add with_oidc_keystore() fn

Author: Turbo87

src/tests/util/test_app.rs

@@ -17,12 +17,13 @@ use crates_io_index::{Credentials, RepositoryConfig};
 use crates_io_team_repo::MockTeamRepo;
 use crates_io_test_db::TestDatabase;
 use crates_io_trustpub::github::test_helpers::AUDIENCE;
+use crates_io_trustpub::keystore::{MockOidcKeyStore, OidcKeyStore};
 use crates_io_worker::Runner;
 use diesel_async::AsyncPgConnection;
 use futures_util::TryStreamExt;
 use oauth2::{ClientId, ClientSecret};
 use regex::Regex;
-use std::collections::HashSet;
+use std::collections::{HashMap, HashSet};
 use std::sync::LazyLock;
 use std::{rc::Rc, sync::Arc, time::Duration};
 use tokio::runtime::Handle;
@@ -102,6 +103,7 @@ impl TestApp {
             use_chaos_proxy: false,
             team_repo: MockTeamRepo::new(),
             github: None,
+            oidc_key_stores: Default::default(),
         }
     }
 
@@ -243,6 +245,7 @@ pub struct TestAppBuilder {
     use_chaos_proxy: bool,
     team_repo: MockTeamRepo,
     github: Option<MockGitHubClient>,
+    oidc_key_stores: HashMap<String, Box<dyn OidcKeyStore>>,
 }
 
 impl TestAppBuilder {
@@ -281,7 +284,7 @@ impl TestAppBuilder {
             (primary_proxy, replica_proxy)
         };
 
-        let (app, router) = build_app(self.config, self.github);
+        let (app, router) = build_app(self.config, self.github, self.oidc_key_stores);
 
         let runner = if self.build_job_runner {
             let index = self
@@ -389,6 +392,17 @@ impl TestAppBuilder {
         self
     }
 
+    /// Add a new OIDC keystore to the application
+    pub fn with_oidc_keystore(
+        mut self,
+        issuer_url: impl Into<String>,
+        keystore: MockOidcKeyStore,
+    ) -> Self {
+        self.oidc_key_stores
+            .insert(issuer_url.into(), Box::new(keystore));
+        self
+    }
+
     pub fn with_team_repo(mut self, team_repo: MockTeamRepo) -> Self {
         self.team_repo = team_repo;
         self
@@ -482,7 +496,11 @@ fn simple_config() -> config::Server {
     }
 }
 
-fn build_app(config: config::Server, github: Option<MockGitHubClient>) -> (Arc<App>, axum::Router) {
+fn build_app(
+    config: config::Server,
+    github: Option<MockGitHubClient>,
+    oidc_key_stores: HashMap<String, Box<dyn OidcKeyStore>>,
+) -> (Arc<App>, axum::Router) {
     // Use the in-memory email backend for all tests, allowing tests to analyze the emails sent by
     // the application. This will also prevent cluttering the filesystem.
     let emails = Emails::new_in_memory();
@@ -494,6 +512,7 @@ fn build_app(config: config::Server, github: Option<MockGitHubClient>) -> (Arc<A
         .databases_from_config(&config.db)
         .github(github)
         .github_oauth_from_config(&config)
+        .oidc_key_stores(oidc_key_stores)
         .emails(emails)
         .storage_from_config(&config.storage)
         .rate_limiter_from_config(config.rate_limiter.clone())