From 05a829715f99cd90b122c64cd2f40157854e467b Mon Sep 17 00:00:00 2001
From: Nikita Pivkin <nikita.pivkin@smartforce.io>
Date: Wed, 21 Aug 2024 08:37:42 +0600
Subject: [PATCH] fix: safely check if the directory exists (#7353)

Signed-off-by: nikpivkin <nikita.pivkin@smartforce.io>
---
 pkg/utils/fsutils/fs.go      | 14 +++-----
 pkg/utils/fsutils/fs_test.go | 62 ++++++++++++++++++++++++++----------
 2 files changed, 49 insertions(+), 27 deletions(-)

diff --git a/pkg/utils/fsutils/fs.go b/pkg/utils/fsutils/fs.go
index e0518236f9a8..b2aea51dda0a 100644
--- a/pkg/utils/fsutils/fs.go
+++ b/pkg/utils/fsutils/fs.go
@@ -1,7 +1,6 @@
 package fsutils
 
 import (
-	"errors"
 	"fmt"
 	"io"
 	"io/fs"
@@ -59,18 +58,13 @@ func CopyFile(src, dst string) (int64, error) {
 }
 
 func DirExists(path string) bool {
-	if f, err := os.Stat(path); os.IsNotExist(err) || !f.IsDir() {
-		return false
-	}
-	return true
+	f, err := os.Stat(path)
+	return err == nil && f.IsDir()
 }
 
 func FileExists(filename string) bool {
-	_, err := os.Stat(filename)
-	if errors.Is(err, os.ErrNotExist) {
-		return false
-	}
-	return err == nil
+	f, err := os.Stat(filename)
+	return err == nil && !f.IsDir()
 }
 
 type WalkDirRequiredFunc func(path string, d fs.DirEntry) bool
diff --git a/pkg/utils/fsutils/fs_test.go b/pkg/utils/fsutils/fs_test.go
index fef4b7f5a32c..42a4b113a5ef 100644
--- a/pkg/utils/fsutils/fs_test.go
+++ b/pkg/utils/fsutils/fs_test.go
@@ -2,29 +2,13 @@ package fsutils
 
 import (
 	"os"
+	"path/filepath"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
 
-func touch(t *testing.T, name string) {
-	f, err := os.Create(name)
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := f.Close(); err != nil {
-		t.Fatal(err)
-	}
-}
-
-func write(t *testing.T, name, content string) {
-	err := os.WriteFile(name, []byte(content), 0666)
-	if err != nil {
-		t.Fatal(err)
-	}
-}
-
 func TestCopyFile(t *testing.T) {
 	type args struct {
 		src string
@@ -72,3 +56,47 @@ func TestCopyFile(t *testing.T) {
 		})
 	}
 }
+
+func TestDirExists(t *testing.T) {
+	t.Run("invalid path", func(t *testing.T) {
+		assert.False(t, DirExists("\000invalid:path"))
+	})
+
+	t.Run("valid path", func(t *testing.T) {
+		assert.True(t, DirExists(t.TempDir()))
+	})
+
+	t.Run("dir not exist", func(t *testing.T) {
+		assert.False(t, DirExists(filepath.Join(t.TempDir(), "tmp")))
+	})
+
+	t.Run("file path", func(t *testing.T) {
+		filePath := filepath.Join(t.TempDir(), "tmp")
+		f, err := os.Create(filePath)
+		require.NoError(t, f.Close())
+		require.NoError(t, err)
+		assert.False(t, DirExists(filePath))
+	})
+}
+
+func TestFileExists(t *testing.T) {
+	t.Run("invalid path", func(t *testing.T) {
+		assert.False(t, FileExists("\000invalid:path"))
+	})
+
+	t.Run("valid path", func(t *testing.T) {
+		filePath := filepath.Join(t.TempDir(), "tmp")
+		f, err := os.Create(filePath)
+		require.NoError(t, f.Close())
+		require.NoError(t, err)
+		assert.True(t, FileExists(filePath))
+	})
+
+	t.Run("file not exist", func(t *testing.T) {
+		assert.False(t, FileExists(filepath.Join(t.TempDir(), "tmp")))
+	})
+
+	t.Run("dir path", func(t *testing.T) {
+		assert.False(t, FileExists(t.TempDir()))
+	})
+}