Ut Ipassigner Linux

Signed-off-by: Rajnish Kumar <rajnishk1@vmware.com>
antrea-io · Oct 4, 2023 · 765269f · 765269f
1 parent c730c77
commit 765269f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/pkg/agent/ipassigner/ip_assigner_linux_test.go b/pkg/agent/ipassigner/ip_assigner_linux_test.go
@@ -210,13 +210,13 @@ func TestIPAssigner_UnAssignIP(t *testing.T) {
 			expectedAssignedIPs: sets.New[string]("2.1.1.1", "2.2.2.1"),
 		},
 		{
-			name:                "Unassign  IPv4",
+			name:                "Unassign IPv4",
 			ip:                  "2.1.1.1",
 			assignedIPs:         sets.New[string]("2.1.1.1", "2.2.2.1"),
 			expectedAssignedIPs: sets.New[string]("2.2.2.1"),
 		},
 		{
-			name:                "Unassign  IPv6",
+			name:                "Unassign IPv6",
 			ip:                  "2001:db8::1",
 			assignedIPs:         sets.New[string]("2.2.2.1", "2001:db8::1"),
 			expectedAssignedIPs: sets.New[string]("2.2.2.1"),
@@ -317,8 +317,11 @@ func TestIPAssigner_InitIPs(t *testing.T) {
 				ndpResponder:      mockResponder,
 			}
 
-			for ipStr := range tt.ips {
-				mockResponder.EXPECT().AddIP(net.ParseIP(ipStr)).Return(nil)
+			if (tt.ips).Has("192.168.1.100") {
+				mockResponder.EXPECT().AddIP(net.ParseIP("192.168.1.100")).Return(nil)
+			}
+			if (tt.ips).Has("192.168.1.101") {
+				mockResponder.EXPECT().AddIP(net.ParseIP("192.168.1.101")).Return(nil)
 			}
 
 			advertiseArpNdp := advertiseResponder

diff --git a/pkg/agent/ipassigner/responder/testing/mock_responder.go b/pkg/agent/ipassigner/responder/testing/mock_responder.go