mithril_aggregator/services/snapshotter/ancillary_signer/
interface.rs

1use async_trait::async_trait;
2
3use mithril_common::crypto_helper::ManifestSignature;
4use mithril_common::entities::AncillaryFilesManifest;
5use mithril_common::StdResult;
6
7#[cfg_attr(test, mockall::automock)]
8#[async_trait]
9/// Define how to sign the ancillary manifest.
10pub trait AncillarySigner: Sync + Send {
11    /// Compute the signature of the ancillary manifest.
12    async fn compute_ancillary_manifest_signature(
13        &self,
14        manifest: &AncillaryFilesManifest,
15    ) -> StdResult<ManifestSignature>;
16}
17
18#[cfg(test)]
19impl MockAncillarySigner {
20    pub(crate) fn that_succeeds_with_signature<T>(signature_to_return: T) -> Self
21    where
22        T: TryInto<ManifestSignature>,
23        T::Error: std::fmt::Debug,
24    {
25        let expected_signature = signature_to_return.try_into().unwrap();
26        let mut mock = MockAncillarySigner::new();
27        mock.expect_compute_ancillary_manifest_signature()
28            .returning(move |_| Ok(expected_signature));
29        mock
30    }
31
32    pub(crate) fn that_fails_with_message<T: Into<String>>(message: T) -> Self {
33        let error = anyhow::anyhow!("{}", message.into());
34        let mut mock = MockAncillarySigner::new();
35        mock.expect_compute_ancillary_manifest_signature()
36            .return_once(move |_| Err(error));
37        mock
38    }
39}