test(utils): add rate limit test case to superuser login

Author: pawcoding

test/utils/get-superuser-token.e2e-spec.ts

@@ -1,4 +1,4 @@
-import { assert, beforeAll, describe, expect, it } from "vitest";
+import { assert, beforeAll, describe, expect, it, vi } from "vitest";
 import { getSuperuserToken } from "../../src/utils/get-superuser-token";
 import { checkE2eConnection } from "../_mocks/check-e2e-connection";
 import { createLoaderContext } from "../_mocks/create-loader-context";
@@ -16,6 +16,7 @@ describe("getSuperuserToken", () => {
       email: "invalid",
       password: "invalid"
     });
+
     expect(result).toBeUndefined();
   });
 
@@ -42,6 +43,37 @@ describe("getSuperuserToken", () => {
       options.url,
       options.superuserCredentials
     );
+
     expect(result).toBeDefined();
   });
+
+  it("should retry on rate limit error", async () => {
+    assert(options.superuserCredentials, "Superuser credentials are not set.");
+    assert(
+      !("impersonateToken" in options.superuserCredentials),
+      "Impersonate token should not be used in tests."
+    );
+
+    vi.useFakeTimers({
+      toFake: ["setTimeout"]
+    });
+    vi.spyOn(global, "fetch").mockResolvedValueOnce(
+      new Response(undefined, { status: 429 })
+    );
+
+    const promise = getSuperuserToken(
+      options.url,
+      options.superuserCredentials
+    );
+
+    // Fast-forward time to speed up retries
+    await vi.runAllTimersAsync();
+
+    const result = await promise;
+
+    expect(result).toBeDefined();
+    expect(global.fetch).toHaveBeenCalledTimes(2);
+
+    vi.useRealTimers();
+  });
 });