Add integration tests for pcap_loop (#337)

rust-pcap · Mar 15, 2024 · 188acb1 · 188acb1
1 parent d1494d0
commit 188acb1
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/tests/capture/activated/mod.rs b/tests/capture/activated/mod.rs
@@ -136,3 +136,22 @@ fn test_filter() {
     let result = capture.next_packet();
     assert!(result.is_ok());
 }
+
+#[test]
+fn read_packet_via_pcap_loop() {
+    let mut packets = 0;
+    let mut capture = capture_from_test_file("packet_snaplen_65535.pcap");
+    capture
+        .for_each(None, |_| {
+            packets += 1;
+        })
+        .unwrap();
+    assert_eq!(packets, 1);
+}
+
+#[test]
+#[should_panic]
+fn panic_in_pcap_loop() {
+    let mut capture = capture_from_test_file("packet_snaplen_65535.pcap");
+    capture.for_each(None, |_| panic!()).unwrap();
+}