aismemory 0.5.1 → 0.5.3

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.
Files changed (42) hide show
  1. package/dist/__tests__/ais-http-error.test.d.ts +1 -0
  2. package/dist/__tests__/ais-http-error.test.js +49 -0
  3. package/dist/__tests__/ais-http-error.test.js.map +1 -0
  4. package/dist/__tests__/auto-enroll-key-e2e.test.d.ts +1 -0
  5. package/dist/__tests__/auto-enroll-key-e2e.test.js +116 -0
  6. package/dist/__tests__/auto-enroll-key-e2e.test.js.map +1 -0
  7. package/dist/__tests__/auto-enroll-key.test.d.ts +1 -0
  8. package/dist/__tests__/auto-enroll-key.test.js +165 -0
  9. package/dist/__tests__/auto-enroll-key.test.js.map +1 -0
  10. package/dist/__tests__/device-flow-recovery.test.js +9 -5
  11. package/dist/__tests__/device-flow-recovery.test.js.map +1 -1
  12. package/dist/__tests__/device-poll.test.d.ts +12 -0
  13. package/dist/__tests__/device-poll.test.js +191 -0
  14. package/dist/__tests__/device-poll.test.js.map +1 -0
  15. package/dist/__tests__/mcp-init-no-auth.test.js +14 -1
  16. package/dist/__tests__/mcp-init-no-auth.test.js.map +1 -1
  17. package/dist/__tests__/memory-write-failure.test.d.ts +1 -0
  18. package/dist/__tests__/memory-write-failure.test.js +148 -0
  19. package/dist/__tests__/memory-write-failure.test.js.map +1 -0
  20. package/dist/__tests__/resolve-agent-preference-e2e.test.d.ts +1 -0
  21. package/dist/__tests__/resolve-agent-preference-e2e.test.js +173 -0
  22. package/dist/__tests__/resolve-agent-preference-e2e.test.js.map +1 -0
  23. package/dist/__tests__/resolve-agent-preference.test.d.ts +1 -0
  24. package/dist/__tests__/resolve-agent-preference.test.js +83 -0
  25. package/dist/__tests__/resolve-agent-preference.test.js.map +1 -0
  26. package/dist/ais-http-error.d.ts +26 -0
  27. package/dist/ais-http-error.js +75 -0
  28. package/dist/ais-http-error.js.map +1 -0
  29. package/dist/device-poll.d.ts +51 -0
  30. package/dist/device-poll.js +77 -0
  31. package/dist/device-poll.js.map +1 -0
  32. package/dist/env-agent.d.ts +27 -0
  33. package/dist/env-agent.js +33 -0
  34. package/dist/env-agent.js.map +1 -1
  35. package/dist/index.js +155 -58
  36. package/dist/index.js.map +1 -1
  37. package/dist/key-auth.d.ts +30 -0
  38. package/dist/key-auth.js +77 -1
  39. package/dist/key-auth.js.map +1 -1
  40. package/dist/telemetry.d.ts +1 -1
  41. package/dist/telemetry.js.map +1 -1
  42. package/package.json +2 -2
