tests(dmq): provide tooling for creating signed fake DMQ messages

Author: jpraynaud

internal/mithril-dmq/src/test/fake_message.rs

@@ -0,0 +1,26 @@
+//! Fake message computation for testing purposes.
+
+use std::sync::Arc;
+
+use mithril_cardano_node_chain::test::double::FakeChainObserver;
+use mithril_common::{crypto_helper::TryToBytes, test::crypto_helper::KesSignerFake};
+
+use crate::{DmqMessage, DmqMessageBuilder, test::payload::DmqMessageTestPayload};
+
+/// Computes a fake DMQ message for testing purposes.
+pub async fn compute_fake_msg(bytes: &[u8], test_directory: &str) -> DmqMessage {
+    let dmq_builder = DmqMessageBuilder::new(
+        {
+            let (kes_signature, operational_certificate) =
+                KesSignerFake::dummy_signature(test_directory);
+            let kes_signer =
+                KesSignerFake::new(vec![Ok((kes_signature, operational_certificate.clone()))]);
+
+            Arc::new(kes_signer)
+        },
+        Arc::new(FakeChainObserver::default()),
+    )
+    .set_ttl(100);
+    let message = DmqMessageTestPayload::new(bytes);
+    dmq_builder.build(&message.to_bytes_vec().unwrap()).await.unwrap()
+}

internal/mithril-dmq/src/test/mod.rs

@@ -5,7 +5,7 @@
 //! This module provides in particular test doubles for the traits defined in this crate.
 
 pub mod double;
-
+pub mod fake_message;
 pub mod payload;
 
 #[cfg(test)]