agent: refine unit test for func GetGuestDetails

since we add extra functionality in func GetGuestDetails for memory hotplug via probe interface, we need to refine its related unit test. Fixes: #442 Signed-off-by: Penny Zheng <[email protected]>
kata-containers · Jan 23, 2019 · ee5b506 · ee5b506
1 parent a6f156c
commit ee5b506
Showing 1 changed file with 27 additions and 0 deletions.
diff --git a/grpc_test.go b/grpc_test.go
@@ -665,11 +665,18 @@ func TestGetGuestDetails(t *testing.T) {
 
 	req := &pb.GuestDetailsRequest{
 		MemBlockSize: true,
+		MemHotplugProbe: true,
 	}
 
 	// sysfsMemoryBlockSizePath exist with error format
 	file, err := ioutil.TempFile("", "test")
 	assert.NoError(err)
+
+	oldsysfsMemoryBlockSizePath := sysfsMemoryBlockSizePath
+	defer func() {
+		sysfsMemoryBlockSizePath = oldsysfsMemoryBlockSizePath
+	}()
+
 	sysfsMemoryBlockSizePath = file.Name()
 	// empty
 	_, err = a.GetGuestDetails(context.TODO(), req)
@@ -699,6 +706,26 @@ func TestGetGuestDetails(t *testing.T) {
 	resp, err = a.GetGuestDetails(context.TODO(), req)
 	assert.NoError(err)
 	assert.Equal(resp.MemBlockSizeBytes, uint64(0))
+
+	// sysfsMemoryHotplugProbePath exist
+	probeFile, err := ioutil.TempFile("", "probe")
+        assert.NoError(err)
+
+	oldSysfsMemoryHotplugProbePath := sysfsMemoryHotplugProbePath
+	defer func() {
+		sysfsMemoryHotplugProbePath = oldSysfsMemoryHotplugProbePath
+	}()
+
+	sysfsMemoryHotplugProbePath = probeFile.Name()
+	resp, err = a.GetGuestDetails(context.TODO(), req)
+	assert.NoError(err)
+	assert.Equal(resp.SupportMemHotplugProbe, true)
+
+	// sysfsMemoryHotplugProbePath does not exist
+	os.Remove(sysfsMemoryHotplugProbePath)
+	resp, err = a.GetGuestDetails(context.TODO(), req)
+	assert.NoError(err)
+	assert.Equal(resp.SupportMemHotplugProbe, false)
 }
 
 func TestGetAgentDetails(t *testing.T) {