add tests for pcap parsing

satta · Aug 22, 2019 · 0653103 · 0653103
1 parent f564e43
commit 0653103
Show file tree

Hide file tree

Showing 7 changed files with 28 additions and 0 deletions.
diff --git a/pcap_flowtuple_source_test.go b/pcap_flowtuple_source_test.go
@@ -0,0 +1,28 @@
+package gommunityid
+
+import (
+	"log"
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestPcapFlowTupleSource(t *testing.T) {
+	testfiles, err := filepath.Glob("testdata/*.pcap")
+	if err != nil {
+		t.Fatal(err)
+	}
+	i := 0
+	for _, testfile := range testfiles {
+		ftChan, err := PcapFlowTupleSource(testfile)
+		if err != nil {
+			log.Fatal(err)
+		}
+		for _ = range ftChan {
+			i++
+		}
+		log.Printf("read %s with %d packets", testfile, i)
+		assert.Greater(t, i, 0)
+	}
+}
diff --git a/testdata/icmp.pcap b/testdata/icmp.pcap
diff --git a/testdata/icmp6.pcap b/testdata/icmp6.pcap
diff --git a/testdata/ipv6.pcap b/testdata/ipv6.pcap
diff --git a/testdata/sctp.pcap b/testdata/sctp.pcap
diff --git a/testdata/tcp.pcap b/testdata/tcp.pcap
diff --git a/testdata/udp.pcap b/testdata/udp.pcap