fixed tests

Author: shadowy-pycoder

layers/arp_test.go

@@ -65,6 +65,8 @@ func TestParseARP(t *testing.T) {
 		SenderIP:         netip.AddrFrom4([4]byte{0x7F, 0x00, 0x00, 0x01}),
 		TargetMAC:        net.HardwareAddr{0x43, 0x40, 0x8d, 0x28, 0xca, 0x0b},
 		TargetIP:         netip.AddrFrom4([4]byte{0x7F, 0x00, 0x00, 0x02}),
+		DstVendor:        net.HardwareAddr{0x43, 0x40, 0x8d, 0x28, 0xca, 0x0b}.String(),
+		SrcVendor:        net.HardwareAddr{0x7b, 0x13, 0x0b, 0x87, 0xea, 0x51}.String(),
 	}
 	arp := &ARPPacket{}
 	packet, close := testPacket(t, "arp")

layers/ethernet_test.go

@@ -26,6 +26,8 @@ func TestParseEthernet(t *testing.T) {
 		SrcMAC:    net.HardwareAddr{0x43, 0x40, 0x8d, 0x28, 0xca, 0x0b},
 		EtherType: &EthernetType{Val: 0x0800, Desc: "IPv4"},
 		Payload:   []byte{},
+		DstVendor: net.HardwareAddr{0x7b, 0x13, 0x0b, 0x87, 0xea, 0x51}.String(),
+		SrcVendor: net.HardwareAddr{0x43, 0x40, 0x8d, 0x28, 0xca, 0x0b}.String(),
 	}
 	eth := &EthernetFrame{}
 	packet, close := testPacket(t, "ethernet")

layers/tcp_test.go

@@ -29,13 +29,23 @@ func TestParseTCP(t *testing.T) {
 		Reserved:   0,
 		Flags: &TCPFlags{
 			Raw: 2,
-			SYN: 1,
+			Val: []*TCPFlag{
+				{Val: 0, Desc: "CWR"},
+				{Val: 0, Desc: "ECE"},
+				{Val: 0, Desc: "URG"},
+				{Val: 0, Desc: "ACK"},
+				{Val: 0, Desc: "PSH"},
+				{Val: 0, Desc: "RST"},
+				{Val: 1, Desc: "SYN"},
+				{Val: 0, Desc: "FIN"},
+			},
 		},
 		WindowSize: 64240,
 		Checksum:   30263,
 		Options: []byte{
 			0x02, 0x04, 0x05, 0xb4, 0x04, 0x02, 0x08, 0x0a, 0xac, 0xf8,
-			0x48, 0x3e, 0x00, 0x00, 0x00, 0x00, 0x01, 0x03, 0x03, 0x07},
+			0x48, 0x3e, 0x00, 0x00, 0x00, 0x00, 0x01, 0x03, 0x03, 0x07,
+		},
 		payload: []byte{},
 	}
 	tcp := &TCPSegment{}