An attempt to fix ChatBot tests.

ansible · Dec 17, 2024 · 47bb672 · 47bb672
1 parent 1a7622f
commit 47bb672
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/ansible_ai_connect_chatbot/src/App.test.tsx b/ansible_ai_connect_chatbot/src/App.test.tsx
@@ -28,8 +28,6 @@ describe("App tests", () => {
     );
   };
 
-  const delay = (ms: number) => new Promise((res) => setTimeout(res, ms));
-
   const createError = (message: string, status: number): AxiosError => {
     const request = { path: "/chat" };
     const headers = new AxiosHeaders({
@@ -255,13 +253,14 @@ describe("App tests", () => {
   });
 
   it("Chat service returns 429 Too Many Requests error", async () => {
+    vi.useFakeTimers();
     mockAxios(429, true);
     renderApp();
     const textArea = screen.getByLabelText("Send a message...");
     await act(async () => userEvent.type(textArea, "Hello"));
     const sendButton = screen.getByLabelText("Send button");
     await act(async () => fireEvent.click(sendButton));
-    await delay(3100);
+    await act(async () => vi.runAllTimers());
     expect(
       screen.getByText("Chatbot service is busy with too many requests. ", {
         exact: false,