@xpert-ai/xpert-sdk 0.0.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/dist/auth/error.d.ts +9 -0
- package/dist/auth/error.js +76 -0
- package/dist/auth/error.js.map +1 -0
- package/dist/auth/error.mjs +76 -0
- package/dist/auth/error.mjs.map +1 -0
- package/dist/auth/index.d.ts +18 -0
- package/dist/auth/index.js +35 -0
- package/dist/auth/index.js.map +1 -0
- package/dist/auth/index.mjs +35 -0
- package/dist/auth/index.mjs.map +1 -0
- package/dist/auth/types.d.ts +297 -0
- package/dist/auth/types.js +2 -0
- package/dist/auth/types.js.map +1 -0
- package/dist/auth/types.mjs +2 -0
- package/dist/auth/types.mjs.map +1 -0
- package/dist/client.d.ts +662 -0
- package/dist/client.js +1233 -0
- package/dist/client.js.map +1 -0
- package/dist/client.mjs +1233 -0
- package/dist/client.mjs.map +1 -0
- package/dist/index.d.ts +7 -0
- package/dist/index.js +3 -0
- package/dist/index.js.map +1 -0
- package/dist/index.mjs +3 -0
- package/dist/index.mjs.map +1 -0
- package/dist/logging/index.d.ts +43 -0
- package/dist/logging/index.js +33 -0
- package/dist/logging/index.js.map +1 -0
- package/dist/logging/index.mjs +33 -0
- package/dist/logging/index.mjs.map +1 -0
- package/dist/schema.d.ts +270 -0
- package/dist/schema.js +2 -0
- package/dist/schema.js.map +1 -0
- package/dist/schema.mjs +2 -0
- package/dist/schema.mjs.map +1 -0
- package/dist/singletons/fetch.d.ts +7 -0
- package/dist/singletons/fetch.js +24 -0
- package/dist/singletons/fetch.js.map +1 -0
- package/dist/singletons/fetch.mjs +24 -0
- package/dist/singletons/fetch.mjs.map +1 -0
- package/dist/tests/fetch.test.d.ts +1 -0
- package/dist/tests/fetch.test.js +154 -0
- package/dist/tests/fetch.test.js.map +1 -0
- package/dist/tests/fetch.test.mjs +154 -0
- package/dist/tests/fetch.test.mjs.map +1 -0
- package/dist/tests/sse.test.d.ts +1 -0
- package/dist/tests/sse.test.js +157 -0
- package/dist/tests/sse.test.js.map +1 -0
- package/dist/tests/sse.test.mjs +157 -0
- package/dist/tests/sse.test.mjs.map +1 -0
- package/dist/types.d.ts +176 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/types.messages.d.ts +86 -0
- package/dist/types.messages.js +2 -0
- package/dist/types.messages.js.map +1 -0
- package/dist/types.messages.mjs +2 -0
- package/dist/types.messages.mjs.map +1 -0
- package/dist/types.mjs +2 -0
- package/dist/types.mjs.map +1 -0
- package/dist/types.stream.d.ts +212 -0
- package/dist/types.stream.js +2 -0
- package/dist/types.stream.js.map +1 -0
- package/dist/types.stream.mjs +2 -0
- package/dist/types.stream.mjs.map +1 -0
- package/dist/ui/branching.d.ts +35 -0
- package/dist/ui/branching.js +145 -0
- package/dist/ui/branching.js.map +1 -0
- package/dist/ui/branching.mjs +145 -0
- package/dist/ui/branching.mjs.map +1 -0
- package/dist/ui/branching.test.d.ts +1 -0
- package/dist/ui/branching.test.js +369 -0
- package/dist/ui/branching.test.js.map +1 -0
- package/dist/ui/branching.test.mjs +369 -0
- package/dist/ui/branching.test.mjs.map +1 -0
- package/dist/ui/errors.d.ts +12 -0
- package/dist/ui/errors.js +10 -0
- package/dist/ui/errors.js.map +1 -0
- package/dist/ui/errors.mjs +10 -0
- package/dist/ui/errors.mjs.map +1 -0
- package/dist/ui/manager.d.ts +87 -0
- package/dist/ui/manager.js +160 -0
- package/dist/ui/manager.js.map +1 -0
- package/dist/ui/manager.mjs +160 -0
- package/dist/ui/manager.mjs.map +1 -0
- package/dist/ui/messages.d.ts +18 -0
- package/dist/ui/messages.js +67 -0
- package/dist/ui/messages.js.map +1 -0
- package/dist/ui/messages.mjs +67 -0
- package/dist/ui/messages.mjs.map +1 -0
- package/dist/ui/utils.d.ts +2 -0
- package/dist/ui/utils.js +11 -0
- package/dist/ui/utils.js.map +1 -0
- package/dist/ui/utils.mjs +11 -0
- package/dist/ui/utils.mjs.map +1 -0
- package/dist/utils/async_caller.d.ts +48 -0
- package/dist/utils/async_caller.js +147 -0
- package/dist/utils/async_caller.js.map +1 -0
- package/dist/utils/async_caller.mjs +147 -0
- package/dist/utils/async_caller.mjs.map +1 -0
- package/dist/utils/env.d.ts +1 -0
- package/dist/utils/env.js +13 -0
- package/dist/utils/env.js.map +1 -0
- package/dist/utils/env.mjs +13 -0
- package/dist/utils/env.mjs.map +1 -0
- package/dist/utils/signals.d.ts +1 -0
- package/dist/utils/signals.js +19 -0
- package/dist/utils/signals.js.map +1 -0
- package/dist/utils/signals.mjs +19 -0
- package/dist/utils/signals.mjs.map +1 -0
- package/dist/utils/sse.d.ts +8 -0
- package/dist/utils/sse.js +151 -0
- package/dist/utils/sse.js.map +1 -0
- package/dist/utils/sse.mjs +151 -0
- package/dist/utils/sse.mjs.map +1 -0
- package/dist/utils/stream.d.ts +13 -0
- package/dist/utils/stream.js +109 -0
- package/dist/utils/stream.js.map +1 -0
- package/dist/utils/stream.mjs +109 -0
- package/dist/utils/stream.mjs.map +1 -0
- package/package.json +56 -0
|
@@ -0,0 +1,154 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
|
|
2
|
+
import { Client } from "../client.js";
|
|
3
|
+
import { overrideFetchImplementation } from "../singletons/fetch.js";
|
|
4
|
+
describe.each([["global"], ["mocked"]])("Client uses %s fetch", (description) => {
|
|
5
|
+
let globalFetchMock;
|
|
6
|
+
let overriddenFetch;
|
|
7
|
+
let expectedFetchMock;
|
|
8
|
+
let unexpectedFetchMock;
|
|
9
|
+
beforeEach(() => {
|
|
10
|
+
globalFetchMock = vi.fn(() => Promise.resolve({
|
|
11
|
+
ok: true,
|
|
12
|
+
json: () => Promise.resolve({
|
|
13
|
+
batch_ingest_config: {
|
|
14
|
+
use_multipart_endpoint: true,
|
|
15
|
+
},
|
|
16
|
+
}),
|
|
17
|
+
text: () => Promise.resolve(""),
|
|
18
|
+
headers: new Headers({}),
|
|
19
|
+
}));
|
|
20
|
+
overriddenFetch = vi.fn(() => Promise.resolve({
|
|
21
|
+
ok: true,
|
|
22
|
+
json: () => Promise.resolve({
|
|
23
|
+
batch_ingest_config: {
|
|
24
|
+
use_multipart_endpoint: true,
|
|
25
|
+
},
|
|
26
|
+
}),
|
|
27
|
+
text: () => Promise.resolve(""),
|
|
28
|
+
headers: new Headers({}),
|
|
29
|
+
}));
|
|
30
|
+
expectedFetchMock =
|
|
31
|
+
description === "mocked" ? overriddenFetch : globalFetchMock;
|
|
32
|
+
unexpectedFetchMock =
|
|
33
|
+
description === "mocked" ? globalFetchMock : overriddenFetch;
|
|
34
|
+
if (description === "mocked") {
|
|
35
|
+
overrideFetchImplementation(overriddenFetch);
|
|
36
|
+
}
|
|
37
|
+
else {
|
|
38
|
+
overrideFetchImplementation(globalFetchMock);
|
|
39
|
+
}
|
|
40
|
+
// Mock global fetch
|
|
41
|
+
globalThis.fetch = globalFetchMock;
|
|
42
|
+
});
|
|
43
|
+
afterEach(() => {
|
|
44
|
+
vi.restoreAllMocks();
|
|
45
|
+
});
|
|
46
|
+
describe("createRuns", () => {
|
|
47
|
+
it("should create an example with the given input and generation", async () => {
|
|
48
|
+
const client = new Client({ apiKey: "test-api-key" });
|
|
49
|
+
const thread = await client.threads.create();
|
|
50
|
+
expect(expectedFetchMock).toHaveBeenCalledTimes(1);
|
|
51
|
+
expect(unexpectedFetchMock).not.toHaveBeenCalled();
|
|
52
|
+
vi.clearAllMocks(); // Clear all mocks before the next operation
|
|
53
|
+
// Then clear & run the function
|
|
54
|
+
await client.runs.create(thread.thread_id, "somegraph", {
|
|
55
|
+
input: { foo: "bar" },
|
|
56
|
+
});
|
|
57
|
+
expect(expectedFetchMock).toHaveBeenCalledTimes(1);
|
|
58
|
+
expect(unexpectedFetchMock).not.toHaveBeenCalled();
|
|
59
|
+
});
|
|
60
|
+
});
|
|
61
|
+
describe("header coalescing", () => {
|
|
62
|
+
it("should properly merge headers with conflicting name casing", async () => {
|
|
63
|
+
const client = new Client({ apiKey: "test-api-key" });
|
|
64
|
+
await client.threads.fetch("/test", {
|
|
65
|
+
headers: { "X-Api-Key": "custom-value" },
|
|
66
|
+
});
|
|
67
|
+
expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
|
|
68
|
+
headers: expect.objectContaining({
|
|
69
|
+
"x-api-key": "custom-value",
|
|
70
|
+
}),
|
|
71
|
+
}));
|
|
72
|
+
});
|
|
73
|
+
it("should properly merge headers from multiple sources", async () => {
|
|
74
|
+
const client = new Client({
|
|
75
|
+
apiKey: "test-api-key",
|
|
76
|
+
defaultHeaders: {
|
|
77
|
+
"x-default": "default-value",
|
|
78
|
+
"x-override": "default-value",
|
|
79
|
+
},
|
|
80
|
+
});
|
|
81
|
+
await client.threads.fetch("/test", {
|
|
82
|
+
headers: {
|
|
83
|
+
"x-custom": "custom-value",
|
|
84
|
+
"x-override": "custom-value",
|
|
85
|
+
},
|
|
86
|
+
});
|
|
87
|
+
expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
|
|
88
|
+
headers: expect.objectContaining({
|
|
89
|
+
"x-api-key": "test-api-key",
|
|
90
|
+
"x-default": "default-value",
|
|
91
|
+
"x-custom": "custom-value",
|
|
92
|
+
"x-override": "custom-value",
|
|
93
|
+
}),
|
|
94
|
+
}));
|
|
95
|
+
vi.clearAllMocks();
|
|
96
|
+
// Test with null/undefined values
|
|
97
|
+
await client.threads.fetch("/test", {
|
|
98
|
+
headers: {
|
|
99
|
+
"x-null": null,
|
|
100
|
+
"x-undefined": undefined,
|
|
101
|
+
"x-empty": "",
|
|
102
|
+
},
|
|
103
|
+
});
|
|
104
|
+
expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
|
|
105
|
+
headers: expect.objectContaining({
|
|
106
|
+
"x-api-key": "test-api-key",
|
|
107
|
+
"x-default": "default-value",
|
|
108
|
+
}),
|
|
109
|
+
}));
|
|
110
|
+
expect(expectedFetchMock).not.toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
|
|
111
|
+
headers: expect.objectContaining({
|
|
112
|
+
"x-null": null,
|
|
113
|
+
"x-undefined": undefined,
|
|
114
|
+
}),
|
|
115
|
+
}));
|
|
116
|
+
});
|
|
117
|
+
it("should handle Headers object input", async () => {
|
|
118
|
+
const client = new Client({ apiKey: "test-api-key" });
|
|
119
|
+
const headers = new Headers();
|
|
120
|
+
headers.append("x-custom", "custom-value");
|
|
121
|
+
headers.append("x-multi", "value1");
|
|
122
|
+
headers.append("x-multi", "value2");
|
|
123
|
+
await client.threads.fetch("/test", { headers });
|
|
124
|
+
expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
|
|
125
|
+
headers: expect.objectContaining({
|
|
126
|
+
"x-api-key": "test-api-key",
|
|
127
|
+
"x-custom": "custom-value",
|
|
128
|
+
"x-multi": "value1, value2",
|
|
129
|
+
}),
|
|
130
|
+
}));
|
|
131
|
+
});
|
|
132
|
+
it("should handle array of header tuples", async () => {
|
|
133
|
+
const client = new Client({
|
|
134
|
+
apiKey: "test-api-key",
|
|
135
|
+
defaultHeaders: {
|
|
136
|
+
"x-custom": "custom-value",
|
|
137
|
+
},
|
|
138
|
+
});
|
|
139
|
+
const headers = [
|
|
140
|
+
["x-multi", "value1"],
|
|
141
|
+
["x-multi", "value2"],
|
|
142
|
+
];
|
|
143
|
+
await client.threads.fetch("/test", { headers });
|
|
144
|
+
expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
|
|
145
|
+
headers: expect.objectContaining({
|
|
146
|
+
"x-api-key": "test-api-key",
|
|
147
|
+
"x-custom": "custom-value",
|
|
148
|
+
"x-multi": "value1, value2",
|
|
149
|
+
}),
|
|
150
|
+
}));
|
|
151
|
+
});
|
|
152
|
+
});
|
|
153
|
+
});
|
|
154
|
+
//# sourceMappingURL=fetch.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fetch.test.js","sourceRoot":"","sources":["../../src/tests/fetch.test.ts"],"names":[],"mappings":"OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ;OACjE,EAAE,MAAM,EAAE;OACV,EAAE,2BAA2B,EAAE;AAEtC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CACrC,sBAAsB,EACtB,CAAC,WAAmB,EAAE,EAAE;IACtB,IAAI,eAAyC,CAAC;IAC9C,IAAI,eAAyC,CAAC;IAE9C,IAAI,iBAA2C,CAAC;IAChD,IAAI,mBAA6C,CAAC;IAElD,UAAU,CAAC,GAAG,EAAE;QACd,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAC3B,OAAO,CAAC,OAAO,CAAC;YACd,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,GAAG,EAAE,CACT,OAAO,CAAC,OAAO,CAAC;gBACd,mBAAmB,EAAE;oBACnB,sBAAsB,EAAE,IAAI;iBAC7B;aACF,CAAC;YACJ,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,CAAC;SACzB,CAAC,CACH,CAAC;QACF,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAC3B,OAAO,CAAC,OAAO,CAAC;YACd,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,GAAG,EAAE,CACT,OAAO,CAAC,OAAO,CAAC;gBACd,mBAAmB,EAAE;oBACnB,sBAAsB,EAAE,IAAI;iBAC7B;aACF,CAAC;YACJ,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,CAAC;SACzB,CAAC,CACH,CAAC;QACF,iBAAiB;YACf,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;QAC/D,mBAAmB;YACjB,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;QAE/D,IAAI,WAAW,KAAK,QAAQ,EAAE,CAAC;YAC7B,2BAA2B,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC;aAAM,CAAC;YACN,2BAA2B,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC;QACD,oBAAoB;QACnB,UAAkB,CAAC,KAAK,GAAG,eAAe,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAC7C,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAEnD,EAAE,CAAC,aAAa,EAAE,CAAC,CAAC,4CAA4C;YAEhE,gCAAgC;YAChC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,WAAW,EAAE;gBACtD,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE;aACtB,CAAC,CAAC;YACH,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YACtD,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC3C,OAAO,EAAE,EAAE,WAAW,EAAE,cAAc,EAAE;aACzC,CAAC,CAAC;YACH,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;iBAC5B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;gBACxB,MAAM,EAAE,cAAc;gBACtB,cAAc,EAAE;oBACd,WAAW,EAAE,eAAe;oBAC5B,YAAY,EAAE,eAAe;iBAC9B;aACF,CAAC,CAAC;YAEH,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC3C,OAAO,EAAE;oBACP,UAAU,EAAE,cAAc;oBAC1B,YAAY,EAAE,cAAc;iBAC7B;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,WAAW,EAAE,eAAe;oBAC5B,UAAU,EAAE,cAAc;oBAC1B,YAAY,EAAE,cAAc;iBAC7B,CAAC;aACH,CAAC,CACH,CAAC;YAEF,EAAE,CAAC,aAAa,EAAE,CAAC;YAEnB,kCAAkC;YAClC,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC3C,OAAO,EAAE;oBACP,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,SAAS;oBACxB,SAAS,EAAE,EAAE;iBACd;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,WAAW,EAAE,eAAe;iBAC7B,CAAC;aACH,CAAC,CACH,CAAC;YACF,MAAM,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAChD,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,SAAS;iBACzB,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YACtD,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;YAC9B,OAAO,CAAC,MAAM,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;YAC3C,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YACpC,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAEpC,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAE1D,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,UAAU,EAAE,cAAc;oBAC1B,SAAS,EAAE,gBAAgB;iBAC5B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;gBACxB,MAAM,EAAE,cAAc;gBACtB,cAAc,EAAE;oBACd,UAAU,EAAE,cAAc;iBAC3B;aACF,CAAC,CAAC;YACH,MAAM,OAAO,GAAG;gBACd,CAAC,SAAS,EAAE,QAAQ,CAAC;gBACrB,CAAC,SAAS,EAAE,QAAQ,CAAC;aACtB,CAAC;YAEF,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAE1D,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,UAAU,EAAE,cAAc;oBAC1B,SAAS,EAAE,gBAAgB;iBAC5B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CACF,CAAC"}
|
|
@@ -0,0 +1,154 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach, vi } from "vitest";
|
|
2
|
+
import { Client } from "../client.mjs";
|
|
3
|
+
import { overrideFetchImplementation } from "../singletons/fetch.mjs";
|
|
4
|
+
describe.each([["global"], ["mocked"]])("Client uses %s fetch", (description) => {
|
|
5
|
+
let globalFetchMock;
|
|
6
|
+
let overriddenFetch;
|
|
7
|
+
let expectedFetchMock;
|
|
8
|
+
let unexpectedFetchMock;
|
|
9
|
+
beforeEach(() => {
|
|
10
|
+
globalFetchMock = vi.fn(() => Promise.resolve({
|
|
11
|
+
ok: true,
|
|
12
|
+
json: () => Promise.resolve({
|
|
13
|
+
batch_ingest_config: {
|
|
14
|
+
use_multipart_endpoint: true,
|
|
15
|
+
},
|
|
16
|
+
}),
|
|
17
|
+
text: () => Promise.resolve(""),
|
|
18
|
+
headers: new Headers({}),
|
|
19
|
+
}));
|
|
20
|
+
overriddenFetch = vi.fn(() => Promise.resolve({
|
|
21
|
+
ok: true,
|
|
22
|
+
json: () => Promise.resolve({
|
|
23
|
+
batch_ingest_config: {
|
|
24
|
+
use_multipart_endpoint: true,
|
|
25
|
+
},
|
|
26
|
+
}),
|
|
27
|
+
text: () => Promise.resolve(""),
|
|
28
|
+
headers: new Headers({}),
|
|
29
|
+
}));
|
|
30
|
+
expectedFetchMock =
|
|
31
|
+
description === "mocked" ? overriddenFetch : globalFetchMock;
|
|
32
|
+
unexpectedFetchMock =
|
|
33
|
+
description === "mocked" ? globalFetchMock : overriddenFetch;
|
|
34
|
+
if (description === "mocked") {
|
|
35
|
+
overrideFetchImplementation(overriddenFetch);
|
|
36
|
+
}
|
|
37
|
+
else {
|
|
38
|
+
overrideFetchImplementation(globalFetchMock);
|
|
39
|
+
}
|
|
40
|
+
// Mock global fetch
|
|
41
|
+
globalThis.fetch = globalFetchMock;
|
|
42
|
+
});
|
|
43
|
+
afterEach(() => {
|
|
44
|
+
vi.restoreAllMocks();
|
|
45
|
+
});
|
|
46
|
+
describe("createRuns", () => {
|
|
47
|
+
it("should create an example with the given input and generation", async () => {
|
|
48
|
+
const client = new Client({ apiKey: "test-api-key" });
|
|
49
|
+
const thread = await client.threads.create();
|
|
50
|
+
expect(expectedFetchMock).toHaveBeenCalledTimes(1);
|
|
51
|
+
expect(unexpectedFetchMock).not.toHaveBeenCalled();
|
|
52
|
+
vi.clearAllMocks(); // Clear all mocks before the next operation
|
|
53
|
+
// Then clear & run the function
|
|
54
|
+
await client.runs.create(thread.thread_id, "somegraph", {
|
|
55
|
+
input: { foo: "bar" },
|
|
56
|
+
});
|
|
57
|
+
expect(expectedFetchMock).toHaveBeenCalledTimes(1);
|
|
58
|
+
expect(unexpectedFetchMock).not.toHaveBeenCalled();
|
|
59
|
+
});
|
|
60
|
+
});
|
|
61
|
+
describe("header coalescing", () => {
|
|
62
|
+
it("should properly merge headers with conflicting name casing", async () => {
|
|
63
|
+
const client = new Client({ apiKey: "test-api-key" });
|
|
64
|
+
await client.threads.fetch("/test", {
|
|
65
|
+
headers: { "X-Api-Key": "custom-value" },
|
|
66
|
+
});
|
|
67
|
+
expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
|
|
68
|
+
headers: expect.objectContaining({
|
|
69
|
+
"x-api-key": "custom-value",
|
|
70
|
+
}),
|
|
71
|
+
}));
|
|
72
|
+
});
|
|
73
|
+
it("should properly merge headers from multiple sources", async () => {
|
|
74
|
+
const client = new Client({
|
|
75
|
+
apiKey: "test-api-key",
|
|
76
|
+
defaultHeaders: {
|
|
77
|
+
"x-default": "default-value",
|
|
78
|
+
"x-override": "default-value",
|
|
79
|
+
},
|
|
80
|
+
});
|
|
81
|
+
await client.threads.fetch("/test", {
|
|
82
|
+
headers: {
|
|
83
|
+
"x-custom": "custom-value",
|
|
84
|
+
"x-override": "custom-value",
|
|
85
|
+
},
|
|
86
|
+
});
|
|
87
|
+
expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
|
|
88
|
+
headers: expect.objectContaining({
|
|
89
|
+
"x-api-key": "test-api-key",
|
|
90
|
+
"x-default": "default-value",
|
|
91
|
+
"x-custom": "custom-value",
|
|
92
|
+
"x-override": "custom-value",
|
|
93
|
+
}),
|
|
94
|
+
}));
|
|
95
|
+
vi.clearAllMocks();
|
|
96
|
+
// Test with null/undefined values
|
|
97
|
+
await client.threads.fetch("/test", {
|
|
98
|
+
headers: {
|
|
99
|
+
"x-null": null,
|
|
100
|
+
"x-undefined": undefined,
|
|
101
|
+
"x-empty": "",
|
|
102
|
+
},
|
|
103
|
+
});
|
|
104
|
+
expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
|
|
105
|
+
headers: expect.objectContaining({
|
|
106
|
+
"x-api-key": "test-api-key",
|
|
107
|
+
"x-default": "default-value",
|
|
108
|
+
}),
|
|
109
|
+
}));
|
|
110
|
+
expect(expectedFetchMock).not.toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
|
|
111
|
+
headers: expect.objectContaining({
|
|
112
|
+
"x-null": null,
|
|
113
|
+
"x-undefined": undefined,
|
|
114
|
+
}),
|
|
115
|
+
}));
|
|
116
|
+
});
|
|
117
|
+
it("should handle Headers object input", async () => {
|
|
118
|
+
const client = new Client({ apiKey: "test-api-key" });
|
|
119
|
+
const headers = new Headers();
|
|
120
|
+
headers.append("x-custom", "custom-value");
|
|
121
|
+
headers.append("x-multi", "value1");
|
|
122
|
+
headers.append("x-multi", "value2");
|
|
123
|
+
await client.threads.fetch("/test", { headers });
|
|
124
|
+
expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
|
|
125
|
+
headers: expect.objectContaining({
|
|
126
|
+
"x-api-key": "test-api-key",
|
|
127
|
+
"x-custom": "custom-value",
|
|
128
|
+
"x-multi": "value1, value2",
|
|
129
|
+
}),
|
|
130
|
+
}));
|
|
131
|
+
});
|
|
132
|
+
it("should handle array of header tuples", async () => {
|
|
133
|
+
const client = new Client({
|
|
134
|
+
apiKey: "test-api-key",
|
|
135
|
+
defaultHeaders: {
|
|
136
|
+
"x-custom": "custom-value",
|
|
137
|
+
},
|
|
138
|
+
});
|
|
139
|
+
const headers = [
|
|
140
|
+
["x-multi", "value1"],
|
|
141
|
+
["x-multi", "value2"],
|
|
142
|
+
];
|
|
143
|
+
await client.threads.fetch("/test", { headers });
|
|
144
|
+
expect(expectedFetchMock).toHaveBeenCalledWith(expect.any(URL), expect.objectContaining({
|
|
145
|
+
headers: expect.objectContaining({
|
|
146
|
+
"x-api-key": "test-api-key",
|
|
147
|
+
"x-custom": "custom-value",
|
|
148
|
+
"x-multi": "value1, value2",
|
|
149
|
+
}),
|
|
150
|
+
}));
|
|
151
|
+
});
|
|
152
|
+
});
|
|
153
|
+
});
|
|
154
|
+
//# sourceMappingURL=fetch.test.mjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fetch.test.mjs","sourceRoot":"","sources":["../../src/tests/fetch.test.ts"],"names":[],"mappings":"OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ;OACjE,EAAE,MAAM,EAAE;OACV,EAAE,2BAA2B,EAAE;AAEtC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CACrC,sBAAsB,EACtB,CAAC,WAAmB,EAAE,EAAE;IACtB,IAAI,eAAyC,CAAC;IAC9C,IAAI,eAAyC,CAAC;IAE9C,IAAI,iBAA2C,CAAC;IAChD,IAAI,mBAA6C,CAAC;IAElD,UAAU,CAAC,GAAG,EAAE;QACd,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAC3B,OAAO,CAAC,OAAO,CAAC;YACd,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,GAAG,EAAE,CACT,OAAO,CAAC,OAAO,CAAC;gBACd,mBAAmB,EAAE;oBACnB,sBAAsB,EAAE,IAAI;iBAC7B;aACF,CAAC;YACJ,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,CAAC;SACzB,CAAC,CACH,CAAC;QACF,eAAe,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAC3B,OAAO,CAAC,OAAO,CAAC;YACd,EAAE,EAAE,IAAI;YACR,IAAI,EAAE,GAAG,EAAE,CACT,OAAO,CAAC,OAAO,CAAC;gBACd,mBAAmB,EAAE;oBACnB,sBAAsB,EAAE,IAAI;iBAC7B;aACF,CAAC;YACJ,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC/B,OAAO,EAAE,IAAI,OAAO,CAAC,EAAE,CAAC;SACzB,CAAC,CACH,CAAC;QACF,iBAAiB;YACf,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;QAC/D,mBAAmB;YACjB,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;QAE/D,IAAI,WAAW,KAAK,QAAQ,EAAE,CAAC;YAC7B,2BAA2B,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC;aAAM,CAAC;YACN,2BAA2B,CAAC,eAAe,CAAC,CAAC;QAC/C,CAAC;QACD,oBAAoB;QACnB,UAAkB,CAAC,KAAK,GAAG,eAAe,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,eAAe,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAC7C,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAEnD,EAAE,CAAC,aAAa,EAAE,CAAC,CAAC,4CAA4C;YAEhE,gCAAgC;YAChC,MAAM,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,EAAE,WAAW,EAAE;gBACtD,KAAK,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE;aACtB,CAAC,CAAC;YACH,MAAM,CAAC,iBAAiB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACnD,MAAM,CAAC,mBAAmB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,4DAA4D,EAAE,KAAK,IAAI,EAAE;YAC1E,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YACtD,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC3C,OAAO,EAAE,EAAE,WAAW,EAAE,cAAc,EAAE;aACzC,CAAC,CAAC;YACH,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;iBAC5B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;gBACxB,MAAM,EAAE,cAAc;gBACtB,cAAc,EAAE;oBACd,WAAW,EAAE,eAAe;oBAC5B,YAAY,EAAE,eAAe;iBAC9B;aACF,CAAC,CAAC;YAEH,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC3C,OAAO,EAAE;oBACP,UAAU,EAAE,cAAc;oBAC1B,YAAY,EAAE,cAAc;iBAC7B;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,WAAW,EAAE,eAAe;oBAC5B,UAAU,EAAE,cAAc;oBAC1B,YAAY,EAAE,cAAc;iBAC7B,CAAC;aACH,CAAC,CACH,CAAC;YAEF,EAAE,CAAC,aAAa,EAAE,CAAC;YAEnB,kCAAkC;YAClC,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE;gBAC3C,OAAO,EAAE;oBACP,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,SAAS;oBACxB,SAAS,EAAE,EAAE;iBACd;aACF,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,WAAW,EAAE,eAAe;iBAC7B,CAAC;aACH,CAAC,CACH,CAAC;YACF,MAAM,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAChD,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,SAAS;iBACzB,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC,CAAC;YACtD,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;YAC9B,OAAO,CAAC,MAAM,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;YAC3C,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YACpC,OAAO,CAAC,MAAM,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;YAEpC,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAE1D,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,UAAU,EAAE,cAAc;oBAC1B,SAAS,EAAE,gBAAgB;iBAC5B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;gBACxB,MAAM,EAAE,cAAc;gBACtB,cAAc,EAAE;oBACd,UAAU,EAAE,cAAc;iBAC3B;aACF,CAAC,CAAC;YACH,MAAM,OAAO,GAAG;gBACd,CAAC,SAAS,EAAE,QAAQ,CAAC;gBACrB,CAAC,SAAS,EAAE,QAAQ,CAAC;aACtB,CAAC;YAEF,MAAO,MAAM,CAAC,OAAe,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;YAE1D,MAAM,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAC5C,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EACf,MAAM,CAAC,gBAAgB,CAAC;gBACtB,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC;oBAC/B,WAAW,EAAE,cAAc;oBAC3B,UAAU,EAAE,cAAc;oBAC1B,SAAS,EAAE,gBAAgB;iBAC5B,CAAC;aACH,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CACF,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,157 @@
|
|
|
1
|
+
import { describe, test, expect } from "vitest";
|
|
2
|
+
import { Readable } from "node:stream";
|
|
3
|
+
import { IterableReadableStream } from "../utils/stream.js";
|
|
4
|
+
import { BytesLineDecoder, SSEDecoder } from "../utils/sse.js";
|
|
5
|
+
const gather = async (stream) => {
|
|
6
|
+
const results = [];
|
|
7
|
+
const iterator = IterableReadableStream.fromReadableStream(stream);
|
|
8
|
+
for await (const chunk of iterator)
|
|
9
|
+
results.push(chunk);
|
|
10
|
+
return results;
|
|
11
|
+
};
|
|
12
|
+
const textEncoder = new TextEncoder();
|
|
13
|
+
const textDecoder = new TextDecoder();
|
|
14
|
+
describe("BytesLineDecoder", () => {
|
|
15
|
+
const createStream = (chunks) => {
|
|
16
|
+
return Readable.toWeb(Readable.from(chunks));
|
|
17
|
+
};
|
|
18
|
+
test("handles single line with newline", async () => {
|
|
19
|
+
const input = createStream([textEncoder.encode("hello\n")]);
|
|
20
|
+
const decoded = input.pipeThrough(BytesLineDecoder());
|
|
21
|
+
const results = await gather(decoded);
|
|
22
|
+
expect(results.length).toBe(1);
|
|
23
|
+
expect(textDecoder.decode(results[0])).toBe("hello");
|
|
24
|
+
});
|
|
25
|
+
test("handles multiple lines", async () => {
|
|
26
|
+
const input = createStream([textEncoder.encode("line1\nline2\nline3\n")]);
|
|
27
|
+
const decoded = input.pipeThrough(BytesLineDecoder());
|
|
28
|
+
const results = await gather(decoded);
|
|
29
|
+
expect(results.length).toBe(3);
|
|
30
|
+
expect(textDecoder.decode(results[0])).toBe("line1");
|
|
31
|
+
expect(textDecoder.decode(results[1])).toBe("line2");
|
|
32
|
+
expect(textDecoder.decode(results[2])).toBe("line3");
|
|
33
|
+
});
|
|
34
|
+
test("handles split chunks", async () => {
|
|
35
|
+
const input = createStream([
|
|
36
|
+
textEncoder.encode("li"),
|
|
37
|
+
textEncoder.encode("ne1\nli"),
|
|
38
|
+
textEncoder.encode("ne2\n"),
|
|
39
|
+
]);
|
|
40
|
+
const decoded = input.pipeThrough(BytesLineDecoder());
|
|
41
|
+
const results = await gather(decoded);
|
|
42
|
+
expect(results.length).toBe(2);
|
|
43
|
+
expect(textDecoder.decode(results[0])).toBe("line1");
|
|
44
|
+
expect(textDecoder.decode(results[1])).toBe("line2");
|
|
45
|
+
});
|
|
46
|
+
test("handles CR LF line endings", async () => {
|
|
47
|
+
const input = createStream([textEncoder.encode("line1\r\nline2\r\n")]);
|
|
48
|
+
const decoded = input.pipeThrough(BytesLineDecoder());
|
|
49
|
+
const results = await gather(decoded);
|
|
50
|
+
expect(results.length).toBe(2);
|
|
51
|
+
expect(textDecoder.decode(results[0])).toBe("line1");
|
|
52
|
+
expect(textDecoder.decode(results[1])).toBe("line2");
|
|
53
|
+
});
|
|
54
|
+
test("handles split CR LF", async () => {
|
|
55
|
+
const input = createStream([
|
|
56
|
+
textEncoder.encode("line1\r"),
|
|
57
|
+
textEncoder.encode("\nline2\r\n"),
|
|
58
|
+
]);
|
|
59
|
+
const decoded = input.pipeThrough(BytesLineDecoder());
|
|
60
|
+
const results = await gather(decoded);
|
|
61
|
+
expect(results.length).toBe(2);
|
|
62
|
+
expect(textDecoder.decode(results[0])).toBe("line1");
|
|
63
|
+
expect(textDecoder.decode(results[1])).toBe("line2");
|
|
64
|
+
});
|
|
65
|
+
test("handles stale line", async () => {
|
|
66
|
+
const input = createStream([textEncoder.encode("hello")]);
|
|
67
|
+
const decoded = input.pipeThrough(BytesLineDecoder());
|
|
68
|
+
const results = await gather(decoded);
|
|
69
|
+
expect(results.length).toBe(1);
|
|
70
|
+
expect(textDecoder.decode(results[0])).toBe("hello");
|
|
71
|
+
});
|
|
72
|
+
});
|
|
73
|
+
describe("SSEDecoder", () => {
|
|
74
|
+
const createStream = (lines) => {
|
|
75
|
+
return Readable.toWeb(Readable.from(lines.map((line) => textEncoder.encode(line))));
|
|
76
|
+
};
|
|
77
|
+
test("decodes simple event", async () => {
|
|
78
|
+
const input = createStream([
|
|
79
|
+
"event: test\n",
|
|
80
|
+
'data: {"message": "hello"}\n',
|
|
81
|
+
"\n",
|
|
82
|
+
]);
|
|
83
|
+
const decoded = input
|
|
84
|
+
.pipeThrough(BytesLineDecoder())
|
|
85
|
+
.pipeThrough(SSEDecoder());
|
|
86
|
+
const results = await gather(decoded);
|
|
87
|
+
expect(results.length).toBe(1);
|
|
88
|
+
expect(results[0]).toEqual({
|
|
89
|
+
event: "test",
|
|
90
|
+
data: { message: "hello" },
|
|
91
|
+
});
|
|
92
|
+
});
|
|
93
|
+
test("ignores comments", async () => {
|
|
94
|
+
const input = createStream([
|
|
95
|
+
": this is a comment\n",
|
|
96
|
+
"event: test\n",
|
|
97
|
+
'data: {"message": "hello"}\n',
|
|
98
|
+
]);
|
|
99
|
+
const decoded = input
|
|
100
|
+
.pipeThrough(BytesLineDecoder())
|
|
101
|
+
.pipeThrough(SSEDecoder());
|
|
102
|
+
const results = await gather(decoded);
|
|
103
|
+
expect(results.length).toBe(1);
|
|
104
|
+
expect(results[0]).toEqual({
|
|
105
|
+
event: "test",
|
|
106
|
+
data: { message: "hello" },
|
|
107
|
+
});
|
|
108
|
+
});
|
|
109
|
+
test("handles multiple events", async () => {
|
|
110
|
+
const input = createStream([
|
|
111
|
+
"event: test1\n",
|
|
112
|
+
'data: {"message": "hello"}\n',
|
|
113
|
+
"\n",
|
|
114
|
+
"event: test2\n",
|
|
115
|
+
'data: {"message": "world"}\n',
|
|
116
|
+
"\n",
|
|
117
|
+
]);
|
|
118
|
+
const decoded = input
|
|
119
|
+
.pipeThrough(BytesLineDecoder())
|
|
120
|
+
.pipeThrough(SSEDecoder());
|
|
121
|
+
const results = await gather(decoded);
|
|
122
|
+
expect(results.length).toBe(2);
|
|
123
|
+
expect(results[0]).toEqual({
|
|
124
|
+
event: "test1",
|
|
125
|
+
data: { message: "hello" },
|
|
126
|
+
});
|
|
127
|
+
expect(results[1]).toEqual({
|
|
128
|
+
event: "test2",
|
|
129
|
+
data: { message: "world" },
|
|
130
|
+
});
|
|
131
|
+
});
|
|
132
|
+
test("end event without data", async () => {
|
|
133
|
+
const input = createStream(["event: test\n"]);
|
|
134
|
+
const decoded = input
|
|
135
|
+
.pipeThrough(BytesLineDecoder())
|
|
136
|
+
.pipeThrough(SSEDecoder());
|
|
137
|
+
const results = await gather(decoded);
|
|
138
|
+
expect(results.length).toBe(1);
|
|
139
|
+
expect(results[0]).toEqual({
|
|
140
|
+
event: "test",
|
|
141
|
+
data: null,
|
|
142
|
+
});
|
|
143
|
+
});
|
|
144
|
+
test("end event without newline", async () => {
|
|
145
|
+
const input = createStream(["event: end"]);
|
|
146
|
+
const decoded = input
|
|
147
|
+
.pipeThrough(BytesLineDecoder())
|
|
148
|
+
.pipeThrough(SSEDecoder());
|
|
149
|
+
const results = await gather(decoded);
|
|
150
|
+
expect(results.length).toBe(1);
|
|
151
|
+
expect(results[0]).toEqual({
|
|
152
|
+
event: "end",
|
|
153
|
+
data: null,
|
|
154
|
+
});
|
|
155
|
+
});
|
|
156
|
+
});
|
|
157
|
+
//# sourceMappingURL=sse.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sse.test.js","sourceRoot":"","sources":["../../src/tests/sse.test.ts"],"names":[],"mappings":"OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,QAAQ;OACxC,EAAE,QAAQ,EAAE,MAAM,aAAa;OAC/B,EAAE,sBAAsB,EAAE;OAC1B,EAAE,gBAAgB,EAAE,UAAU,EAAE;AAEvC,MAAM,MAAM,GAAG,KAAK,EAAK,MAAyB,EAAgB,EAAE;IAClE,MAAM,OAAO,GAAQ,EAAE,CAAC;IACxB,MAAM,QAAQ,GAAG,sBAAsB,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;IACnE,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,QAAQ;QAAE,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxD,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;AACtC,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAC;AAEtC,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,MAAM,YAAY,GAAG,CAAC,MAAoB,EAAE,EAAE;QAC5C,OAAO,QAAQ,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAA+B,CAAC;IAC7E,CAAC,CAAC;IAEF,IAAI,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAClD,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;QAC5D,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,uBAAuB,CAAC,CAAC,CAAC,CAAC;QAC1E,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,KAAK,GAAG,YAAY,CAAC;YACzB,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC;YACxB,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC;YAC7B,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC;SAC5B,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC5C,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC;QACvE,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;QACrC,MAAM,KAAK,GAAG,YAAY,CAAC;YACzB,WAAW,CAAC,MAAM,CAAC,SAAS,CAAC;YAC7B,WAAW,CAAC,MAAM,CAAC,aAAa,CAAC;SAClC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QACpC,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,KAAK,CAAC,WAAW,CAAC,gBAAgB,EAAE,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QAEtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,MAAM,YAAY,GAAG,CAAC,KAAe,EAAE,EAAE;QACvC,OAAO,QAAQ,CAAC,KAAK,CACnB,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAC/B,CAAC;IAClC,CAAC,CAAC;IAEF,IAAI,CAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,KAAK,GAAG,YAAY,CAAC;YACzB,eAAe;YACf,8BAA8B;YAC9B,IAAI;SACL,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,KAAK;aAClB,WAAW,CAAC,gBAAgB,EAAE,CAAC;aAC/B,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,MAAM;YACb,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;SAC3B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE;QAClC,MAAM,KAAK,GAAG,YAAY,CAAC;YACzB,uBAAuB;YACvB,eAAe;YACf,8BAA8B;SAC/B,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,KAAK;aAClB,WAAW,CAAC,gBAAgB,EAAE,CAAC;aAC/B,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,MAAM;YACb,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;SAC3B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACzC,MAAM,KAAK,GAAG,YAAY,CAAC;YACzB,gBAAgB;YAChB,8BAA8B;YAC9B,IAAI;YACJ,gBAAgB;YAChB,8BAA8B;YAC9B,IAAI;SACL,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,KAAK;aAClB,WAAW,CAAC,gBAAgB,EAAE,CAAC;aAC/B,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;SAC3B,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE;SAC3B,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACxC,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,KAAK;aAClB,WAAW,CAAC,gBAAgB,EAAE,CAAC;aAC/B,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,MAAM;YACb,IAAI,EAAE,IAAI;SACX,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAI,CAAC,2BAA2B,EAAE,KAAK,IAAI,EAAE;QAC3C,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC;QAC3C,MAAM,OAAO,GAAG,KAAK;aAClB,WAAW,CAAC,gBAAgB,EAAE,CAAC;aAC/B,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;QAE7B,MAAM,OAAO,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC;QACtC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;YACzB,KAAK,EAAE,KAAK;YACZ,IAAI,EAAE,IAAI;SACX,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,157 @@
|
|
|
1
|
+
import { describe, test, expect } from "vitest";
|
|
2
|
+
import { Readable } from "node:stream";
|
|
3
|
+
import { IterableReadableStream } from "../utils/stream.mjs";
|
|
4
|
+
import { BytesLineDecoder, SSEDecoder } from "../utils/sse.mjs";
|
|
5
|
+
const gather = async (stream) => {
|
|
6
|
+
const results = [];
|
|
7
|
+
const iterator = IterableReadableStream.fromReadableStream(stream);
|
|
8
|
+
for await (const chunk of iterator)
|
|
9
|
+
results.push(chunk);
|
|
10
|
+
return results;
|
|
11
|
+
};
|
|
12
|
+
const textEncoder = new TextEncoder();
|
|
13
|
+
const textDecoder = new TextDecoder();
|
|
14
|
+
describe("BytesLineDecoder", () => {
|
|
15
|
+
const createStream = (chunks) => {
|
|
16
|
+
return Readable.toWeb(Readable.from(chunks));
|
|
17
|
+
};
|
|
18
|
+
test("handles single line with newline", async () => {
|
|
19
|
+
const input = createStream([textEncoder.encode("hello\n")]);
|
|
20
|
+
const decoded = input.pipeThrough(BytesLineDecoder());
|
|
21
|
+
const results = await gather(decoded);
|
|
22
|
+
expect(results.length).toBe(1);
|
|
23
|
+
expect(textDecoder.decode(results[0])).toBe("hello");
|
|
24
|
+
});
|
|
25
|
+
test("handles multiple lines", async () => {
|
|
26
|
+
const input = createStream([textEncoder.encode("line1\nline2\nline3\n")]);
|
|
27
|
+
const decoded = input.pipeThrough(BytesLineDecoder());
|
|
28
|
+
const results = await gather(decoded);
|
|
29
|
+
expect(results.length).toBe(3);
|
|
30
|
+
expect(textDecoder.decode(results[0])).toBe("line1");
|
|
31
|
+
expect(textDecoder.decode(results[1])).toBe("line2");
|
|
32
|
+
expect(textDecoder.decode(results[2])).toBe("line3");
|
|
33
|
+
});
|
|
34
|
+
test("handles split chunks", async () => {
|
|
35
|
+
const input = createStream([
|
|
36
|
+
textEncoder.encode("li"),
|
|
37
|
+
textEncoder.encode("ne1\nli"),
|
|
38
|
+
textEncoder.encode("ne2\n"),
|
|
39
|
+
]);
|
|
40
|
+
const decoded = input.pipeThrough(BytesLineDecoder());
|
|
41
|
+
const results = await gather(decoded);
|
|
42
|
+
expect(results.length).toBe(2);
|
|
43
|
+
expect(textDecoder.decode(results[0])).toBe("line1");
|
|
44
|
+
expect(textDecoder.decode(results[1])).toBe("line2");
|
|
45
|
+
});
|
|
46
|
+
test("handles CR LF line endings", async () => {
|
|
47
|
+
const input = createStream([textEncoder.encode("line1\r\nline2\r\n")]);
|
|
48
|
+
const decoded = input.pipeThrough(BytesLineDecoder());
|
|
49
|
+
const results = await gather(decoded);
|
|
50
|
+
expect(results.length).toBe(2);
|
|
51
|
+
expect(textDecoder.decode(results[0])).toBe("line1");
|
|
52
|
+
expect(textDecoder.decode(results[1])).toBe("line2");
|
|
53
|
+
});
|
|
54
|
+
test("handles split CR LF", async () => {
|
|
55
|
+
const input = createStream([
|
|
56
|
+
textEncoder.encode("line1\r"),
|
|
57
|
+
textEncoder.encode("\nline2\r\n"),
|
|
58
|
+
]);
|
|
59
|
+
const decoded = input.pipeThrough(BytesLineDecoder());
|
|
60
|
+
const results = await gather(decoded);
|
|
61
|
+
expect(results.length).toBe(2);
|
|
62
|
+
expect(textDecoder.decode(results[0])).toBe("line1");
|
|
63
|
+
expect(textDecoder.decode(results[1])).toBe("line2");
|
|
64
|
+
});
|
|
65
|
+
test("handles stale line", async () => {
|
|
66
|
+
const input = createStream([textEncoder.encode("hello")]);
|
|
67
|
+
const decoded = input.pipeThrough(BytesLineDecoder());
|
|
68
|
+
const results = await gather(decoded);
|
|
69
|
+
expect(results.length).toBe(1);
|
|
70
|
+
expect(textDecoder.decode(results[0])).toBe("hello");
|
|
71
|
+
});
|
|
72
|
+
});
|
|
73
|
+
describe("SSEDecoder", () => {
|
|
74
|
+
const createStream = (lines) => {
|
|
75
|
+
return Readable.toWeb(Readable.from(lines.map((line) => textEncoder.encode(line))));
|
|
76
|
+
};
|
|
77
|
+
test("decodes simple event", async () => {
|
|
78
|
+
const input = createStream([
|
|
79
|
+
"event: test\n",
|
|
80
|
+
'data: {"message": "hello"}\n',
|
|
81
|
+
"\n",
|
|
82
|
+
]);
|
|
83
|
+
const decoded = input
|
|
84
|
+
.pipeThrough(BytesLineDecoder())
|
|
85
|
+
.pipeThrough(SSEDecoder());
|
|
86
|
+
const results = await gather(decoded);
|
|
87
|
+
expect(results.length).toBe(1);
|
|
88
|
+
expect(results[0]).toEqual({
|
|
89
|
+
event: "test",
|
|
90
|
+
data: { message: "hello" },
|
|
91
|
+
});
|
|
92
|
+
});
|
|
93
|
+
test("ignores comments", async () => {
|
|
94
|
+
const input = createStream([
|
|
95
|
+
": this is a comment\n",
|
|
96
|
+
"event: test\n",
|
|
97
|
+
'data: {"message": "hello"}\n',
|
|
98
|
+
]);
|
|
99
|
+
const decoded = input
|
|
100
|
+
.pipeThrough(BytesLineDecoder())
|
|
101
|
+
.pipeThrough(SSEDecoder());
|
|
102
|
+
const results = await gather(decoded);
|
|
103
|
+
expect(results.length).toBe(1);
|
|
104
|
+
expect(results[0]).toEqual({
|
|
105
|
+
event: "test",
|
|
106
|
+
data: { message: "hello" },
|
|
107
|
+
});
|
|
108
|
+
});
|
|
109
|
+
test("handles multiple events", async () => {
|
|
110
|
+
const input = createStream([
|
|
111
|
+
"event: test1\n",
|
|
112
|
+
'data: {"message": "hello"}\n',
|
|
113
|
+
"\n",
|
|
114
|
+
"event: test2\n",
|
|
115
|
+
'data: {"message": "world"}\n',
|
|
116
|
+
"\n",
|
|
117
|
+
]);
|
|
118
|
+
const decoded = input
|
|
119
|
+
.pipeThrough(BytesLineDecoder())
|
|
120
|
+
.pipeThrough(SSEDecoder());
|
|
121
|
+
const results = await gather(decoded);
|
|
122
|
+
expect(results.length).toBe(2);
|
|
123
|
+
expect(results[0]).toEqual({
|
|
124
|
+
event: "test1",
|
|
125
|
+
data: { message: "hello" },
|
|
126
|
+
});
|
|
127
|
+
expect(results[1]).toEqual({
|
|
128
|
+
event: "test2",
|
|
129
|
+
data: { message: "world" },
|
|
130
|
+
});
|
|
131
|
+
});
|
|
132
|
+
test("end event without data", async () => {
|
|
133
|
+
const input = createStream(["event: test\n"]);
|
|
134
|
+
const decoded = input
|
|
135
|
+
.pipeThrough(BytesLineDecoder())
|
|
136
|
+
.pipeThrough(SSEDecoder());
|
|
137
|
+
const results = await gather(decoded);
|
|
138
|
+
expect(results.length).toBe(1);
|
|
139
|
+
expect(results[0]).toEqual({
|
|
140
|
+
event: "test",
|
|
141
|
+
data: null,
|
|
142
|
+
});
|
|
143
|
+
});
|
|
144
|
+
test("end event without newline", async () => {
|
|
145
|
+
const input = createStream(["event: end"]);
|
|
146
|
+
const decoded = input
|
|
147
|
+
.pipeThrough(BytesLineDecoder())
|
|
148
|
+
.pipeThrough(SSEDecoder());
|
|
149
|
+
const results = await gather(decoded);
|
|
150
|
+
expect(results.length).toBe(1);
|
|
151
|
+
expect(results[0]).toEqual({
|
|
152
|
+
event: "end",
|
|
153
|
+
data: null,
|
|
154
|
+
});
|
|
155
|
+
});
|
|
156
|
+
});
|
|
157
|
+
//# sourceMappingURL=sse.test.mjs.map
|