fix tests

Author: yujonglee

apps/desktop/src/store/tinybase/persister/calendar.test.ts

@@ -58,7 +58,9 @@ describe("createCalendarPersister", () => {
       };
       vi.mocked(readTextFile).mockResolvedValue(JSON.stringify(mockData));
 
-      const persister = createCalendarPersister<Schemas>(store);
+      const persister = createCalendarPersister<Schemas>(store, {
+        mode: "load-and-save",
+      });
       await persister.load();
 
       expect(readTextFile).toHaveBeenCalledWith(
@@ -74,7 +76,9 @@ describe("createCalendarPersister", () => {
         new Error("No such file or directory"),
       );
 
-      const persister = createCalendarPersister<Schemas>(store);
+      const persister = createCalendarPersister<Schemas>(store, {
+        mode: "load-and-save",
+      });
       await persister.load();
 
       expect(store.getTable("calendars")).toEqual({});

apps/desktop/src/store/tinybase/persister/chat.ts

@@ -75,8 +75,8 @@ export function createChatPersister<Schemas extends OptionalSchemas>(
   const saveFn =
     config.mode === "load-only"
       ? async () => {}
-      : async (getContent: () => unknown) => {
-          const [tables] = getContent() as [TablesContent | undefined, unknown];
+      : async () => {
+          const tables = store.getTables() as TablesContent | undefined;
           const dataDir = await getDataDir();
 
           const messagesByChatGroup = collectMessagesByChatGroup(tables);

apps/desktop/src/store/tinybase/persister/events.test.ts

@@ -63,7 +63,9 @@ describe("createEventPersister", () => {
       };
       vi.mocked(readTextFile).mockResolvedValue(JSON.stringify(mockData));
 
-      const persister = createEventPersister<Schemas>(store);
+      const persister = createEventPersister<Schemas>(store, {
+        mode: "load-and-save",
+      });
       await persister.load();
 
       expect(readTextFile).toHaveBeenCalledWith(
@@ -79,7 +81,9 @@ describe("createEventPersister", () => {
         new Error("No such file or directory"),
       );
 
-      const persister = createEventPersister<Schemas>(store);
+      const persister = createEventPersister<Schemas>(store, {
+        mode: "load-and-save",
+      });
       await persister.load();
 
       expect(store.getTable("events")).toEqual({});

apps/desktop/src/store/tinybase/persister/human.test.ts

@@ -58,7 +58,9 @@ describe("createHumanPersister", () => {
       };
       vi.mocked(readTextFile).mockResolvedValue(JSON.stringify(mockData));
 
-      const persister = createHumanPersister<Schemas>(store);
+      const persister = createHumanPersister<Schemas>(store, {
+        mode: "load-and-save",
+      });
       await persister.load();
 
       expect(readTextFile).toHaveBeenCalledWith(
@@ -74,7 +76,9 @@ describe("createHumanPersister", () => {
         new Error("No such file or directory"),
       );
 
-      const persister = createHumanPersister<Schemas>(store);
+      const persister = createHumanPersister<Schemas>(store, {
+        mode: "load-and-save",
+      });
       await persister.load();
 
       expect(store.getTable("humans")).toEqual({});

apps/desktop/src/store/tinybase/persister/note.ts

@@ -29,8 +29,8 @@ export function createNotePersister<Schemas extends OptionalSchemas>(
   const saveFn =
     config.mode === "load-only"
       ? async () => {}
-      : async (getContent: () => unknown) => {
-          const [tables] = getContent() as [TablesContent | undefined, unknown];
+      : async () => {
+          const tables = store.getTables() as TablesContent | undefined;
           const dataDir = await getDataDir();
 
           const enhancedNotes = collectEnhancedNoteBatchItems(

apps/desktop/src/store/tinybase/persister/organization.test.ts

@@ -53,7 +53,9 @@ describe("createOrganizationPersister", () => {
       };
       vi.mocked(readTextFile).mockResolvedValue(JSON.stringify(mockData));
 
-      const persister = createOrganizationPersister<Schemas>(store);
+      const persister = createOrganizationPersister<Schemas>(store, {
+        mode: "load-and-save",
+      });
       await persister.load();
 
       expect(readTextFile).toHaveBeenCalledWith(
@@ -69,7 +71,9 @@ describe("createOrganizationPersister", () => {
         new Error("No such file or directory"),
       );
 
-      const persister = createOrganizationPersister<Schemas>(store);
+      const persister = createOrganizationPersister<Schemas>(store, {
+        mode: "load-and-save",
+      });
       await persister.load();
 
       expect(store.getTable("organizations")).toEqual({});

apps/desktop/src/store/tinybase/persister/session.test.ts

@@ -228,7 +228,9 @@ describe("createSessionPersister", () => {
       };
       vi.mocked(readTextFile).mockResolvedValue(JSON.stringify(mockMeta));
 
-      const persister = createSessionPersister<Schemas>(store);
+      const persister = createSessionPersister<Schemas>(store, {
+        mode: "load-and-save",
+      });
       await persister.load();
 
       expect(store.getTable("sessions")).toEqual({
@@ -277,7 +279,9 @@ describe("createSessionPersister", () => {
       };
       vi.mocked(readTextFile).mockResolvedValue(JSON.stringify(mockMeta));
 
-      const persister = createSessionPersister<Schemas>(store);
+      const persister = createSessionPersister<Schemas>(store, {
+        mode: "load-and-save",
+      });
       await persister.load();
 
       expect(store.getTable("tags")).toEqual({});
@@ -324,7 +328,9 @@ describe("createSessionPersister", () => {
           }),
         );
 
-      const persister = createSessionPersister<Schemas>(store);
+      const persister = createSessionPersister<Schemas>(store, {
+        mode: "load-and-save",
+      });
       await persister.load();
 
       const tags = store.getTable("tags");

apps/desktop/src/store/tinybase/persister/transcript.ts

@@ -86,8 +86,8 @@ export function createTranscriptPersister<Schemas extends OptionalSchemas>(
   const saveFn =
     config.mode === "load-only"
       ? async () => {}
-      : async (getContent: () => unknown) => {
-          const [tables] = getContent() as [TablesContent | undefined, unknown];
+      : async () => {
+          const tables = store.getTables() as TablesContent | undefined;
           const dataDir = await getDataDir();
 
           const transcriptsBySession = collectTranscriptsBySession(tables);