Fix Database and IndexController tests

Author: samuel27m

tests/controllers/IndexController.spec.ts

@@ -1,15 +1,20 @@
 import { IndexController } from '../../src/controllers/IndexController';
-import { Request, Response } from 'express';
+import { NextFunction, Request, Response } from 'express';
 
 describe('IndexController', () => {
     it('should respond as expected', () => {
         const request = {};
         const response = {
             json: jest.fn(),
         };
+        const next: NextFunction = jest.fn();
 
-        IndexController(request as Request, response as unknown as Response);
+        IndexController(request as Request, response as unknown as Response, next);
 
-        expect(response.json).toBeCalledWith(['ok']);
+        expect(response.json).toBeCalledWith(
+            expect.objectContaining({
+                status: 200,
+            }),
+        );
     });
 });

tests/database/index.spec.ts

@@ -1,11 +1,34 @@
 import { AppDataSource, initialiseDataSource, destroyDataSource } from '../../src/database';
 
 describe('DataSource Utility', () => {
+    let initCount: number = 0;
     const originalConsoleError = console.error;
 
     beforeEach(async () => {
         console.error = jest.fn();
 
+        jest.spyOn(AppDataSource, 'initialize').mockImplementation(async () => {
+            initCount++;
+            if (initCount > 1) {
+                throw new Error('initialize called more than once before destroying.');
+            }
+            Object.defineProperty(AppDataSource, 'isInitialized', {
+                value: true,
+                writable: true,
+                configurable: true,
+            });
+            return AppDataSource;
+        });
+
+        jest.spyOn(AppDataSource, 'destroy').mockImplementation(async () => {
+            initCount = 0;
+            Object.defineProperty(AppDataSource, 'isInitialized', {
+                value: false,
+                writable: true,
+                configurable: true,
+            });
+        });
+
         if (!AppDataSource.isInitialized) {
             await initialiseDataSource();
         }