fix: fix test and add delay between requests

Signed-off-by: Jeff MAURY <[email protected]>
containers · Feb 5, 2024 · b2ce138 · b2ce138
1 parent d9dbeef
commit b2ce138
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/packages/backend/src/managers/playground.spec.ts b/packages/backend/src/managers/playground.spec.ts
@@ -69,7 +69,9 @@ beforeEach(() => {
     {} as PodmanConnection,
   );
   originalFetch = globalThis.fetch;
-  globalThis.fetch = vi.fn().mockResolvedValue({});
+  globalThis.fetch = vi.fn().mockResolvedValue({
+    ok: true,
+  });
 });
 
 afterEach(() => {

diff --git a/packages/backend/src/managers/playground.ts b/packages/backend/src/managers/playground.ts
@@ -33,6 +33,7 @@ import type { PlaygroundState, PlaygroundStatus } from '@shared/src/models/IPlay
 import type { ContainerRegistry } from '../registries/ContainerRegistry';
 import type { PodmanConnection } from './podmanConnection';
 import OpenAI from 'openai';
+import { timeout } from '../utils/utils';
 
 const LABEL_MODEL_ID = 'ai-studio-model-id';
 const LABEL_MODEL_PORT = 'ai-studio-model-port';
@@ -216,6 +217,7 @@ export class PlayGroundManager {
       try {
         const response = await fetch(`http://localhost:${freePort}`);
         contacted = response.ok;
+        await timeout(1000);
       } catch (err: unknown) {
         /* empty */
       }