fix challenge period tests

polytope-labs · Jul 14, 2023 · 7d29de3 · 7d29de3
1 parent 2c38f55
commit 7d29de3
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pallet-ismp/src/tests.rs b/pallet-ismp/src/tests.rs
@@ -218,6 +218,7 @@ fn should_reject_updates_within_challenge_period() {
     ext.execute_with(|| {
         set_timestamp(None);
         let host = Host::<Test>::default();
+        host.store_challenge_period(MOCK_CONSENSUS_STATE_ID, 1_000_000).unwrap();
         check_challenge_period(&host).unwrap()
     })
 }
@@ -229,6 +230,7 @@ fn should_reject_messages_for_frozen_state_machines() {
     ext.execute_with(|| {
         set_timestamp(None);
         let host = Host::<Test>::default();
+        host.store_challenge_period(MOCK_CONSENSUS_STATE_ID, 1_000_000).unwrap();
         frozen_check(&host).unwrap()
     })
 }
@@ -241,6 +243,7 @@ fn should_reject_expired_check_clients() {
         set_timestamp(None);
         let host = Host::<Test>::default();
         host.store_unbonding_period(MOCK_CONSENSUS_STATE_ID, 1_000_000).unwrap();
+        host.store_challenge_period(MOCK_CONSENSUS_STATE_ID, 1_000_000).unwrap();
         check_client_expiry(&host).unwrap()
     })
 }
@@ -253,6 +256,7 @@ fn should_handle_post_request_timeouts_correctly() {
         set_timestamp(None);
         let host = Host::<Test>::default();
         let dispatcher = Dispatcher::<Test>::default();
+        host.store_challenge_period(MOCK_CONSENSUS_STATE_ID, 1_000_000).unwrap();
         timeout_post_processing_check(&host, &dispatcher).unwrap()
     })
 }
@@ -263,6 +267,7 @@ fn should_handle_get_request_timeouts_correctly() {
     ext.execute_with(|| {
         let host = Host::<Test>::default();
         setup_mock_client::<_, Test>(&host);
+        host.store_challenge_period(MOCK_CONSENSUS_STATE_ID, 1_000_000).unwrap();
         let requests = (0..2)
             .into_iter()
             .map(|i| {
@@ -306,6 +311,7 @@ fn should_handle_get_request_responses_correctly() {
     let mut ext = new_test_ext();
     ext.execute_with(|| {
         let host = Host::<Test>::default();
+        host.store_challenge_period(MOCK_CONSENSUS_STATE_ID, 1_000_000).unwrap();
         setup_mock_client::<_, Test>(&host);
         let requests = (0..2)
             .into_iter()