chore: add dataplane interface for tests

npm/pkg/dataplane/Makefile

@@ -0,0 +1,12 @@
+REPO_ROOT = $(shell git rev-parse --show-toplevel)
+TOOLS_DIR = $(REPO_ROOT)/build/tools
+TOOLS_BIN_DIR = $(REPO_ROOT)/build/tools/bin
+MOCKGEN = $(TOOLS_BIN_DIR)/mockgen
+
+.PHONY: generate
+
+generate: $(MOCKGEN) ## Generate mock clients
+	$(MOCKGEN) -source=$(REPO_ROOT)/npm/pkg/dataplane/types.go -copyright_file=$(REPO_ROOT)/npm/pkg/dataplane/ignore_headers.txt -package=mocks > mocks/genericdataplane_generated.go
+
+$(MOCKGEN):
+	@make -C $(REPO_ROOT) $(MOCKGEN)

npm/pkg/dataplane/dataplane_test.go

@@ -5,6 +5,9 @@ import (
 
 	"github.com/Azure/azure-container-networking/npm/metrics"
 	"github.com/Azure/azure-container-networking/npm/pkg/dataplane/ipsets"
+	"github.com/Azure/azure-container-networking/npm/pkg/dataplane/mocks"
+	"github.com/golang/mock/gomock"
+	"github.com/stretchr/testify/require"
 )
 
 func TestNewDataPlane(t *testing.T) {
@@ -151,3 +154,13 @@ func TestAddToSet(t *testing.T) {
 		}
 	}
 }
+
+// gomock sample usage for generated mock dataplane
+func TestAddToList(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	m := mocks.NewMockGenericDataplane(ctrl)
+	m.EXPECT().AddToList("test", []string{"test"}).Return(nil)
+	require.NoError(t, m.AddToList("test", []string{"test"}))
+}

npm/pkg/dataplane/ignore_headers.txt

@@ -0,0 +1 @@
+autogenerated

npm/pkg/dataplane/types.go

@@ -0,0 +1,23 @@
+package dataplane
+
+import (
+	"github.com/Azure/azure-container-networking/npm"
+	"github.com/Azure/azure-container-networking/npm/pkg/dataplane/ipsets"
+	"github.com/Azure/azure-container-networking/npm/pkg/dataplane/policies"
+)
+
+type GenericDataplane interface {
+	InitializeDataPlane() error
+	ResetDataPlane() error
+	CreateIPSet(setName string, setType ipsets.SetType) error
+	DeleteIPSet(name string) error
+	AddToSet(setNames []string, ip, podKey string) error
+	RemoveFromSet(setNames []string, ip, podKey string) error
+	AddToList(listName string, setNames []string) error
+	RemoveFromList(listName string, setNames []string) error
+	UpdatePod(pod *npm.NpmPod) error
+	ApplyDataPlane() error
+	AddPolicy(policies *policies.NPMNetworkPolicy) error
+	RemovePolicy(policyName string) error
+	UpdatePolicy(policies *policies.NPMNetworkPolicy) error
+}