fix: update tests to catch llm fails

upstash · Nov 6, 2024 · 6bc2413 · 6bc2413
1 parent 8dedafe
commit 6bc2413
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/src/client/api/llm.test.ts b/src/client/api/llm.test.ts
@@ -42,6 +42,7 @@ describe("llm", () => {
           "upstash-forward-authorization": null,
           "upstash-callback": callback,
           "upstash-method": "POST",
+          "content-type": "application/json",
         },
       },
     });
@@ -77,6 +78,7 @@ describe("llm", () => {
           "upstash-forward-helicone-auth": `Bearer ${analyticsToken}`,
           "upstash-callback": callback,
           "upstash-method": "POST",
+          "content-type": "application/json",
         },
       },
     });
@@ -110,6 +112,7 @@ describe("llm", () => {
           "upstash-forward-authorization": `Bearer ${llmToken}`,
           "upstash-callback": callback,
           "upstash-method": "POST",
+          "content-type": "application/json",
         },
       },
     });
@@ -146,6 +149,7 @@ describe("llm", () => {
           "upstash-forward-helicone-target-url": "https://api.openai.com",
           "upstash-callback": callback,
           "upstash-method": "POST",
+          "content-type": "application/json",
         },
       },
     });
@@ -179,6 +183,7 @@ describe("llm", () => {
           "upstash-forward-x-api-key": llmToken,
           "upstash-callback": callback,
           "upstash-method": "POST",
+          "content-type": "application/json",
         },
       },
     });
@@ -215,6 +220,7 @@ describe("llm", () => {
           "upstash-forward-helicone-target-url": "https://api.anthropic.com",
           "upstash-callback": callback,
           "upstash-method": "POST",
+          "content-type": "application/json",
         },
       },
     });
@@ -248,6 +254,7 @@ describe("llm", () => {
           "upstash-forward-authorization": `Bearer ${llmToken}`,
           "upstash-callback": callback,
           "upstash-method": "POST",
+          "content-type": "application/json",
         },
       },
     });
@@ -284,6 +291,7 @@ describe("llm", () => {
           "upstash-forward-helicone-target-url": customBaseUrl,
           "upstash-callback": callback,
           "upstash-method": "POST",
+          "content-type": "application/json",
         },
       },
     });

diff --git a/src/client/llm/chat.test.ts b/src/client/llm/chat.test.ts
@@ -132,6 +132,14 @@ describe("Test QStash chat", () => {
       callback: "https://example.com",
     });
     expect(result.messageId).toBeTruthy();
+
+    // sleep before checking the message
+    // eslint-disable-next-line @typescript-eslint/no-magic-numbers
+    await new Promise((r) => setTimeout(r, 3000));
+
+    const { events } = await client.events({ filter: { messageId: result.messageId } });
+    const deliveredEvent = events.find((event) => event.state === "DELIVERED");
+    expect(deliveredEvent).not.toBeUndefined();
   });
 
   test("should batch with llm api", async () => {
@@ -298,6 +306,14 @@ describe("Test QStash chat with third party LLMs", () => {
       callback: "https://oz.requestcatcher.com/",
     });
     expect(result.messageId).toBeTruthy();
+
+    // sleep before checking the message
+    // eslint-disable-next-line @typescript-eslint/no-magic-numbers
+    await new Promise((r) => setTimeout(r, 3000));
+
+    const { events } = await client.events({ filter: { messageId: result.messageId } });
+    const deliveredEvent = events.find((event) => event.state === "DELIVERED");
+    expect(deliveredEvent).not.toBeUndefined();
   });
 
   test("should publish with llm api", () => {