wehub-resource-sync
f99010fae1
Desktop Artifacts / Desktop Build (Linux) (push) Waiting to run
Desktop Artifacts / Desktop Build (Windows) (push) Waiting to run
Desktop Artifacts / Desktop Build (Linux (arm64)) (push) Waiting to run
Desktop Artifacts (macOS) / Desktop Build (macOS (aarch64)) (push) Waiting to run
Desktop Artifacts (macOS) / Desktop Build (macOS (x86_64)) (push) Waiting to run
CI / lint (push) Failing after 1s
CI / frontend (push) Failing after 1s
CI / scripts (push) Failing after 1s
CI / Go Test (ubuntu-latest) (push) Failing after 0s
CI / frontend-node-25 (push) Failing after 1s
CI / docs (push) Failing after 0s
CI / coverage (push) Failing after 0s
CI / e2e (push) Failing after 0s
Docker / build-and-push (push) Failing after 1s
CI / integration (push) Failing after 4m43s
CI / Go Test (windows-latest) (push) Has been cancelled
CI / Desktop Unit Tests (Windows) (push) Has been cancelled
79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
//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?")
|
|
}
|