test(storage): cover Claude provenance round trips

Author: mjacobs

internal/duckdb/store_contract_test.go

@@ -61,7 +61,7 @@ func TestDuckDBSessionDateFilterIncludesOverlappingSessions(t *testing.T) {
 	assert.Equal(t, []string{"open", "spanning"}, ids)
 }
 
-func TestSessionIdentity(t *testing.T) {
+func TestClaudeProvenanceRoundTrip(t *testing.T) {
 	ctx := context.Background()
 	local := newLocalDB(t)
 	startedAt := "2024-06-15T08:00:00Z"
@@ -74,13 +74,22 @@ func TestSessionIdentity(t *testing.T) {
 		Agent:            "claude",
 		AgentLabel:       "Claude Triage",
 		Entrypoint:       "sdk-cli",
+		SessionKind:      "bg",
 		SessionName:      &sessionName,
 		StartedAt:        &startedAt,
 		EndedAt:          &endedAt,
 		CreatedAt:        startedAt,
 		MessageCount:     1,
 		UserMessageCount: 1,
 	}), "upsert identity session")
+	require.NoError(t, local.InsertMessages([]db.Message{{
+		SessionID:     "duck-identity",
+		Ordinal:       0,
+		Role:          "user",
+		Content:       "hello",
+		ContentLength: 5,
+		PromptSource:  "typed",
+	}}), "insert identity message")
 
 	syncer := newInMemoryTestSync(t, local, SyncOptions{})
 	require.NoError(t, createSchema(ctx, syncer.DB()))
@@ -96,8 +105,19 @@ func TestSessionIdentity(t *testing.T) {
 	assert.Equal(t, "duck-identity", index.Sessions[0].ID)
 	assert.Equal(t, "Claude Triage", index.Sessions[0].AgentLabel)
 	assert.Equal(t, "sdk-cli", index.Sessions[0].Entrypoint)
+	assert.Equal(t, "bg", index.Sessions[0].SessionKind)
 	require.NotNil(t, index.Sessions[0].DisplayName)
 	assert.Equal(t, "Agent Title", *index.Sessions[0].DisplayName)
+
+	session, err := store.GetSession(ctx, "duck-identity")
+	require.NoError(t, err)
+	require.NotNil(t, session)
+	assert.Equal(t, "bg", session.SessionKind)
+
+	messages, err := store.GetMessages(ctx, "duck-identity", 0, 10, true)
+	require.NoError(t, err)
+	require.Len(t, messages, 1)
+	assert.Equal(t, "typed", messages[0].PromptSource)
 }
 
 func TestDuckDBStoreContract(t *testing.T) {

internal/postgres/session_identity_pgtest_test.go

@@ -14,7 +14,7 @@ import (
 	"go.kenn.io/agentsview/internal/db"
 )
 
-func TestPGSessionIdentityVisibleInReadPaths(t *testing.T) {
+func TestPGClaudeProvenanceVisibleInReadPaths(t *testing.T) {
 	pgURL := testPGURL(t)
 
 	const schema = "agentsview_session_identity_test"
@@ -38,13 +38,22 @@ func TestPGSessionIdentityVisibleInReadPaths(t *testing.T) {
 		Agent:            "claude",
 		AgentLabel:       "triage",
 		Entrypoint:       "sdk-cli",
+		SessionKind:      "bg",
 		MessageCount:     1,
 		UserMessageCount: 1,
 		CreatedAt:        "2026-01-01T00:00:00Z",
 		StartedAt:        strPtr("2026-01-01T00:00:00Z"),
 		EndedAt:          strPtr("2026-01-01T01:00:00Z"),
 	}
 	require.NoError(t, localDB.UpsertSession(sess), "UpsertSession")
+	require.NoError(t, localDB.InsertMessages([]db.Message{{
+		SessionID:     sess.ID,
+		Ordinal:       0,
+		Role:          "user",
+		Content:       "hello",
+		ContentLength: 5,
+		PromptSource:  "typed",
+	}}), "InsertMessages")
 
 	sync := &Sync{
 		pg:         pg,
@@ -56,14 +65,15 @@ func TestPGSessionIdentityVisibleInReadPaths(t *testing.T) {
 	_, pushErr := sync.Push(ctx, true, nil)
 	require.NoError(t, pushErr, "Push")
 
-	var agentLabel sql.NullString
-	var entrypoint sql.NullString
+	var agentLabel, entrypoint, sessionKind sql.NullString
 	require.NoError(t, pg.QueryRowContext(ctx,
-		`SELECT agent_label, entrypoint FROM sessions WHERE id = $1`,
+		`SELECT agent_label, entrypoint, session_kind
+		 FROM sessions WHERE id = $1`,
 		sess.ID,
-	).Scan(&agentLabel, &entrypoint), "query raw PG row")
+	).Scan(&agentLabel, &entrypoint, &sessionKind), "query raw PG row")
 	assert.Equal(t, "triage", agentLabel.String)
 	assert.Equal(t, "sdk-cli", entrypoint.String)
+	assert.Equal(t, "bg", sessionKind.String)
 
 	store, err := NewStore(pgURL, schema, true)
 	require.NoError(t, err, "NewStore")
@@ -76,10 +86,17 @@ func TestPGSessionIdentityVisibleInReadPaths(t *testing.T) {
 	require.Len(t, idx.Sessions, 1, "expected one session in sidebar index")
 	assert.Equal(t, "triage", idx.Sessions[0].AgentLabel)
 	assert.Equal(t, "sdk-cli", idx.Sessions[0].Entrypoint)
+	assert.Equal(t, "bg", idx.Sessions[0].SessionKind)
 
 	full, err := store.GetSession(ctx, sess.ID)
 	require.NoError(t, err, "GetSession")
 	require.NotNil(t, full, "GetSession must return the session")
 	assert.Equal(t, "triage", full.AgentLabel)
 	assert.Equal(t, "sdk-cli", full.Entrypoint)
+	assert.Equal(t, "bg", full.SessionKind)
+
+	messages, err := store.GetMessages(ctx, sess.ID, 0, 10, true)
+	require.NoError(t, err, "GetMessages")
+	require.Len(t, messages, 1)
+	assert.Equal(t, "typed", messages[0].PromptSource)
 }