]> git.maquefel.me Git - brevno-suite/hugo/commitdiff
helpers: simplify path tests with T.TempDir
authorOleksandr Redko <Oleksandr_Redko@epam.com>
Mon, 15 May 2023 08:52:06 +0000 (11:52 +0300)
committerBjørn Erik Pedersen <bjorn.erik.pedersen@gmail.com>
Tue, 16 May 2023 16:35:19 +0000 (18:35 +0200)
helpers/path_test.go

index 85081c5bec96d86961405fecad7dd9f4d1d94fa8..45b69292382f4828cfd553c59c7a4991584ed990 100644 (file)
@@ -239,42 +239,35 @@ func TestIsDir(t *testing.T) {
        }
 }
 
-func createZeroSizedFileInTempDir() (*os.File, error) {
+func createZeroSizedFileInTempDir(t *testing.T) *os.File {
+       t.Helper()
+
        filePrefix := "_path_test_"
-       f, e := os.CreateTemp("", filePrefix) // dir is os.TempDir()
-       if e != nil {
-               // if there was an error no file was created.
-               // => no requirement to delete the file
-               return nil, e
+       f, err := os.CreateTemp(t.TempDir(), filePrefix)
+       if err != nil {
+               t.Error(err)
        }
-       return f, nil
+       if err := f.Close(); err != nil {
+               t.Error(err)
+       }
+       return f
 }
 
-func createNonZeroSizedFileInTempDir() (*os.File, error) {
-       f, err := createZeroSizedFileInTempDir()
-       if err != nil {
-               // no file ??
-               return nil, err
-       }
+func createNonZeroSizedFileInTempDir(t *testing.T) *os.File {
+       t.Helper()
+
+       f := createZeroSizedFileInTempDir(t)
        byteString := []byte("byteString")
-       err = os.WriteFile(f.Name(), byteString, 0644)
+       err := os.WriteFile(f.Name(), byteString, 0644)
        if err != nil {
-               // delete the file
-               deleteFileInTempDir(f)
-               return nil, err
+               t.Error(err)
        }
-       return f, nil
-}
-
-func deleteFileInTempDir(f *os.File) {
-       _ = os.Remove(f.Name())
+       return f
 }
 
 func TestExists(t *testing.T) {
-       zeroSizedFile, _ := createZeroSizedFileInTempDir()
-       defer deleteFileInTempDir(zeroSizedFile)
-       nonZeroSizedFile, _ := createNonZeroSizedFileInTempDir()
-       defer deleteFileInTempDir(nonZeroSizedFile)
+       zeroSizedFile := createZeroSizedFileInTempDir(t)
+       nonZeroSizedFile := createNonZeroSizedFileInTempDir(t)
        emptyDirectory := t.TempDir()
        nonExistentFile := os.TempDir() + "/this-file-does-not-exist.txt"
        nonExistentDir := os.TempDir() + "/this/directory/does/not/exist/"
@@ -420,8 +413,7 @@ func TestFindCWD(t *testing.T) {
 }
 
 func TestSafeWriteToDisk(t *testing.T) {
-       emptyFile, _ := createZeroSizedFileInTempDir()
-       defer deleteFileInTempDir(emptyFile)
+       emptyFile := createZeroSizedFileInTempDir(t)
        tmpDir := t.TempDir()
 
        randomString := "This is a random string!"
@@ -461,8 +453,7 @@ func TestSafeWriteToDisk(t *testing.T) {
 }
 
 func TestWriteToDisk(t *testing.T) {
-       emptyFile, _ := createZeroSizedFileInTempDir()
-       defer deleteFileInTempDir(emptyFile)
+       emptyFile := createZeroSizedFileInTempDir(t)
        tmpDir := t.TempDir()
 
        randomString := "This is a random string!"