pkg/fileutils: add unit test for IsRegularFile

Signed-off-by: Zhou Hao <[email protected]>
dragonflyoss · Feb 14, 2020 · e33b08f · e33b08f
1 parent 1481da3
commit e33b08f
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/pkg/fileutils/fileutils_test.go b/pkg/fileutils/fileutils_test.go
@@ -276,3 +276,17 @@ func (s *FileUtilTestSuite) TestLoadYaml(c *check.C) {
 
 	}
 }
+
+func (s *FileUtilTestSuite) TestIsRegularFile(c *check.C) {
+	pathStr := filepath.Join(s.tmpDir, "TestIsRegularFile")
+	c.Assert(IsRegularFile(pathStr), check.Equals, false)
+
+	os.Create(pathStr)
+	c.Assert(IsRegularFile(pathStr), check.Equals, true)
+	os.Remove(pathStr)
+
+	// Don't set mode to create a non-regular file
+	os.OpenFile(pathStr, 0, 0666)
+	c.Assert(IsRegularFile(pathStr), check.Equals, false)
+	os.Remove(pathStr)
+}