diff --git a/pkg/kubelet/cm/deviceplugin/manager_test.go b/pkg/kubelet/cm/deviceplugin/manager_test.go index 71d7324e46e..f25c1e82166 100644 --- a/pkg/kubelet/cm/deviceplugin/manager_test.go +++ b/pkg/kubelet/cm/deviceplugin/manager_test.go @@ -19,6 +19,8 @@ package deviceplugin import ( "flag" "fmt" + "io/ioutil" + "os" "reflect" "sync/atomic" "testing" @@ -262,7 +264,12 @@ func TestCheckpoint(t *testing.T) { resourceName1 := "domain1.com/resource1" resourceName2 := "domain2.com/resource2" + as := assert.New(t) + tmpDir, err := ioutil.TempDir("", "checkpoint") + as.Nil(err) + defer os.RemoveAll(tmpDir) testManager := &ManagerImpl{ + socketdir: tmpDir, allDevices: make(map[string]sets.String), allocatedDevices: make(map[string]sets.String), podDevices: make(podDevices), @@ -301,8 +308,7 @@ func TestCheckpoint(t *testing.T) { expectedAllocatedDevices := testManager.podDevices.devices() expectedAllDevices := testManager.allDevices - err := testManager.writeCheckpoint() - as := assert.New(t) + err = testManager.writeCheckpoint() as.Nil(err) testManager.podDevices = make(podDevices) @@ -388,7 +394,11 @@ func TestPodContainerDeviceAllocation(t *testing.T) { nodeInfo := &schedulercache.NodeInfo{} nodeInfo.SetNode(cachedNode) + tmpDir, err := ioutil.TempDir("", "checkpoint") + as.Nil(err) + defer os.RemoveAll(tmpDir) testManager := &ManagerImpl{ + socketdir: tmpDir, callback: monitorCallback, allDevices: make(map[string]sets.String), allocatedDevices: make(map[string]sets.String), @@ -489,7 +499,7 @@ func TestPodContainerDeviceAllocation(t *testing.T) { } podsStub.updateActivePods([]*v1.Pod{pod}) - err := testManager.Allocate(nodeInfo, &lifecycle.PodAdmitAttributes{Pod: pod}) + err = testManager.Allocate(nodeInfo, &lifecycle.PodAdmitAttributes{Pod: pod}) as.Nil(err) runContainerOpts := testManager.GetDeviceRunContainerOptions(pod, &pod.Spec.Containers[0]) as.NotNil(runContainerOpts)