opencode/packages/app/src/context/sync-session-diff.test.ts at 78cdbb82469c9af584c2603dc4fa9c256facc9e9 · jeanibarz/opencode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
import { describe, expect, mock, test } from "bun:test"
import { createRoot } from "solid-js"
import { createStore } from "solid-js/store"
import type { FileDiff } from "@opencode-ai/sdk/v2/client"
import type { State } from "./global-sync/types"
import { createSyncContextValue } from "./sync"

type SyncDeps = Parameters<typeof createSyncContextValue>[0]

function baseState(): State {
  return {
    status: "complete",
    agent: [],
    command: [],
    project: "",
    projectMeta: undefined,
    icon: undefined,
    provider_ready: false,
    provider: { all: [], connected: [], default: {} },
    config: {},
    path: { state: "", config: "", worktree: "", directory: "/tmp/project", home: "" },
    session: [],
    sessionTotal: 0,
    session_status: {},
    session_diff: {},
    todo: {},
    permission: {},
    question: {},
    mcp_ready: false,
    mcp: {},
    lsp_ready: false,
    lsp: [],
    vcs: undefined,
    limit: 5,
    message: {},
    part: {},
  }
}

describe("sync session diff recovery", () => {
  test("re-fetches malformed cached diffs until a valid array is stored", async () => {
    const sessionID = "ses_1"
    const valid = [{ file: "a.ts", before: "", after: "", additions: 1, deletions: 0 }] as FileDiff[]
    const [store, setStore] = createStore({
      ...baseState(),
      session_diff: { [sessionID]: {} as State["session_diff"][string] },
    })

    const diff = mock(async () => {
      const count = diff.mock.calls.length
      return count === 1 ? ({ data: {} } as const) : ({ data: valid } as const)
    })

    const globalSync = {
      child: () => [store, setStore],
      data: {
        project: [],
        session_todo: {},
      },
      todo: {
        set() {},
      },
    }

    const sdk = {
      directory: "/tmp/project",
      client: {
        session: {
          diff,
        },
      },
    }

    await new Promise<void>((resolve, reject) => {
      createRoot((dispose) => {
        const sync = createSyncContextValue({
          globalSync: globalSync as unknown as SyncDeps["globalSync"],
          sdk: sdk as unknown as SyncDeps["sdk"],
        })
        void (async () => {
          try {
            await sync.session.diff(sessionID)
            expect(diff).toHaveBeenCalledTimes(1)
            expect(store.session_diff[sessionID]).toBeUndefined()

            await sync.session.diff(sessionID)
            expect(diff).toHaveBeenCalledTimes(2)
            expect(store.session_diff[sessionID]).toEqual(valid)

            await sync.session.diff(sessionID)
            expect(diff).toHaveBeenCalledTimes(2)
            resolve()
          } catch (error) {
            reject(error)
          } finally {
            dispose()
          }
        })()
      })
    })
  })
})