Add unit tests for various packages including config, buffer, debug, pathfinder, rate, and resolver.

pkg/debug/debug_test.go

@@ -0,0 +1,185 @@
+package debug
+
+import (
+	"flag"
+	"testing"
+)
+
+func TestInit(t *testing.T) {
+	originalFlag := flag.CommandLine
+	defer func() {
+		flag.CommandLine = originalFlag
+		initialized = false
+	}()
+
+	flag.CommandLine = flag.NewFlagSet("test", flag.ContinueOnError)
+	debugLevel = flag.Int("debug", 3, "debug level")
+
+	Init()
+
+	if !initialized {
+		t.Error("Init() should set initialized to true")
+	}
+
+	if GetLogger() == nil {
+		t.Error("GetLogger() should return non-nil logger after Init()")
+	}
+}
+
+func TestGetLogger(t *testing.T) {
+	originalFlag := flag.CommandLine
+	defer func() {
+		flag.CommandLine = originalFlag
+		initialized = false
+	}()
+
+	flag.CommandLine = flag.NewFlagSet("test", flag.ContinueOnError)
+	debugLevel = flag.Int("debug", 3, "debug level")
+	initialized = false
+
+	logger := GetLogger()
+	if logger == nil {
+		t.Error("GetLogger() should return non-nil logger")
+	}
+
+	if !initialized {
+		t.Error("GetLogger() should initialize if not already initialized")
+	}
+}
+
+func TestLog(t *testing.T) {
+	originalFlag := flag.CommandLine
+	defer func() {
+		flag.CommandLine = originalFlag
+		initialized = false
+	}()
+
+	flag.CommandLine = flag.NewFlagSet("test", flag.ContinueOnError)
+	debugLevel = flag.Int("debug", 7, "debug level")
+	initialized = false
+
+	Log(DEBUG_INFO, "test message", "key", "value")
+}
+
+func TestSetDebugLevel(t *testing.T) {
+	originalFlag := flag.CommandLine
+	defer func() {
+		flag.CommandLine = originalFlag
+		initialized = false
+	}()
+
+	flag.CommandLine = flag.NewFlagSet("test", flag.ContinueOnError)
+	debugLevel = flag.Int("debug", 3, "debug level")
+	initialized = false
+
+	SetDebugLevel(5)
+	if GetDebugLevel() != 5 {
+		t.Errorf("SetDebugLevel(5) did not set level correctly, got %d", GetDebugLevel())
+	}
+}
+
+func TestGetDebugLevel(t *testing.T) {
+	originalFlag := flag.CommandLine
+	defer func() {
+		flag.CommandLine = originalFlag
+		initialized = false
+	}()
+
+	flag.CommandLine = flag.NewFlagSet("test", flag.ContinueOnError)
+	debugLevel = flag.Int("debug", 4, "debug level")
+
+	level := GetDebugLevel()
+	if level != 4 {
+		t.Errorf("GetDebugLevel() = %d, want 4", level)
+	}
+}
+
+func TestLog_LevelFiltering(t *testing.T) {
+	originalFlag := flag.CommandLine
+	defer func() {
+		flag.CommandLine = originalFlag
+		initialized = false
+	}()
+
+	flag.CommandLine = flag.NewFlagSet("test", flag.ContinueOnError)
+	debugLevel = flag.Int("debug", 3, "debug level")
+	initialized = false
+
+	Log(DEBUG_TRACE, "trace message")
+	Log(DEBUG_INFO, "info message")
+	Log(DEBUG_ERROR, "error message")
+}
+
+func TestConstants(t *testing.T) {
+	if DEBUG_CRITICAL != 1 {
+		t.Errorf("DEBUG_CRITICAL = %d, want 1", DEBUG_CRITICAL)
+	}
+	if DEBUG_ERROR != 2 {
+		t.Errorf("DEBUG_ERROR = %d, want 2", DEBUG_ERROR)
+	}
+	if DEBUG_INFO != 3 {
+		t.Errorf("DEBUG_INFO = %d, want 3", DEBUG_INFO)
+	}
+	if DEBUG_VERBOSE != 4 {
+		t.Errorf("DEBUG_VERBOSE = %d, want 4", DEBUG_VERBOSE)
+	}
+	if DEBUG_TRACE != 5 {
+		t.Errorf("DEBUG_TRACE = %d, want 5", DEBUG_TRACE)
+	}
+	if DEBUG_PACKETS != 6 {
+		t.Errorf("DEBUG_PACKETS = %d, want 6", DEBUG_PACKETS)
+	}
+	if DEBUG_ALL != 7 {
+		t.Errorf("DEBUG_ALL = %d, want 7", DEBUG_ALL)
+	}
+}
+
+func TestLog_WithArgs(t *testing.T) {
+	originalFlag := flag.CommandLine
+	defer func() {
+		flag.CommandLine = originalFlag
+		initialized = false
+	}()
+
+	flag.CommandLine = flag.NewFlagSet("test", flag.ContinueOnError)
+	debugLevel = flag.Int("debug", 7, "debug level")
+	initialized = false
+
+	Log(DEBUG_INFO, "test message", "key1", "value1", "key2", "value2")
+}
+
+func TestInit_MultipleCalls(t *testing.T) {
+	originalFlag := flag.CommandLine
+	defer func() {
+		flag.CommandLine = originalFlag
+		initialized = false
+	}()
+
+	flag.CommandLine = flag.NewFlagSet("test", flag.ContinueOnError)
+	debugLevel = flag.Int("debug", 3, "debug level")
+	initialized = false
+
+	Init()
+	firstLogger := GetLogger()
+
+	Init()
+	secondLogger := GetLogger()
+
+	if firstLogger != secondLogger {
+		t.Error("Multiple Init() calls should not create new loggers")
+	}
+}
+
+func TestLog_DisabledLevel(t *testing.T) {
+	originalFlag := flag.CommandLine
+	defer func() {
+		flag.CommandLine = originalFlag
+		initialized = false
+	}()
+
+	flag.CommandLine = flag.NewFlagSet("test", flag.ContinueOnError)
+	debugLevel = flag.Int("debug", 1, "debug level")
+	initialized = false
+
+	Log(DEBUG_TRACE, "this should be filtered")
+}