@@ -0,0 +1,191 @@
1
+ /**
2
+ * S2 — client-side device-poll loop tests
3
+ *
4
+ * Tests for runPollLoop() in device-poll.ts:
5
+ * - Returns access_token on approval
6
+ * - Emits heartbeat every HEARTBEAT_INTERVAL_MS during pending
7
+ * - When pending_reason==='page_open', heartbeat contains 'click'
8
+ * - On expired_token, throws DeviceFlowRestartedError (auto-restart,
9
+ * never throws a plain expired error)
10
+ * - Throws on access_denied
11
+ */
12
+ import { describe, it, expect } from 'vitest';
13
+ import { runPollLoop, DeviceFlowRestartedError, HEARTBEAT_INTERVAL_MS, } from '../device-poll.js';
14
+ // ─── helpers ─────────────────────────────────────────────────────────────────
15
+ const HANDLE = {
16
+ device_code: 'test-device-code-abc123',
17
+ verification_uri_complete: 'https://agentsandswarms.ai/activate?user_code=BCDF-GHJK',
18
+ user_code: 'BCDF-GHJK',
19
+ };
20
+ const NEW_HANDLE = {
21
+ device_code: 'new-device-code-xyz789',
22
+ verification_uri_complete: 'https://agentsandswarms.ai/activate?user_code=MNPQ-RSTU',
23
+ user_code: 'MNPQ-RSTU',
24
+ };
25
+ function makePollOnce(impl) {
26
+ return impl;
27
+ }
28
+ function makeStartNewFlow(impl) {
29
+ return impl;
30
+ }
31
+ function makeDeps(overrides = {}) {
32
+ const stderrLines = [];
33
+ const deps = {
34
+ pollOnce: makePollOnce(async () => ({ access_token: 'default-token' })),
35
+ startNewFlow: makeStartNewFlow(async () => NEW_HANDLE),
36
+ emitStderr: (msg) => { stderrLines.push(msg); },
37
+ nowMs: () => Date.now(),
38
+ pollIntervalMs: 0,
39
+ ...overrides,
40
+ };
41
+ return { deps, stderrLines };
42
+ }
43
+ // ─── tests ───────────────────────────────────────────────────────────────────
44
+ describe('runPollLoop()', () => {
45
+ describe('success path', () => {
46
+ it('returns the token response when access_token is present', async () => {
47
+ const { deps } = makeDeps({
48
+ pollOnce: makePollOnce(async () => ({ access_token: 'jwt-token-abc', expires_in: 604800 })),
49
+ });
50
+ const result = await runPollLoop(HANDLE, deps);
51
+ expect(result.access_token).toBe('jwt-token-abc');
52
+ expect(result.expires_in).toBe(604800);
53
+ });
54
+ it('retries on authorization_pending then succeeds', async () => {
55
+ let count = 0;
56
+ const { deps } = makeDeps({
57
+ pollOnce: makePollOnce(async () => {
58
+ count++;
59
+ if (count < 3)
60
+ return { error: 'authorization_pending', pending_reason: 'page_not_opened' };
61
+ return { access_token: 'approved-jwt', expires_in: 604800 };
62
+ }),
63
+ });
64
+ const result = await runPollLoop(HANDLE, deps);
65
+ expect(result.access_token).toBe('approved-jwt');
66
+ expect(count).toBe(3);
67
+ });
68
+ });
69
+ describe('heartbeat behavior', () => {
70
+ it('emits a heartbeat line after HEARTBEAT_INTERVAL_MS elapses during pending', async () => {
71
+ const startTime = 1_000_000;
72
+ let callCount = 0;
73
+ const times = [
74
+ startTime, // deadline calc
75
+ startTime + HEARTBEAT_INTERVAL_MS + 1, // pending check — fires heartbeat
76
+ startTime + HEARTBEAT_INTERVAL_MS + 1, // deadline check
77
+ startTime + HEARTBEAT_INTERVAL_MS + 2,
78
+ ];
79
+ let timeIdx = 0;
80
+ const { deps, stderrLines } = makeDeps({
81
+ nowMs: () => times[Math.min(timeIdx++, times.length - 1)] ?? startTime,
82
+ pollOnce: makePollOnce(async () => {
83
+ callCount++;
84
+ if (callCount === 1)
85
+ return { error: 'authorization_pending', pending_reason: 'page_not_opened' };
86
+ return { access_token: 'jwt-after-heartbeat', expires_in: 604800 };
87
+ }),
88
+ });
89
+ await runPollLoop(HANDLE, deps);
90
+ expect(stderrLines.some((l) => l.includes('Waiting'))).toBe(true);
91
+ });
92
+ it('emits a "click" coaching message when pending_reason is page_open', async () => {
93
+ const startTime = 2_000_000;
94
+ let callCount = 0;
95
+ const times = [
96
+ startTime,
97
+ startTime + HEARTBEAT_INTERVAL_MS + 1,
98
+ startTime + HEARTBEAT_INTERVAL_MS + 1,
99
+ startTime + HEARTBEAT_INTERVAL_MS + 2,
100
+ ];
101
+ let timeIdx = 0;
102
+ const { deps, stderrLines } = makeDeps({
103
+ nowMs: () => times[Math.min(timeIdx++, times.length - 1)] ?? startTime,
104
+ pollOnce: makePollOnce(async () => {
105
+ callCount++;
106
+ if (callCount === 1)
107
+ return { error: 'authorization_pending', pending_reason: 'page_open' };
108
+ return { access_token: 'jwt-after-click', expires_in: 604800 };
109
+ }),
110
+ });
111
+ await runPollLoop(HANDLE, deps);
112
+ expect(stderrLines.some((l) => l.toLowerCase().includes('click'))).toBe(true);
113
+ });
114
+ it('does NOT emit a heartbeat on every poll (throttled)', async () => {
115
+ // Time frozen at 1ms: gap from lastHeartbeatAt=0 is 1ms < 30_000ms
116
+ let callCount = 0;
117
+ const { deps, stderrLines } = makeDeps({
118
+ nowMs: () => 1,
119
+ pollOnce: makePollOnce(async () => {
120
+ callCount++;
121
+ if (callCount < 5)
122
+ return { error: 'authorization_pending', pending_reason: 'page_not_opened' };
123
+ return { access_token: 'jwt-final', expires_in: 604800 };
124
+ }),
125
+ });
126
+ await runPollLoop(HANDLE, deps);
127
+ // 4 pending responses but 0 heartbeats (time never passed the interval)
128
+ expect(stderrLines.filter((l) => l.includes('Waiting')).length).toBe(0);
129
+ });
130
+ });
131
+ describe('auto-restart on expired_token', () => {
132
+ it('throws DeviceFlowRestartedError (not a plain Error) on expired_token', async () => {
133
+ const { deps } = makeDeps({
134
+ pollOnce: makePollOnce(async () => ({ error: 'expired_token' })),
135
+ });
136
+ await expect(runPollLoop(HANDLE, deps)).rejects.toBeInstanceOf(DeviceFlowRestartedError);
137
+ });
138
+ it('starts a new flow on expired_token', async () => {
139
+ let startCalled = 0;
140
+ const { deps } = makeDeps({
141
+ pollOnce: makePollOnce(async () => ({ error: 'expired_token' })),
142
+ startNewFlow: makeStartNewFlow(async () => { startCalled++; return NEW_HANDLE; }),
143
+ });
144
+ await expect(runPollLoop(HANDLE, deps)).rejects.toThrow();
145
+ expect(startCalled).toBe(1);
146
+ });
147
+ it('DeviceFlowRestartedError carries the new verification URI', async () => {
148
+ const { deps } = makeDeps({
149
+ pollOnce: makePollOnce(async () => ({ error: 'expired_token' })),
150
+ });
151
+ let caught;
152
+ try {
153
+ await runPollLoop(HANDLE, deps);
154
+ }
155
+ catch (err) {
156
+ caught = err;
157
+ }
158
+ expect(caught).toBeInstanceOf(DeviceFlowRestartedError);
159
+ const restarted = caught;
160
+ expect(restarted.newVerificationUri).toBe(NEW_HANDLE.verification_uri_complete);
161
+ expect(restarted.newUserCode).toBe(NEW_HANDLE.user_code);
162
+ });
163
+ });
164
+ describe('access_denied', () => {
165
+ it('throws a plain Error on access_denied (not DeviceFlowRestartedError)', async () => {
166
+ let startCalled = 0;
167
+ const { deps } = makeDeps({
168
+ pollOnce: makePollOnce(async () => ({ error: 'access_denied' })),
169
+ startNewFlow: makeStartNewFlow(async () => { startCalled++; return NEW_HANDLE; }),
170
+ });
171
+ await expect(runPollLoop(HANDLE, deps)).rejects.toThrow('denied');
172
+ expect(startCalled).toBe(0);
173
+ });
174
+ });
175
+ describe('slow_down', () => {
176
+ it('retries after slow_down and eventually succeeds', async () => {
177
+ let count = 0;
178
+ const { deps } = makeDeps({
179
+ pollOnce: makePollOnce(async () => {
180
+ count++;
181
+ if (count === 1)
182
+ return { error: 'slow_down' };
183
+ return { access_token: 'jwt-after-slowdown', expires_in: 604800 };
184
+ }),
185
+ });
186
+ const result = await runPollLoop(HANDLE, deps);
187
+ expect(result.access_token).toBe('jwt-after-slowdown');
188
+ });
189
+ });
190
+ });
191
+ //# sourceMappingURL=device-poll.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"device-poll.test.js","sourceRoot":"","sources":["../../src/__tests__/device-poll.test.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EACL,WAAW,EACX,wBAAwB,EACxB,qBAAqB,GAItB,MAAM,mBAAmB,CAAC;AAE3B,gFAAgF;AAEhF,MAAM,MAAM,GAAqB;IAC/B,WAAW,EAAE,yBAAyB;IACtC,yBAAyB,EAAE,yDAAyD;IACpF,SAAS,EAAE,WAAW;CACvB,CAAC;AACF,MAAM,UAAU,GAAqB;IACnC,WAAW,EAAE,wBAAwB;IACrC,yBAAyB,EAAE,yDAAyD;IACpF,SAAS,EAAE,WAAW;CACvB,CAAC;AAEF,SAAS,YAAY,CACnB,IAAwD;IAExD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,gBAAgB,CACvB,IAAqC;IAErC,OAAO,IAAI,CAAC;AACd,CAAC;AAOD,SAAS,QAAQ,CAAC,YAAmC,EAAE;IACrD,MAAM,WAAW,GAAa,EAAE,CAAC;IAEjC,MAAM,IAAI,GAAiB;QACzB,QAAQ,EAAE,YAAY,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,YAAY,EAAE,eAAe,EAAE,CAAC,CAAC;QACvE,YAAY,EAAE,gBAAgB,CAAC,KAAK,IAAI,EAAE,CAAC,UAAU,CAAC;QACtD,UAAU,EAAE,CAAC,GAAW,EAAQ,EAAE,GAAG,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAC7D,KAAK,EAAE,GAAW,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE;QAC/B,cAAc,EAAE,CAAC;QACjB,GAAG,SAAS;KACb,CAAC;IAEF,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC;AAC/B,CAAC;AAED,gFAAgF;AAEhF,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,EAAE,IAAI,EAAE,GAAG,QAAQ,CAAC;gBACxB,QAAQ,EAAE,YAAY,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,YAAY,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CAAC;aAC5F,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAE/C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,MAAM,EAAE,IAAI,EAAE,GAAG,QAAQ,CAAC;gBACxB,QAAQ,EAAE,YAAY,CAAC,KAAK,IAAI,EAAE;oBAChC,KAAK,EAAE,CAAC;oBACR,IAAI,KAAK,GAAG,CAAC;wBAAE,OAAO,EAAE,KAAK,EAAE,uBAAuB,EAAE,cAAc,EAAE,iBAAiB,EAAE,CAAC;oBAC5F,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;gBAC9D,CAAC,CAAC;aACH,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAE/C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACjD,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACxB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YACzF,MAAM,SAAS,GAAG,SAAS,CAAC;YAC5B,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,MAAM,KAAK,GAAG;gBACZ,SAAS,EAA6B,gBAAgB;gBACtD,SAAS,GAAG,qBAAqB,GAAG,CAAC,EAAE,kCAAkC;gBACzE,SAAS,GAAG,qBAAqB,GAAG,CAAC,EAAE,iBAAiB;gBACxD,SAAS,GAAG,qBAAqB,GAAG,CAAC;aACtC,CAAC;YACF,IAAI,OAAO,GAAG,CAAC,CAAC;YAEhB,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,QAAQ,CAAC;gBACrC,KAAK,EAAE,GAAW,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,SAAS;gBAC9E,QAAQ,EAAE,YAAY,CAAC,KAAK,IAAI,EAAE;oBAChC,SAAS,EAAE,CAAC;oBACZ,IAAI,SAAS,KAAK,CAAC;wBAAE,OAAO,EAAE,KAAK,EAAE,uBAAuB,EAAE,cAAc,EAAE,iBAAiB,EAAE,CAAC;oBAClG,OAAO,EAAE,YAAY,EAAE,qBAAqB,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;gBACrE,CAAC,CAAC;aACH,CAAC,CAAC;YAEH,MAAM,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAEhC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,MAAM,SAAS,GAAG,SAAS,CAAC;YAC5B,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,MAAM,KAAK,GAAG;gBACZ,SAAS;gBACT,SAAS,GAAG,qBAAqB,GAAG,CAAC;gBACrC,SAAS,GAAG,qBAAqB,GAAG,CAAC;gBACrC,SAAS,GAAG,qBAAqB,GAAG,CAAC;aACtC,CAAC;YACF,IAAI,OAAO,GAAG,CAAC,CAAC;YAEhB,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,QAAQ,CAAC;gBACrC,KAAK,EAAE,GAAW,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,SAAS;gBAC9E,QAAQ,EAAE,YAAY,CAAC,KAAK,IAAI,EAAE;oBAChC,SAAS,EAAE,CAAC;oBACZ,IAAI,SAAS,KAAK,CAAC;wBAAE,OAAO,EAAE,KAAK,EAAE,uBAAuB,EAAE,cAAc,EAAE,WAAW,EAAE,CAAC;oBAC5F,OAAO,EAAE,YAAY,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;gBACjE,CAAC,CAAC;aACH,CAAC,CAAC;YAEH,MAAM,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAEhC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,mEAAmE;YACnE,IAAI,SAAS,GAAG,CAAC,CAAC;YAElB,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,GAAG,QAAQ,CAAC;gBACrC,KAAK,EAAE,GAAW,EAAE,CAAC,CAAC;gBACtB,QAAQ,EAAE,YAAY,CAAC,KAAK,IAAI,EAAE;oBAChC,SAAS,EAAE,CAAC;oBACZ,IAAI,SAAS,GAAG,CAAC;wBAAE,OAAO,EAAE,KAAK,EAAE,uBAAuB,EAAE,cAAc,EAAE,iBAAiB,EAAE,CAAC;oBAChG,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;gBAC3D,CAAC,CAAC;aACH,CAAC,CAAC;YAEH,MAAM,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAEhC,wEAAwE;YACxE,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,MAAM,EAAE,IAAI,EAAE,GAAG,QAAQ,CAAC;gBACxB,QAAQ,EAAE,YAAY,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;aACjE,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,wBAAwB,CAAC,CAAC;QAC3F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,MAAM,EAAE,IAAI,EAAE,GAAG,QAAQ,CAAC;gBACxB,QAAQ,EAAE,YAAY,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;gBAChE,YAAY,EAAE,gBAAgB,CAAC,KAAK,IAAI,EAAE,GAAG,WAAW,EAAE,CAAC,CAAC,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC;aAClF,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAC1D,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,EAAE,IAAI,EAAE,GAAG,QAAQ,CAAC;gBACxB,QAAQ,EAAE,YAAY,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;aACjE,CAAC,CAAC;YAEH,IAAI,MAAe,CAAC;YACpB,IAAI,CAAC;gBACH,MAAM,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAClC,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,GAAG,GAAG,CAAC;YACf,CAAC;YAED,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,wBAAwB,CAAC,CAAC;YACxD,MAAM,SAAS,GAAG,MAAkC,CAAC;YACrD,MAAM,CAAC,SAAS,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC;YAChF,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,MAAM,EAAE,IAAI,EAAE,GAAG,QAAQ,CAAC;gBACxB,QAAQ,EAAE,YAAY,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,CAAC,CAAC;gBAChE,YAAY,EAAE,gBAAgB,CAAC,KAAK,IAAI,EAAE,GAAG,WAAW,EAAE,CAAC,CAAC,OAAO,UAAU,CAAC,CAAC,CAAC,CAAC;aAClF,CAAC,CAAC;YAEH,MAAM,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YAClE,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,MAAM,EAAE,IAAI,EAAE,GAAG,QAAQ,CAAC;gBACxB,QAAQ,EAAE,YAAY,CAAC,KAAK,IAAI,EAAE;oBAChC,KAAK,EAAE,CAAC;oBACR,IAAI,KAAK,KAAK,CAAC;wBAAE,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC;oBAC/C,OAAO,EAAE,YAAY,EAAE,oBAAoB,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC;gBACpE,CAAC,CAAC;aACH,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACzD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -1,8 +1,20 @@
1
1
  import { spawn } from 'node:child_process';
