test: fix unit test nil pointer.

Add filesystem to qemu object. Fix mock_hypervisor Signed-off-by: Jose Carlos Venegas Munoz <jose.carlos.venegas.munoz@intel.com>
kata-containers · Oct 2, 2018 · 1f5792e · 1f5792e
1 parent 4697cf3
commit 1f5792e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/virtcontainers/mock_hypervisor.go b/virtcontainers/mock_hypervisor.go
@@ -64,6 +64,9 @@ func (m *mockHypervisor) hotplugAddDevice(devInfo interface{}, devType deviceTyp
 	switch devType {
 	case cpuDev:
 		return m.vCPUs, nil
+	case memoryDev:
+		memdev := devInfo.(*memoryDevice)
+		return memdev.sizeMB, nil
 	}
 	return nil, nil
 }

diff --git a/virtcontainers/qemu_test.go b/virtcontainers/qemu_test.go
@@ -348,8 +348,10 @@ func TestHotplugRemoveMemory(t *testing.T) {
 	assert := assert.New(t)
 
 	qemuConfig := newQemuConfig()
+	fs := &filesystem{}
 	q := &qemu{
-		config: qemuConfig,
+		config:  qemuConfig,
+		storage: fs,
 	}
 
 	_, err := q.hotplugRemoveDevice(&memoryDevice{0, 128}, memoryDev)
@@ -360,8 +362,10 @@ func TestHotplugUnsupportedDeviceType(t *testing.T) {
 	assert := assert.New(t)
 
 	qemuConfig := newQemuConfig()
+	fs := &filesystem{}
 	q := &qemu{
-		config: qemuConfig,
+		config:  qemuConfig,
+		storage: fs,
 	}
 
 	_, err := q.hotplugAddDevice(&memoryDevice{0, 128}, fsDev)