Skip to content

Commit 73b2fed

Browse files
committed
test: refactor test_cli_runner return values to improve test readability
1 parent fb450e6 commit 73b2fed

File tree

2 files changed

+167
-97
lines changed

2 files changed

+167
-97
lines changed

mithril-common/src/chain_observer/cli_observer.rs

Lines changed: 38 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -572,7 +572,10 @@ mod tests {
572572
use std::collections::BTreeMap;
573573

574574
use super::*;
575-
use crate::{chain_observer::test_cli_runner::TestCliRunner, crypto_helper::ColdKeyGenerator};
575+
use crate::{
576+
chain_observer::test_cli_runner::{test_cli_runner_values, TestCliRunner},
577+
crypto_helper::ColdKeyGenerator,
578+
};
576579

577580
use kes_summed_ed25519::{kes::Sum6Kes, traits::KesSk};
578581

@@ -581,15 +584,18 @@ mod tests {
581584
let observer = CardanoCliChainObserver::new(Box::<TestCliRunner>::default());
582585
let era = observer.get_current_era().await.unwrap().unwrap();
583586

584-
assert_eq!("Conway".to_string(), era);
587+
assert_eq!(
588+
test_cli_runner_values::LAUNCH_ERA_EXPECTED_ERA.to_string(),
589+
era
590+
);
585591
}
586592

587593
#[tokio::test]
588594
async fn test_get_current_epoch() {
589595
let observer = CardanoCliChainObserver::new(Box::<TestCliRunner>::default());
590596
let epoch = observer.get_current_epoch().await.unwrap().unwrap();
591597

592-
assert_eq!(Epoch(120), epoch);
598+
assert_eq!(test_cli_runner_values::LAUNCH_EPOCH_EXPECTED_EPOCH, epoch);
593599
}
594600

595601
#[tokio::test]
@@ -599,9 +605,9 @@ mod tests {
599605

600606
assert_eq!(
601607
ChainPoint {
602-
slot_number: SlotNumber(25886617),
603-
block_number: BlockNumber(1270276),
604-
block_hash: "7383b17d7b05b0953cf0649abff60173995eb9febe556889333e20e1e5b7ca84"
608+
slot_number: test_cli_runner_values::LAUNCH_CHAIN_POINT_EXPECTED_SLOT_NUMBER,
609+
block_number: test_cli_runner_values::LAUNCH_CHAIN_POINT_EXPECTED_BLOCK_NUMBER,
610+
block_hash: test_cli_runner_values::LAUNCH_CHAIN_POINT_EXPECTED_BLOCK_HASH
605611
.to_string(),
606612
},
607613
chain_point
@@ -655,7 +661,16 @@ mod tests {
655661
let observer = CardanoCliChainObserver::new(Box::<TestCliRunner>::default());
656662
let address = "addrtest_123456".to_string();
657663
let datums = observer.get_current_datums(&address).await.unwrap();
658-
assert_eq!(vec![TxDatum(r#"{"constructor":0,"fields":[{"bytes":"5b0a20207b0a20202020226e616d65223a20227468616c6573222c0a202020202265706f6368223a203132330a20207d2c0a20207b0a20202020226e616d65223a20227079746861676f726173222c0a202020202265706f6368223a206e756c6c0a20207d0a5d0a"}]}"#.to_string())], datums);
664+
assert_eq!(
665+
vec![TxDatum(
666+
format!(
667+
r#"{{"constructor":0,"fields":[{{"bytes":"{}"}}]}}"#,
668+
test_cli_runner_values::LAUNCH_UTXO_EXPECTED_BYTES
669+
)
670+
.to_string()
671+
)],
672+
datums
673+
);
659674
}
660675

661676
#[tokio::test]
@@ -665,13 +680,19 @@ mod tests {
665680
.get_current_stake_value("pool1qqyjr9pcrv97gwrueunug829fs5znw6p2wxft3fvqkgu5f4qlrg")
666681
.await
667682
.expect("get current stake value should not fail");
668-
assert_eq!(3_000_000, stake);
683+
assert_eq!(
684+
test_cli_runner_values::LAUNCH_STAKE_SNAPSHOT_EXPECTED_DEFAULT_POOL_STAKE_MARK,
685+
stake
686+
);
669687

670688
let stake = observer
671-
.get_current_stake_value("pool1qpqvz90w7qsex2al2ejjej0rfgrwsguch307w8fraw7a7adf6g8")
689+
.get_current_stake_value(test_cli_runner_values::LAUNCH_STAKE_SNAPSHOT_POOL_ID_SPECIFIC)
672690
.await
673691
.expect("get current stake value should not fail");
674-
assert_eq!(0, stake);
692+
assert_eq!(
693+
test_cli_runner_values::LAUNCH_STAKE_SNAPSHOT_EXPECTED_POOL_STAKE_MARK_FOR_POOL_ID_SPECIFIC,
694+
stake
695+
);
675696
}
676697

677698
#[tokio::test]
@@ -709,15 +730,15 @@ mod tests {
709730
let mut expected_stake_distribution = StakeDistribution::new();
710731
expected_stake_distribution.insert(
711732
"pool1qqqqqdk4zhsjuxxd8jyvwncf5eucfskz0xjjj64fdmlgj735lr9".to_string(),
712-
300000000001,
733+
test_cli_runner_values::LAUNCH_STAKE_SNAPSHOT_ALL_POOLS_EXPECTED_STAKE_MARK_POOL_1,
713734
);
714735
expected_stake_distribution.insert(
715736
"pool1qqqqpanw9zc0rzh0yp247nzf2s35uvnsm7aaesfl2nnejaev0uc".to_string(),
716-
600000000001,
737+
test_cli_runner_values::LAUNCH_STAKE_SNAPSHOT_ALL_POOLS_EXPECTED_STAKE_MARK_POOL_2,
717738
);
718739
expected_stake_distribution.insert(
719740
"pool1qqqqzyqf8mlm70883zht60n4q6uqxg4a8x266sewv8ad2grkztl".to_string(),
720-
1200000000001,
741+
test_cli_runner_values::LAUNCH_STAKE_SNAPSHOT_ALL_POOLS_EXPECTED_STAKE_MARK_POOL_3,
721742
);
722743
assert_eq!(
723744
BTreeMap::from_iter(
@@ -806,6 +827,9 @@ mod tests {
806827
.await
807828
.unwrap()
808829
.unwrap();
809-
assert_eq!(404, kes_period);
830+
assert_eq!(
831+
test_cli_runner_values::LAUNCH_KES_PERIOD_EXPECTED_KES_PERIOD,
832+
kes_period
833+
);
810834
}
811835
}

0 commit comments

Comments
 (0)