//go:build pgtest

package postgres

import (
	"context"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"go.kenn.io/agentsview/internal/db"
)

// TestPushThinkingText_SanitizesNullAndInvalidUTF8 verifies that
// bulkInsertMessages runs ThinkingText through sanitizePG before
// sending it to PostgreSQL. Without the sanitize call, a NUL byte
// or invalid UTF-8 in a thinking block would make the PG INSERT
// reject the entire batch and stall the push.
func TestPushThinkingText_SanitizesNullAndInvalidUTF8(t *testing.T) {
	pgURL := testPGURL(t)
	cleanPGSchema(t, pgURL)
	t.Cleanup(func() { cleanPGSchema(t, pgURL) })

	local := testDB(t)
	ps, err := New(
		pgURL, "agentsview", local,
		"thinking-test-machine", true,
		SyncOptions{},
	)
	require.NoError(t, err, "creating sync")
	defer ps.Close()

	ctx := context.Background()
	require.NoError(t, ps.EnsureSchema(ctx), "ensure schema")

	started := time.Now().UTC().Format(time.RFC3339)
	first := "hello"
	sess := db.Session{
		ID:           "think-1",
		Project:      "proj",
		Machine:      "local",
		Agent:        "claude",
		FirstMessage: &first,
		StartedAt:    &started,
		MessageCount: 1,
	}
	require.NoError(t, local.UpsertSession(sess), "upsert")

	// Message whose thinking_text contains a NUL byte and a
	// truncated multi-byte UTF-8 sequence. Before the fix the
	// insert would fail with "invalid byte sequence".
	thinking := "plan\x00step\xe2"
	require.NoError(t, local.InsertMessages([]db.Message{{
		SessionID:    "think-1",
		Ordinal:      0,
		Role:         "assistant",
		Content:      "ok",
		ThinkingText: thinking,
		HasThinking:  true,
	}}), "insert local message")

	_, err = ps.Push(ctx, false, nil)
	require.NoError(t, err, "push")

	store, err := NewStore(pgURL, "agentsview", true)
	require.NoError(t, err, "NewStore")
	defer store.Close()

	msgs, err := store.GetMessages(ctx, "think-1", 0, 10, true)
	require.NoError(t, err, "GetMessages")
	require.Len(t, msgs, 1)
	// NUL bytes and invalid UTF-8 must be stripped; the
	// remaining text stays intact and in order.
	assert.Equal(t, "planstep", msgs[0].ThinkingText,
		"sanitize skipped?")
}