replicate/internal/log_test.go

package internal_test

import (
	"bytes"
	"log/slog"
	"strings"
	"testing"

	"github.com/hanzoai/replicate/internal"
)

func TestInitLog_PrettyHandler(t *testing.T) {
	var buf bytes.Buffer
	internal.InitLog(&buf, "INFO", "pretty", false)
	slog.Default().Info("test message")

	output := buf.String()
	if strings.Contains(output, "\x1b[") {
		t.Fatalf("expected no ANSI codes for non-TTY writer, got: %q", output)
	}
}

func TestInitLog_TextHandler(t *testing.T) {
	var buf bytes.Buffer
	internal.InitLog(&buf, "INFO", "text", false)
	slog.Default().Info("test message")
}

func TestInitLog_JSONHandler(t *testing.T) {
	var buf bytes.Buffer
	internal.InitLog(&buf, "INFO", "json", false)
	slog.Default().Info("test message")
}

func TestInitLog_AddSource(t *testing.T) {
	var buf bytes.Buffer
	internal.InitLog(&buf, "INFO", "text", true)
	slog.Default().Info("test message")

	output := buf.String()
	if !strings.Contains(output, "source=") {
		t.Fatalf("expected source= in output, got: %s", output)
	}
}

func TestInitLog_PrettyAddSource(t *testing.T) {
	var buf bytes.Buffer
	internal.InitLog(&buf, "INFO", "pretty", true)
	slog.Default().Info("test message")

	output := buf.String()
	if !strings.Contains(output, "log_test.go") {
		t.Fatalf("expected source file in output, got: %s", output)
	}
}

func TestInitLog_AllLevels(t *testing.T) {
	for _, level := range []string{"TRACE", "DEBUG", "INFO", "WARN", "WARNING", "ERROR"} {
		t.Run(level, func(t *testing.T) {
			var buf bytes.Buffer
			internal.InitLog(&buf, level, "text", false)
		})
	}
}

func TestReplaceAttr_TraceLevel(t *testing.T) {
	a := slog.Attr{Key: slog.LevelKey, Value: slog.AnyValue(internal.LevelTrace)}
	got := internal.ReplaceAttr(nil, a)
	if got.Value.String() != "TRACE" {
		t.Fatalf("expected TRACE, got %s", got.Value.String())
	}
}

func TestReplaceAttr_NonTraceLevelUnchanged(t *testing.T) {
	a := slog.Attr{Key: slog.LevelKey, Value: slog.AnyValue(slog.LevelInfo)}
	got := internal.ReplaceAttr(nil, a)
	if got.Value.Any() != slog.LevelInfo {
		t.Fatalf("expected INFO level unchanged, got %v", got.Value.Any())
	}
}

func TestReplaceAttr_SourceCleaning(t *testing.T) {
	src := &slog.Source{
		Function: "github.com/hanzoai/replicate/db.(*DB).Replicate",
		File:     "github.com/hanzoai/replicate/db.go",
		Line:     211,
	}
	a := slog.Attr{Key: slog.SourceKey, Value: slog.AnyValue(src)}
	got := internal.ReplaceAttr(nil, a)
	gotSrc := got.Value.Any().(*slog.Source)
	if gotSrc.File != "db.go" {
		t.Fatalf("expected db.go, got %s", gotSrc.File)
	}
}

func TestCleanSourcePath(t *testing.T) {
	tests := []struct {
		name string
		file string
		want string
	}{
		{
			name: "module prefix stripped",
			file: "github.com/hanzoai/replicate/db.go",
			want: "db.go",
		},
		{
			name: "nested path",
			file: "github.com/hanzoai/replicate/internal/internal.go",
			want: "internal/internal.go",
		},
		{
			name: "absolute with module path",
			file: "/Users/dev/go/pkg/mod/github.com/hanzoai/[email protected]/db.go",
			want: "db.go",
		},
		{
			name: "absolute checkout path",
			file: "/Users/dev/projects/replicate/db.go",
			want: "db.go",
		},
		{
			name: "absolute checkout nested path",
			file: "/workspace/replicate/internal/internal.go",
			want: "internal/internal.go",
		},
		{
			name: "no module prefix",
			file: "other/package/file.go",
			want: "other/package/file.go",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			src := &slog.Source{File: tt.file}
			internal.CleanSourcePath(src)
			if src.File != tt.want {
				t.Fatalf("got %s, want %s", src.File, tt.want)
			}
		})
	}
}