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)

        this.sources = sources;
    }

    /**
     * Adds a source to the message.
     *
     * @param source the source to add
     */
    public void addSource(final ChatSource source) {
        if (sources == null) {
            sources = new ArrayList<>();
        }
        sources.add(source);
    }

    /**
     * Checks if this is a user message.

        doc1.put("title", "Installation Guide");
        doc1.put("url", "http://example.com/install");
        msg.addSource(new ChatSource(1, doc1));
        final Map<String, Object> doc2 = new HashMap<>();
        doc2.put("title", "Quick Start");
        doc2.put("url", "http://example.com/start");
        msg.addSource(new ChatSource(2, doc2));

pkg log/slog, type Handler interface, WithAttrs([]Attr) Handler #56345
pkg log/slog, type Handler interface, WithGroup(string) Handler #56345
pkg log/slog, type HandlerOptions struct #56345
pkg log/slog, type HandlerOptions struct, AddSource bool #56345
pkg log/slog, type HandlerOptions struct, Level Leveler #56345
pkg log/slog, type HandlerOptions struct, ReplaceAttr func([]string, Attr) Attr #56345
pkg log/slog, type JSONHandler struct #56345

        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() {

            final ChatSource source = new ChatSource(i + 1, sourceList.get(i));
            source.setGoUrl(buildGoUrl(contextPath, source.getDocId(), queryId, requestedTime, i));
            assistantMessage.addSource(source);
        }
    }

    /**
     * Populates the url_link field in the document map if not already present.
     *
     * @param doc the document map
     */