Fixing a test compilation error introduced by my previous commit

near · Aug 14, 2019 · ea785c7 · ea785c7
1 parent 00c562a
commit ea785c7
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 13 deletions.
diff --git a/chain/chain/src/types.rs b/chain/chain/src/types.rs
@@ -63,7 +63,7 @@ pub enum ShardFullChunkOrOnePart<'a> {
     NoChunk,
 }
 
-#[derive(Eq, PartialEq)]
+#[derive(Eq, PartialEq, Debug)]
 pub enum ValidatorSignatureVerificationResult {
     Valid,
     Invalid,

diff --git a/near/src/runtime.rs b/near/src/runtime.rs
@@ -603,6 +603,7 @@ mod test {
 
     use crate::config::{TESTING_INIT_BALANCE, TESTING_INIT_STAKE};
     use crate::{get_store_path, GenesisConfig, NightshadeRuntime};
+    use near_chain::types::ValidatorSignatureVerificationResult;
 
     fn stake(nonce: Nonce, sender: &BlockProducer, amount: Balance) -> SignedTransaction {
         TransactionBody::Stake(StakeTransaction {
@@ -1124,12 +1125,15 @@ mod test {
         let data = [0; 32];
         let signer = InMemorySigner::from_seed(&validators[0], &validators[0]);
         let signature = signer.sign(&data);
-        assert!(env.runtime.verify_validator_signature(
-            &EpochId::default(),
-            &validators[0],
-            &data,
-            &signature
-        ));
+        assert_eq!(
+            ValidatorSignatureVerificationResult::Valid,
+            env.runtime.verify_validator_signature(
+                &EpochId::default(),
+                &validators[0],
+                &data,
+                &signature
+            )
+        );
     }
 
     #[test]
@@ -1139,12 +1143,15 @@ mod test {
         let data = [0; 32];
         let signer = InMemorySigner::from_seed(&validators[0], &validators[0]);
         let signature = signer.sign(&data);
-        assert!(!env.runtime.verify_validator_signature(
-            &EpochId::default(),
-            &validators[1],
-            &data,
-            &signature
-        ));
+        assert_eq!(
+            ValidatorSignatureVerificationResult::Invalid,
+            env.runtime.verify_validator_signature(
+                &EpochId::default(),
+                &validators[1],
+                &data,
+                &signature
+            )
+        );
     }
 
     #[test]