Skip to content

Commit

Permalink
test: add sample pcap and update test cases for reader.go
Browse files Browse the repository at this point in the history
  • Loading branch information
md-irohas committed Jul 1, 2023
1 parent 5fce112 commit fa28ea7
Show file tree
Hide file tree
Showing 2 changed files with 12 additions and 5 deletions.
17 changes: 12 additions & 5 deletions rcap/reader_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,16 @@ import (
"github.com/google/gopacket/pcap"
)

func makeReader() *Reader {
func makeReader(t *testing.T) *Reader {
var err error

reader := &Reader{}
reader.handle, _ = pcap.OpenOffline("testdata/sample.pcap")
reader.handle, err = pcap.OpenOffline("testdata/sample.pcap")

if err != nil {
t.Fatalf("failed to make Reader for test: %v", err)
}

return reader
}

Expand All @@ -23,7 +30,7 @@ func TestNewReader(t *testing.T) {
}

func TestReaderLinkType(t *testing.T) {
r := makeReader()
r := makeReader(t)
if r.LinkType() != layers.LinkTypeEthernet {
t.Errorf("'%v' is expected, but got '%v'.", layers.LinkTypeEthernet, r.LinkType())
}
Expand All @@ -50,7 +57,7 @@ func TestReaderResetNumPackets(t *testing.T) {
}

func TestReaderReadPacket(t *testing.T) {
r := makeReader()
r := makeReader(t)

expectedPayload := []byte("this is a test packet.\n")
expectedTime := int64(1688205150)
Expand All @@ -75,6 +82,6 @@ func TestReaderReadPacket(t *testing.T) {
}

func TestReaderClose(t *testing.T) {
r := makeReader()
r := makeReader(t)
r.Close()
}
Binary file added rcap/testdata/sample.pcap
Binary file not shown.

0 comments on commit fa28ea7

Please sign in to comment.