fix test compilation error

cloudnativelabs · Oct 7, 2023 · 12561f9 · 12561f9
1 parent e5f272e
commit 12561f9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/controllers/netpol/network_policy_controller_test.go b/pkg/controllers/netpol/network_policy_controller_test.go
@@ -898,7 +898,11 @@ func TestNetworkPolicyController(t *testing.T) {
 	for _, test := range testCases {
 		t.Run(test.name, func(t *testing.T) {
 			// TODO: Handle IPv6
-			_, err := NewNetworkPolicyController(client, test.config, podInformer, netpolInformer, nsInformer, &sync.Mutex{})
+			iptablesHandlers := make(map[v1.IPFamily]utils.IPTablesHandler, 1)
+			iptablesHandlers[v1.IPv4Protocol] = newFakeIPTables(iptables.ProtocolIPv4)
+			ipSetHandlers := make(map[v1.IPFamily]utils.IPSetHandler, 1)
+			ipSetHandlers[v1.IPv4Protocol] = &fakeIPSet{}
+			_, err := NewNetworkPolicyController(client, test.config, podInformer, netpolInformer, nsInformer, &sync.Mutex{}, iptablesHandlers, ipSetHandlers)
 			if err == nil && test.expectError {
 				t.Error("This config should have failed, but it was successful instead")
 			} else if err != nil {