2
- import { mkdtempSync, rmSync, existsSync } from 'node:fs';
2
+ import { mkdtempSync, rmSync, existsSync, readFileSync } from 'node:fs';
3
3
  import { tmpdir } from 'node:os';
4
4
  import { join, resolve } from 'path';
5
5
  import { describe, it, expect } from 'vitest';
6
+ function readPackageVersion() {
7
+ const packageJsonPath = resolve(__dirname, '..', '..', 'package.json');
8
+ const packageJson = JSON.parse(readFileSync(packageJsonPath, 'utf-8'));
9
+ if (typeof packageJson !== 'object' ||
10
+ packageJson === null ||
11
+ !('version' in packageJson) ||
12
+ typeof packageJson.version !== 'string' ||
13
+ packageJson.version.length === 0) {
14
+ throw new Error('Unable to load aismemory package version from package.json');
15
+ }
16
+ return packageJson.version;
17
+ }
6
18
  /**
7
19
  * Regression test for the v0.2.0/v0.3.0 bug where `npx aismemory` blocked the
8
20
  * MCP transport handshake on user device-flow approval, exceeding Claude
@@ -64,6 +76,7 @@ describe('MCP transport boots without blocking on auth', () => {
64
76
  const parsed = JSON.parse(responseLine);
65
77
  expect(elapsed).toBeLessThan(3000);
66
78
  expect(parsed.result?.serverInfo?.name).toBe('aismemory');
79
+ expect(parsed.result?.serverInfo?.version).toBe(readPackageVersion());
67
80
  }
68
81
  finally {
69
82
  if (proc && !proc.killed)
@@ -1 +1 @@
1
- {"version":3,"file":"mcp-init-no-auth.test.js","sourceRoot":"","sources":["../../src/__tests__/mcp-init-no-auth.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAuC,MAAM,oBAAoB,CAAC;AAChF,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAC1D,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAE9C;;;;;;;;;;GAUG;AACH,QAAQ,CAAC,8CAA8C,EAAE,GAAG,EAAE;IAC5D,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACpE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,+DAA+D,QAAQ,GAAG,CAAC,CAAC;QAC9F,CAAC;QAED,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,cAAc,CAAC,CAAC,CAAC;QAC7D,IAAI,IAAI,GAA0C,IAAI,CAAC;QACvD,IAAI,CAAC;YACH,IAAI,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE;gBAC/B,GAAG,EAAE;oBACH,GAAG,OAAO,CAAC,GAAG;oBACd,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,uBAAuB,EAAE,oCAAoC;iBACvE;gBACD,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;aAChC,CAAC,CAAC;YAEH,IAAI,SAAS,GAAG,EAAE,CAAC;YACnB,IAAI,SAAS,GAAG,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE,GAAG,SAAS,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACtE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE,GAAG,SAAS,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAEtE,MAAM,WAAW,GACf,IAAI,CAAC,SAAS,CAAC;gBACb,OAAO,EAAE,KAAK;gBACd,EAAE,EAAE,CAAC;gBACL,MAAM,EAAE,YAAY;gBACpB,MAAM,EAAE;oBACN,eAAe,EAAE,YAAY;oBAC7B,YAAY,EAAE,EAAE;oBAChB,UAAU,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE;iBACpD;aACF,CAAC,GAAG,IAAI,CAAC;YAEZ,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAE9B,MAAM,YAAY,GAAG,MAAM,IAAI,OAAO,CAAS,CAAC,YAAY,EAAE,MAAM,EAAE,EAAE;gBACtE,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;oBAC9B,MAAM,CAAC,IAAI,KAAK,CAAC,mCAAmC,SAAS,EAAE,CAAC,CAAC,CAAC;gBACpE,CAAC,EAAE,IAAI,CAAC,CAAC;gBACT,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,EAAE;oBAChC,MAAM,IAAI,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;oBACxE,IAAI,IAAI,EAAE,CAAC;wBACT,YAAY,CAAC,OAAO,CAAC,CAAC;wBACtB,aAAa,CAAC,QAAQ,CAAC,CAAC;wBACxB,YAAY,CAAC,IAAI,CAAC,CAAC;oBACrB,CAAC;gBACH,CAAC,EAAE,EAAE,CAAC,CAAC;YACT,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAErC,CAAC;YAEF,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC5D,CAAC;gBAAS,CAAC;YACT,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM;gBAAE,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC/C,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC;IACH,CAAC,EAAE,MAAM,CAAC,CAAC;AACb,CAAC,CAAC,CAAC"}
1
+ {"version":3,"file":"mcp-init-no-auth.test.js","sourceRoot":"","sources":["../../src/__tests__/mcp-init-no-auth.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAuC,MAAM,oBAAoB,CAAC;AAChF,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACxE,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAE9C,SAAS,kBAAkB;IACzB,MAAM,eAAe,GAAG,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,cAAc,CAAC,CAAC;IACvE,MAAM,WAAW,GAAY,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC,CAAC;IAChF,IACE,OAAO,WAAW,KAAK,QAAQ;QAC/B,WAAW,KAAK,IAAI;QACpB,CAAC,CAAC,SAAS,IAAI,WAAW,CAAC;QAC3B,OAAO,WAAW,CAAC,OAAO,KAAK,QAAQ;QACvC,WAAW,CAAC,OAAO,CAAC,MAAM,KAAK,CAAC,EAChC,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,4DAA4D,CAAC,CAAC;IAChF,CAAC;IACD,OAAO,WAAW,CAAC,OAAO,CAAC;AAC7B,CAAC;AAED;;;;;;;;;;GAUG;AACH,QAAQ,CAAC,8CAA8C,EAAE,GAAG,EAAE;IAC5D,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACxF,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACpE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,+DAA+D,QAAQ,GAAG,CAAC,CAAC;QAC9F,CAAC;QAED,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,cAAc,CAAC,CAAC,CAAC;QAC7D,IAAI,IAAI,GAA0C,IAAI,CAAC;QACvD,IAAI,CAAC;YACH,IAAI,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE;gBAC/B,GAAG,EAAE;oBACH,GAAG,OAAO,CAAC,GAAG;oBACd,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,uBAAuB,EAAE,oCAAoC;iBACvE;gBACD,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;aAChC,CAAC,CAAC;YAEH,IAAI,SAAS,GAAG,EAAE,CAAC;YACnB,IAAI,SAAS,GAAG,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE,GAAG,SAAS,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACtE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE,GAAG,SAAS,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAEtE,MAAM,WAAW,GACf,IAAI,CAAC,SAAS,CAAC;gBACb,OAAO,EAAE,KAAK;gBACd,EAAE,EAAE,CAAC;gBACL,MAAM,EAAE,YAAY;gBACpB,MAAM,EAAE;oBACN,eAAe,EAAE,YAAY;oBAC7B,YAAY,EAAE,EAAE;oBAChB,UAAU,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE;iBACpD;aACF,CAAC,GAAG,IAAI,CAAC;YAEZ,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACtB,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;YAE9B,MAAM,YAAY,GAAG,MAAM,IAAI,OAAO,CAAS,CAAC,YAAY,EAAE,MAAM,EAAE,EAAE;gBACtE,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;oBAC9B,MAAM,CAAC,IAAI,KAAK,CAAC,mCAAmC,SAAS,EAAE,CAAC,CAAC,CAAC;gBACpE,CAAC,EAAE,IAAI,CAAC,CAAC;gBACT,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,EAAE;oBAChC,MAAM,IAAI,GAAG,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC;oBACxE,IAAI,IAAI,EAAE,CAAC;wBACT,YAAY,CAAC,OAAO,CAAC,CAAC;wBACtB,aAAa,CAAC,QAAQ,CAAC,CAAC;wBACxB,YAAY,CAAC,IAAI,CAAC,CAAC;oBACrB,CAAC;gBACH,CAAC,EAAE,EAAE,CAAC,CAAC;YACT,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC;YAChC,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAErC,CAAC;YAEF,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC1D,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,kBAAkB,EAAE,CAAC,CAAC;QACxE,CAAC;gBAAS,CAAC;YACT,IAAI,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM;gBAAE,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC/C,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC;IACH,CAAC,EAAE,MAAM,CAAC,CAAC;AACb,CAAC,CAAC,CAAC"}
@@ -0,0 +1 @@
1
+ export {};
@@ -0,0 +1,148 @@
1
+ import { createServer } from 'node:http';
2
+ import { spawn } from 'node:child_process';
3
+ import { mkdtempSync, mkdirSync, writeFileSync, existsSync } from 'node:fs';
4
+ import { tmpdir } from 'node:os';
5
+ import { join, resolve } from 'path';
6
+ import { describe, it, expect, beforeAll, afterAll } from 'vitest';
7
+ import { generateAndSaveKeypair } from '../key-auth.js';
8
+ /**
9
+ * Regression: a failed memory write must NOT look like a success.
10
+ *
11
+ * aisPost() used to return res.json() even on a non-2xx status, so a 422
12
+ * (content rejected) or 429 (over quota) response was handed back as an
13
+ * ordinary tool result and `remember` recorded it as stored. The agent
14
+ * believed it had remembered something it had not — the documented
15
+ * "AIS memory writes fail silently" symptom.
16
+ *
17
+ * After the fix, any non-2xx (other than a token-rejection, which feeds the
18
+ * re-auth loop) throws AisHttpError, which the MCP handler renders as
19
+ * `isError: true` with an unmistakable failure message.
20
+ */
21
+ const TOKEN = mintJwt('fresh-token');
22
+ function mintJwt(marker) {
23
+ const header = Buffer.from(JSON.stringify({ alg: 'HS256', typ: 'JWT' })).toString('base64url');
24
+ const payload = Buffer.from(JSON.stringify({ sub: 'user-1', user_id: 'user-1', tenant_id: 'tenant-1', marker })).toString('base64url');
25
+ return `${header}.${payload}.fakesig-${marker}`;
26
+ }
27
+ describe('aismemory MCP surfaces failed memory writes', () => {
28
+ let fakeAis;
29
+ let port = 0;
30
+ beforeAll(async () => {
31
+ fakeAis = createServer((req, res) => {
32
+ req.on('data', () => { });
33
+ req.on('end', () => {
34
+ // The one path under test: a memory write the server rejects (422).
35
+ if (req.url === '/v1/agents/agent-1/memory' && req.method === 'POST') {
36
+ res.writeHead(422, { 'content-type': 'application/json' });
37
+ res.end(JSON.stringify({
38
+ success: false,
39
+ error: { code: 'CONTENT_FLAGGED', message: 'content rejected by safety pipeline' },
40
+ }));
41
+ return;
42
+ }
43
+ // Everything else (whoami / hydration / refresh) succeeds so the
44
+ // binary reaches a steady state and the remember call is the only
45
+ // failing request.
46
+ res.writeHead(200, { 'content-type': 'application/json' });
47
+ res.end(JSON.stringify({
48
+ success: true,
49
+ data: { id: 'agent-1', name: 'TestAgent', status: 'active', agents: [], recentMemories: [], goals: [] },
50
+ }));
51
+ });
52
+ });
53
+ await new Promise((r) => {
54
+ fakeAis.listen(0, '127.0.0.1', () => {
55
+ const addr = fakeAis.address();
56
+ if (addr && typeof addr === 'object')
57
+ port = addr.port;
58
+ r();
59
+ });
60
+ });
61
+ });
62
+ afterAll(async () => {
63
+ await new Promise((r) => fakeAis?.close(() => r()));
64
+ });
65
+ it('returns an MCP error (not a fake success) when the write is rejected', async () => {
66
+ const distPath = resolve(__dirname, '..', '..', 'dist', 'index.js');
67
+ if (!existsSync(distPath)) {
68
+ throw new Error(`dist/index.js missing — run \`pnpm build\` first (looked at ${distPath})`);
69
+ }
70
+ const fakeHome = mkdtempSync(join(tmpdir(), 'aismem-writefail-'));
71
+ const aismemDir = join(fakeHome, '.aismemory');
72
+ mkdirSync(aismemDir, { recursive: true });
73
+ writeFileSync(join(aismemDir, 'credentials.json'), JSON.stringify({
74
+ token: TOKEN,
75
+ agentId: 'agent-1',
76
+ tenantId: 'tenant-1',
77
+ expiresAt: new Date(Date.now() + 30 * 24 * 3600 * 1000).toISOString(),
78
+ }));
79
+ await generateAndSaveKeypair({
80
+ userId: 'user-1',
81
+ userDid: 'did:web:test:users:user-1',
82
+ keysDir: join(aismemDir, 'keys'),
83
+ });
84
+ let proc = null;
85
+ try {
86
+ proc = spawn('node', [distPath], {
87
+ env: { ...process.env, HOME: fakeHome, AIS_URL: `http://127.0.0.1:${port}` },
88
+ stdio: ['pipe', 'pipe', 'pipe'],
89
+ });
90
+ let stdoutBuf = '';
91
+ proc.stdout.on('data', (d) => { stdoutBuf += d.toString(); });
92
+ proc.stderr.on('data', () => { });
93
+ proc.stdin.write(JSON.stringify({
94
+ jsonrpc: '2.0', id: 1, method: 'initialize',
95
+ params: { protocolVersion: '2024-11-05', capabilities: {}, clientInfo: { name: 'writefail-test', version: '0.0.0' } },
96
+ }) + '\n');
97
+ await waitFor(() => stdoutBuf.includes('"id":1'), 5000, 'initialize response');
98
+ proc.stdin.write(JSON.stringify({
99
+ jsonrpc: '2.0', id: 2, method: 'tools/call',
100
+ params: { name: 'remember', arguments: { content: 'a fact the server will reject', type: 'fact' } },
101
+ }) + '\n');
102
+ await waitFor(() => stdoutBuf.includes('"id":2'), 15000, 'remember response');
103
+ const result = extractResult(stdoutBuf, 2);
104
+ // The failure must be visible to the agent, not masked as success.
105
+ expect(result.isError).toBe(true);
106
+ expect(result.text).toMatch(/did NOT succeed/i);
107
+ expect(result.text).toContain('422');
108
+ }
109
+ finally {
110
+ proc?.kill();
111
+ }
112
+ }, 30000);
113
+ });
114
+ function waitFor(predicate, ms, label) {
115
+ return new Promise((resolveOuter, reject) => {
116
+ const start = Date.now();
117
+ const tick = () => {
118
+ if (predicate()) {
119
+ resolveOuter();
120
+ return;
121
+ }
122
+ if (Date.now() - start > ms) {
123
+ reject(new Error(`timeout waiting for ${label}`));
124
+ return;
125
+ }
126
+ setTimeout(tick, 50);
127
+ };
128
+ tick();
129
+ });
130
+ }
131
+ function extractResult(buf, id) {
132
+ for (const line of buf.split('\n')) {
133
+ if (!line.includes(`"id":${id}`))
134
+ continue;
135
+ try {
136
+ const parsed = JSON.parse(line);
137
+ if (parsed.result) {
138
+ return {
139
+ isError: parsed.result.isError === true,
140
+ text: parsed.result.content?.[0]?.text ?? '',
141
+ };
142
+ }
143
+ }
144
+ catch { /* malformed line; skip */ }
145
+ }
146
+ return { isError: false, text: '' };
147
+ }
148
+ //# sourceMappingURL=memory-write-failure.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"memory-write-failure.test.js","sourceRoot":"","sources":["../../src/__tests__/memory-write-failure.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAA0D,MAAM,WAAW,CAAC;AACjG,OAAO,EAAE,KAAK,EAAuC,MAAM,oBAAoB,CAAC;AAChF,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAC5E,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AACnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AAExD;;;;;;;;;;;;GAYG;AAEH,MAAM,KAAK,GAAG,OAAO,CAAC,aAAa,CAAC,CAAC;AAErC,SAAS,OAAO,CAAC,MAAc;IAC7B,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;IAC/F,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CACzB,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CACpF,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;IACxB,OAAO,GAAG,MAAM,IAAI,OAAO,YAAY,MAAM,EAAE,CAAC;AAClD,CAAC;AAED,QAAQ,CAAC,6CAA6C,EAAE,GAAG,EAAE;IAC3D,IAAI,OAA2B,CAAC;IAChC,IAAI,IAAI,GAAG,CAAC,CAAC;IAEb,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,OAAO,GAAG,YAAY,CAAC,CAAC,GAAoB,EAAE,GAAmB,EAAE,EAAE;YACnE,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,GAA+C,CAAC,CAAC,CAAC;YACtE,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBACjB,oEAAoE;gBACpE,IAAI,GAAG,CAAC,GAAG,KAAK,2BAA2B,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;oBACrE,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;oBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;wBACrB,OAAO,EAAE,KAAK;wBACd,KAAK,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,qCAAqC,EAAE;qBACnF,CAAC,CAAC,CAAC;oBACJ,OAAO;gBACT,CAAC;gBACD,iEAAiE;gBACjE,kEAAkE;gBAClE,mBAAmB;gBACnB,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;gBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;oBACrB,OAAO,EAAE,IAAI;oBACb,IAAI,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,cAAc,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;iBACxG,CAAC,CAAC,CAAC;YACN,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE;YAC5B,OAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE;gBACnC,MAAM,IAAI,GAAG,OAAQ,CAAC,OAAO,EAAE,CAAC;gBAChC,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;oBAAE,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;gBACvD,CAAC,EAAE,CAAC;YACN,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;QACpF,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACpE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,+DAA+D,QAAQ,GAAG,CAAC,CAAC;QAC9F,CAAC;QAED,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,mBAAmB,CAAC,CAAC,CAAC;QAClE,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAC/C,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,aAAa,CACX,IAAI,CAAC,SAAS,EAAE,kBAAkB,CAAC,EACnC,IAAI,CAAC,SAAS,CAAC;YACb,KAAK,EAAE,KAAK;YACZ,OAAO,EAAE,SAAS;YAClB,QAAQ,EAAE,UAAU;YACpB,SAAS,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE;SACtE,CAAC,CACH,CAAC;QACF,MAAM,sBAAsB,CAAC;YAC3B,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE,2BAA2B;YACpC,OAAO,EAAE,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC;SACjC,CAAC,CAAC;QAEH,IAAI,IAAI,GAA0C,IAAI,CAAC;QACvD,IAAI,CAAC;YACH,IAAI,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE;gBAC/B,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,oBAAoB,IAAI,EAAE,EAAE;gBAC5E,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;aAChC,CAAC,CAAC;YACH,IAAI,SAAS,GAAG,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE,GAAG,SAAS,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACtE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,GAAc,CAAC,CAAC,CAAC;YAE7C,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;gBAC9B,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,YAAY;gBAC3C,MAAM,EAAE,EAAE,eAAe,EAAE,YAAY,EAAE,YAAY,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,gBAAgB,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE;aACtH,CAAC,GAAG,IAAI,CAAC,CAAC;YACX,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,qBAAqB,CAAC,CAAC;YAE/E,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;gBAC9B,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,YAAY;gBAC3C,MAAM,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,OAAO,EAAE,+BAA+B,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE;aACpG,CAAC,GAAG,IAAI,CAAC,CAAC;YACX,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,mBAAmB,CAAC,CAAC;YAE9E,MAAM,MAAM,GAAG,aAAa,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;YAC3C,mEAAmE;YACnE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;YAChD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACvC,CAAC;gBAAS,CAAC;YACT,IAAI,EAAE,IAAI,EAAE,CAAC;QACf,CAAC;IACH,CAAC,EAAE,KAAK,CAAC,CAAC;AACZ,CAAC,CAAC,CAAC;AAEH,SAAS,OAAO,CAAC,SAAwB,EAAE,EAAU,EAAE,KAAa;IAClE,OAAO,IAAI,OAAO,CAAC,CAAC,YAAY,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,GAAS,EAAE;YACtB,IAAI,SAAS,EAAE,EAAE,CAAC;gBAAC,YAAY,EAAE,CAAC;gBAAC,OAAO;YAAC,CAAC;YAC5C,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,GAAG,EAAE,EAAE,CAAC;gBAAC,MAAM,CAAC,IAAI,KAAK,CAAC,uBAAuB,KAAK,EAAE,CAAC,CAAC,CAAC;gBAAC,OAAO;YAAC,CAAC;YAC3F,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACvB,CAAC,CAAC;QACF,IAAI,EAAE,CAAC;IACT,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,aAAa,CAAC,GAAW,EAAE,EAAU;IAC5C,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACnC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC;YAAE,SAAS;QAC3C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAE7B,CAAC;YACF,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBAClB,OAAO;oBACL,OAAO,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO,KAAK,IAAI;oBACvC,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE;iBAC7C,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,MAAM,CAAC,CAAC,0BAA0B,CAAC,CAAC;IACxC,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC;AACtC,CAAC"}
@@ -0,0 +1,173 @@
1
+ /**
2
+ * CORBOT-B0FA4A79 e2e: after a silent key re-auth, the MCP reconnects to the
3
+ * agent recorded in credentials.json — not the first active agent in the
4
+ * tenant.
5
+ *
6
+ * Spawns the built binary with near-expiry credentials bonded to agent-a and
7
+ * a key file present (silent re-auth path), against a fake AIS whose
8
+ * /v1/agents list returns agent-b FIRST. Before the fix the session
9
+ * reconnected to agent-b (apparent memory loss); after the fix it must come
10
+ * back as agent-a.
11
+ */
12
+ import { createServer } from 'node:http';
13
+ import { spawn } from 'node:child_process';
14
+ import { mkdtempSync, mkdirSync, writeFileSync, rmSync, existsSync, readFileSync } from 'node:fs';
15
+ import { tmpdir } from 'node:os';
16
+ import { join, resolve } from 'path';
17
+ import { describe, it, expect, beforeAll, afterAll } from 'vitest';
18
+ import { generateAndSaveKeypair } from '../key-auth.js';
19
+ const NEW_TOKEN = mintJwt('fresh-after-reauth');
20
+ function mintJwt(marker) {
21
+ const header = Buffer.from(JSON.stringify({ alg: 'HS256', typ: 'JWT' })).toString('base64url');
22
+ const payload = Buffer.from(JSON.stringify({ sub: 'user-1', user_id: 'user-1', tenant_id: 'tenant-1', marker })).toString('base64url');
23
+ return `${header}.${payload}.fakesig-${marker}`;
24
+ }
25
+ describe('re-auth reconnects to the previously bonded agent', () => {
26
+ let fakeAis;
27
+ let port = 0;
28
+ beforeAll(async () => {
29
+ fakeAis = createServer((req, res) => {
30
+ req.on('data', () => { });
31
+ req.on('end', () => {
32
+ if (req.url === '/v1/auth/challenge' && req.method === 'POST') {
33
+ res.writeHead(200, { 'content-type': 'application/json' });
34
+ res.end(JSON.stringify({
35
+ success: true,
36
+ data: { nonce: 'test-nonce', exp: Math.floor(Date.now() / 1000) + 60, hmac: 'test-hmac' },
37
+ }));
38
+ return;
39
+ }
40
+ if (req.url === '/v1/auth/did-prove' && req.method === 'POST') {
41
+ res.writeHead(200, { 'content-type': 'application/json' });
42
+ res.end(JSON.stringify({
43
+ success: true,
44
+ data: {
45
+ bearerToken: NEW_TOKEN,
46
+ expiresAt: new Date(Date.now() + 7 * 24 * 3600 * 1000).toISOString(),
47
+ },
48
+ }));
49
+ return;
50
+ }
51
+ if (req.url === '/v1/agents' && req.method === 'GET') {
52
+ // The previously bonded agent is deliberately NOT first.
53
+ res.writeHead(200, { 'content-type': 'application/json' });
54
+ res.end(JSON.stringify({
55
+ success: true,
56
+ data: {
57
+ agents: [
58
+ { id: 'agent-b', name: 'OtherAgent', status: 'active' },
59
+ { id: 'agent-a', name: 'MyAgent', status: 'active' },
60
+ ],
61
+ },
62
+ }));
63
+ return;
64
+ }
65
+ if (req.url?.startsWith('/v1/agents/agent-') && req.method === 'GET') {
66
+ const id = req.url.split('/')[3];
67
+ res.writeHead(200, { 'content-type': 'application/json' });
68
+ res.end(JSON.stringify({
69
+ success: true,
70
+ data: { id, name: id === 'agent-a' ? 'MyAgent' : 'OtherAgent', status: 'active' },
71
+ }));
72
+ return;
73
+ }
74
+ res.writeHead(404, { 'content-type': 'application/json' });
75
+ res.end(JSON.stringify({ success: false, error: { code: 'NOT_FOUND' } }));
76
+ });
77
+ });
78
+ await new Promise((r) => {
79
+ fakeAis.listen(0, '127.0.0.1', () => {
80
+ const addr = fakeAis.address();
81
+ if (addr && typeof addr === 'object')
82
+ port = addr.port;
83
+ r();
84
+ });
85
+ });
86
+ });
87
+ afterAll(async () => {
88
+ await new Promise((r) => fakeAis?.close(() => r()));
89
+ });
90
+ it('whoami answers as the bonded agent, and credentials keep its id', async () => {
91
+ const distPath = resolve(__dirname, '..', '..', 'dist', 'index.js');
92
+ if (!existsSync(distPath)) {
93
+ throw new Error(`dist/index.js missing — run \`pnpm build\` first (looked at ${distPath})`);
94
+ }
95
+ const fakeHome = mkdtempSync(join(tmpdir(), 'aismem-agent-pref-'));
96
+ const aismemDir = join(fakeHome, '.aismemory');
97
+ mkdirSync(aismemDir, { recursive: true });
98
+ const credsPath = join(aismemDir, 'credentials.json');
99
+ writeFileSync(credsPath, JSON.stringify({
100
+ token: mintJwt('old-session-token'),
101
+ agentId: 'agent-a',
102
+ tenantId: 'tenant-1',
103
+ // Inside the staleness safety margin → forces silent key re-auth.
104
+ expiresAt: new Date(Date.now() + 60 * 1000).toISOString(),
105
+ }));
106
+ await generateAndSaveKeypair({
107
+ userId: 'user-1',
108
+ userDid: 'did:web:test:users:user-1',
109
+ keysDir: join(aismemDir, 'keys'),
110
+ });
111
+ let proc = null;
112
+ try {
113
+ proc = spawn('node', [distPath], {
114
+ env: { ...process.env, HOME: fakeHome, AIS_URL: `http://127.0.0.1:${port}` },
115
+ stdio: ['pipe', 'pipe', 'pipe'],
116
+ });
117
+ let stdoutBuf = '';
118
+ proc.stdout.on('data', (d) => { stdoutBuf += d.toString(); });
119
+ proc.stderr.on('data', () => { });
120
+ proc.stdin.write(JSON.stringify({
121
+ jsonrpc: '2.0', id: 1, method: 'initialize',
122
+ params: { protocolVersion: '2024-11-05', capabilities: {}, clientInfo: { name: 'agent-pref-test', version: '0.0.0' } },
123
+ }) + '\n');
124
+ await waitFor(() => stdoutBuf.includes('"id":1'), 5000, 'initialize response');
125
+ proc.stdin.write(JSON.stringify({
126
+ jsonrpc: '2.0', id: 2, method: 'tools/call',
127
+ params: { name: 'whoami', arguments: {} },
128
+ }) + '\n');
129
+ await waitFor(() => stdoutBuf.includes('"id":2'), 15000, 'whoami response');
130
+ const whoami = extractResponseText(stdoutBuf, 2);
131
+ expect(whoami).toContain('agent-a');
132
+ expect(whoami).not.toContain('agent-b');
133
+ const savedCreds = JSON.parse(readFileSync(credsPath, 'utf-8'));
134
+ expect(savedCreds.agentId).toBe('agent-a');
135
+ }
136
+ finally {
137
+ proc?.kill();
138
+ rmSync(fakeHome, { recursive: true, force: true });
139
+ }
140
+ }, 30000);
141
+ });
142
+ function waitFor(predicate, ms, label) {
143
+ return new Promise((resolveOuter, reject) => {
144
+ const start = Date.now();
145
+ const tick = () => {
146
+ if (predicate()) {
147
+ resolveOuter();
148
+ return;
149
+ }
150
+ if (Date.now() - start > ms) {
151
+ reject(new Error(`timeout waiting for ${label}`));
152
+ return;
153
+ }
154
+ setTimeout(tick, 50);
155
+ };
156
+ tick();
157
+ });
158
+ }
159
+ function extractResponseText(buf, id) {
160
+ for (const line of buf.split('\n')) {
161
+ if (!line.includes(`"id":${id}`))
162
+ continue;
163
+ try {
164
+ const parsed = JSON.parse(line);
165
+ const text = parsed.result?.content?.[0]?.text;
166
+ if (typeof text === 'string')
167
+ return text;
168
+ }
169
+ catch { /* malformed line; skip */ }
170
+ }
171
+ return '';
172
+ }
173
+ //# sourceMappingURL=resolve-agent-preference-e2e.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"resolve-agent-preference-e2e.test.js","sourceRoot":"","sources":["../../src/__tests__/resolve-agent-preference-e2e.test.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AACH,OAAO,EAAE,YAAY,EAA0D,MAAM,WAAW,CAAC;AACjG,OAAO,EAAE,KAAK,EAAuC,MAAM,oBAAoB,CAAC;AAChF,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAClG,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAC;AAEnE,OAAO,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AAExD,MAAM,SAAS,GAAG,OAAO,CAAC,oBAAoB,CAAC,CAAC;AAEhD,SAAS,OAAO,CAAC,MAAc;IAC7B,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;IAC/F,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CACzB,IAAI,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,EAAE,CAAC,CACpF,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;IACxB,OAAO,GAAG,MAAM,IAAI,OAAO,YAAY,MAAM,EAAE,CAAC;AAClD,CAAC;AAED,QAAQ,CAAC,mDAAmD,EAAE,GAAG,EAAE;IACjE,IAAI,OAA2B,CAAC;IAChC,IAAI,IAAI,GAAG,CAAC,CAAC;IAEb,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,OAAO,GAAG,YAAY,CAAC,CAAC,GAAoB,EAAE,GAAmB,EAAE,EAAE;YACnE,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,GAAmB,CAAC,CAAC,CAAC;YAC1C,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBACjB,IAAI,GAAG,CAAC,GAAG,KAAK,oBAAoB,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;oBAC9D,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;oBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;wBACrB,OAAO,EAAE,IAAI;wBACb,IAAI,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,GAAG,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE;qBAC1F,CAAC,CAAC,CAAC;oBACJ,OAAO;gBACT,CAAC;gBACD,IAAI,GAAG,CAAC,GAAG,KAAK,oBAAoB,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;oBAC9D,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;oBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;wBACrB,OAAO,EAAE,IAAI;wBACb,IAAI,EAAE;4BACJ,WAAW,EAAE,SAAS;4BACtB,SAAS,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE;yBACrE;qBACF,CAAC,CAAC,CAAC;oBACJ,OAAO;gBACT,CAAC;gBACD,IAAI,GAAG,CAAC,GAAG,KAAK,YAAY,IAAI,GAAG,CAAC,MAAM,KAAK,KAAK,EAAE,CAAC;oBACrD,yDAAyD;oBACzD,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;oBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;wBACrB,OAAO,EAAE,IAAI;wBACb,IAAI,EAAE;4BACJ,MAAM,EAAE;gCACN,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,QAAQ,EAAE;gCACvD,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE;6BACrD;yBACF;qBACF,CAAC,CAAC,CAAC;oBACJ,OAAO;gBACT,CAAC;gBACD,IAAI,GAAG,CAAC,GAAG,EAAE,UAAU,CAAC,mBAAmB,CAAC,IAAI,GAAG,CAAC,MAAM,KAAK,KAAK,EAAE,CAAC;oBACrE,MAAM,EAAE,GAAG,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAE,CAAC;oBAClC,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;oBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;wBACrB,OAAO,EAAE,IAAI;wBACb,IAAI,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,EAAE,MAAM,EAAE,QAAQ,EAAE;qBAClF,CAAC,CAAC,CAAC;oBACJ,OAAO;gBACT,CAAC;gBACD,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;gBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC;YAC5E,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QACH,MAAM,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE;YAC5B,OAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE;gBACnC,MAAM,IAAI,GAAG,OAAQ,CAAC,OAAO,EAAE,CAAC;gBAChC,IAAI,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;oBAAE,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;gBACvD,CAAC,EAAE,CAAC;YACN,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,IAAI,OAAO,CAAO,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;QACpE,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,+DAA+D,QAAQ,GAAG,CAAC,CAAC;QAC9F,CAAC;QAED,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,oBAAoB,CAAC,CAAC,CAAC;QACnE,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;QAC/C,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,EAAE,kBAAkB,CAAC,CAAC;QACtD,aAAa,CACX,SAAS,EACT,IAAI,CAAC,SAAS,CAAC;YACb,KAAK,EAAE,OAAO,CAAC,mBAAmB,CAAC;YACnC,OAAO,EAAE,SAAS;YAClB,QAAQ,EAAE,UAAU;YACpB,kEAAkE;YAClE,SAAS,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,WAAW,EAAE;SAC1D,CAAC,CACH,CAAC;QACF,MAAM,sBAAsB,CAAC;YAC3B,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE,2BAA2B;YACpC,OAAO,EAAE,IAAI,CAAC,SAAS,EAAE,MAAM,CAAC;SACjC,CAAC,CAAC;QAEH,IAAI,IAAI,GAA0C,IAAI,CAAC;QACvD,IAAI,CAAC;YACH,IAAI,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE;gBAC/B,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,GAAG,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,oBAAoB,IAAI,EAAE,EAAE;gBAC5E,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;aAChC,CAAC,CAAC;YACH,IAAI,SAAS,GAAG,EAAE,CAAC;YACnB,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE,GAAG,SAAS,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACtE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,GAAc,CAAC,CAAC,CAAC;YAE7C,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;gBAC9B,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,YAAY;gBAC3C,MAAM,EAAE,EAAE,eAAe,EAAE,YAAY,EAAE,YAAY,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE;aACvH,CAAC,GAAG,IAAI,CAAC,CAAC;YACX,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,qBAAqB,CAAC,CAAC;YAE/E,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC;gBAC9B,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,YAAY;gBAC3C,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE,EAAE;aAC1C,CAAC,GAAG,IAAI,CAAC,CAAC;YACX,MAAM,OAAO,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,iBAAiB,CAAC,CAAC;YAE5E,MAAM,MAAM,GAAG,mBAAmB,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;YAExC,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAwB,CAAC;YACvF,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC;gBAAS,CAAC;YACT,IAAI,EAAE,IAAI,EAAE,CAAC;YACb,MAAM,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,CAAC;IACH,CAAC,EAAE,KAAK,CAAC,CAAC;AACZ,CAAC,CAAC,CAAC;AAEH,SAAS,OAAO,CAAC,SAAwB,EAAE,EAAU,EAAE,KAAa;IAClE,OAAO,IAAI,OAAO,CAAC,CAAC,YAAY,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,GAAS,EAAE;YACtB,IAAI,SAAS,EAAE,EAAE,CAAC;gBAAC,YAAY,EAAE,CAAC;gBAAC,OAAO;YAAC,CAAC;YAC5C,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,GAAG,EAAE,EAAE,CAAC;gBAAC,MAAM,CAAC,IAAI,KAAK,CAAC,uBAAuB,KAAK,EAAE,CAAC,CAAC,CAAC;gBAAC,OAAO;YAAC,CAAC;YAC3F,UAAU,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACvB,CAAC,CAAC;QACF,IAAI,EAAE,CAAC;IACT,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,mBAAmB,CAAC,GAAW,EAAE,EAAU;IAClD,KAAK,MAAM,IAAI,IAAI,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;QACnC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,CAAC;YAAE,SAAS;QAC3C,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAwD,CAAC;YACvF,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC;YAC/C,IAAI,OAAO,IAAI,KAAK,QAAQ;gBAAE,OAAO,IAAI,CAAC;QAC5C,CAAC;QAAC,MAAM,CAAC,CAAC,0BAA0B,CAAC,CAAC;IACxC,CAAC;IACD,OAAO,EAAE,CAAC;AACZ,CAAC"}
@@ -0,0 +1 @@
1
+ export {};