Use real avks instead of mock strings in tests

Author: Alenar

mithril-aggregator/src/database/provider/certificate.rs

@@ -82,7 +82,7 @@ impl CertificateRecord {
             parent_certificate_id: Some(parent_id.to_string()),
             message: "message".to_string(),
             signature: fake_keys::multi_signature()[0].to_owned(),
-            aggregate_verification_key: "avk".to_string(),
+            aggregate_verification_key: fake_keys::aggregate_verification_key()[0].to_owned(),
             epoch: beacon.epoch,
             beacon,
             protocol_version: "protocol_version".to_string(),

mithril-aggregator/src/tools/certificates_hash_migrator.rs

@@ -342,41 +342,41 @@ mod test {
         let expected = vec![
             (
                 dummy_genesis(
-                    "b9263d7c2e94f6a62802b6521b30967e7e7a7fdccf02026dc70a0ffbc46da393",
+                    "98b44c52ac3c82adcbc5aea27a0c99cbba716048dddaf401b27acded80f1abcd",
                     1,
                     1,
                 ),
                 None,
             ),
             (
                 dummy_certificate(
-                    "5d73905d314abc86451a399d4f1820201d4c20076d8fb0e4b277ffe5f5e4dc07",
-                    "b9263d7c2e94f6a62802b6521b30967e7e7a7fdccf02026dc70a0ffbc46da393",
+                    "0b7c12daffe63ca93a9cded424e300361a5234ab8c52b3e7029ff9dfbd8a16bd",
+                    "98b44c52ac3c82adcbc5aea27a0c99cbba716048dddaf401b27acded80f1abcd",
                     1,
                     2,
                 ),
                 Some(SignedEntityRecord {
                     signed_entity_id: "signed_entity_id".to_string(),
                     signed_entity_type: MithrilStakeDistribution(Epoch(1)),
                     certificate_id:
-                        "5d73905d314abc86451a399d4f1820201d4c20076d8fb0e4b277ffe5f5e4dc07"
+                        "0b7c12daffe63ca93a9cded424e300361a5234ab8c52b3e7029ff9dfbd8a16bd"
                             .to_string(),
                     artifact: "".to_string(),
                     created_at: Default::default(),
                 }),
             ),
             (
                 dummy_certificate(
-                    "583803ea5050f781de38c25d8773628ccf5c6fd737b4d984db2d41444bfdbf39",
-                    "5d73905d314abc86451a399d4f1820201d4c20076d8fb0e4b277ffe5f5e4dc07",
+                    "0aa74b928d3eade6548b22349435a12e9591f0c34c9ca2b0abfbeeefbb1acb37",
+                    "0b7c12daffe63ca93a9cded424e300361a5234ab8c52b3e7029ff9dfbd8a16bd",
                     2,
                     3,
                 ),
                 Some(SignedEntityRecord {
                     signed_entity_id: "signed_entity_id".to_string(),
                     signed_entity_type: MithrilStakeDistribution(Epoch(2)),
                     certificate_id:
-                        "583803ea5050f781de38c25d8773628ccf5c6fd737b4d984db2d41444bfdbf39"
+                        "0aa74b928d3eade6548b22349435a12e9591f0c34c9ca2b0abfbeeefbb1acb37"
                             .to_string(),
                     artifact: "".to_string(),
                     created_at: Default::default(),

mithril-common/src/entities/certificate.rs

@@ -200,7 +200,7 @@ mod tests {
         );
         protocol_message.set_message_part(
             ProtocolMessagePartKey::NextAggregateVerificationKey,
-            "next-avk-123".to_string(),
+            fake_keys::aggregate_verification_key()[1].to_owned(),
         );
 
         protocol_message
@@ -209,7 +209,7 @@ mod tests {
     #[test]
     fn test_certificate_compute_hash() {
         const HASH_EXPECTED: &str =
-            "255d59cef74aae5bc2e83e87612dea41309551dde0c770d46bf6607971bb9765";
+            "af0965134ef5b2c2a33005cf401c58ca882dead8efda358540dac0575098b54e";
 
         let initiated_at = DateTime::parse_from_rfc3339("2024-02-12T13:11:47.0123043Z")
             .unwrap()
@@ -227,7 +227,7 @@ mod tests {
                 get_signers_with_stake(),
             ),
             get_protocol_message(),
-            "aggregate_verification_key".to_string(),
+            fake_keys::aggregate_verification_key()[0].to_owned(),
             CertificateSignature::MultiSignature(
                 fake_keys::multi_signature()[0].try_into().unwrap(),
             ),
@@ -272,7 +272,9 @@ mod tests {
                     let mut protocol_message_modified = certificate.protocol_message.clone();
                     protocol_message_modified.set_message_part(
                         ProtocolMessagePartKey::NextAggregateVerificationKey,
-                        "next-avk-456".to_string(),
+                        fake_keys::aggregate_verification_key()[2]
+                            .try_into()
+                            .unwrap(),
                     );
 
                     protocol_message_modified
@@ -285,7 +287,9 @@ mod tests {
         assert_ne!(
             HASH_EXPECTED,
             Certificate {
-                aggregate_verification_key: "aggregate_verification_key-modified".to_string(),
+                aggregate_verification_key: fake_keys::aggregate_verification_key()[2]
+                    .try_into()
+                    .unwrap(),
                 ..certificate.clone()
             }
             .compute_hash(),
@@ -306,7 +310,7 @@ mod tests {
     #[test]
     fn test_genesis_certificate_compute_hash() {
         const HASH_EXPECTED: &str =
-            "bbb265e74082896873d3fbe568e4b0118ddcf9a63b4f4b369b92773439e80159";
+            "e8cc8885ef7a76f35216a6ef603ab0387dcacb892e0a33df9de9f5bf98cf203b";
 
         let initiated_at = DateTime::parse_from_rfc3339("2024-02-12T13:11:47.0123043Z")
             .unwrap()
@@ -324,7 +328,7 @@ mod tests {
                 get_signers_with_stake(),
             ),
             get_protocol_message(),
-            "aggregate_verification_key".to_string(),
+            fake_keys::aggregate_verification_key()[1].to_owned(),
             CertificateSignature::GenesisSignature(
                 fake_keys::genesis_signature()[0].try_into().unwrap(),
             ),

mithril-common/src/messages/certificate.rs

@@ -59,7 +59,7 @@ impl CertificateMessage {
         );
         protocol_message.set_message_part(
             ProtocolMessagePartKey::NextAggregateVerificationKey,
-            "next-avk-123".to_string(),
+            fake_keys::aggregate_verification_key()[1].to_owned(),
         );
         Self {
             hash: "hash".to_string(),
@@ -68,7 +68,7 @@ impl CertificateMessage {
             metadata: CertificateMetadataMessagePart::dummy(),
             protocol_message: protocol_message.clone(),
             signed_message: "signed_message".to_string(),
-            aggregate_verification_key: "aggregate_verification_key".to_string(),
+            aggregate_verification_key: fake_keys::aggregate_verification_key()[0].to_owned(),
             multi_signature: fake_keys::multi_signature()[0].to_owned(),
             genesis_signature: String::new(),
         }

mithril-common/src/test_utils/fake_data.rs

@@ -123,7 +123,7 @@ pub fn certificate(certificate_hash: String) -> entities::Certificate {
     );
 
     // Protocol message
-    let next_aggregate_verification_key = "next-avk-123".to_string();
+    let next_aggregate_verification_key = fake_keys::aggregate_verification_key()[2].to_owned();
     let mut protocol_message = ProtocolMessage::new();
     let snapshot_digest = format!("1{}", beacon.immutable_file_number).repeat(20);
     protocol_message.set_message_part(ProtocolMessagePartKey::SnapshotDigest, snapshot_digest);
@@ -134,7 +134,7 @@ pub fn certificate(certificate_hash: String) -> entities::Certificate {
 
     // Certificate
     let previous_hash = format!("{certificate_hash}0");
-    let aggregate_verification_key = format!("AVK{}", beacon.immutable_file_number).repeat(5);
+    let aggregate_verification_key = fake_keys::aggregate_verification_key()[1].to_owned();
     let multi_signature = fake_keys::multi_signature()[0].try_into().unwrap();
 
     entities::Certificate {