mithril_aggregator/test/double/
dummies.rs

1use chrono::Utc;
2use uuid::Uuid;
3
4use mithril_common::test::double::{Dummy, fake_data};
5
6mod record {
7    use mithril_common::entities::{ProtocolMessage, SignedEntityType};
8
9    use crate::database::record::{ImmutableFileDigestRecord, OpenMessageRecord};
10
11    use super::*;
12
13    impl Dummy for ImmutableFileDigestRecord {
14        /// Create a dumb ImmutableFileDigestRecord instance mainly for test purposes
15        fn dummy() -> Self {
16            Self {
17                immutable_file_name: "123.chunk".to_string(),
18                digest: "dummy_digest".to_string(),
19            }
20        }
21    }
22
23    impl Dummy for OpenMessageRecord {
24        /// Create a dumb OpenMessage instance mainly for test purposes
25        fn dummy() -> Self {
26            let beacon = fake_data::beacon();
27            let epoch = beacon.epoch;
28            let signed_entity_type = SignedEntityType::CardanoImmutableFilesFull(beacon);
29
30            Self {
31                open_message_id: Uuid::parse_str("193d1442-e89b-43cf-9519-04d8db9a12ff").unwrap(),
32                epoch,
33                signed_entity_type,
34                protocol_message: ProtocolMessage::new(),
35                is_certified: false,
36                is_expired: false,
37                created_at: Utc::now(),
38                expires_at: None,
39            }
40        }
41    }
42}
43
44mod entities {
45    use mithril_common::entities::{
46        CardanoTransactionsSigningConfig, ProtocolMessage, SignedEntityType,
47    };
48
49    use crate::entities::{AggregatorEpochSettings, LeaderAggregatorEpochSettings, OpenMessage};
50
51    use super::*;
52
53    impl Dummy for AggregatorEpochSettings {
54        /// Create a dummy `AggregatorEpochSettings`
55        fn dummy() -> Self {
56            let protocol_parameters = fake_data::protocol_parameters();
57            let cardano_transactions_signing_config = CardanoTransactionsSigningConfig::dummy();
58
59            // Aggregator Epoch settings
60            AggregatorEpochSettings {
61                protocol_parameters,
62                cardano_transactions_signing_config,
63            }
64        }
65    }
66
67    impl Dummy for LeaderAggregatorEpochSettings {
68        /// Create a dummy `LeaderAggregatorEpochSettings`
69        fn dummy() -> Self {
70            // Beacon
71            let beacon = fake_data::beacon();
72
73            // Registration protocol parameters
74            let registration_protocol_parameters = fake_data::protocol_parameters();
75
76            // Signers
77            let signers = fake_data::signers(5);
78            let current_signers = signers[1..3].to_vec();
79            let next_signers = signers[2..5].to_vec();
80
81            // Cardano transactions signing configuration
82            let cardano_transactions_signing_config =
83                Some(CardanoTransactionsSigningConfig::dummy());
84
85            // Signer Epoch settings
86            LeaderAggregatorEpochSettings {
87                epoch: beacon.epoch,
88                registration_protocol_parameters,
89                current_signers,
90                next_signers,
91                cardano_transactions_signing_config,
92            }
93        }
94    }
95
96    impl Dummy for OpenMessage {
97        fn dummy() -> Self {
98            let beacon = fake_data::beacon();
99            let epoch = beacon.epoch;
100            let signed_entity_type = SignedEntityType::CardanoImmutableFilesFull(beacon);
101
102            Self {
103                epoch,
104                signed_entity_type,
105                protocol_message: ProtocolMessage::new(),
106                is_certified: false,
107                is_expired: false,
108                single_signatures: vec![
109                    fake_data::single_signature(vec![1, 4, 5]),
110                    fake_data::single_signature(vec![2, 3, 8]),
111                ],
112                created_at: Utc::now(),
113                expires_at: None,
114            }
115        }
116    }
117}
mithril_aggregator/test/double/dummies.rs

mithril_aggregator/test/double/
dummies.rs
