test: Add a test for os.Exit to the logger

cmd/logger/main.go

@@ -8,6 +8,8 @@ import (
 	"github.com/Dobefu/csb/cmd/color"
 )
 
+var osExit = os.Exit
+
 var (
 	LOG_VERBOSE byte = 0
 	LOG_INFO    byte = 1
@@ -70,7 +72,7 @@ func Fatal(format string, a ...any) string {
 	output := logMessage(LOG_FATAL, format, a...)
 
 	if EXIT_ON_FATAL {
-		os.Exit(1)
+		osExit(1)
 	}
 
 	return output

cmd/logger/main_test.go

@@ -1,6 +1,7 @@
 package logger
 
 import (
+	"os"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -85,3 +86,17 @@ func TestLogFatal(t *testing.T) {
 	SetLogLevel(LOG_FATAL)
 	assert.NotEmpty(t, Fatal("test"))
 }
+
+func TestLogFatalWithExit(t *testing.T) {
+	SetExitOnFatal(true)
+	isExitCalled := false
+
+	osExit = func(code int) {
+		isExitCalled = true
+	}
+
+	defer func() { osExit = os.Exit }()
+
+	assert.NotEmpty(t, Fatal("test"))
+	assert.True(t, isExitCalled, "os.Exit should have been called")
+}