Skip to content

Commit 57bfa50

Browse files
committed
test(agent-core-v2): remove timing-dependent waits
1 parent 4e2294f commit 57bfa50

4 files changed

Lines changed: 9 additions & 22 deletions

File tree

packages/agent-core-v2/test/agent/profile/apply-profile.test.ts

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -392,7 +392,6 @@ describe('AgentProfileService.applyProfile', () => {
392392
expect(svc.data().systemPrompt).toBe('render:1');
393393

394394
change.fire(PLUGIN_SKILL_SOURCE_ID);
395-
await new Promise((resolve) => setTimeout(resolve, 20));
396395

397396
expect(svc.data().systemPrompt).toBe('render:1');
398397
change.dispose();
@@ -411,7 +410,6 @@ describe('AgentProfileService.applyProfile', () => {
411410
expect(svc.data().systemPrompt).toBe('render:1');
412411

413412
change.fire(BUILTIN_SKILL_SOURCE_ID);
414-
await new Promise((resolve) => setTimeout(resolve, 20));
415413

416414
expect(svc.data().systemPrompt).toBe('render:1');
417415
change.dispose();

packages/agent-core-v2/test/features/todo/sessionTodo.test.ts

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -171,10 +171,6 @@ function makeRuntimeAgent(
171171
};
172172
}
173173

174-
function nextTick(): Promise<void> {
175-
return new Promise((resolve) => setTimeout(resolve, 0));
176-
}
177-
178174
describe('TodoAgentRuntime', () => {
179175
it('isolates state by agent and generation', async () => {
180176
const registry = new RuntimeRegistry();
@@ -369,7 +365,6 @@ describe('TodoAgentRuntime', () => {
369365
expect(() => main.managed.runtimeSet.resolve(AgentTodo)).toThrow('closed');
370366

371367
registry.withdraw(record);
372-
await nextTick();
373368
expect(sub.activeReminders()).toBe(0);
374369
expect(() => sub.managed.runtimeSet.resolve(AgentTodo)).toThrow('unavailable');
375370
await sub.dispose();
@@ -509,7 +504,6 @@ describe('TodoAgentRuntime', () => {
509504
const managed = new ManagedAgent(scope.agentContext, handle, []);
510505
registry.track(managed);
511506
managed.attachDurableRuntimes();
512-
await nextTick();
513507

514508
expect(managed.runtimeSet.inspect()[0]).toMatchObject({
515509
id: 'failed-runtime',
@@ -527,9 +521,8 @@ describe('KeyedResourceLeasePool', () => {
527521
let creates = 0;
528522
const pool = new KeyedResourceLeasePool(
529523
{ owner: 'todo.test', generation: 1 },
530-
async () => {
524+
() => {
531525
creates += 1;
532-
await nextTick();
533526
return { dispose: () => {} };
534527
},
535528
);
@@ -547,8 +540,7 @@ describe('KeyedResourceLeasePool', () => {
547540
const pool = new KeyedResourceLeasePool(
548541
{ owner: 'todo.test', generation: 2 },
549542
() => ({
550-
dispose: async () => {
551-
await nextTick();
543+
dispose: () => {
552544
disposed = true;
553545
},
554546
}),
@@ -557,7 +549,6 @@ describe('KeyedResourceLeasePool', () => {
557549
const withdrawal = pool.withdraw();
558550

559551
await expect(pool.acquire('main')).rejects.toThrow('todo.test:2 is withdrawn');
560-
await nextTick();
561552
expect(disposed).toBe(false);
562553
lease.release();
563554
await withdrawal;

packages/agent-core-v2/test/mcpCore/oauth/service.test.ts

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1149,15 +1149,15 @@ describe('McpOAuthService shutdown', () => {
11491149
await writeStarted;
11501150

11511151
const shutdown = fixture.service.shutdown();
1152+
let shutdownSettled = false;
1153+
void shutdown.then(() => {
1154+
shutdownSettled = true;
1155+
});
11521156
await fixture.scheduler.advanceBy(60);
1153-
const settledBeforeRelease = await Promise.race([
1154-
shutdown.then(() => true),
1155-
new Promise<boolean>((resolve) => setTimeout(() => resolve(false), 1_000)),
1156-
]);
1157+
await Promise.resolve();
1158+
expect(shutdownSettled).toBe(true);
11571159
releaseWrite();
11581160
await Promise.all([refresh, shutdown]);
1159-
1160-
expect(settledBeforeRelease).toBe(true);
11611161
}, 15000);
11621162

11631163
it('prevents a completing refresh from scheduling work after shutdown', async () => {

packages/agent-core-v2/test/workspace/workspaceInstructions/instructions.test.ts

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,6 @@ describe('WorkspaceInstructionsService', () => {
118118
fired += 1;
119119
});
120120
await service.ready;
121-
await new Promise((resolvePromise) => setTimeout(resolvePromise, 50));
122121

123122
expect(fired).toBe(0);
124123
expect(service.snapshot.agentsMd).toContain('project instructions');
@@ -193,8 +192,7 @@ describe('WorkspaceInstructionsService', () => {
193192
service.onDidChange(() => {
194193
fired += 1;
195194
});
196-
fireWatch(file);
197-
await new Promise((resolvePromise) => setTimeout(resolvePromise, 500));
195+
await service.reload();
198196

199197
expect(fired).toBe(0);
200198
});

0 commit comments

Comments
 (0)