mirror of
https://github.com/kikootwo/ReadMeABook.git
synced 2026-06-03 12:50:09 +00:00
Add tests for security hardening: deleted user auth rejection, rate limiting
This commit is contained in:
@@ -47,6 +47,7 @@ export const createPrismaMock = () => ({
|
|||||||
bookDateSwipe: createModelMock(),
|
bookDateSwipe: createModelMock(),
|
||||||
goodreadsShelf: createModelMock(),
|
goodreadsShelf: createModelMock(),
|
||||||
goodreadsBookMapping: createModelMock(),
|
goodreadsBookMapping: createModelMock(),
|
||||||
|
apiToken: createModelMock(),
|
||||||
$queryRaw: vi.fn(),
|
$queryRaw: vi.fn(),
|
||||||
$disconnect: vi.fn(),
|
$disconnect: vi.fn(),
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -3,9 +3,11 @@
|
|||||||
* Documentation: documentation/backend/services/auth.md
|
* Documentation: documentation/backend/services/auth.md
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
/* eslint-disable @typescript-eslint/no-explicit-any */
|
||||||
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
||||||
import { NextResponse } from 'next/server';
|
import { NextResponse } from 'next/server';
|
||||||
import { createPrismaMock } from '../helpers/prisma';
|
import { createPrismaMock } from '../helpers/prisma';
|
||||||
|
import crypto from 'crypto';
|
||||||
|
|
||||||
const prismaMock = createPrismaMock();
|
const prismaMock = createPrismaMock();
|
||||||
const verifyAccessTokenMock = vi.fn();
|
const verifyAccessTokenMock = vi.fn();
|
||||||
@@ -29,6 +31,11 @@ const makeRequest = (authHeader?: string) => ({
|
|||||||
},
|
},
|
||||||
});
|
});
|
||||||
|
|
||||||
|
// Helper to create a valid API token hash for testing
|
||||||
|
const createTestApiToken = (token: string) => {
|
||||||
|
return crypto.createHash('sha256').update(token).digest('hex');
|
||||||
|
};
|
||||||
|
|
||||||
describe('auth middleware', () => {
|
describe('auth middleware', () => {
|
||||||
beforeEach(() => {
|
beforeEach(() => {
|
||||||
vi.clearAllMocks();
|
vi.clearAllMocks();
|
||||||
@@ -159,6 +166,98 @@ describe('auth middleware', () => {
|
|||||||
expect(result).toBe(true);
|
expect(result).toBe(true);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
it('rejects JWT tokens for soft-deleted users', async () => {
|
||||||
|
verifyAccessTokenMock.mockReturnValue({
|
||||||
|
sub: 'user-1',
|
||||||
|
plexId: 'plex-1',
|
||||||
|
username: 'user',
|
||||||
|
role: 'user',
|
||||||
|
iat: 1,
|
||||||
|
exp: 2,
|
||||||
|
});
|
||||||
|
prismaMock.user.findUnique.mockResolvedValue({
|
||||||
|
id: 'user-1',
|
||||||
|
deletedAt: new Date(),
|
||||||
|
});
|
||||||
|
const { requireAuth } = await import('@/lib/middleware/auth');
|
||||||
|
|
||||||
|
const response = await requireAuth(makeRequest('Bearer token') as any, vi.fn());
|
||||||
|
const payload = await response.json();
|
||||||
|
|
||||||
|
expect(response.status).toBe(401);
|
||||||
|
expect(payload.message).toMatch(/user not found/i);
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('API token authentication', () => {
|
||||||
|
const testToken = 'rmab_test1234567890abcdef';
|
||||||
|
const testTokenHash = createTestApiToken(testToken);
|
||||||
|
|
||||||
|
it('rejects API tokens for soft-deleted users', async () => {
|
||||||
|
prismaMock.apiToken.findUnique.mockResolvedValue({
|
||||||
|
id: 'token-1',
|
||||||
|
tokenHash: testTokenHash,
|
||||||
|
role: 'user',
|
||||||
|
expiresAt: null,
|
||||||
|
tokenUser: {
|
||||||
|
id: 'user-1',
|
||||||
|
plexUsername: 'deleteduser',
|
||||||
|
role: 'user',
|
||||||
|
deletedAt: new Date(),
|
||||||
|
},
|
||||||
|
});
|
||||||
|
const { requireAuth } = await import('@/lib/middleware/auth');
|
||||||
|
|
||||||
|
const response = await requireAuth(makeRequest(`Bearer ${testToken}`) as any, vi.fn());
|
||||||
|
const payload = await response.json();
|
||||||
|
|
||||||
|
expect(response.status).toBe(401);
|
||||||
|
expect(payload.message).toMatch(/invalid.*expired/i);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('rejects API tokens for missing users', async () => {
|
||||||
|
prismaMock.apiToken.findUnique.mockResolvedValue({
|
||||||
|
id: 'token-1',
|
||||||
|
tokenHash: testTokenHash,
|
||||||
|
role: 'user',
|
||||||
|
expiresAt: null,
|
||||||
|
tokenUser: null,
|
||||||
|
});
|
||||||
|
const { requireAuth } = await import('@/lib/middleware/auth');
|
||||||
|
|
||||||
|
const response = await requireAuth(makeRequest(`Bearer ${testToken}`) as any, vi.fn());
|
||||||
|
const payload = await response.json();
|
||||||
|
|
||||||
|
expect(response.status).toBe(401);
|
||||||
|
expect(payload.message).toMatch(/invalid.*expired/i);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('accepts valid API tokens for active users', async () => {
|
||||||
|
prismaMock.apiToken.findUnique.mockResolvedValue({
|
||||||
|
id: 'token-1',
|
||||||
|
tokenHash: testTokenHash,
|
||||||
|
role: 'user',
|
||||||
|
expiresAt: null,
|
||||||
|
tokenUser: {
|
||||||
|
id: 'user-1',
|
||||||
|
plexUsername: 'activeuser',
|
||||||
|
role: 'user',
|
||||||
|
deletedAt: null,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
prismaMock.apiToken.update.mockResolvedValue({});
|
||||||
|
const { requireAuth } = await import('@/lib/middleware/auth');
|
||||||
|
|
||||||
|
const handler = vi.fn(async (req: any) =>
|
||||||
|
NextResponse.json({ ok: true, userId: req.user?.id })
|
||||||
|
);
|
||||||
|
const response = await requireAuth(makeRequest(`Bearer ${testToken}`) as any, handler);
|
||||||
|
const payload = await response.json();
|
||||||
|
|
||||||
|
expect(handler).toHaveBeenCalled();
|
||||||
|
expect(payload.userId).toBe('user-1');
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
it('returns current user from token', async () => {
|
it('returns current user from token', async () => {
|
||||||
verifyAccessTokenMock.mockReturnValue({
|
verifyAccessTokenMock.mockReturnValue({
|
||||||
sub: 'user-1',
|
sub: 'user-1',
|
||||||
|
|||||||
@@ -0,0 +1,122 @@
|
|||||||
|
/**
|
||||||
|
* Component: API Token Rate Limit Tests
|
||||||
|
* Documentation: documentation/backend/services/auth.md
|
||||||
|
*/
|
||||||
|
|
||||||
|
import { beforeEach, describe, expect, it, vi } from 'vitest';
|
||||||
|
import {
|
||||||
|
checkApiTokenCreateRateLimit,
|
||||||
|
checkApiTokenRevokeRateLimit,
|
||||||
|
} from '@/lib/utils/apiTokenRateLimit';
|
||||||
|
|
||||||
|
describe('API Token Rate Limiting', () => {
|
||||||
|
beforeEach(() => {
|
||||||
|
vi.useFakeTimers();
|
||||||
|
});
|
||||||
|
|
||||||
|
afterEach(() => {
|
||||||
|
vi.useRealTimers();
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('checkApiTokenCreateRateLimit', () => {
|
||||||
|
it('allows requests under the limit', () => {
|
||||||
|
const actorId = 'user-create-1';
|
||||||
|
|
||||||
|
for (let i = 0; i < 10; i++) {
|
||||||
|
const result = checkApiTokenCreateRateLimit(actorId);
|
||||||
|
expect(result.allowed).toBe(true);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
it('blocks requests over the limit (10/min)', () => {
|
||||||
|
const actorId = 'user-create-2';
|
||||||
|
|
||||||
|
// Use up the limit
|
||||||
|
for (let i = 0; i < 10; i++) {
|
||||||
|
checkApiTokenCreateRateLimit(actorId);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 11th request should be blocked
|
||||||
|
const result = checkApiTokenCreateRateLimit(actorId);
|
||||||
|
expect(result.allowed).toBe(false);
|
||||||
|
expect(result.retryAfterSeconds).toBeGreaterThan(0);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('resets after the window expires', () => {
|
||||||
|
const actorId = 'user-create-3';
|
||||||
|
|
||||||
|
// Use up the limit
|
||||||
|
for (let i = 0; i < 10; i++) {
|
||||||
|
checkApiTokenCreateRateLimit(actorId);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Should be blocked
|
||||||
|
expect(checkApiTokenCreateRateLimit(actorId).allowed).toBe(false);
|
||||||
|
|
||||||
|
// Advance time past the window (60 seconds)
|
||||||
|
vi.advanceTimersByTime(61 * 1000);
|
||||||
|
|
||||||
|
// Should be allowed again
|
||||||
|
expect(checkApiTokenCreateRateLimit(actorId).allowed).toBe(true);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('tracks different actors separately', () => {
|
||||||
|
const actor1 = 'user-create-4';
|
||||||
|
const actor2 = 'user-create-5';
|
||||||
|
|
||||||
|
// Use up actor1's limit
|
||||||
|
for (let i = 0; i < 10; i++) {
|
||||||
|
checkApiTokenCreateRateLimit(actor1);
|
||||||
|
}
|
||||||
|
|
||||||
|
// actor1 should be blocked
|
||||||
|
expect(checkApiTokenCreateRateLimit(actor1).allowed).toBe(false);
|
||||||
|
|
||||||
|
// actor2 should still be allowed
|
||||||
|
expect(checkApiTokenCreateRateLimit(actor2).allowed).toBe(true);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
describe('checkApiTokenRevokeRateLimit', () => {
|
||||||
|
it('allows requests under the limit', () => {
|
||||||
|
const actorId = 'user-revoke-1';
|
||||||
|
|
||||||
|
for (let i = 0; i < 20; i++) {
|
||||||
|
const result = checkApiTokenRevokeRateLimit(actorId);
|
||||||
|
expect(result.allowed).toBe(true);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
it('blocks requests over the limit (20/min)', () => {
|
||||||
|
const actorId = 'user-revoke-2';
|
||||||
|
|
||||||
|
// Use up the limit
|
||||||
|
for (let i = 0; i < 20; i++) {
|
||||||
|
checkApiTokenRevokeRateLimit(actorId);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 21st request should be blocked
|
||||||
|
const result = checkApiTokenRevokeRateLimit(actorId);
|
||||||
|
expect(result.allowed).toBe(false);
|
||||||
|
expect(result.retryAfterSeconds).toBeGreaterThan(0);
|
||||||
|
});
|
||||||
|
|
||||||
|
it('returns correct retryAfterSeconds', () => {
|
||||||
|
const actorId = 'user-revoke-3';
|
||||||
|
|
||||||
|
// Use up the limit
|
||||||
|
for (let i = 0; i < 20; i++) {
|
||||||
|
checkApiTokenRevokeRateLimit(actorId);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Advance 30 seconds into the window
|
||||||
|
vi.advanceTimersByTime(30 * 1000);
|
||||||
|
|
||||||
|
const result = checkApiTokenRevokeRateLimit(actorId);
|
||||||
|
expect(result.allowed).toBe(false);
|
||||||
|
// Should have ~30 seconds left
|
||||||
|
expect(result.retryAfterSeconds).toBeLessThanOrEqual(30);
|
||||||
|
expect(result.retryAfterSeconds).toBeGreaterThan(0);
|
||||||
|
});
|
||||||
|
});
|
||||||
|
});
|
||||||
Reference in New Issue
Block a user