test: cover convo history download routes

Author: rohan-uiuc

__tests__/pages/api/UIUC-api/__tests__/downloadConvoHistory.test.ts

@@ -57,4 +57,55 @@ describe('UIUC-api/downloadConvoHistory', () => {
     expect(res2.status).toHaveBeenCalledWith(200)
     expect(res2.send).toHaveBeenCalled()
   })
+
+  it('handles JSON response that is not the S3 download case', async () => {
+    hoisted.axiosGet.mockResolvedValueOnce({
+      headers: { 'content-type': 'application/json' },
+      data: Buffer.from(JSON.stringify({ response: 'Not S3' })),
+    })
+    const res = createMockRes()
+    await handler(
+      createMockReq({ method: 'GET', query: { course_name: 'CS101' } }) as any,
+      res as any,
+    )
+    expect(res.status).toHaveBeenCalledWith(200)
+    expect(res.json).toHaveBeenCalledWith(
+      expect.objectContaining({
+        message: 'Your conversation history is ready for download.',
+      }),
+    )
+  })
+
+  it('handles unexpected content types from the backend', async () => {
+    hoisted.axiosGet.mockResolvedValueOnce({
+      headers: { 'content-type': 'text/plain' },
+      data: Buffer.from('nope'),
+    })
+    const res = createMockRes()
+    await handler(
+      createMockReq({ method: 'GET', query: { course_name: 'CS101' } }) as any,
+      res as any,
+    )
+    expect(res.status).toHaveBeenCalledWith(500)
+    expect(res.json).toHaveBeenCalledWith(
+      expect.objectContaining({
+        message: expect.stringContaining('Unexpected response format'),
+      }),
+    )
+  })
+
+  it('returns 500 when axios throws', async () => {
+    hoisted.axiosGet.mockRejectedValueOnce(new Error('boom'))
+    const res = createMockRes()
+    await handler(
+      createMockReq({ method: 'GET', query: { course_name: 'CS101' } }) as any,
+      res as any,
+    )
+    expect(res.status).toHaveBeenCalledWith(500)
+    expect(res.json).toHaveBeenCalledWith(
+      expect.objectContaining({
+        message: 'Error exporting conversation history.',
+      }),
+    )
+  })
 })

__tests__/pages/api/UIUC-api/__tests__/downloadConvoHistoryUser.test.ts

@@ -75,6 +75,19 @@ describe('UIUC-api/downloadConvoHistoryUser', () => {
     expect(res4.status).toHaveBeenCalledWith(400)
   })
 
+  it('returns 400 when projectName is not a string', async () => {
+    const res = createWritableRes()
+    await handler(
+      createMockReq({
+        method: 'GET',
+        query: { projectName: ['CS101'] },
+        user: { email: 'a@b.com' },
+      }) as any,
+      res as any,
+    )
+    expect(res.status).toHaveBeenCalledWith(400)
+  })
+
   it('streams backend response to res on success', async () => {
     const fetchSpy = vi
       .spyOn(globalThis, 'fetch')
@@ -94,4 +107,115 @@ describe('UIUC-api/downloadConvoHistoryUser', () => {
     expect(res.end).toHaveBeenCalled()
     fetchSpy.mockRestore()
   })
+
+  it('propagates backend non-ok status codes', async () => {
+    const fetchSpy = vi.spyOn(globalThis, 'fetch').mockResolvedValueOnce(
+      new Response('nope', {
+        status: 502,
+        statusText: 'Bad Gateway',
+        headers: { 'content-type': 'application/json' },
+      }),
+    )
+
+    const res = createWritableRes()
+    await handler(
+      createMockReq({
+        method: 'GET',
+        query: { projectName: 'CS101' },
+        user: { email: 'a@b.com' },
+      }) as any,
+      res as any,
+    )
+
+    expect(res.status).toHaveBeenCalledWith(502)
+    expect(res.json).toHaveBeenCalledWith(
+      expect.objectContaining({
+        error: expect.stringContaining('Backend error'),
+      }),
+    )
+    fetchSpy.mockRestore()
+  })
+
+  it('returns 500 when backend response has no body', async () => {
+    const fetchSpy = vi.spyOn(globalThis, 'fetch').mockResolvedValueOnce(
+      new Response(null, {
+        status: 200,
+        headers: { 'content-type': 'application/zip' },
+      }),
+    )
+
+    const res = createWritableRes()
+    await handler(
+      createMockReq({
+        method: 'GET',
+        query: { projectName: 'CS101' },
+        user: { email: 'a@b.com' },
+      }) as any,
+      res as any,
+    )
+    expect(res.status).toHaveBeenCalledWith(500)
+    expect(res.json).toHaveBeenCalledWith(
+      expect.objectContaining({ error: 'No response body received' }),
+    )
+    fetchSpy.mockRestore()
+  })
+
+  it('returns 504 on TimeoutError and 408 on AbortError', async () => {
+    const timeoutErr = Object.assign(new Error('took too long'), {
+      name: 'TimeoutError',
+    })
+    const abortErr = Object.assign(new Error('aborted'), { name: 'AbortError' })
+
+    const fetchSpy = vi
+      .spyOn(globalThis, 'fetch')
+      .mockRejectedValueOnce(timeoutErr)
+      .mockRejectedValueOnce(abortErr)
+
+    const res1 = createWritableRes()
+    await handler(
+      createMockReq({
+        method: 'GET',
+        query: { projectName: 'CS101' },
+        user: { email: 'a@b.com' },
+      }) as any,
+      res1 as any,
+    )
+    expect(res1.status).toHaveBeenCalledWith(504)
+
+    const res2 = createWritableRes()
+    await handler(
+      createMockReq({
+        method: 'GET',
+        query: { projectName: 'CS101' },
+        user: { email: 'a@b.com' },
+      }) as any,
+      res2 as any,
+    )
+    expect(res2.status).toHaveBeenCalledWith(408)
+
+    fetchSpy.mockRestore()
+  })
+
+  it('returns 500 on generic errors', async () => {
+    const fetchSpy = vi
+      .spyOn(globalThis, 'fetch')
+      .mockRejectedValueOnce(new Error('boom'))
+
+    const res = createWritableRes()
+    await handler(
+      createMockReq({
+        method: 'GET',
+        query: { projectName: 'CS101' },
+        user: { email: 'a@b.com' },
+      }) as any,
+      res as any,
+    )
+    expect(res.status).toHaveBeenCalledWith(500)
+    expect(res.json).toHaveBeenCalledWith(
+      expect.objectContaining({
+        error: 'Internal server error while exporting documents',
+      }),
+    )
+    fetchSpy.mockRestore()
+  })
 })