Add tests for empty proof and empty pub inputs

lambdaclass · Sep 19, 2024 · 1d5265b · 1d5265b
1 parent 7a2b555
commit 1d5265b
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/batcher/aligned-batcher/src/mina/mod.rs b/batcher/aligned-batcher/src/mina/mod.rs
@@ -22,9 +22,9 @@ mod test {
     use super::verify_proof_integrity;
 
     const PROTOCOL_STATE_PROOF_BYTES: &[u8] =
-        include_bytes!("../../../../batcher/aligned/test_files/mina/protocol_state.proof");
+        include_bytes!("../../../../scripts/test_files/mina/mina_state.proof");
     const PROTOCOL_STATE_PUB_BYTES: &[u8] =
-        include_bytes!("../../../../batcher/aligned/test_files/mina/protocol_state.pub");
+        include_bytes!("../../../../scripts/test_files/mina/mina_state.pub");
 
     #[test]
     fn verify_proof_integrity_does_not_fail() {
@@ -33,4 +33,14 @@ mod test {
             PROTOCOL_STATE_PUB_BYTES,
         ));
     }
+
+    #[test]
+    fn verify_empty_proof() {
+        assert!(!verify_proof_integrity(&[], PROTOCOL_STATE_PUB_BYTES));
+    }
+
+    #[test]
+    fn verify_empty_pub_inputs() {
+        assert!(!verify_proof_integrity(PROTOCOL_STATE_PROOF_BYTES, &[]));
+    }
 }