Files
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
chore: import upstream snapshot with attribution
2026-07-13 12:30:36 +08:00

89 lines
2.7 KiB
Go

//go:build pgtest
package postgres
import (
"context"
"encoding/json"
"path/filepath"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"go.kenn.io/agentsview/internal/db"
)
func TestPushSessionTranscriptRevisionRoundTrip(t *testing.T) {
pgURL := testPGURL(t)
const schema = "agentsview_read_progress_test"
pg, err := Open(pgURL, schema, true)
require.NoError(t, err)
defer pg.Close()
ctx := context.Background()
_, err = pg.Exec(`DROP SCHEMA IF EXISTS ` + schema + ` CASCADE`)
require.NoError(t, err)
require.NoError(t, EnsureSchema(ctx, pg, schema))
localDB, err := db.Open(filepath.Join(t.TempDir(), "local.db"))
require.NoError(t, err)
defer localDB.Close()
syncer := &Sync{
pg: pg, local: localDB, machine: "test-machine",
schema: schema, schemaDone: true,
}
modified := "2026-07-12T12:00:00Z"
sess := db.Session{
ID: "read-progress", Project: "project", Machine: "test-machine",
Agent: "claude", MessageCount: 1, UserMessageCount: 1,
CreatedAt: "2026-07-12T12:00:00Z", LocalModifiedAt: &modified,
}
require.NoError(t, localDB.UpsertSession(sess))
require.NoError(t, localDB.InsertMessages([]db.Message{{
SessionID: sess.ID, Ordinal: 0, Role: "user",
Content: "hello", ContentLength: len("hello"),
}}))
result, err := syncer.Push(ctx, false, nil)
require.NoError(t, err)
assert.Equal(t, 1, result.SessionsPushed)
// Simulate a row left at the schema default by an older push, while the
// normal incremental watermark says there is no local work. A missing
// backfill marker must force one full push and repair that row.
_, err = pg.ExecContext(ctx,
`UPDATE sessions SET transcript_revision = '0' WHERE id = $1`, sess.ID,
)
require.NoError(t, err)
require.NoError(t, localDB.SetSyncState(
transcriptRevisionBackfillStateKey, "",
))
result, err = syncer.Push(ctx, false, nil)
require.NoError(t, err)
assert.Equal(t, 1, result.SessionsPushed)
store, err := NewStore(pgURL, schema, true)
require.NoError(t, err)
defer store.Close()
detail, err := store.GetSession(ctx, sess.ID)
require.NoError(t, err)
require.NotNil(t, detail)
assertPGJSONTranscriptRevision(t, detail, "1")
index, err := store.GetSidebarSessionIndex(ctx, db.SessionFilter{Limit: 50})
require.NoError(t, err)
require.Len(t, index.Sessions, 1)
assertPGJSONTranscriptRevision(t, index.Sessions[0], "1")
}
func assertPGJSONTranscriptRevision(t *testing.T, value any, want string) {
t.Helper()
raw, err := json.Marshal(value)
require.NoError(t, err)
var fields map[string]any
require.NoError(t, json.Unmarshal(raw, &fields))
assert.Equal(t, want, fields["transcript_revision"])
assert.NotContains(t, fields, "file_hash")
assert.NotContains(t, fields, "local_modified_at")
}