浏览代码

Fix the unit test not to remove /tmp

Signed-off-by: Yohei Ueda <yohei@jp.ibm.com>
Yohei Ueda 10 年之前
父节点
当前提交
2329354dc3
共有 1 个文件被更改,包括 6 次插入5 次删除
  1. 6 5
      pkg/parsers/operatingsystem/operatingsystem_test.go

+ 6 - 5
pkg/parsers/operatingsystem/operatingsystem_test.go

@@ -38,12 +38,13 @@ BUG_REPORT_URL="http://bugs.launchpad.net/ubuntu/"`)
 	)
 
 	dir := os.TempDir()
+	etcOsRelease = filepath.Join(dir, "etcOsRelease")
+
 	defer func() {
+		os.Remove(etcOsRelease)
 		etcOsRelease = backup
-		os.RemoveAll(dir)
 	}()
 
-	etcOsRelease = filepath.Join(dir, "etcOsRelease")
 	for expect, osRelease := range map[string][]byte{
 		"Ubuntu 14.04 LTS": ubuntuTrusty,
 		"Gentoo/Linux":     gentoo,
@@ -92,13 +93,13 @@ func TestIsContainerized(t *testing.T) {
 	)
 
 	dir := os.TempDir()
+	proc1Cgroup = filepath.Join(dir, "proc1Cgroup")
+
 	defer func() {
+		os.Remove(proc1Cgroup)
 		proc1Cgroup = backup
-		os.RemoveAll(dir)
 	}()
 
-	proc1Cgroup = filepath.Join(dir, "proc1Cgroup")
-
 	if err := ioutil.WriteFile(proc1Cgroup, nonContainerizedProc1Cgroup, 0600); err != nil {
 		t.Fatalf("failed to write to %s: %v", proc1Cgroup, err)
 	}