import (
	"bytes"
	"context"
	"log/slog"
	"strings"
	"testing"
	"time"
)

func TestSlogLogger(t *testing.T) {
	buf := &bytes.Buffer{}
	handler := slog.NewTextHandler(buf, &slog.HandlerOptions{AddSource: true})
	logger := NewSlogLogger(slog.New(handler), Config{LogLevel: Info})

	logger.Trace(context.Background(), time.Now(), func() (string, int64) {
		return "select count(*) from users", 0
	}, nil)

        final ChatMessage message = new ChatMessage();
        final ChatSource source = new ChatSource();
        source.setIndex(1);
        source.setTitle("Test Document");

        message.addSource(source);

        assertEquals(1, message.getSources().size());
        assertEquals("Test Document", message.getSources().get(0).getTitle());
    }

    @Test
    public void test_addSourceWithNullSources() {