test: update server tests

versatiles-org · Mar 7, 2024 · a9ec83a · a9ec83a
1 parent 91abdec
commit a9ec83a
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/src/lib/__mocks__/global_fetch.ts b/src/lib/__mocks__/global_fetch.ts
@@ -4,6 +4,8 @@ export function mockFetchResponse(data: unknown): void {
 	// @ts-expect-error too lazy
 	global.fetch = jest.fn(async () => Promise.resolve({
 		arrayBuffer: async () => Promise.resolve(getAsBuffer()),
+		// eslint-disable-next-line @typescript-eslint/naming-convention
+		headers: new Headers({ 'content-type': 'text/plain' }),
 		json: async () => Promise.resolve(getAsJSON()),
 		status: 200,
 	}));

diff --git a/src/server/server.test.ts b/src/server/server.test.ts
@@ -24,12 +24,15 @@ describe('Server', () => {
 	let getFunction: (req: IncomingMessage, res: ServerResponse) => void;
 
 	async function fakeRequest(path: string): Promise<{
+		header: jest.Mock<(key: string, value: string) => void>;
 		status: jest.Mock<(status: number) => void>;
 		end: jest.Mock<(data: Buffer | string) => void>;
 	}> {
 		return new Promise(resolve => {
 			const response = {
-				status: jest.fn().mockReturnThis(), end: jest.fn(() => {
+				status: jest.fn().mockReturnThis(),
+				header: jest.fn().mockReturnThis(),
+				end: jest.fn(() => {
 					resolve(response);
 				}),
 			};
@@ -63,22 +66,40 @@ describe('Server', () => {
 	it('should serve files from the file system', async () => {
 		const res = await fakeRequest('/existingFile.txt');
 
+		expect(res.header).toHaveBeenCalledTimes(1);
+		expect(res.header).toHaveBeenCalledWith('content-type', 'text/plain');
+
+		expect(res.status).toHaveBeenCalledTimes(1);
 		expect(res.status).toHaveBeenCalledWith(200);
+
+		expect(res.end).toHaveBeenCalledTimes(1);
 		expect(res.end).toHaveBeenCalledWith(Buffer.from('file content'));
 	});
 
 	it('should return 404 for non-existing files', async () => {
 		const res = await fakeRequest('/nonExistingFile.txt');
 
+		expect(res.header).toHaveBeenCalledTimes(0);
+
+		expect(res.status).toHaveBeenCalledTimes(1);
 		expect(res.status).toHaveBeenCalledWith(404);
+
+		expect(res.end).toHaveBeenCalledTimes(1);
 		expect(res.end).toHaveBeenCalledWith(expect.stringContaining('not found'));
 	});
 
 	it('should proxy requests based on configuration', async () => {
 		const res = await fakeRequest('/api/data');
 
 		expect(fetch).toHaveBeenCalledWith('http://example.com/api/data');
+
+		expect(res.header).toHaveBeenCalledTimes(1);
+		expect(res.header).toHaveBeenCalledWith('content-type', 'text/plain');
+
+		expect(res.status).toHaveBeenCalledTimes(1);
 		expect(res.status).toHaveBeenCalledWith(200);
+
+		expect(res.end).toHaveBeenCalledTimes(1);
 		expect(res.end).toHaveBeenCalledWith(Buffer.from('response'));
 	});
 });