Fix function calls in unit tests

Signed-off-by: Stu Gott <[email protected]>
attlee-wang · Feb 25, 2020 · 95f4705 · 95f4705
1 parent 5d9a2f0
commit 95f4705
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/pkg/virt-controller/services/template_test.go b/pkg/virt-controller/services/template_test.go
@@ -48,6 +48,7 @@ import (
 const namespaceKubevirt = "kubevirt"
 
 var _ = Describe("Template", func() {
+	var qemuGid int64 = 107
 
 	log.Log.SetIOWriter(GinkgoWriter)
 
@@ -77,6 +78,7 @@ var _ = Describe("Template", func() {
 			pvcCache,
 			virtClient,
 			config,
+			qemuGid,
 		)
 
 		// Set up mock clients

diff --git a/pkg/virt-controller/watch/migration_test.go b/pkg/virt-controller/watch/migration_test.go
@@ -66,6 +66,7 @@ var _ = Describe("Migration watcher", func() {
 	var kubeClient *fake.Clientset
 	var networkClient *fakenetworkclient.Clientset
 	var pvcInformer cache.SharedIndexInformer
+	var qemuGid int64 = 107
 
 	shouldExpectPodCreation := func(uid types.UID, migrationUid types.UID, expectedAntiAffinityCount int, expectedAffinityCount int, expectedNodeAffinityCount int) {
 		// Expect pod creation
@@ -165,7 +166,7 @@ var _ = Describe("Migration watcher", func() {
 		config, _, _ := testutils.NewFakeClusterConfig(&k8sv1.ConfigMap{})
 
 		controller = NewMigrationController(
-			services.NewTemplateService("a", "b", "c", "d", "e", "f", pvcInformer.GetStore(), virtClient, config),
+			services.NewTemplateService("a", "b", "c", "d", "e", "f", pvcInformer.GetStore(), virtClient, config, qemuGid),
 			vmiInformer,
 			podInformer,
 			migrationInformer,

diff --git a/pkg/virt-controller/watch/vmi_test.go b/pkg/virt-controller/watch/vmi_test.go
@@ -70,6 +70,7 @@ var _ = Describe("VirtualMachineInstance watcher", func() {
 	var dataVolumeSource *framework.FakeControllerSource
 	var dataVolumeInformer cache.SharedIndexInformer
 	var dataVolumeFeeder *testutils.DataVolumeFeeder
+	var qemuGid int64 = 107
 
 	shouldExpectPodCreation := func(uid types.UID) {
 		// Expect pod creation
@@ -165,12 +166,13 @@ var _ = Describe("VirtualMachineInstance watcher", func() {
 		config, _, _ := testutils.NewFakeClusterConfig(&k8sv1.ConfigMap{})
 		pvcInformer, _ = testutils.NewFakeInformerFor(&k8sv1.PersistentVolumeClaim{})
 		controller = NewVMIController(
-			services.NewTemplateService("a", "b", "c", "d", "e", "f", pvcInformer.GetStore(), virtClient, config),
+			services.NewTemplateService("a", "b", "c", "d", "e", "f", pvcInformer.GetStore(), virtClient, config, qemuGid),
 			vmiInformer,
 			podInformer,
 			recorder,
 			virtClient,
-			dataVolumeInformer)
+			dataVolumeInformer,
+		)
 		// Wrap our workqueue to have a way to detect when we are done processing updates
 		mockQueue = testutils.NewMockWorkQueue(controller.Queue)
 		controller.Queue = mockQueue