fix: update logs test

Author: tejastn10

logs/logs_test.go

@@ -7,23 +7,30 @@ import (
 )
 
 // Helper function to clean up after each test (if necessary)
-func cleanup() {
-	os.Remove("argus.log") // Delete the argus.log file after tests to avoid issues in subsequent runs
+func cleanup(t *testing.T) {
+	// Remove the log file if it exists
+	os.Remove("output/argus.log")
+
+	// Remove the output folder if it is empty
+	if err := os.RemoveAll("output"); err != nil {
+		// Log an error if the folder removal fails
+		t.Errorf("Failed to remove output folder: %v", err)
+	}
 }
 
 // TestLoggingToFile verifies that log messages are correctly logged to a file.
 func TestLoggingToFile(t *testing.T) {
 	// Clean up from any previous test runs
-	cleanup()
+	cleanup(t)
 
-	// Initialize logger to log to file
+	// Initialize logger to log to file inside the output folder
 	Init(true, false)
 
 	// Log an info message
 	Info("File logging test")
 
-	// Check if the log message is written to the argus.log file
-	fileContent, err := os.ReadFile("argus.log")
+	// Check if the log message is written to the output/argus.log file
+	fileContent, err := os.ReadFile("output/argus.log")
 	if err != nil {
 		t.Fatalf("Failed to read log file: %v", err)
 	}
@@ -34,7 +41,7 @@ func TestLoggingToFile(t *testing.T) {
 	}
 
 	// Clean up
-	cleanup()
+	cleanup(t)
 }
 
 // Helper function to check if a byte slice contains a specific string