From 37ebbf14f91280cb90433d28cdf2dbe730ad3352 Mon Sep 17 00:00:00 2001
From: Yuval Kohavi <yuval.kohavi@gmail.com>
Date: Wed, 17 Apr 2019 18:44:33 -0400
Subject: [PATCH] fix compile

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

diff --git a/pkg/controller/controller_test.go b/pkg/controller/controller_test.go
index 8946773a..28ac9fcd 100644
--- a/pkg/controller/controller_test.go
+++ b/pkg/controller/controller_test.go
@@ -1,6 +1,9 @@
 package controller
 
 import (
+	"sync"
+	"time"
+
 	"github.com/weaveworks/flagger/pkg/apis/flagger/v1alpha3"
 	istiov1alpha1 "github.com/weaveworks/flagger/pkg/apis/istio/common/v1alpha1"
 	istiov1alpha3 "github.com/weaveworks/flagger/pkg/apis/istio/v1alpha3"
@@ -21,8 +24,6 @@ import (
 	"k8s.io/client-go/kubernetes/fake"
 	"k8s.io/client-go/tools/record"
 	"k8s.io/client-go/util/workqueue"
-	"sync"
-	"time"
 )
 
 var (
@@ -82,6 +83,9 @@ func SetupMocks(abtest bool) Mocks {
 	flaggerInformerFactory := informers.NewSharedInformerFactory(flaggerClient, noResyncPeriodFunc())
 	flaggerInformer := flaggerInformerFactory.Flagger().V1alpha3().Canaries()
 
+	// init router
+	rf := router.NewFactory(nil, kubeClient, flaggerClient, logger, flaggerClient)
+
 	ctrl := &Controller{
 		kubeClient:    kubeClient,
 		istioClient:   flaggerClient,
@@ -96,11 +100,10 @@ func SetupMocks(abtest bool) Mocks {
 		deployer:      deployer,
 		observer:      observer,
 		recorder:      metrics.NewRecorder(controllerAgentName, false),
+		routerFactory: rf,
 	}
 	ctrl.flaggerSynced = alwaysReady
 
-	// init router
-	rf := router.NewFactory(kubeClient, flaggerClient, logger, flaggerClient)
 	meshRouter := rf.MeshRouter("istio")
 
 	return Mocks{