Function TestResolveSessionID_ExcludesSubSessions

pkg/session/store_test.go:1063–1108

(t *testing.T)

}

func TestResolveSessionID_ExcludesSubSessions(t *testing.T) {
	t.Parallel()

	tempDB := filepath.Join(t.TempDir(), "test_resolve_subsessions.db")

	store, err := NewSQLiteSessionStore(t.Context(), tempDB)
	require.NoError(t, err)
	defer store.(*SQLiteSessionStore).Close()

	sqliteStore := store.(*SQLiteSessionStore)

	// Create a parent session
	parent := &Session{
		ID:        "parent",
		CreatedAt: time.Now().Add(-2 * time.Hour),
	}
	err = store.AddSession(t.Context(), parent)
	require.NoError(t, err)

	// Create a sub-session directly in the database to avoid the AddSubSession complexity
	// This simulates an existing sub-session without going through the full API
	subSessionTime := time.Now().Add(-1 * time.Hour)
	_, err = sqliteStore.db.ExecContext(t.Context(),
		`INSERT INTO sessions (id, tools_approved, input_tokens, output_tokens, title, cost, send_user_message, max_iterations, working_dir, created_at, starred, permissions, agent_model_overrides, custom_models_used, thinking, parent_id)
		 VALUES (?, 0, 0, 0, '', 0, 1, 0, '', ?, 0, '', '{}', '[]', 1, ?)`,
		"subsession", subSessionTime.Format(time.RFC3339), "parent")
	require.NoError(t, err)

	// Create another root session (most recent root)
	root2 := &Session{
		ID:        "root2",
		CreatedAt: time.Now(),
	}
	err = store.AddSession(t.Context(), root2)
	require.NoError(t, err)

	// -1 should resolve to root2, not the subsession
	id, err := ResolveSessionID(t.Context(), store, "-1")
	require.NoError(t, err)
	assert.Equal(t, "root2", id)

	// -2 should resolve to parent
	id, err = ResolveSessionID(t.Context(), store, "-2")
	require.NoError(t, err)
	assert.Equal(t, "parent", id)
}

func TestResolveSessionID_InMemory(t *testing.T) {
	t.Parallel()