From c4b8787898788525f114b9ebda17fedfbe56faa8 Mon Sep 17 00:00:00 2001
From: Milwad <milwad.dev@gmail.com>
Date: Tue, 24 Sep 2024 17:30:01 +0330
Subject: [PATCH] add `TestCleanDirectory`

---
 filei_test.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/filei_test.go b/filei_test.go
index fe9e175..251ac4e 100644
--- 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")
+	}
+}