Increase tests coverage.

internal/db/sqlite/prompt_test.go

@@ -194,3 +194,96 @@ func TestPromptStore_GetPromptsByIDs_EmptyInput(t *testing.T) {
 	require.NoError(t, err)
 	assert.Nil(t, prompts)
 }
+
+func TestPromptStore_FindRecentPromptByText(t *testing.T) {
+	promptStore, store, cleanup := testPromptStore(t)
+	defer cleanup()
+
+	ctx := context.Background()
+
+	// Create a session
+	seedSession(t, storeDB(store), "claude-1", "sdk-1", "test-project")
+
+	// Save a prompt
+	_, err := promptStore.SaveUserPromptWithMatches(ctx, "claude-1", 1, "Help me fix this bug in the code", 0)
+	require.NoError(t, err)
+
+	// Find the prompt by text - returns (id, promptNumber, found)
+	id, promptNum, found := promptStore.FindRecentPromptByText(ctx, "claude-1", "Help me fix this bug in the code", 60)
+	assert.True(t, found, "should find the exact prompt text")
+	assert.Greater(t, id, int64(0))
+	assert.Equal(t, 1, promptNum)
+
+	// Try to find non-existent prompt
+	_, _, found = promptStore.FindRecentPromptByText(ctx, "claude-1", "This prompt does not exist", 60)
+	assert.False(t, found, "should not find non-existent prompt")
+
+	// Try with different session
+	_, _, found = promptStore.FindRecentPromptByText(ctx, "claude-2", "Help me fix this bug in the code", 60)
+	assert.False(t, found, "should not find prompt for different session")
+}
+
+func TestPromptStore_FindRecentPromptByText_WindowSeconds(t *testing.T) {
+	promptStore, store, cleanup := testPromptStore(t)
+	defer cleanup()
+
+	ctx := context.Background()
+
+	// Create a session
+	seedSession(t, storeDB(store), "claude-1", "sdk-1", "test-project")
+
+	// Save a prompt with an old timestamp
+	oldEpoch := time.Now().Add(-2 * time.Hour).UnixMilli()
+	_, err := storeDB(store).Exec(`
+		INSERT INTO user_prompts (claude_session_id, prompt_number, prompt_text, created_at, created_at_epoch)
+		VALUES (?, ?, ?, datetime('now'), ?)
+	`, "claude-1", 1, "Old prompt text", oldEpoch)
+	require.NoError(t, err)
+
+	// Search within last hour - should not find old prompt
+	_, _, found := promptStore.FindRecentPromptByText(ctx, "claude-1", "Old prompt text", 3600)
+	assert.False(t, found, "should not find prompt outside window")
+
+	// Search within last 3 hours - should find old prompt
+	_, _, found = promptStore.FindRecentPromptByText(ctx, "claude-1", "Old prompt text", 3*3600)
+	assert.True(t, found, "should find prompt within extended window")
+}
+
+func TestPromptStore_SaveMultiplePrompts(t *testing.T) {
+	promptStore, store, cleanup := testPromptStore(t)
+	defer cleanup()
+
+	ctx := context.Background()
+
+	// Create sessions
+	seedSession(t, storeDB(store), "claude-1", "sdk-1", "project-x")
+	seedSession(t, storeDB(store), "claude-2", "sdk-2", "project-y")
+
+	tests := []struct {
+		claudeSessionID string
+		promptNum       int
+		text            string
+		matches         int
+	}{
+		{"claude-1", 1, "First prompt", 5},
+		{"claude-1", 2, "Second prompt", 3},
+		{"claude-2", 1, "Third prompt", 0},
+		{"claude-1", 3, "Fourth prompt", 10},
+	}
+
+	for _, tt := range tests {
+		id, err := promptStore.SaveUserPromptWithMatches(ctx, tt.claudeSessionID, tt.promptNum, tt.text, tt.matches)
+		require.NoError(t, err)
+		assert.Greater(t, id, int64(0))
+	}
+
+	// Verify counts
+	var count int
+	err := storeDB(store).QueryRow("SELECT COUNT(*) FROM user_prompts WHERE claude_session_id = 'claude-1'").Scan(&count)
+	require.NoError(t, err)
+	assert.Equal(t, 3, count)
+
+	err = storeDB(store).QueryRow("SELECT COUNT(*) FROM user_prompts WHERE claude_session_id = 'claude-2'").Scan(&count)
+	require.NoError(t, err)
+	assert.Equal(t, 1, count)
+}