João Salavisa
119 lines
4.4 KiB
TypeScript
119 lines
4.4 KiB
TypeScript
import { describe, it, expect, beforeEach } from "vitest";
|
|
import { create } from "zustand";
|
|
import { immer } from "zustand/middleware/immer";
|
|
import { createUISlice } from "@/lib/state/slices/ui/ui-slice";
|
|
import { getQuickChatSetupSessionId } from "@/lib/state/slices/ui/quick-chat-session";
|
|
import { getStoredQuickChatNames } from "@/lib/local-storage";
|
|
import type { UISlice } from "@/lib/state/slices/ui/types";
|
|
|
|
const SESSION_ID = "sess-1";
|
|
const WORKSPACE_ID = "ws-1";
|
|
const PROFILE_ID = "profile-pass";
|
|
|
|
function makeStore() {
|
|
return create<UISlice>()(immer(createUISlice));
|
|
}
|
|
|
|
function findSession(store: ReturnType<typeof makeStore>) {
|
|
return store.getState().quickChat.sessions.find((s) => s.sessionId === SESSION_ID);
|
|
}
|
|
|
|
describe("openQuickChat agentProfileId persistence", () => {
|
|
it("stores agentProfileId on a new session entry", () => {
|
|
const store = makeStore();
|
|
store.getState().openQuickChat(SESSION_ID, WORKSPACE_ID, PROFILE_ID);
|
|
expect(findSession(store)?.agentProfileId).toBe(PROFILE_ID);
|
|
expect(store.getState().quickChat.activeSessionId).toBe(SESSION_ID);
|
|
});
|
|
|
|
it("backfills agentProfileId on an existing session entry without one", () => {
|
|
const store = makeStore();
|
|
store.getState().openQuickChat(SESSION_ID, WORKSPACE_ID);
|
|
expect(findSession(store)?.agentProfileId).toBeUndefined();
|
|
store.getState().openQuickChat(SESSION_ID, WORKSPACE_ID, PROFILE_ID);
|
|
expect(findSession(store)?.agentProfileId).toBe(PROFILE_ID);
|
|
});
|
|
|
|
it("does not duplicate sessions when reopening with the same id", () => {
|
|
const store = makeStore();
|
|
store.getState().openQuickChat(SESSION_ID, WORKSPACE_ID, "profile-a");
|
|
store.getState().openQuickChat(SESSION_ID, WORKSPACE_ID, "profile-a");
|
|
expect(
|
|
store.getState().quickChat.sessions.filter((s) => s.sessionId === SESSION_ID),
|
|
).toHaveLength(1);
|
|
});
|
|
|
|
it("keeps existing agentProfileId when reopened without one", () => {
|
|
const store = makeStore();
|
|
store.getState().openQuickChat(SESSION_ID, WORKSPACE_ID, PROFILE_ID);
|
|
store.getState().openQuickChat(SESSION_ID, WORKSPACE_ID); // no profile
|
|
expect(findSession(store)?.agentProfileId).toBe(PROFILE_ID);
|
|
});
|
|
});
|
|
|
|
describe("Quick Chat workspace isolation", () => {
|
|
it("keeps typed setup tabs isolated by workspace", () => {
|
|
const store = makeStore();
|
|
store.getState().openQuickChat("", "ws-a");
|
|
store.getState().openQuickChat("", "ws-b");
|
|
|
|
expect(store.getState().quickChat.sessions).toEqual([
|
|
{
|
|
sessionId: getQuickChatSetupSessionId("ws-a", "chat"),
|
|
workspaceId: "ws-a",
|
|
kind: "chat",
|
|
},
|
|
{
|
|
sessionId: getQuickChatSetupSessionId("ws-b", "chat"),
|
|
workspaceId: "ws-b",
|
|
kind: "chat",
|
|
},
|
|
]);
|
|
expect(store.getState().quickChat.activeSessionId).toBe(
|
|
getQuickChatSetupSessionId("ws-b", "chat"),
|
|
);
|
|
});
|
|
|
|
it("falls back to a tab from the closed session workspace", () => {
|
|
const store = makeStore();
|
|
store.getState().openQuickChat("session-a", "ws-a");
|
|
store.getState().openQuickChat("session-b-1", "ws-b");
|
|
store.getState().openQuickChat("session-b-2", "ws-b");
|
|
|
|
store.getState().closeQuickChatSession("session-b-2");
|
|
|
|
expect(store.getState().quickChat.activeSessionId).toBe("session-b-1");
|
|
});
|
|
});
|
|
|
|
describe("renameQuickChatSession", () => {
|
|
beforeEach(() => {
|
|
window.localStorage.clear();
|
|
});
|
|
|
|
it("updates the session name in the store", () => {
|
|
const store = makeStore();
|
|
store.getState().openQuickChat(SESSION_ID, WORKSPACE_ID);
|
|
store.getState().renameQuickChatSession(SESSION_ID, "My renamed chat");
|
|
expect(findSession(store)?.name).toBe("My renamed chat");
|
|
});
|
|
|
|
// The name now lives on the backing task title (see persistQuickChatRename).
|
|
// Writing it to localStorage here too would let a stale local entry override
|
|
// a rename made on another device — the divergence this whole flow fixes.
|
|
it("does not pin the name in localStorage", () => {
|
|
const store = makeStore();
|
|
store.getState().openQuickChat(SESSION_ID, WORKSPACE_ID);
|
|
store.getState().renameQuickChatSession(SESSION_ID, "Persisted name");
|
|
expect(getStoredQuickChatNames()).toEqual({});
|
|
});
|
|
|
|
it("ignores a session that does not exist in state", () => {
|
|
const store = makeStore();
|
|
expect(() =>
|
|
store.getState().renameQuickChatSession("ghost-session", "Whatever"),
|
|
).not.toThrow();
|
|
expect(findSession(store)).toBeUndefined();
|
|
});
|
|
});
|