diff --git a/tests/unit/cursor/cursor-models.test.ts b/tests/unit/cursor/cursor-models.test.ts index 6c5b3cc9..ce6d6635 100644 --- a/tests/unit/cursor/cursor-models.test.ts +++ b/tests/unit/cursor/cursor-models.test.ts @@ -168,20 +168,21 @@ describe('fetchModelsFromDaemon', () => { res.end(oversizedPayload); }); - await new Promise((resolve) => server.listen(0, '127.0.0.1', resolve)); - const address = server.address(); - if (!address || typeof address === 'string') { - throw new Error('Unable to resolve test server port'); - } + await new Promise((resolve) => server.listen(0, '127.0.0.1', resolve)); + const address = server.address(); + if (!address || typeof address === 'string') { + throw new Error('Unable to resolve test server port'); + } try { const models = await fetchModelsFromDaemon(address.port); expect(models).toEqual(DEFAULT_CURSOR_MODELS); } finally { + server.closeAllConnections?.(); await new Promise((resolve) => server.close(() => resolve())); } }, - 10000 + 30000 ); });