@animalabs/agent-framework 0.1.0 → 0.3.0
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/dist/src/framework.d.ts +13 -0
- package/dist/src/framework.d.ts.map +1 -1
- package/dist/src/framework.js +133 -10
- package/dist/src/framework.js.map +1 -1
- package/dist/src/gate/event-gate.d.ts +5 -0
- package/dist/src/gate/event-gate.d.ts.map +1 -1
- package/dist/src/gate/event-gate.js +104 -6
- package/dist/src/gate/event-gate.js.map +1 -1
- package/dist/src/gate/types.d.ts +31 -0
- package/dist/src/gate/types.d.ts.map +1 -1
- package/dist/src/index.d.ts +3 -0
- package/dist/src/index.d.ts.map +1 -1
- package/dist/src/index.js +4 -0
- package/dist/src/index.js.map +1 -1
- package/dist/src/mcpl/channel-registry.d.ts +16 -2
- package/dist/src/mcpl/channel-registry.d.ts.map +1 -1
- package/dist/src/mcpl/channel-registry.js +101 -6
- package/dist/src/mcpl/channel-registry.js.map +1 -1
- package/dist/src/mcpl/push-handler.js +1 -1
- package/dist/src/mcpl/push-handler.js.map +1 -1
- package/dist/src/mcpl/server-connection.d.ts +17 -2
- package/dist/src/mcpl/server-connection.d.ts.map +1 -1
- package/dist/src/mcpl/server-connection.js +66 -3
- package/dist/src/mcpl/server-connection.js.map +1 -1
- package/dist/src/mcpl/tool-policy.d.ts +24 -0
- package/dist/src/mcpl/tool-policy.d.ts.map +1 -0
- package/dist/src/mcpl/tool-policy.js +46 -0
- package/dist/src/mcpl/tool-policy.js.map +1 -0
- package/dist/src/mcpl/types.d.ts +31 -0
- package/dist/src/mcpl/types.d.ts.map +1 -1
- package/dist/src/mcpl/types.js.map +1 -1
- package/dist/src/modules/workspace/index.d.ts +43 -1
- package/dist/src/modules/workspace/index.d.ts.map +1 -1
- package/dist/src/modules/workspace/index.js +246 -32
- package/dist/src/modules/workspace/index.js.map +1 -1
- package/dist/src/modules/workspace/sync.d.ts +7 -2
- package/dist/src/modules/workspace/sync.d.ts.map +1 -1
- package/dist/src/modules/workspace/sync.js +2 -2
- package/dist/src/modules/workspace/sync.js.map +1 -1
- package/dist/src/modules/workspace/types.d.ts +55 -3
- package/dist/src/modules/workspace/types.d.ts.map +1 -1
- package/dist/src/modules/workspace/types.js +8 -1
- package/dist/src/modules/workspace/types.js.map +1 -1
- package/dist/src/modules/workspace/watcher.d.ts +40 -3
- package/dist/src/modules/workspace/watcher.d.ts.map +1 -1
- package/dist/src/modules/workspace/watcher.js +80 -13
- package/dist/src/modules/workspace/watcher.js.map +1 -1
- package/dist/src/types/trace.d.ts +20 -0
- package/dist/src/types/trace.d.ts.map +1 -1
- package/dist/src/usage/index.d.ts +3 -0
- package/dist/src/usage/index.d.ts.map +1 -0
- package/dist/src/usage/index.js +2 -0
- package/dist/src/usage/index.js.map +1 -0
- package/dist/src/usage/types.d.ts +28 -0
- package/dist/src/usage/types.d.ts.map +1 -0
- package/dist/src/usage/types.js +2 -0
- package/dist/src/usage/types.js.map +1 -0
- package/dist/src/usage/usage-tracker.d.ts +33 -0
- package/dist/src/usage/usage-tracker.d.ts.map +1 -0
- package/dist/src/usage/usage-tracker.js +115 -0
- package/dist/src/usage/usage-tracker.js.map +1 -0
- package/dist/test/channel-registry-typing.test.d.ts +2 -0
- package/dist/test/channel-registry-typing.test.d.ts.map +1 -0
- package/dist/test/channel-registry-typing.test.js +82 -0
- package/dist/test/channel-registry-typing.test.js.map +1 -0
- package/dist/test/event-gate.test.js +135 -16
- package/dist/test/event-gate.test.js.map +1 -1
- package/dist/test/mcpl-gate-roundtrip.test.d.ts +2 -0
- package/dist/test/mcpl-gate-roundtrip.test.d.ts.map +1 -0
- package/dist/test/mcpl-gate-roundtrip.test.js +140 -0
- package/dist/test/mcpl-gate-roundtrip.test.js.map +1 -0
- package/dist/test/tool-policy.test.d.ts +2 -0
- package/dist/test/tool-policy.test.d.ts.map +1 -0
- package/dist/test/tool-policy.test.js +51 -0
- package/dist/test/tool-policy.test.js.map +1 -0
- package/dist/test/usage-tracker.test.d.ts +2 -0
- package/dist/test/usage-tracker.test.d.ts.map +1 -0
- package/dist/test/usage-tracker.test.js +129 -0
- package/dist/test/usage-tracker.test.js.map +1 -0
- package/dist/test/workspace-watcher.test.d.ts +2 -0
- package/dist/test/workspace-watcher.test.d.ts.map +1 -0
- package/dist/test/workspace-watcher.test.js +117 -0
- package/dist/test/workspace-watcher.test.js.map +1 -0
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +3 -3
- package/src/framework.ts +148 -10
- package/src/gate/event-gate.ts +106 -6
- package/src/gate/types.ts +28 -0
- package/src/index.ts +7 -0
- package/src/mcpl/channel-registry.ts +120 -8
- package/src/mcpl/push-handler.ts +1 -1
- package/src/mcpl/server-connection.ts +78 -3
- package/src/mcpl/tool-policy.ts +52 -0
- package/src/mcpl/types.ts +34 -0
- package/src/modules/workspace/index.ts +269 -38
- package/src/modules/workspace/sync.ts +11 -4
- package/src/modules/workspace/types.ts +70 -3
- package/src/modules/workspace/watcher.ts +104 -15
- package/src/types/trace.ts +26 -1
- package/src/usage/index.ts +2 -0
- package/src/usage/types.ts +27 -0
- package/src/usage/usage-tracker.ts +141 -0
|
@@ -0,0 +1,140 @@
|
|
|
1
|
+
import { describe, it, beforeEach, afterEach } from 'node:test';
|
|
2
|
+
import assert from 'node:assert';
|
|
3
|
+
import { existsSync, mkdirSync, writeFileSync, rmSync } from 'node:fs';
|
|
4
|
+
import { join } from 'node:path';
|
|
5
|
+
import { EventGate } from '../src/gate/event-gate.js';
|
|
6
|
+
import { ChannelRegistry } from '../src/mcpl/channel-registry.js';
|
|
7
|
+
const TMP_DIR = join(import.meta.dirname, '../.test-tmp-roundtrip');
|
|
8
|
+
beforeEach(() => {
|
|
9
|
+
mkdirSync(TMP_DIR, { recursive: true });
|
|
10
|
+
});
|
|
11
|
+
afterEach(() => {
|
|
12
|
+
if (existsSync(TMP_DIR))
|
|
13
|
+
rmSync(TMP_DIR, { recursive: true, force: true });
|
|
14
|
+
});
|
|
15
|
+
function makeGate(config) {
|
|
16
|
+
const configPath = join(TMP_DIR, 'gate.json');
|
|
17
|
+
writeFileSync(configPath, JSON.stringify(config));
|
|
18
|
+
return new EventGate({
|
|
19
|
+
configPath,
|
|
20
|
+
emitTrace: () => { },
|
|
21
|
+
addMessage: () => '',
|
|
22
|
+
requestInference: () => { },
|
|
23
|
+
getAgentNames: () => ['agent'],
|
|
24
|
+
});
|
|
25
|
+
}
|
|
26
|
+
function makeRegistry(shouldTriggerInference) {
|
|
27
|
+
const pushed = [];
|
|
28
|
+
const registry = new ChannelRegistry({}, {}, (ev) => pushed.push(ev), () => { }, shouldTriggerInference ? { shouldTriggerInference } : undefined);
|
|
29
|
+
return { registry, pushed };
|
|
30
|
+
}
|
|
31
|
+
function incomingParams(channelId, text) {
|
|
32
|
+
return {
|
|
33
|
+
messages: [{
|
|
34
|
+
channelId,
|
|
35
|
+
messageId: 'msg-1',
|
|
36
|
+
author: { id: '42', name: 'Alice' },
|
|
37
|
+
timestamp: new Date().toISOString(),
|
|
38
|
+
content: [{ type: 'text', text }],
|
|
39
|
+
}],
|
|
40
|
+
};
|
|
41
|
+
}
|
|
42
|
+
// ---------------------------------------------------------------------------
|
|
43
|
+
// Contract: ChannelRegistry emits eventType='mcpl:channel-incoming' to the gate
|
|
44
|
+
// ---------------------------------------------------------------------------
|
|
45
|
+
describe('ChannelRegistry → shouldTriggerInference contract', () => {
|
|
46
|
+
it('emits eventType="mcpl:channel-incoming" in callback metadata', () => {
|
|
47
|
+
const seen = [];
|
|
48
|
+
const { registry } = makeRegistry((_content, metadata) => {
|
|
49
|
+
seen.push(metadata);
|
|
50
|
+
return true;
|
|
51
|
+
});
|
|
52
|
+
registry.handleIncoming('zulip', incomingParams('zulip:tracker-miner-f', 'hi'));
|
|
53
|
+
assert.strictEqual(seen.length, 1);
|
|
54
|
+
assert.strictEqual(seen[0].eventType, 'mcpl:channel-incoming');
|
|
55
|
+
assert.strictEqual(seen[0].serverId, 'zulip');
|
|
56
|
+
assert.strictEqual(seen[0].channelId, 'zulip:tracker-miner-f');
|
|
57
|
+
});
|
|
58
|
+
});
|
|
59
|
+
// ---------------------------------------------------------------------------
|
|
60
|
+
// Roundtrip: a recipe-shaped channel policy fires for a channel-incoming event
|
|
61
|
+
// ---------------------------------------------------------------------------
|
|
62
|
+
describe('ChannelRegistry + EventGate roundtrip', () => {
|
|
63
|
+
it('channel-scoped policy triggers inference on matching channel-incoming', () => {
|
|
64
|
+
const gate = makeGate({
|
|
65
|
+
default: 'skip',
|
|
66
|
+
policies: [
|
|
67
|
+
{
|
|
68
|
+
name: 'tracker-channel',
|
|
69
|
+
match: { scope: ['mcpl:channel-incoming'], channel: 'zulip:tracker-miner-f' },
|
|
70
|
+
behavior: 'always',
|
|
71
|
+
},
|
|
72
|
+
],
|
|
73
|
+
});
|
|
74
|
+
const { registry, pushed } = makeRegistry(gate.asShouldTriggerCallback());
|
|
75
|
+
registry.handleIncoming('zulip', incomingParams('zulip:tracker-miner-f', 'question in channel'));
|
|
76
|
+
assert.strictEqual(pushed.length, 1);
|
|
77
|
+
const event = pushed[0];
|
|
78
|
+
assert.strictEqual(event.type, 'mcpl:channel-incoming');
|
|
79
|
+
assert.strictEqual(event.triggerInference, true);
|
|
80
|
+
});
|
|
81
|
+
it('default:skip wins when no policy matches the channel', () => {
|
|
82
|
+
const gate = makeGate({
|
|
83
|
+
default: 'skip',
|
|
84
|
+
policies: [
|
|
85
|
+
{
|
|
86
|
+
name: 'tracker-channel',
|
|
87
|
+
match: { scope: ['mcpl:channel-incoming'], channel: 'zulip:tracker-miner-f' },
|
|
88
|
+
behavior: 'always',
|
|
89
|
+
},
|
|
90
|
+
],
|
|
91
|
+
});
|
|
92
|
+
const { registry, pushed } = makeRegistry(gate.asShouldTriggerCallback());
|
|
93
|
+
registry.handleIncoming('zulip', incomingParams('zulip:other-channel', 'noise'));
|
|
94
|
+
assert.strictEqual(pushed.length, 1);
|
|
95
|
+
const event = pushed[0];
|
|
96
|
+
assert.strictEqual(event.triggerInference, false);
|
|
97
|
+
});
|
|
98
|
+
});
|
|
99
|
+
// ---------------------------------------------------------------------------
|
|
100
|
+
// ChannelRegistry: per-server channelSubscription policy gates auto-open
|
|
101
|
+
// ---------------------------------------------------------------------------
|
|
102
|
+
describe('ChannelRegistry subscriptionPolicy', () => {
|
|
103
|
+
function makeServerStub() {
|
|
104
|
+
const opens = [];
|
|
105
|
+
const server = {
|
|
106
|
+
sendChannelsOpen: async (args) => { opens.push(args); },
|
|
107
|
+
};
|
|
108
|
+
const serverRegistry = { getServer: () => server };
|
|
109
|
+
return { opens, serverRegistry };
|
|
110
|
+
}
|
|
111
|
+
const channels = [
|
|
112
|
+
{ id: 'zulip:tracker-miner-f', type: 'zulip-stream', address: { streamId: 1 }, name: 'tracker-miner-f' },
|
|
113
|
+
{ id: 'zulip:general', type: 'zulip-stream', address: { streamId: 2 }, name: 'general' },
|
|
114
|
+
{ id: 'zulip:random', type: 'zulip-stream', address: { streamId: 3 }, name: 'random' },
|
|
115
|
+
];
|
|
116
|
+
const registerParams = { channels };
|
|
117
|
+
it("default 'auto' opens every registered channel", async () => {
|
|
118
|
+
const { opens, serverRegistry } = makeServerStub();
|
|
119
|
+
const registry = new ChannelRegistry(serverRegistry, {}, () => { }, () => { });
|
|
120
|
+
await registry.handleRegister('zulip', registerParams);
|
|
121
|
+
assert.strictEqual(opens.length, 3);
|
|
122
|
+
});
|
|
123
|
+
it("'manual' opens nothing but still registers the channels", async () => {
|
|
124
|
+
const { opens, serverRegistry } = makeServerStub();
|
|
125
|
+
const registry = new ChannelRegistry(serverRegistry, {}, () => { }, () => { });
|
|
126
|
+
registry.setSubscriptionPolicy('zulip', 'manual');
|
|
127
|
+
await registry.handleRegister('zulip', registerParams);
|
|
128
|
+
assert.strictEqual(opens.length, 0);
|
|
129
|
+
assert.strictEqual(registry.getOpenChannels().length, 0);
|
|
130
|
+
});
|
|
131
|
+
it('string[] allow-list opens only matching channels', async () => {
|
|
132
|
+
const { opens, serverRegistry } = makeServerStub();
|
|
133
|
+
const registry = new ChannelRegistry(serverRegistry, {}, () => { }, () => { });
|
|
134
|
+
registry.setSubscriptionPolicy('zulip', ['zulip:tracker-miner-f']);
|
|
135
|
+
await registry.handleRegister('zulip', registerParams);
|
|
136
|
+
assert.strictEqual(opens.length, 1);
|
|
137
|
+
assert.deepStrictEqual(opens[0].address, { streamId: 1 });
|
|
138
|
+
});
|
|
139
|
+
});
|
|
140
|
+
//# sourceMappingURL=mcpl-gate-roundtrip.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"mcpl-gate-roundtrip.test.js","sourceRoot":"","sources":["../../test/mcpl-gate-roundtrip.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAChE,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACvE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,OAAO,EAAE,SAAS,EAAE,MAAM,2BAA2B,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,iCAAiC,CAAC;AAKlE,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,wBAAwB,CAAC,CAAC;AAEpE,UAAU,CAAC,GAAG,EAAE;IACd,SAAS,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;AAC1C,CAAC,CAAC,CAAC;AAEH,SAAS,CAAC,GAAG,EAAE;IACb,IAAI,UAAU,CAAC,OAAO,CAAC;QAAE,MAAM,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AAC7E,CAAC,CAAC,CAAC;AAEH,SAAS,QAAQ,CAAC,MAAkB;IAClC,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAC9C,aAAa,CAAC,UAAU,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC;IAClD,OAAO,IAAI,SAAS,CAAC;QACnB,UAAU;QACV,SAAS,EAAE,GAAG,EAAE,GAAE,CAAC;QACnB,UAAU,EAAE,GAAG,EAAE,CAAC,EAAE;QACpB,gBAAgB,EAAE,GAAG,EAAE,GAAE,CAAC;QAC1B,aAAa,EAAE,GAAG,EAAE,CAAC,CAAC,OAAO,CAAC;KAC/B,CAAC,CAAC;AACL,CAAC;AAED,SAAS,YAAY,CAAC,sBAA2E;IAC/F,MAAM,MAAM,GAAmB,EAAE,CAAC;IAClC,MAAM,QAAQ,GAAG,IAAI,eAAe,CAClC,EAAW,EACX,EAAW,EACX,CAAC,EAAE,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,EACvB,GAAG,EAAE,GAAE,CAAC,EACR,sBAAsB,CAAC,CAAC,CAAC,EAAE,sBAAsB,EAAE,CAAC,CAAC,CAAC,SAAS,CAChE,CAAC;IACF,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC;AAC9B,CAAC;AAED,SAAS,cAAc,CAAC,SAAiB,EAAE,IAAY;IACrD,OAAO;QACL,QAAQ,EAAE,CAAC;gBACT,SAAS;gBACT,SAAS,EAAE,OAAO;gBAClB,MAAM,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE;gBACnC,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;gBACnC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;aAClC,CAAC;KACH,CAAC;AACJ,CAAC;AAED,8EAA8E;AAC9E,gFAAgF;AAChF,8EAA8E;AAE9E,QAAQ,CAAC,mDAAmD,EAAE,GAAG,EAAE;IACjE,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,IAAI,GAAmC,EAAE,CAAC;QAChD,MAAM,EAAE,QAAQ,EAAE,GAAG,YAAY,CAAC,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE;YACvD,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACpB,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,QAAQ,CAAC,cAAc,CAAC,OAAO,EAAE,cAAc,CAAC,uBAAuB,EAAE,IAAI,CAAC,CAAC,CAAC;QAEhF,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACnC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,uBAAuB,CAAC,CAAC;QAC/D,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAC9C,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,uBAAuB,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,+EAA+E;AAC/E,8EAA8E;AAE9E,QAAQ,CAAC,uCAAuC,EAAE,GAAG,EAAE;IACrD,EAAE,CAAC,uEAAuE,EAAE,GAAG,EAAE;QAC/E,MAAM,IAAI,GAAG,QAAQ,CAAC;YACpB,OAAO,EAAE,MAAM;YACf,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,iBAAiB;oBACvB,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,uBAAuB,CAAC,EAAE,OAAO,EAAE,uBAAuB,EAAE;oBAC7E,QAAQ,EAAE,QAAQ;iBACnB;aACF;SACF,CAAC,CAAC;QAEH,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,YAAY,CAAC,IAAI,CAAC,uBAAuB,EAAE,CAAC,CAAC;QAE1E,QAAQ,CAAC,cAAc,CAAC,OAAO,EAAE,cAAc,CAAC,uBAAuB,EAAE,qBAAqB,CAAC,CAAC,CAAC;QAEjG,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACrC,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAiD,CAAC;QACxE,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,EAAE,uBAAuB,CAAC,CAAC;QACxD,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,IAAI,GAAG,QAAQ,CAAC;YACpB,OAAO,EAAE,MAAM;YACf,QAAQ,EAAE;gBACR;oBACE,IAAI,EAAE,iBAAiB;oBACvB,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,uBAAuB,CAAC,EAAE,OAAO,EAAE,uBAAuB,EAAE;oBAC7E,QAAQ,EAAE,QAAQ;iBACnB;aACF;SACF,CAAC,CAAC;QAEH,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,YAAY,CAAC,IAAI,CAAC,uBAAuB,EAAE,CAAC,CAAC;QAE1E,QAAQ,CAAC,cAAc,CAAC,OAAO,EAAE,cAAc,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC,CAAC;QAEjF,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACrC,MAAM,KAAK,GAAG,MAAM,CAAC,CAAC,CAAiD,CAAC;QACxE,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,yEAAyE;AACzE,8EAA8E;AAE9E,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;IAClD,SAAS,cAAc;QACrB,MAAM,KAAK,GAA8C,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG;YACb,gBAAgB,EAAE,KAAK,EAAE,IAAwC,EAAE,EAAE,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;SAC5F,CAAC;QACF,MAAM,cAAc,GAAG,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,MAAM,EAAiE,CAAC;QAClH,OAAO,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;IACnC,CAAC;IAED,MAAM,QAAQ,GAAG;QACf,EAAE,EAAE,EAAE,uBAAuB,EAAE,IAAI,EAAE,cAAuB,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE;QACjH,EAAE,EAAE,EAAE,eAAe,EAAE,IAAI,EAAE,cAAuB,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;QACjG,EAAE,EAAE,EAAE,cAAc,EAAE,IAAI,EAAE,cAAuB,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;KAChG,CAAC;IACF,MAAM,cAAc,GAAG,EAAE,QAAQ,EAAuC,CAAC;IAEzE,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;QAC7D,MAAM,EAAE,KAAK,EAAE,cAAc,EAAE,GAAG,cAAc,EAAE,CAAC;QACnD,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,cAAc,EAAE,EAAW,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAEtF,MAAM,QAAQ,CAAC,cAAc,CAAC,OAAO,EAAE,cAAuB,CAAC,CAAC;QAEhE,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACvE,MAAM,EAAE,KAAK,EAAE,cAAc,EAAE,GAAG,cAAc,EAAE,CAAC;QACnD,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,cAAc,EAAE,EAAW,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACtF,QAAQ,CAAC,qBAAqB,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;QAElD,MAAM,QAAQ,CAAC,cAAc,CAAC,OAAO,EAAE,cAAuB,CAAC,CAAC;QAEhE,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,WAAW,CAAC,QAAQ,CAAC,eAAe,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,EAAE,KAAK,EAAE,cAAc,EAAE,GAAG,cAAc,EAAE,CAAC;QACnD,MAAM,QAAQ,GAAG,IAAI,eAAe,CAAC,cAAc,EAAE,EAAW,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACtF,QAAQ,CAAC,qBAAqB,CAAC,OAAO,EAAE,CAAC,uBAAuB,CAAC,CAAC,CAAC;QAEnE,MAAM,QAAQ,CAAC,cAAc,CAAC,OAAO,EAAE,cAAuB,CAAC,CAAC;QAEhE,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACpC,MAAM,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-policy.test.d.ts","sourceRoot":"","sources":["../../test/tool-policy.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
import { describe, it } from 'node:test';
|
|
2
|
+
import assert from 'node:assert';
|
|
3
|
+
import { isToolAllowed } from '../src/mcpl/tool-policy.js';
|
|
4
|
+
describe('isToolAllowed', () => {
|
|
5
|
+
it('allows everything when no policy is set', () => {
|
|
6
|
+
assert.strictEqual(isToolAllowed('upload_file', undefined), true);
|
|
7
|
+
assert.strictEqual(isToolAllowed('upload_file', {}), true);
|
|
8
|
+
});
|
|
9
|
+
it('allow-list: only listed tools pass', () => {
|
|
10
|
+
const policy = { enabledTools: ['list_files', 'read_file'] };
|
|
11
|
+
assert.strictEqual(isToolAllowed('list_files', policy), true);
|
|
12
|
+
assert.strictEqual(isToolAllowed('read_file', policy), true);
|
|
13
|
+
assert.strictEqual(isToolAllowed('upload_file', policy), false);
|
|
14
|
+
});
|
|
15
|
+
it('deny-list: listed tools blocked, rest allowed', () => {
|
|
16
|
+
const policy = { disabledTools: ['delete_file', 'rename'] };
|
|
17
|
+
assert.strictEqual(isToolAllowed('delete_file', policy), false);
|
|
18
|
+
assert.strictEqual(isToolAllowed('rename', policy), false);
|
|
19
|
+
assert.strictEqual(isToolAllowed('read_file', policy), true);
|
|
20
|
+
});
|
|
21
|
+
it('deny wins over allow on overlap', () => {
|
|
22
|
+
const policy = { enabledTools: ['*'], disabledTools: ['upload_*'] };
|
|
23
|
+
assert.strictEqual(isToolAllowed('upload_file', policy), false);
|
|
24
|
+
assert.strictEqual(isToolAllowed('read_file', policy), true);
|
|
25
|
+
});
|
|
26
|
+
it('wildcards: prefix, suffix, and bare *', () => {
|
|
27
|
+
assert.strictEqual(isToolAllowed('read_file', { enabledTools: ['read_*'] }), true);
|
|
28
|
+
assert.strictEqual(isToolAllowed('write_file', { enabledTools: ['read_*'] }), false);
|
|
29
|
+
assert.strictEqual(isToolAllowed('read_file', { enabledTools: ['*_file'] }), true);
|
|
30
|
+
assert.strictEqual(isToolAllowed('read_dir', { enabledTools: ['*_file'] }), false);
|
|
31
|
+
assert.strictEqual(isToolAllowed('anything_at_all', { enabledTools: ['*'] }), true);
|
|
32
|
+
});
|
|
33
|
+
it('literal patterns require exact match (not substring)', () => {
|
|
34
|
+
const policy = { enabledTools: ['read'] };
|
|
35
|
+
assert.strictEqual(isToolAllowed('read', policy), true);
|
|
36
|
+
assert.strictEqual(isToolAllowed('read_file', policy), false);
|
|
37
|
+
});
|
|
38
|
+
it('regex metacharacters in patterns are treated literally', () => {
|
|
39
|
+
const policy = { enabledTools: ['get.file', 'foo+bar', 'baz(qux)'] };
|
|
40
|
+
assert.strictEqual(isToolAllowed('get.file', policy), true);
|
|
41
|
+
assert.strictEqual(isToolAllowed('getXfile', policy), false, '. is not a regex wildcard');
|
|
42
|
+
assert.strictEqual(isToolAllowed('foo+bar', policy), true);
|
|
43
|
+
assert.strictEqual(isToolAllowed('foobar', policy), false, '+ is not a regex quantifier');
|
|
44
|
+
assert.strictEqual(isToolAllowed('baz(qux)', policy), true);
|
|
45
|
+
});
|
|
46
|
+
it('empty arrays behave like absent fields', () => {
|
|
47
|
+
assert.strictEqual(isToolAllowed('anything', { enabledTools: [] }), true);
|
|
48
|
+
assert.strictEqual(isToolAllowed('anything', { disabledTools: [] }), true);
|
|
49
|
+
});
|
|
50
|
+
});
|
|
51
|
+
//# sourceMappingURL=tool-policy.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-policy.test.js","sourceRoot":"","sources":["../../test/tool-policy.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,aAAa,EAAE,MAAM,4BAA4B,CAAC;AAE3D,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,aAAa,EAAE,SAAS,CAAC,EAAE,IAAI,CAAC,CAAC;QAClE,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,aAAa,EAAE,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;IAC7D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,MAAM,GAAG,EAAE,YAAY,EAAE,CAAC,YAAY,EAAE,WAAW,CAAC,EAAE,CAAC;QAC7D,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,YAAY,EAAE,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;QAC9D,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,WAAW,EAAE,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;QAC7D,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,aAAa,EAAE,MAAM,CAAC,EAAE,KAAK,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,MAAM,GAAG,EAAE,aAAa,EAAE,CAAC,aAAa,EAAE,QAAQ,CAAC,EAAE,CAAC;QAC5D,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,aAAa,EAAE,MAAM,CAAC,EAAE,KAAK,CAAC,CAAC;QAChE,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,KAAK,CAAC,CAAC;QAC3D,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,WAAW,EAAE,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,MAAM,GAAG,EAAE,YAAY,EAAE,CAAC,GAAG,CAAC,EAAE,aAAa,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC;QACpE,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,aAAa,EAAE,MAAM,CAAC,EAAE,KAAK,CAAC,CAAC;QAChE,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,WAAW,EAAE,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,WAAW,EAAE,EAAE,YAAY,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;QACnF,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,YAAY,EAAE,EAAE,YAAY,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;QACrF,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,WAAW,EAAE,EAAE,YAAY,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;QACnF,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,YAAY,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;QACnF,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,iBAAiB,EAAE,EAAE,YAAY,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;IACtF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,MAAM,GAAG,EAAE,YAAY,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1C,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;QACxD,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,WAAW,EAAE,MAAM,CAAC,EAAE,KAAK,CAAC,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,MAAM,GAAG,EAAE,YAAY,EAAE,CAAC,UAAU,EAAE,SAAS,EAAE,UAAU,CAAC,EAAE,CAAC;QACrE,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,UAAU,EAAE,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;QAC5D,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,UAAU,EAAE,MAAM,CAAC,EAAE,KAAK,EAAE,2BAA2B,CAAC,CAAC;QAC1F,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,SAAS,EAAE,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;QAC3D,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,EAAE,KAAK,EAAE,6BAA6B,CAAC,CAAC;QAC1F,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,UAAU,EAAE,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;QAC1E,MAAM,CAAC,WAAW,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;IAC7E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"usage-tracker.test.d.ts","sourceRoot":"","sources":["../../test/usage-tracker.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,129 @@
|
|
|
1
|
+
import { describe, it } from 'node:test';
|
|
2
|
+
import assert from 'node:assert/strict';
|
|
3
|
+
import { UsageTracker } from '../src/usage/usage-tracker.js';
|
|
4
|
+
describe('UsageTracker', () => {
|
|
5
|
+
function createTracker() {
|
|
6
|
+
const traces = [];
|
|
7
|
+
const tracker = new UsageTracker({ emitTrace: (e) => traces.push(e) });
|
|
8
|
+
return { tracker, traces };
|
|
9
|
+
}
|
|
10
|
+
it('accumulates tokens across multiple inferences', () => {
|
|
11
|
+
const { tracker } = createTracker();
|
|
12
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 100, outputTokens: 50 });
|
|
13
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 200, outputTokens: 75, cacheCreationTokens: 10, cacheReadTokens: 30 });
|
|
14
|
+
const snap = tracker.getSnapshot();
|
|
15
|
+
assert.equal(snap.totals.inputTokens, 300);
|
|
16
|
+
assert.equal(snap.totals.outputTokens, 125);
|
|
17
|
+
assert.equal(snap.totals.cacheCreationTokens, 10);
|
|
18
|
+
assert.equal(snap.totals.cacheReadTokens, 30);
|
|
19
|
+
assert.equal(snap.inferenceCount, 2);
|
|
20
|
+
});
|
|
21
|
+
it('tracks per-agent breakdown', () => {
|
|
22
|
+
const { tracker } = createTracker();
|
|
23
|
+
tracker.onInferenceCompleted('alice', { inputTokens: 100, outputTokens: 50 });
|
|
24
|
+
tracker.onInferenceCompleted('bob', { inputTokens: 200, outputTokens: 75 });
|
|
25
|
+
tracker.onInferenceCompleted('alice', { inputTokens: 150, outputTokens: 60 });
|
|
26
|
+
const snap = tracker.getSnapshot();
|
|
27
|
+
assert.equal(snap.byAgent.length, 2);
|
|
28
|
+
const alice = snap.byAgent.find(a => a.agentName === 'alice');
|
|
29
|
+
assert.equal(alice.usage.inputTokens, 250);
|
|
30
|
+
assert.equal(alice.inferenceCount, 2);
|
|
31
|
+
const bob = snap.byAgent.find(a => a.agentName === 'bob');
|
|
32
|
+
assert.equal(bob.usage.inputTokens, 200);
|
|
33
|
+
assert.equal(bob.inferenceCount, 1);
|
|
34
|
+
});
|
|
35
|
+
it('accumulates estimated cost', () => {
|
|
36
|
+
const { tracker } = createTracker();
|
|
37
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 100, outputTokens: 50 }, { total: 0.01, currency: 'USD' });
|
|
38
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 200, outputTokens: 75 }, { total: 0.02, currency: 'USD' });
|
|
39
|
+
const snap = tracker.getSnapshot();
|
|
40
|
+
assert.ok(snap.totals.estimatedCost);
|
|
41
|
+
assert.ok(Math.abs(snap.totals.estimatedCost.total - 0.03) < 1e-10);
|
|
42
|
+
assert.equal(snap.totals.estimatedCost.currency, 'USD');
|
|
43
|
+
});
|
|
44
|
+
it('drops cost on currency mismatch', () => {
|
|
45
|
+
const { tracker } = createTracker();
|
|
46
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 100, outputTokens: 50 }, { total: 0.01, currency: 'USD' });
|
|
47
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 200, outputTokens: 75 }, { total: 0.02, currency: 'EUR' });
|
|
48
|
+
const snap = tracker.getSnapshot();
|
|
49
|
+
assert.equal(snap.totals.estimatedCost, undefined);
|
|
50
|
+
});
|
|
51
|
+
it('leaves estimatedCost undefined when no cost data provided', () => {
|
|
52
|
+
const { tracker } = createTracker();
|
|
53
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 100, outputTokens: 50 });
|
|
54
|
+
const snap = tracker.getSnapshot();
|
|
55
|
+
assert.equal(snap.totals.estimatedCost, undefined);
|
|
56
|
+
});
|
|
57
|
+
it('returns deep-copied snapshots', () => {
|
|
58
|
+
const { tracker } = createTracker();
|
|
59
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 100, outputTokens: 50 }, { total: 0.01, currency: 'USD' });
|
|
60
|
+
const snap1 = tracker.getSnapshot();
|
|
61
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 200, outputTokens: 75 }, { total: 0.02, currency: 'USD' });
|
|
62
|
+
// snap1 should not have been mutated
|
|
63
|
+
assert.equal(snap1.totals.inputTokens, 100);
|
|
64
|
+
assert.ok(snap1.totals.estimatedCost);
|
|
65
|
+
assert.ok(Math.abs(snap1.totals.estimatedCost.total - 0.01) < 1e-10);
|
|
66
|
+
assert.equal(snap1.inferenceCount, 1);
|
|
67
|
+
});
|
|
68
|
+
it('emits deep-copied trace events', () => {
|
|
69
|
+
const { tracker, traces } = createTracker();
|
|
70
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 100, outputTokens: 50 }, { total: 0.01, currency: 'USD' });
|
|
71
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 200, outputTokens: 75 }, { total: 0.02, currency: 'USD' });
|
|
72
|
+
// First trace should not have been mutated by the second inference
|
|
73
|
+
assert.equal(traces[0].totals.inputTokens, 100);
|
|
74
|
+
assert.ok(traces[0].totals.estimatedCost);
|
|
75
|
+
assert.ok(Math.abs(traces[0].totals.estimatedCost.total - 0.01) < 1e-10);
|
|
76
|
+
});
|
|
77
|
+
it('emits usage:updated trace on each inference', () => {
|
|
78
|
+
const { tracker, traces } = createTracker();
|
|
79
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 100, outputTokens: 50 });
|
|
80
|
+
assert.equal(traces.length, 1);
|
|
81
|
+
assert.equal(traces[0].type, 'usage:updated');
|
|
82
|
+
assert.equal(traces[0].agentName, 'agent');
|
|
83
|
+
assert.equal(traces[0].inferenceCount, 1);
|
|
84
|
+
});
|
|
85
|
+
it('round-trips through toJSON/restore', () => {
|
|
86
|
+
const { tracker } = createTracker();
|
|
87
|
+
tracker.onInferenceCompleted('alice', { inputTokens: 100, outputTokens: 50, cacheCreationTokens: 5, cacheReadTokens: 20 }, { total: 0.01, currency: 'USD' });
|
|
88
|
+
tracker.onInferenceCompleted('bob', { inputTokens: 200, outputTokens: 75 }, { total: 0.02, currency: 'USD' });
|
|
89
|
+
tracker.onInferenceCompleted('alice', { inputTokens: 150, outputTokens: 60 }, { total: 0.015, currency: 'USD' });
|
|
90
|
+
const persisted = tracker.toJSON();
|
|
91
|
+
// Simulate JSON round-trip (as Chronicle does)
|
|
92
|
+
const restored = new UsageTracker({
|
|
93
|
+
emitTrace: () => { },
|
|
94
|
+
restored: JSON.parse(JSON.stringify(persisted)),
|
|
95
|
+
});
|
|
96
|
+
const snap = restored.getSnapshot();
|
|
97
|
+
assert.equal(snap.totals.inputTokens, 450);
|
|
98
|
+
assert.equal(snap.totals.outputTokens, 185);
|
|
99
|
+
assert.equal(snap.totals.cacheCreationTokens, 5);
|
|
100
|
+
assert.equal(snap.totals.cacheReadTokens, 20);
|
|
101
|
+
assert.equal(snap.inferenceCount, 3);
|
|
102
|
+
assert.ok(snap.totals.estimatedCost);
|
|
103
|
+
assert.ok(Math.abs(snap.totals.estimatedCost.total - 0.045) < 1e-10);
|
|
104
|
+
assert.equal(snap.totals.estimatedCost.currency, 'USD');
|
|
105
|
+
assert.equal(snap.byAgent.length, 2);
|
|
106
|
+
const alice = snap.byAgent.find(a => a.agentName === 'alice');
|
|
107
|
+
assert.equal(alice.usage.inputTokens, 250);
|
|
108
|
+
assert.equal(alice.inferenceCount, 2);
|
|
109
|
+
const bob = snap.byAgent.find(a => a.agentName === 'bob');
|
|
110
|
+
assert.equal(bob.usage.inputTokens, 200);
|
|
111
|
+
assert.equal(bob.inferenceCount, 1);
|
|
112
|
+
});
|
|
113
|
+
it('continues accumulating after restore', () => {
|
|
114
|
+
const { tracker } = createTracker();
|
|
115
|
+
tracker.onInferenceCompleted('agent', { inputTokens: 100, outputTokens: 50 }, { total: 0.01, currency: 'USD' });
|
|
116
|
+
const restored = new UsageTracker({
|
|
117
|
+
emitTrace: () => { },
|
|
118
|
+
restored: JSON.parse(JSON.stringify(tracker.toJSON())),
|
|
119
|
+
});
|
|
120
|
+
restored.onInferenceCompleted('agent', { inputTokens: 200, outputTokens: 75 }, { total: 0.02, currency: 'USD' });
|
|
121
|
+
const snap = restored.getSnapshot();
|
|
122
|
+
assert.equal(snap.totals.inputTokens, 300);
|
|
123
|
+
assert.equal(snap.totals.outputTokens, 125);
|
|
124
|
+
assert.equal(snap.inferenceCount, 2);
|
|
125
|
+
assert.ok(snap.totals.estimatedCost);
|
|
126
|
+
assert.ok(Math.abs(snap.totals.estimatedCost.total - 0.03) < 1e-10);
|
|
127
|
+
});
|
|
128
|
+
});
|
|
129
|
+
//# sourceMappingURL=usage-tracker.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"usage-tracker.test.js","sourceRoot":"","sources":["../../test/usage-tracker.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAG7D,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;IAC5B,SAAS,aAAa;QACpB,MAAM,MAAM,GAAwB,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QACvE,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC;IAC7B,CAAC;IAED,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,EAAE,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QAC9E,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,mBAAmB,EAAE,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC,CAAC;QAE5H,MAAM,IAAI,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;QAC5C,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,EAAE,CAAC,CAAC;QAClD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,EAAE,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QAC9E,OAAO,CAAC,oBAAoB,CAAC,KAAK,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QAC5E,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QAE9E,MAAM,IAAI,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAErC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,OAAO,CAAE,CAAC;QAC/D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;QAEtC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,KAAK,CAAE,CAAC;QAC3D,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,EAAE,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAChH,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEhH,MAAM,IAAI,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACnC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACrC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;QACpE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,EAAE,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAChH,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEhH,MAAM,IAAI,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,EAAE,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QAE9E,MAAM,IAAI,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACnC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,SAAS,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,EAAE,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEhH,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEhH,qCAAqC;QACrC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAC5C,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACtC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;QACrE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,aAAa,EAAE,CAAC;QAC5C,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAChH,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEhH,mEAAmE;QACnE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAChD,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QAC1C,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,aAAa,EAAE,CAAC;QAC5C,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QAE9E,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAC5C,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,EAAE,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,mBAAmB,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7J,OAAO,CAAC,oBAAoB,CAAC,KAAK,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAC9G,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEjH,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QACnC,+CAA+C;QAC/C,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC;YAChC,SAAS,EAAE,GAAG,EAAE,GAAE,CAAC;YACnB,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;SAChD,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;QAC5C,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC,CAAC;QACjD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;QAC9C,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACrC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,GAAG,KAAK,CAAC,GAAG,KAAK,CAAC,CAAC;QACrE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAExD,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACrC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,OAAO,CAAE,CAAC;QAC/D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;QAEtC,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,KAAK,CAAE,CAAC;QAC3D,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,EAAE,OAAO,EAAE,GAAG,aAAa,EAAE,CAAC;QACpC,OAAO,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEhH,MAAM,QAAQ,GAAG,IAAI,YAAY,CAAC;YAChC,SAAS,EAAE,GAAG,EAAE,GAAE,CAAC;YACnB,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;SACvD,CAAC,CAAC;QACH,QAAQ,CAAC,oBAAoB,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC,CAAC;QAEjH,MAAM,IAAI,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;QACpC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAC3C,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;QAC5C,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC;QACrC,MAAM,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"workspace-watcher.test.d.ts","sourceRoot":"","sources":["../../test/workspace-watcher.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,117 @@
|
|
|
1
|
+
import { describe, it, beforeEach, afterEach } from 'node:test';
|
|
2
|
+
import assert from 'node:assert';
|
|
3
|
+
import { existsSync, mkdtempSync, writeFileSync, unlinkSync, rmSync } from 'node:fs';
|
|
4
|
+
import { tmpdir } from 'node:os';
|
|
5
|
+
import { join } from 'node:path';
|
|
6
|
+
import { MountWatcher } from '../src/modules/workspace/watcher.js';
|
|
7
|
+
// Short debounce keeps each test under ~200ms.
|
|
8
|
+
const DEBOUNCE_MS = 50;
|
|
9
|
+
// Chokidar's awaitWriteFinish (stabilityThreshold:100) + our debounce means
|
|
10
|
+
// each emission takes ~150ms to land; wait at least 300ms.
|
|
11
|
+
const SETTLE_MS = 350;
|
|
12
|
+
let tmp = '';
|
|
13
|
+
beforeEach(() => {
|
|
14
|
+
tmp = mkdtempSync(join(tmpdir(), 'mw-test-'));
|
|
15
|
+
});
|
|
16
|
+
afterEach(() => {
|
|
17
|
+
if (tmp)
|
|
18
|
+
rmSync(tmp, { recursive: true, force: true });
|
|
19
|
+
});
|
|
20
|
+
function makeConfig() {
|
|
21
|
+
return {
|
|
22
|
+
name: 'test',
|
|
23
|
+
path: tmp,
|
|
24
|
+
mode: 'read-write',
|
|
25
|
+
watch: 'always',
|
|
26
|
+
watchDebounceMs: DEBOUNCE_MS,
|
|
27
|
+
};
|
|
28
|
+
}
|
|
29
|
+
function collect() {
|
|
30
|
+
const batches = [];
|
|
31
|
+
const watcher = new MountWatcher(makeConfig(), (changes) => {
|
|
32
|
+
batches.push(changes);
|
|
33
|
+
});
|
|
34
|
+
watcher.start();
|
|
35
|
+
return { watcher, batches };
|
|
36
|
+
}
|
|
37
|
+
async function wait(ms) {
|
|
38
|
+
await new Promise(r => setTimeout(r, ms));
|
|
39
|
+
}
|
|
40
|
+
describe('MountWatcher mergeOp', () => {
|
|
41
|
+
it('unlink + add within debounce window → modified (atomic save)', async () => {
|
|
42
|
+
// Seed a file before starting the watcher, so add+unlink+add reduces to
|
|
43
|
+
// unlink+add (atomic save shape).
|
|
44
|
+
const file = join(tmp, 'ticket.md');
|
|
45
|
+
writeFileSync(file, 'v1');
|
|
46
|
+
const { watcher, batches } = collect();
|
|
47
|
+
await wait(100); // let chokidar initialize
|
|
48
|
+
// Atomic save: unlink then write
|
|
49
|
+
unlinkSync(file);
|
|
50
|
+
writeFileSync(file, 'v2');
|
|
51
|
+
await wait(SETTLE_MS);
|
|
52
|
+
await watcher.stop();
|
|
53
|
+
const flat = batches.flat();
|
|
54
|
+
const forFile = flat.filter(c => c.path === 'ticket.md');
|
|
55
|
+
// Expect exactly one batch entry for the file, with op=modified.
|
|
56
|
+
assert.strictEqual(forFile.length, 1, `expected 1 change, got ${JSON.stringify(flat)}`);
|
|
57
|
+
assert.strictEqual(forFile[0].op, 'modified');
|
|
58
|
+
});
|
|
59
|
+
it('create + modify within window → created (net new)', async () => {
|
|
60
|
+
const { watcher, batches } = collect();
|
|
61
|
+
await wait(100);
|
|
62
|
+
const file = join(tmp, 'new.md');
|
|
63
|
+
writeFileSync(file, 'hello');
|
|
64
|
+
// awaitWriteFinish coalesces rapid writes, so spacing the second write
|
|
65
|
+
// slightly increases the chance chokidar surfaces both add and change.
|
|
66
|
+
await wait(10);
|
|
67
|
+
writeFileSync(file, 'hello updated');
|
|
68
|
+
await wait(SETTLE_MS);
|
|
69
|
+
await watcher.stop();
|
|
70
|
+
const forFile = batches.flat().filter(c => c.path === 'new.md');
|
|
71
|
+
assert.strictEqual(forFile.length, 1);
|
|
72
|
+
assert.strictEqual(forFile[0].op, 'created');
|
|
73
|
+
});
|
|
74
|
+
it('delete + create of a pre-existing file → modified (covers atomic-save shape without relying on chokidar add/unlink coalesce timing)', async () => {
|
|
75
|
+
// Seed so the watcher's first observation is the unlink of a real file.
|
|
76
|
+
const file = join(tmp, 'a.md');
|
|
77
|
+
writeFileSync(file, 'v1');
|
|
78
|
+
const { watcher, batches } = collect();
|
|
79
|
+
await wait(100);
|
|
80
|
+
// Double-flip: unlink + recreate + unlink + recreate inside one window.
|
|
81
|
+
unlinkSync(file);
|
|
82
|
+
writeFileSync(file, 'v2');
|
|
83
|
+
await wait(SETTLE_MS);
|
|
84
|
+
await watcher.stop();
|
|
85
|
+
const forFile = batches.flat().filter(c => c.path === 'a.md');
|
|
86
|
+
assert.strictEqual(forFile.length, 1);
|
|
87
|
+
assert.strictEqual(forFile[0].op, 'modified');
|
|
88
|
+
});
|
|
89
|
+
});
|
|
90
|
+
describe('MountWatcher lifecycle', () => {
|
|
91
|
+
it('mkdir -p covers missing parent dir at attach time (case C repro)', async () => {
|
|
92
|
+
const nested = join(tmp, 'parent-also-missing', 'tickets');
|
|
93
|
+
assert.ok(!existsSync(nested), 'precondition: nested path should not exist');
|
|
94
|
+
const batches = [];
|
|
95
|
+
let ready = false;
|
|
96
|
+
const watcher = new MountWatcher({ name: 'tickets', path: nested, mode: 'read-write', watch: 'always', watchDebounceMs: DEBOUNCE_MS }, (changes) => { batches.push(changes); }, { onReady: () => { ready = true; } });
|
|
97
|
+
watcher.start();
|
|
98
|
+
await wait(150);
|
|
99
|
+
assert.ok(existsSync(nested), 'start() should have created the watched path');
|
|
100
|
+
writeFileSync(join(nested, 'ticket.md'), 'hi');
|
|
101
|
+
await wait(SETTLE_MS);
|
|
102
|
+
await watcher.stop();
|
|
103
|
+
assert.ok(ready, 'onReady should have fired');
|
|
104
|
+
const created = batches.flat().find(c => c.path === 'ticket.md');
|
|
105
|
+
assert.ok(created, `expected created event for ticket.md, got ${JSON.stringify(batches.flat())}`);
|
|
106
|
+
assert.strictEqual(created.op, 'created');
|
|
107
|
+
});
|
|
108
|
+
it('read-only mount does not create the watched path', async () => {
|
|
109
|
+
const roPath = join(tmp, 'read-only-missing');
|
|
110
|
+
const watcher = new MountWatcher({ name: 'ro', path: roPath, mode: 'read-only', watch: 'always', watchDebounceMs: DEBOUNCE_MS }, () => { });
|
|
111
|
+
watcher.start();
|
|
112
|
+
await wait(50);
|
|
113
|
+
assert.ok(!existsSync(roPath), 'read-only mount must not mkdir -p behind the user\'s back');
|
|
114
|
+
await watcher.stop();
|
|
115
|
+
});
|
|
116
|
+
});
|
|
117
|
+
//# sourceMappingURL=workspace-watcher.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"workspace-watcher.test.js","sourceRoot":"","sources":["../../test/workspace-watcher.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AAChE,OAAO,MAAM,MAAM,aAAa,CAAC;AACjC,OAAO,EAAE,UAAU,EAAE,WAAW,EAAa,aAAa,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAChG,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,OAAO,EAAE,YAAY,EAAiB,MAAM,qCAAqC,CAAC;AAGlF,+CAA+C;AAC/C,MAAM,WAAW,GAAG,EAAE,CAAC;AACvB,4EAA4E;AAC5E,2DAA2D;AAC3D,MAAM,SAAS,GAAG,GAAG,CAAC;AAEtB,IAAI,GAAG,GAAG,EAAE,CAAC;AAEb,UAAU,CAAC,GAAG,EAAE;IACd,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,UAAU,CAAC,CAAC,CAAC;AAChD,CAAC,CAAC,CAAC;AAEH,SAAS,CAAC,GAAG,EAAE;IACb,IAAI,GAAG;QAAE,MAAM,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;AACzD,CAAC,CAAC,CAAC;AAEH,SAAS,UAAU;IACjB,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,IAAI,EAAE,GAAG;QACT,IAAI,EAAE,YAAY;QAClB,KAAK,EAAE,QAAQ;QACf,eAAe,EAAE,WAAW;KAC7B,CAAC;AACJ,CAAC;AAED,SAAS,OAAO;IACd,MAAM,OAAO,GAAiB,EAAE,CAAC;IACjC,MAAM,OAAO,GAAG,IAAI,YAAY,CAAC,UAAU,EAAE,EAAE,CAAC,OAAO,EAAE,EAAE;QACzD,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxB,CAAC,CAAC,CAAC;IACH,OAAO,CAAC,KAAK,EAAE,CAAC;IAChB,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;AAC9B,CAAC;AAED,KAAK,UAAU,IAAI,CAAC,EAAU;IAC5B,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;AAC5C,CAAC;AAED,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,wEAAwE;QACxE,kCAAkC;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;QACpC,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAE1B,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,OAAO,EAAE,CAAC;QACvC,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,0BAA0B;QAE3C,iCAAiC;QACjC,UAAU,CAAC,IAAI,CAAC,CAAC;QACjB,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAE1B,MAAM,IAAI,CAAC,SAAS,CAAC,CAAC;QACtB,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QAErB,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QACzD,iEAAiE;QACjE,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,0BAA0B,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxF,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,OAAO,EAAE,CAAC;QACvC,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;QAEhB,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;QACjC,aAAa,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAC7B,uEAAuE;QACvE,uEAAuE;QACvE,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,aAAa,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;QAErC,MAAM,IAAI,CAAC,SAAS,CAAC,CAAC;QACtB,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QAErB,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC;QAChE,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qIAAqI,EAAE,KAAK,IAAI,EAAE;QACnJ,wEAAwE;QACxE,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAC/B,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAE1B,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,OAAO,EAAE,CAAC;QACvC,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;QAEhB,wEAAwE;QACxE,UAAU,CAAC,IAAI,CAAC,CAAC;QACjB,aAAa,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAE1B,MAAM,IAAI,CAAC,SAAS,CAAC,CAAC;QACtB,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QAErB,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;QAC9D,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;QACtC,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,qBAAqB,EAAE,SAAS,CAAC,CAAC;QAC3D,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,4CAA4C,CAAC,CAAC;QAE7E,MAAM,OAAO,GAAiB,EAAE,CAAC;QACjC,IAAI,KAAK,GAAG,KAAK,CAAC;QAClB,MAAM,OAAO,GAAG,IAAI,YAAY,CAC9B,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,QAAQ,EAAE,eAAe,EAAE,WAAW,EAAE,EACpG,CAAC,OAAO,EAAE,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EACvC,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,EAAE,CACrC,CAAC;QACF,OAAO,CAAC,KAAK,EAAE,CAAC;QAEhB,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;QAChB,MAAM,CAAC,EAAE,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,8CAA8C,CAAC,CAAC;QAE9E,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE,IAAI,CAAC,CAAC;QAC/C,MAAM,IAAI,CAAC,SAAS,CAAC,CAAC;QACtB,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;QAErB,MAAM,CAAC,EAAE,CAAC,KAAK,EAAE,2BAA2B,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QACjE,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,6CAA6C,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAClG,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,mBAAmB,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAG,IAAI,YAAY,CAC9B,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,eAAe,EAAE,WAAW,EAAE,EAC9F,GAAG,EAAE,GAAE,CAAC,CACT,CAAC;QACF,OAAO,CAAC,KAAK,EAAE,CAAC;QAChB,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC;QACf,MAAM,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,2DAA2D,CAAC,CAAC;QAC5F,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|