fix(test): rewrite SDK reader tests to use mock client objects instead of mock.module
This commit is contained in:
98
src/hooks/session-recovery/storage/readers-from-sdk.test.ts
Normal file
98
src/hooks/session-recovery/storage/readers-from-sdk.test.ts
Normal file
@@ -0,0 +1,98 @@
|
||||
import { describe, expect, it } from "bun:test"
|
||||
import { readMessagesFromSDK, readPartsFromSDK } from "../storage"
|
||||
import { readMessages } from "./messages-reader"
|
||||
import { readParts } from "./parts-reader"
|
||||
|
||||
function createMockClient(handlers: {
|
||||
messages?: (sessionID: string) => unknown[]
|
||||
message?: (sessionID: string, messageID: string) => unknown
|
||||
}) {
|
||||
return {
|
||||
session: {
|
||||
messages: async (opts: { path: { id: string } }) => {
|
||||
if (handlers.messages) {
|
||||
return { data: handlers.messages(opts.path.id) }
|
||||
}
|
||||
throw new Error("not implemented")
|
||||
},
|
||||
message: async (opts: { path: { id: string; messageID: string } }) => {
|
||||
if (handlers.message) {
|
||||
return { data: handlers.message(opts.path.id, opts.path.messageID) }
|
||||
}
|
||||
throw new Error("not implemented")
|
||||
},
|
||||
},
|
||||
} as unknown
|
||||
}
|
||||
|
||||
describe("session-recovery storage SDK readers", () => {
|
||||
it("readPartsFromSDK returns empty array when fetch fails", async () => {
|
||||
//#given a client that throws on request
|
||||
const client = createMockClient({}) as Parameters<typeof readPartsFromSDK>[0]
|
||||
|
||||
//#when readPartsFromSDK is called
|
||||
const result = await readPartsFromSDK(client, "ses_test", "msg_test")
|
||||
|
||||
//#then it returns empty array
|
||||
expect(result).toEqual([])
|
||||
})
|
||||
|
||||
it("readPartsFromSDK returns stored parts from SDK response", async () => {
|
||||
//#given a client that returns a message with parts
|
||||
const sessionID = "ses_test"
|
||||
const messageID = "msg_test"
|
||||
const storedParts = [
|
||||
{ id: "prt_1", sessionID, messageID, type: "text", text: "hello" },
|
||||
]
|
||||
|
||||
const client = createMockClient({
|
||||
message: (_sid, _mid) => ({ parts: storedParts }),
|
||||
}) as Parameters<typeof readPartsFromSDK>[0]
|
||||
|
||||
//#when readPartsFromSDK is called
|
||||
const result = await readPartsFromSDK(client, sessionID, messageID)
|
||||
|
||||
//#then it returns the parts
|
||||
expect(result).toEqual(storedParts)
|
||||
})
|
||||
|
||||
it("readMessagesFromSDK normalizes and sorts messages", async () => {
|
||||
//#given a client that returns messages list
|
||||
const sessionID = "ses_test"
|
||||
const client = createMockClient({
|
||||
messages: () => [
|
||||
{ id: "msg_b", role: "assistant", time: { created: 2 } },
|
||||
{ id: "msg_a", role: "user", time: { created: 1 } },
|
||||
{ id: "msg_c" },
|
||||
],
|
||||
}) as Parameters<typeof readMessagesFromSDK>[0]
|
||||
|
||||
//#when readMessagesFromSDK is called
|
||||
const result = await readMessagesFromSDK(client, sessionID)
|
||||
|
||||
//#then it returns sorted StoredMessageMeta with defaults
|
||||
expect(result).toEqual([
|
||||
{ id: "msg_c", sessionID, role: "user", time: { created: 0 } },
|
||||
{ id: "msg_a", sessionID, role: "user", time: { created: 1 } },
|
||||
{ id: "msg_b", sessionID, role: "assistant", time: { created: 2 } },
|
||||
])
|
||||
})
|
||||
|
||||
it("readParts returns empty array for nonexistent message", () => {
|
||||
//#given a message ID that has no stored parts
|
||||
//#when readParts is called
|
||||
const parts = readParts("msg_nonexistent")
|
||||
|
||||
//#then it returns empty array
|
||||
expect(parts).toEqual([])
|
||||
})
|
||||
|
||||
it("readMessages returns empty array for nonexistent session", () => {
|
||||
//#given a session ID that has no stored messages
|
||||
//#when readMessages is called
|
||||
const messages = readMessages("ses_nonexistent")
|
||||
|
||||
//#then it returns empty array
|
||||
expect(messages).toEqual([])
|
||||
})
|
||||
})
|
||||
Reference in New Issue
Block a user