From 7b4153113e79fe4764ed2ce056c8ebd998c251c3 Mon Sep 17 00:00:00 2001
From: stefanprodan <stefan.prodan@gmail.com>
Date: Sun, 17 Mar 2019 11:14:47 +0200
Subject: [PATCH] Fix router tests

---
 pkg/controller/controller_test.go | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/pkg/controller/controller_test.go b/pkg/controller/controller_test.go
index deacdeca..b70439a8 100644
--- 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,8 +100,7 @@ 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,
@@ -109,7 +108,7 @@ func SetupMocks(abtest bool) Mocks {
 		deployer:      deployer,
 		logger:        logger,
 		flaggerClient: flaggerClient,
-		istioClient:   flaggerClient,
+		meshClient:    flaggerClient,
 		kubeClient:    kubeClient,
 		ctrl:          ctrl,
 		router:        meshRouter,