openai-api-mock 0.1.28 → 0.1.29

package/__tests__/mock_delay.test.js

@@ -0,0 +1,28 @@
+import fetch from 'node-fetch';
+import { mockOpenAIResponse, stopMocking } from '../src/index.js';
+import OpenAI from 'openai'; 
+
+const openai = new OpenAI({ apiKey: "OPENAI_API_KEY" });
+
+describe('Mock OpenAI Chat with a delay', () => {
+
+    beforeEach(() => {
+        mockOpenAIResponse(true, { latency: 1000 , logRequests: true });
+    });
+
+    afterEach(() => {
+        stopMocking();
+    });
+
+    test('should mock OpenAI Chat with a delay', async () => {
+        const response = await openai.chat.completions.create({
+            model: "gpt-3.5",
+            messages: [
+                { role: 'system', content: "You're an expert chef" },
+                { role: 'user', content: "Suggest at least 5 recipes" },
+            ]
+        });
+
+        expect(response).toBeDefined();
+    });
+});

package/dist/index.cjs

@@ -2199,8 +2199,7 @@ function mockOpenAIResponse(force = false, options = {}) {
   if (env !== "development" && !force) {
     return { isActive: false, stopMocking };
   }
-  const mockWithDelay = latency > 0 ? nock(OPEN_AI_BASE_URL).delay(latency) : nock(OPEN_AI_BASE_URL);
-  mockWithDelay.post(CHAT_COMPLETIONS_ENDPOINT).reply(function(uri, requestBody) {
+  nock(OPEN_AI_BASE_URL).post(CHAT_COMPLETIONS_ENDPOINT).delay(latency).reply(function(uri, requestBody) {
     if (logRequests) {
       console.log(`[openai-api-mock] Chat request:`, JSON.stringify(requestBody, null, 2));
     }
@@ -2222,7 +2221,7 @@ function mockOpenAIResponse(force = false, options = {}) {
       return [500, { error: { message: "Internal server error in mock" } }];
     }
   });
-  mockWithDelay.post(IMAGE_GENERATIONS_ENDPOINT).reply(function(uri, requestBody) {
+  nock(OPEN_AI_BASE_URL).post(IMAGE_GENERATIONS_ENDPOINT).delay(latency).reply(function(uri, requestBody) {
     if (logRequests) {
       console.log(`[openai-api-mock] Image request:`, JSON.stringify(requestBody, null, 2));
     }