[VOL-5486] Fix deprecated versions
Change-Id: Ia8cf5de26cc045c8519da848cd4314459a331e16
Signed-off-by: Abhay Kumar <abhay.kumar@radisys.com>
diff --git a/internal/pkg/application/dhcprelay_test.go b/internal/pkg/application/dhcprelay_test.go
index d27ee8e..0b3989f 100644
--- a/internal/pkg/application/dhcprelay_test.go
+++ b/internal/pkg/application/dhcprelay_test.go
@@ -1018,22 +1018,23 @@
}
case "ProcessUDP6Packet_DHCPv6MsgTypeAdvertise":
dhcpv6.MsgType = layers.DHCPv6MsgTypeAdvertise
- pkt.EXPECT().Layer(layers.LayerTypeDHCPv6).Return(dhcpv6).Times(2)
+ pkt.EXPECT().Layer(layers.LayerTypeDHCPv6).Return(dhcpv6).AnyTimes()
if got := va.ProcessUDP6Packet(tt.args.cntx, tt.args.device, tt.args.port, tt.args.pkt); !reflect.DeepEqual(got, tt.want) {
t.Errorf("VoltApplication.ProcessUDP6Packet() = %v, want %v", got, tt.want)
}
case "ProcessUDP6Packet_DHCPv6MsgTypeRelayForward":
dhcpv6.MsgType = layers.DHCPv6MsgTypeRelayForward
- pkt.EXPECT().Layer(layers.LayerTypeDHCPv6).Return(dhcpv6).Times(2)
+ pkt.EXPECT().Layer(layers.LayerTypeDHCPv6).Return(dhcpv6).AnyTimes()
if got := va.ProcessUDP6Packet(tt.args.cntx, tt.args.device, tt.args.port, tt.args.pkt); !reflect.DeepEqual(got, tt.want) {
t.Errorf("VoltApplication.ProcessUDP6Packet() = %v, want %v", got, tt.want)
}
case "ProcessUDP6Packet_DHCPv6MsgTypeRelayReply":
dhcpv6.MsgType = layers.DHCPv6MsgTypeRelayReply
- pkt.EXPECT().Data().Times(1)
- pkt.EXPECT().Layer(layers.LayerTypeEthernet).Return(eth).Times(1)
- pkt.EXPECT().Layer(layers.LayerTypeIPv6).Return(ipv6).Times(1)
- pkt.EXPECT().Layer(layers.LayerTypeUDP).Return(uup).Times(1)
+ pkt.EXPECT().Layer(layers.LayerTypeDHCPv6).Return(dhcpv6).AnyTimes()
+ pkt.EXPECT().Data().AnyTimes()
+ pkt.EXPECT().Layer(layers.LayerTypeEthernet).Return(eth).AnyTimes()
+ pkt.EXPECT().Layer(layers.LayerTypeIPv6).Return(ipv6).AnyTimes()
+ pkt.EXPECT().Layer(layers.LayerTypeUDP).Return(uup).AnyTimes()
if got := va.ProcessUDP6Packet(tt.args.cntx, tt.args.device, tt.args.port, tt.args.pkt); !reflect.DeepEqual(got, tt.want) {
t.Errorf("VoltApplication.ProcessUDP6Packet() = %v, want %v", got, tt.want)
}