trustpub: Add MockOidcKeyStore::with_test_key() fn

Author: Turbo87

crates/crates_io_trustpub/Cargo.toml

@@ -8,7 +8,7 @@ edition = "2024"
 workspace = true
 
 [features]
-test-helpers = ["dep:mockall"]
+test-helpers = ["dep:mockall", "dep:serde_json"]
 
 [dependencies]
 anyhow = "=1.0.98"
@@ -18,6 +18,7 @@ mockall = { version = "=0.13.1", optional = true }
 reqwest = { version = "=0.12.15", features = ["gzip", "json"] }
 regex = "=1.11.1"
 serde = { version = "=1.0.219", features = ["derive"] }
+serde_json = { version = "=1.0.140", optional = true }
 thiserror = "=2.0.12"
 tokio = { version = "=1.45.0", features = ["sync"] }
 

crates/crates_io_trustpub/src/keystore/mod.rs

@@ -18,3 +18,21 @@ pub trait OidcKeyStore: Send + Sync {
     /// is an error while fetching the key, it will return an error.
     async fn get_oidc_key(&self, key_id: &str) -> anyhow::Result<Option<DecodingKey>>;
 }
+
+#[cfg(feature = "test-helpers")]
+impl MockOidcKeyStore {
+    /// Creates a new instance of [`MockOidcKeyStore`] based on the RSA keys
+    /// provided in the [`crate::test_keys`] module.
+    pub fn with_test_key() -> Self {
+        use crate::test_keys::{DECODING_KEY, KEY_ID};
+        use mockall::predicate::*;
+
+        let mut mock = Self::new();
+
+        mock.expect_get_oidc_key()
+            .with(eq(KEY_ID))
+            .returning(|_| Ok(Some(DECODING_KEY.clone())));
+
+        mock
+    }
+}

crates/crates_io_trustpub/src/lib.rs

@@ -2,5 +2,5 @@
 
 pub mod github;
 pub mod keystore;
-#[cfg(test)]
+#[cfg(any(test, feature = "test-helpers"))]
 pub mod test_keys;

crates/crates_io_trustpub/src/test_keys.rs

@@ -54,7 +54,6 @@ pub(crate) const KEY_ID: &str = "c0ffee";
 static ENCODING_KEY: LazyLock<EncodingKey> =
     LazyLock::new(|| EncodingKey::from_rsa_pem(PRIVATE_KEY_PEM).unwrap());
 
-#[allow(unused)]
 pub(crate) static DECODING_KEY: LazyLock<DecodingKey> = LazyLock::new(|| {
     let jwk = serde_json::from_str(PUBLIC_KEY_JWK).unwrap();
     DecodingKey::from_jwk(&jwk).unwrap()