[VOL-2778] Adding pbits to DHCP packets
Change-Id: I3bfb80c566a08ec69f05f90e33b8196ad401b33d
diff --git a/internal/bbsim/packetHandlers/packet_tags_test.go b/internal/bbsim/packetHandlers/packet_tags_test.go
index b2696a6..231cb71 100644
--- a/internal/bbsim/packetHandlers/packet_tags_test.go
+++ b/internal/bbsim/packetHandlers/packet_tags_test.go
@@ -48,14 +48,18 @@
)
untaggedPkt := gopacket.NewPacket(buffer.Bytes(), layers.LayerTypeEthernet, gopacket.Default)
- taggedPkt, err := packetHandlers.PushSingleTag(111, untaggedPkt)
+
+ taggedPkt, err := packetHandlers.PushSingleTag(111, untaggedPkt, 7)
if err != nil {
t.Fail()
t.Logf("Error in PushSingleTag: %v", err)
}
vlan, _ := packetHandlers.GetVlanTag(taggedPkt)
+ pbit, _ := packetHandlers.GetPbit(taggedPkt)
+
assert.Equal(t, vlan, uint16(111))
+ assert.Equal(t, pbit, uint8(7))
}
func TestPushDoubleTag(t *testing.T) {
@@ -80,7 +84,7 @@
)
untaggedPkt := gopacket.NewPacket(buffer.Bytes(), layers.LayerTypeEthernet, gopacket.Default)
- taggedPkt, err := packetHandlers.PushDoubleTag(900, 800, untaggedPkt)
+ taggedPkt, err := packetHandlers.PushDoubleTag(900, 800, untaggedPkt, 0)
if err != nil {
t.Fail()
t.Logf("Error in PushSingleTag: %v", err)