Mirrors/Stirling-PDF
Mirrors/Stirling-PDF
1
0
Fork 0
mirror of https://github.com/Frooodle/Stirling-PDF.git synced 2025-08-11 13:48:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix tests

Browse Source
This commit is contained in:
James 2025-08-06 17:00:48 +01:00
parent addaf6f751
commit 201f6cb6ab
1 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
frontend/src/hooks/tools/sanitize/useSanitizeOperation.test.ts
View File

@ -20,6 +20,16 @@ vi.mock('react-i18next', () => ({
})
}));
// Mock FileContext
vi.mock('../../../contexts/FileContext', () => ({
useFileContext: () => ({
recordOperation: vi.fn(),
markOperationApplied: vi.fn(),
markOperationFailed: vi.fn(),
addFiles: vi.fn()
})
}));
// Mock fetch
const mockFetch = vi.fn();
globalThis.fetch = mockFetch;
@ -29,6 +39,11 @@ globalThis.URL.createObjectURL = vi.fn(() => 'mock-blob-url');
globalThis.URL.revokeObjectURL = vi.fn();
describe('useSanitizeOperation', () => {
const mockGenerateSanitizedFileName = (originalFileName?: string): string => {
const baseName = originalFileName?.replace(/\.[^/.]+$/, '') || 'document';
return `sanitized_${baseName}.pdf`;
};
beforeEach(() => {
vi.clearAllMocks();
});
@ -68,7 +83,7 @@ describe('useSanitizeOperation', () => {
const testFile = new File(['test'], 'test.pdf', { type: 'application/pdf' });
await act(async () => {
await result.current.executeOperation(parameters, [testFile]);
await result.current.executeOperation(parameters, [testFile], mockGenerateSanitizedFileName);
});
expect(mockFetch).toHaveBeenCalledWith('/api/v1/security/sanitize-pdf', {
@ -104,14 +119,14 @@ describe('useSanitizeOperation', () => {
await act(async () => {
try {
await result.current.executeOperation(parameters, [testFile]);
await result.current.executeOperation(parameters, [testFile], mockGenerateSanitizedFileName);
} catch (error) {
// Expected to throw
}
});
expect(result.current.isLoading).toBe(false);
expect(result.current.errorMessage).toBe('Sanitization failed: Server error');
expect(result.current.errorMessage).toBe('Failed to sanitize all files: test.pdf');
expect(result.current.downloadUrl).toBe(null);
expect(result.current.status).toBe(null);
});
@ -131,7 +146,7 @@ describe('useSanitizeOperation', () => {
let thrownError: Error | null = null;
await act(async () => {
try {
await result.current.executeOperation(parameters, []);
await result.current.executeOperation(parameters, [], mockGenerateSanitizedFileName);
} catch (error) {
thrownError = error as Error;
}
@ -165,7 +180,7 @@ describe('useSanitizeOperation', () => {
const testFile = new File(['test'], 'test.pdf', { type: 'application/pdf' });
await act(async () => {
await result.current.executeOperation(parameters, [testFile]);
await result.current.executeOperation(parameters, [testFile], mockGenerateSanitizedFileName);
});
const [url, options] = mockFetch.mock.calls[0];
@ -219,13 +234,13 @@ describe('useSanitizeOperation', () => {
// Trigger an API error
await act(async () => {
try {
await result.current.executeOperation(parameters, [testFile]);
await result.current.executeOperation(parameters, [testFile], mockGenerateSanitizedFileName);
} catch (error) {
// Expected to throw
}
});
expect(result.current.errorMessage).toBeTruthy();
expect(result.current.errorMessage).toBe('Failed to sanitize all files: test.pdf');
act(() => {
result.current.clearError();