LFG

internal/logger/logger_test.go

@@ -0,0 +1,185 @@
+package logger
+
+import (
+	"bytes"
+	"log"
+	"os"
+	"strings"
+	"testing"
+)
+
+func TestNewLogger(t *testing.T) {
+	logger := NewLogger()
+	if logger == nil {
+		t.Error("Expected logger to be created, got nil")
+	}
+}
+
+func TestLoggerDebug(t *testing.T) {
+	// Create a buffer to capture log output
+	var buf bytes.Buffer
+	log.SetOutput(&buf)
+	defer func() {
+		log.SetOutput(os.Stderr)
+	}()
+
+	logger := NewLogger()
+	logger.Debug("Debug message", String("key", "value"))
+
+	output := buf.String()
+	if !strings.Contains(output, "DEBUG: Debug message") {
+		t.Errorf("Expected log output to contain 'DEBUG: Debug message', got %s", output)
+	}
+	if !strings.Contains(output, "key=value") {
+		t.Errorf("Expected log output to contain 'key=value', got %s", output)
+	}
+}
+
+func TestLoggerInfo(t *testing.T) {
+	// Create a buffer to capture log output
+	var buf bytes.Buffer
+	log.SetOutput(&buf)
+	defer func() {
+		log.SetOutput(os.Stderr)
+	}()
+
+	logger := NewLogger()
+	logger.Info("Info message", Int("number", 42))
+
+	output := buf.String()
+	if !strings.Contains(output, "INFO: Info message") {
+		t.Errorf("Expected log output to contain 'INFO: Info message', got %s", output)
+	}
+	if !strings.Contains(output, "number=42") {
+		t.Errorf("Expected log output to contain 'number=42', got %s", output)
+	}
+}
+
+func TestLoggerWarn(t *testing.T) {
+	// Create a buffer to capture log output
+	var buf bytes.Buffer
+	log.SetOutput(&buf)
+	defer func() {
+		log.SetOutput(os.Stderr)
+	}()
+
+	logger := NewLogger()
+	logger.Warn("Warning message", Bool("flag", true))
+
+	output := buf.String()
+	if !strings.Contains(output, "WARN: Warning message") {
+		t.Errorf("Expected log output to contain 'WARN: Warning message', got %s", output)
+	}
+	if !strings.Contains(output, "flag=true") {
+		t.Errorf("Expected log output to contain 'flag=true', got %s", output)
+	}
+}
+
+func TestLoggerError(t *testing.T) {
+	// Create a buffer to capture log output
+	var buf bytes.Buffer
+	log.SetOutput(&buf)
+	defer func() {
+		log.SetOutput(os.Stderr)
+	}()
+
+	logger := NewLogger()
+	err := os.ErrNotExist
+	logger.Error("Error message", Error(err))
+
+	output := buf.String()
+	if !strings.Contains(output, "ERROR: Error message") {
+		t.Errorf("Expected log output to contain 'ERROR: Error message', got %s", output)
+	}
+	if !strings.Contains(output, "error=file does not exist") {
+		t.Errorf("Expected log output to contain 'error=file does not exist', got %s", output)
+	}
+}
+
+func TestLoggerFatal(t *testing.T) {
+	// Create a buffer to capture log output
+	var buf bytes.Buffer
+	log.SetOutput(&buf)
+	defer func() {
+		log.SetOutput(os.Stderr)
+	}()
+
+	// Mock os.Exit to prevent the test from exiting
+	exitCalled := false
+	exitFunc := func(code int) {
+		exitCalled = true
+	}
+	osExit = exitFunc
+	defer func() {
+		osExit = realOsExit
+	}()
+
+	logger := NewLogger()
+	logger.Fatal("Fatal message", String("reason", "testing"))
+
+	output := buf.String()
+	if !strings.Contains(output, "ERROR: Fatal message") {
+		t.Errorf("Expected log output to contain 'ERROR: Fatal message', got %s", output)
+	}
+	if !strings.Contains(output, "reason=testing") {
+		t.Errorf("Expected log output to contain 'reason=testing', got %s", output)
+	}
+	if !exitCalled {
+		t.Error("Expected os.Exit to be called")
+	}
+}
+
+func TestLoggerMultipleFields(t *testing.T) {
+	// Create a buffer to capture log output
+	var buf bytes.Buffer
+	log.SetOutput(&buf)
+	defer func() {
+		log.SetOutput(os.Stderr)
+	}()
+
+	logger := NewLogger()
+	logger.Info("Multiple fields", 
+		String("string", "value"),
+		Int("int", 123),
+		Bool("bool", false))
+
+	output := buf.String()
+	if !strings.Contains(output, "INFO: Multiple fields") {
+		t.Errorf("Expected log output to contain 'INFO: Multiple fields', got %s", output)
+	}
+	if !strings.Contains(output, "string=value") {
+		t.Errorf("Expected log output to contain 'string=value', got %s", output)
+	}
+	if !strings.Contains(output, "int=123") {
+		t.Errorf("Expected log output to contain 'int=123', got %s", output)
+	}
+	if !strings.Contains(output, "bool=false") {
+		t.Errorf("Expected log output to contain 'bool=false', got %s", output)
+	}
+}
+
+func TestLoggerNoFields(t *testing.T) {
+	// Create a buffer to capture log output
+	var buf bytes.Buffer
+	log.SetOutput(&buf)
+	defer func() {
+		log.SetOutput(os.Stderr)
+	}()
+
+	logger := NewLogger()
+	logger.Info("No fields")
+
+	output := buf.String()
+	if !strings.Contains(output, "INFO: No fields") {
+		t.Errorf("Expected log output to contain 'INFO: No fields', got %s", output)
+	}
+	if strings.Contains(output, "[") {
+		t.Error("Expected log output to not contain field brackets when no fields are provided")
+	}
+}
+
+// Mock os.Exit for testing
+var (
+	osExit    = func(code int) { os.Exit(code) }
+	realOsExit = osExit
+)