From d25f1d9bfb61c489048336dd8207642bd0821e30 Mon Sep 17 00:00:00 2001
From: "google-labs-jules[bot]"
 <161369871+google-labs-jules[bot]@users.noreply.github.com>
Date: Fri, 12 Jun 2026 06:49:44 +0000
Subject: [PATCH] test: add unit tests for handleRateLimit in apiUtils

Co-authored-by: is0692vs <135803462+is0692vs@users.noreply.github.com>
---
 src/lib/__tests__/apiUtils.test.ts | 44 ++++++++++++++++++++++++++++--
 1 file changed, 42 insertions(+), 2 deletions(-)

diff --git a/src/lib/__tests__/apiUtils.test.ts b/src/lib/__tests__/apiUtils.test.ts
index fbd65baa..5e4ad68f 100644
--- a/src/lib/__tests__/apiUtils.test.ts
+++ b/src/lib/__tests__/apiUtils.test.ts
@@ -1,5 +1,6 @@
-import { describe, it, expect, vi, beforeEach } from 'vitest';
-import { handleErrorResponse, getAuthenticatedUser } from '../apiUtils';
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { handleErrorResponse, getAuthenticatedUser, handleRateLimit } from '../apiUtils';
+import { RateLimitError } from '../types';
 import { NextResponse } from 'next/server';
 import { getServerSession } from "next-auth";
 
@@ -56,6 +57,45 @@ describe('apiUtils', () => {
     });
   });
 
+  describe('handleRateLimit', () => {
+    beforeEach(() => {
+      vi.useFakeTimers();
+      vi.setSystemTime(new Date(1000000));
+    });
+
+    afterEach(() => {
+      vi.useRealTimers();
+    });
+
+    it('should throw RateLimitError with timestamp from X-RateLimit-Reset header', () => {
+      const res = new Response(null, {
+        headers: { 'X-RateLimit-Reset': '2000' }
+      });
+      expect(() => handleRateLimit(res)).toThrowError(expect.objectContaining({
+        name: 'RateLimitError',
+        resetAt: new Date(2000000)
+      }));
+    });
+
+    it('should throw RateLimitError with fallback timestamp if header is missing', () => {
+      const res = new Response();
+      expect(() => handleRateLimit(res)).toThrowError(expect.objectContaining({
+        name: 'RateLimitError',
+        resetAt: new Date(4600000)
+      }));
+    });
+
+    it('should throw RateLimitError with fallback timestamp if header is invalid', () => {
+      const res = new Response(null, {
+        headers: { 'X-RateLimit-Reset': 'invalid' }
+      });
+      expect(() => handleRateLimit(res)).toThrowError(expect.objectContaining({
+        name: 'RateLimitError',
+        resetAt: new Date(4600000)
+      }));
+    });
+  });
+
   describe('getAuthenticatedUser', () => {
     it('should return user object if session is valid', async () => {
       vi.mocked(getServerSession).mockResolvedValueOnce({