Fixed Force() in tests using fake Config

jetstack · Jul 31, 2018 · 6d96c4c · 6d96c4c
1 parent 8da9e77
commit 6d96c4c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/pkg/tarmak/cluster/cluster_test.go b/pkg/tarmak/cluster/cluster_test.go
@@ -60,6 +60,8 @@ func newFakeCluster(t *testing.T, cluster *clusterv1alpha1.Cluster) *fakeCluster
 	c.fakeProvider.EXPECT().Cloud().Return("provider").AnyTimes()
 	c.fakeProvider.EXPECT().Name().Return("provider-name").AnyTimes()
 
+	c.fakeConfig.EXPECT().Force().Return(false).AnyTimes()
+
 	c.fakeTarmak.EXPECT().Config().AnyTimes().Return(c.fakeConfig)
 
 	return c

diff --git a/pkg/tarmak/tarmak_test.go b/pkg/tarmak/tarmak_test.go
@@ -64,6 +64,7 @@ func (tt *testTarmak) fakeAWSProvider(name string) {
 	}
 
 	tt.fakeConfig.EXPECT().Provider(name).AnyTimes().Return(&tarmakv1alpha1.Provider{}, nil)
+	tt.fakeConfig.EXPECT().Force().Return(false).AnyTimes()
 }
 
 func (tt *testTarmak) addEnvironment(env *tarmakv1alpha1.Environment) {
-Original file line number
+Diff line change
@@ Expand Up / @@ -64,6 +64,7 @@ func (tt *testTarmak) fakeAWSProvider(name string) { @@
     	}
     	tt.fakeConfig.EXPECT().Provider(name).AnyTimes().Return(&tarmakv1alpha1.Provider{}, nil)
+    	tt.fakeConfig.EXPECT().Force().Return(false).AnyTimes()
     }
     func (tt *testTarmak) addEnvironment(env *tarmakv1alpha1.Environment) {
@@ Expand Down @@