Fix router tests

fluxcd · Mar 17, 2019 · 7b41531 · 7b41531
1 parent 7d340c5
commit 7b41531
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/pkg/controller/controller_test.go b/pkg/controller/controller_test.go
@@ -31,7 +31,7 @@ var (
 type Mocks struct {
 	canary        *v1alpha3.Canary
 	kubeClient    kubernetes.Interface
-	istioClient   clientset.Interface
+	meshClient    clientset.Interface
 	flaggerClient clientset.Interface
 	deployer      CanaryDeployer
 	observer      CanaryObserver
@@ -100,16 +100,15 @@ func SetupMocks(abtest bool) Mocks {
 
 	// init router
 	rf := router.NewFactory(kubeClient, flaggerClient, logger, flaggerClient)
-	var meshRouter router.Interface
-	meshRouter = rf.IstioRouter()
+	meshRouter := rf.MeshRouter("istio")
 
 	return Mocks{
 		canary:        canary,
 		observer:      observer,
 		deployer:      deployer,
 		logger:        logger,
 		flaggerClient: flaggerClient,
-		istioClient:   flaggerClient,
+		meshClient:    flaggerClient,
 		kubeClient:    kubeClient,
 		ctrl:          ctrl,
 		router:        meshRouter,