add TestCleanDirectory

milwad-dev · Sep 24, 2024 · c4b8787 · c4b8787
1 parent ac75e18
commit c4b8787
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/filei_test.go b/filei_test.go
@@ -68,3 +68,24 @@ func TestExistsFile(t *testing.T) {
 		t.Error("Exists() error data must be false")
 	}
 }
+
+func TestCleanDirectory(t *testing.T) {
+	directory := "testdata/dir"
+
+	defer os.Create(directory + "/data.txt")
+	defer os.Create(directory + "/item.txt")
+
+	dirData, _ := os.ReadDir(directory)
+
+	if len(dirData) != 2 {
+		t.Fatal("CleanDirectory() error data must have 2 files")
+	}
+
+	CleanDirectory(directory)
+
+	dirData, _ = os.ReadDir(directory)
+
+	if len(dirData) == 2 {
+		t.Fatal("CleanDirectory() error data must have 0 files")
+	}
+}