Tests and chunk size

Author: lucidLuckylee

src/chunker.rs

@@ -195,7 +195,7 @@ impl Chunker {
 
         self.call_stack.extend(removed_scripts.into_iter().rev());
         assert!(num_unclosed_ifs >= 0, "More OP_ENDIF's than OP_IF's after undo step. (This means there is a bug in the undo logic.)");
-        assert_eq!(num_unclosed_ifs, 0, "Unable to make up for the OP_IF's in this chunk. Consider a larger target size or more tolerance. Unclosed OP_IF's: {:?}, removed_len: {}, undo.call_stack: {:?}, chunks: {:?}", num_unclosed_ifs, removed_len, undo_info.call_stack, self.chunks);
+        assert_eq!(num_unclosed_ifs, 0, "Unable to make up for the OP_IF's in this chunk. Consider a larger target size or more tolerance. Unclosed OP_IF's: {:?}, removed_len: {}, undo.call_stack: {:?}, chunks: {:?}", num_unclosed_ifs, removed_len, undo_info.call_stack, self.chunks.iter().map(|chunk| chunk.size).collect::<Vec<_>>());
         (undo_info.call_stack, removed_len)
     }
 

tests/test.rs

@@ -310,6 +310,39 @@ fn test_if_positions() {
     assert_eq!(script.extra_endif_positions(), vec![]);
 }
 
+#[test]
+fn test_if_positions_opif() {
+    let script = script! {
+        OP_IF
+    };
+
+    assert_eq!(script.num_unclosed_ifs(), 1);
+    assert_eq!(script.unclosed_if_positions(), vec![0]);
+    assert_eq!(script.extra_endif_positions(), vec![]);
+}
+
+#[test]
+fn test_if_positions_opnotif() {
+    let script = script! {
+        OP_NOTIF
+    };
+
+    assert_eq!(script.num_unclosed_ifs(), 1);
+    assert_eq!(script.unclosed_if_positions(), vec![0]);
+    assert_eq!(script.extra_endif_positions(), vec![]);
+}
+
+#[test]
+fn test_if_positions_opendif() {
+    let script = script! {
+        OP_ENDIF
+    };
+
+    assert_eq!(script.num_unclosed_ifs(), -1);
+    assert_eq!(script.unclosed_if_positions(), vec![]);
+    assert_eq!(script.extra_endif_positions(), vec![0]);
+}
+
 pub fn if_sub_script() -> Script {
     script! {
         OP_IF