chore: add fake antrea client.

Signed-off-by: Asklv <[email protected]>
antrea-io · Jun 15, 2024 · ec414a1 · ec414a1
1 parent 4eae08c
commit ec414a1
Show file tree

Hide file tree

Showing 3 changed files with 75 additions and 3 deletions.
diff --git a/hack/update-codegen-dockerized.sh b/hack/update-codegen-dockerized.sh
@@ -43,6 +43,7 @@ MOCKGEN_TARGETS=(
   "pkg/agent/interfacestore InterfaceStore testing"
   "pkg/agent/memberlist Interface testing"
   "pkg/agent/memberlist Memberlist ."
+  "pkg/agent/client AntreaClientProvider testing"
   "pkg/agent/multicast RouteInterface testing"
   "pkg/agent/types McastNetworkPolicyController testing"
   "pkg/agent/monitortool PacketListener testing"

diff --git a/pkg/agent/client/testing/mock_client.go b/pkg/agent/client/testing/mock_client.go
diff --git a/pkg/agent/monitortool/monitor_test.go b/pkg/agent/monitortool/monitor_test.go
@@ -33,16 +33,16 @@ import (
 	"k8s.io/apimachinery/pkg/runtime"
 	"k8s.io/client-go/informers"
 	"k8s.io/client-go/kubernetes/fake"
-	componentbaseconfig "k8s.io/component-base/config"
 	"k8s.io/utils/clock"
 	clocktesting "k8s.io/utils/clock/testing"
 
-	"antrea.io/antrea/pkg/agent/client"
+	clienttesting "antrea.io/antrea/pkg/agent/client/testing"
 	"antrea.io/antrea/pkg/agent/config"
 	monitortesting "antrea.io/antrea/pkg/agent/monitortool/testing"
 	"antrea.io/antrea/pkg/agent/util/nettest"
 	crdv1alpha1 "antrea.io/antrea/pkg/apis/crd/v1alpha1"
 	fakeversioned "antrea.io/antrea/pkg/client/clientset/versioned/fake"
+	statsfake "antrea.io/antrea/pkg/client/clientset/versioned/fake"
 	crdinformers "antrea.io/antrea/pkg/client/informers/externalversions"
 	"antrea.io/antrea/pkg/util/ip"
 )
@@ -157,12 +157,14 @@ func newTestMonitor(
 ) *testMonitor {
 	ctrl := gomock.NewController(t)
 	clientset := fake.NewSimpleClientset(objects...)
+	statsClientset := statsfake.NewSimpleClientset(objects...)
 	informerFactory := informers.NewSharedInformerFactory(clientset, 0)
 	nodeInformer := informerFactory.Core().V1().Nodes()
 	crdClientset := fakeversioned.NewSimpleClientset(crdObjects...)
 	crdInformerFactory := crdinformers.NewSharedInformerFactory(crdClientset, 0)
 	nlmInformer := crdInformerFactory.Crd().V1alpha1().NodeLatencyMonitors()
-	antreaClientProvider, _ := client.NewAntreaClientProvider(componentbaseconfig.ClientConnectionConfiguration{}, clientset)
+	antreaClientProvider := clienttesting.NewMockAntreaClientProvider(ctrl)
+	antreaClientProvider.EXPECT().GetAntreaClient().Return(statsClientset, nil).AnyTimes()
 	m := NewNodeLatencyMonitor(antreaClientProvider, nodeInformer, nlmInformer, nodeConfig, trafficEncapMode)
 	fakeClock := newFakeClock(clockT)
 	m.clock = fakeClock