finished testing route ai/delete

__tests__/ai/__snapshots__/delete.test.js.snap

+// Vitest Snapshot v1, https://vitest.dev/guide/snapshot.html
+
+exports[`ai delete model > given a user tries to access > should respond with a proper body 1`] = `
+{
+  "message": "Access Forbidden",
+}
+`;
+
+exports[`ai delete model > given no jwt sended > should respond with a proper body 1`] = `
+{
+  "message": "No access token found. Access denied.",
+}
+`;
+
+exports[`ai delete model > given no matching model found > should respond with a proper body 1`] = `
+{
+  "message": "pull model manifest: file does not exist",
+}
+`;
+
+exports[`ai delete model > given no valid jwt sended > should respond with a proper body 1`] = `
+{
+  "message": "Access token is no longer valid. Access denied.",
+}
+`;
+
+exports[`ai delete model > given required fields are missing > should respond with a proper body 1`] = `
+{
+  "message": "Validation errors. Please check the error messages.",
+  "validationErrors": {
+    "model": "Required",
+  },
+}
+`;
+
+exports[`ai delete model > given the inputs are valid > should respond with a proper body 1`] = `
+{
+  "status": "success",
+}
+`;

logs/__tests__/ai/delete.test.js → __tests__/ai/delete.test.js

@@ -2,10 +2,8 @@
 import { vi, beforeAll, beforeEach, describe, expect, expectTypeOf, test, it, afterEach } from 'vitest';
 import supertest from "supertest";
 import app from "../../app.js";
-// ignore expiration of the (self-signed) certificate
-process.env.NODE_TLS_REJECT_UNAUTHORIZED = 0;
-// set timeout
-const BEFORE_ALL_TIMEOUT = 30000; // 30 sec
+import jwt from 'jsonwebtoken';
+
 // set route
 const ROUTE = '/ai/models';
 // prepare response of each test
@@ -14,27 +12,37 @@ let response;
 // ############################
 //  OBJECTS
 // ############################
-
-const noModelFoundResponse = {
-  code: 500,
-  message: "no such file or directory",
-  data: {}
+const mockedVals = vi.hoisted(() => {
+  return {
+    foundUser: {
+      _id: '66a29da2942b3eb',
+      username: 'snoopy',
+      name: 'My User',
+      email: 'user@mail.local',
+      verified: true,
+      role: 4,
+      createdAt: '2024-07 - 25T18: 46: 58.982Z',
+      updatedAt: '2024-07 - 25T18: 46: 58.982Z',
+      __v: 0,
+      fullname: '',
+      id: '66a29da2942b3ebcaf047f07'
+    },
+    validInput: {
+      model: 'llama3'
+    }
   };
+});
+
 // ############################
 //  MOCKS
 // ############################
-// import PocketBase Service
-import * as pbService from '../../utils/pocketbase/handlePocketBase.js';
-// mock pbService
-vi.mock('../../utils/pocketbase/handlePocketBase.js', async (importOriginal) => {
+import * as dbService from '../../utils/handleDB.js';
+// mock dbService
+vi.mock('../../utils/handleDB.js', async (importOriginal) => {
   return {
     ...await importOriginal(),
-    pbVerifyAccessToken: vi.fn().mockImplementation((req, res, next) => {
-      next();
-    }),
-    gateKeeper: vi.fn().mockImplementation((req, res, next) => {
-      return next();
-    })
+    dbConnection: vi.fn(() => 'mocked'),
+    findOneRecord: vi.fn(() => mockedVals.foundUser),
   };
 });
 
@@ -53,103 +61,127 @@ vi.mock('../../utils/handleAI.js', async (importOriginal) => {
 // ############################
 
 describe('ai delete model', () => {
+  const _jwt = (id, role) => {
+    return jwt.sign({ id, role }, process.env.JWT_SECRET_KEY, { expiresIn: process.env.JWT_TTL });
+  };
+
   describe('given the inputs are valid', async () => {
+    beforeAll(async () => {
       response = await supertest(app)
         .delete(ROUTE)
-      .set('Authorization', 'Bearer 123valid')
-      .send({ model: 'validModelName' });
-    it('should have called the gateKeeper', () => {
-      expect(pbService.gateKeeper).toHaveBeenCalledTimes(1);
+        .set('Authorization', `Bearer ${_jwt(mockedVals.foundUser.id, mockedVals.foundUser.role)}`)
+        .send(mockedVals.validInput);
     });
-    it('should return a proper status code', () => {
+
+    it('should return a proper status code status', () => {
       expect(response.status).toBe(200);
     });
-    it('should respond with a proper message', () => {
-      expect(response.body).toEqual({ status: 'success' });
+    it('should respond with a proper body', () => {
+      expect(response.body).toMatchSnapshot();
     });
   });
 
   // ############################
 
-  describe('given no valid JWT sent', () => {
+  describe('given no matching model found', () => {
     beforeAll(async () => {
-      pbService.pbVerifyAccessToken.mockImplementationOnce((req, res, next) => {
-        return res.status(403).json({ message: 'You are not logged in.' });
-      });
+      const input = { ...mockedVals.validInput, model: 'unknownModel' };
+
+      let error = new Error("pull model manifest: file does not exist");
+      error.name = 'ResponseError';
+      error.status = 500;
+      aiService.aiDeleteModel.mockImplementation(() => { throw error; });
 
       response = await supertest(app)
         .delete(ROUTE)
-        .send({ model: 'validModelName' });
-    }, BEFORE_ALL_TIMEOUT);
-    it('should return a proper status code', () => {
-      expect(response.status).toBe(403);
+        .set('Authorization', `Bearer ${_jwt(mockedVals.foundUser.id, mockedVals.foundUser.role)}`)
+        .send(input);
     });
-    it('should respond with a proper message', () => {
-      expect(response.body.message).toEqual('You are not logged in.');
+
+    it('should return a proper status code status', () => {
+      expect(response.status).toBe(500);
+    });
+    it('should respond with a proper body', () => {
+      expect(response.body).toMatchSnapshot();
     });
   });
 
 
   // ############################
 
-  describe('given no access granted', () => {
+  describe('given required fields are missing', () => {
     beforeAll(async () => {
-      pbService.gateKeeper.mockImplementationOnce((req, res, next) => {
-        return res.status(403).json({ message: 'Access Forbidden' });
+      const { model, ...input } = mockedVals.validInput;
+
+      response = await supertest(app)
+        .delete(ROUTE)
+        .set('Authorization', `Bearer ${_jwt(mockedVals.foundUser.id, mockedVals.foundUser.role)}`)
+        .send(input);
+    });
+
+    it('should return a proper status code status', () => {
+      expect(response.status).toBe(400);
+    });
+    it('should respond with a proper body', () => {
+      expect(response.body).toMatchSnapshot();
+    });
+  });
+
+  // ############################
+
+  describe('given a user tries to access', () => {
+    beforeAll(async () => {
+
+      dbService.findOneRecord.mockImplementationOnce(async () => {
+        return { ...mockedVals.foundUser, role: 0 };
       });
 
       response = await supertest(app)
         .delete(ROUTE)
-        .set('Authorization', 'Bearer 123valid');
-    }, BEFORE_ALL_TIMEOUT);
+        .set('Authorization', `Bearer ${_jwt(mockedVals.foundUser.id, 0)}`)
+        .send(mockedVals.validInput);
+    });
 
-    it('should return a proper status code', () => {
+    it('should return a proper status code status', () => {
       expect(response.status).toBe(403);
     });
-    it('should respond with a proper message', () => {
-      expect(response.body.message).toEqual('Access Forbidden');
+    it('should respond with a proper body', () => {
+      expect(response.body).toMatchSnapshot();
     });
   });
 
   // ############################
 
-  describe('given no model name sent', () => {
+  describe('given no valid jwt sended', () => {
     beforeAll(async () => {
       response = await supertest(app)
         .delete(ROUTE)
-        .set('Authorization', 'Bearer 123valid');
-    }, BEFORE_ALL_TIMEOUT);
-    it('should return a proper status code', () => {
-      expect(response.status).toBe(400);
+        .set('Authorization', `Bearer invalid`)
+        .send(mockedVals.validInput);
+    });
+
+    it('should return a proper status code status', () => {
+      expect(response.status).toBe(403);
     });
-    it('should respond with a proper message', () => {
-      expect(response.body.validationErrors.model).toEqual('Required');
+    it('should respond with a proper body', () => {
+      expect(response.body).toMatchSnapshot();
     });
   });
 
   // ############################
 
-  describe('given no matching model found', () => {
+  describe('given no jwt sended', () => {
     beforeAll(async () => {
-      let error = new Error('no such file or directory');
-      error.name = 'ResponseError';
-      error.response = noModelFoundResponse;
-      error.status = 500;
-      aiService.aiDeleteModel.mockImplementation(() => { throw error; });
-
       response = await supertest(app)
         .delete(ROUTE)
-        .set('Authorization', 'Bearer 123valid')
-        .send({ model: 'invalidModelName' });
-    }, BEFORE_ALL_TIMEOUT);
-    it('should force aiDeleteModel to throw an error', () => {
-      expect(aiService.aiDeleteModel).toThrowError();
+        .send(mockedVals.validInput);
     });
-    it('should return a proper status code', () => {
-      expect(response.status).toBe(500);
+
+    it('should return a proper status code status', () => {
+      expect(response.status).toBe(401);
     });
-    it('should respond with a proper message', () => {
-      expect(response.body.message).toEqual('no such file or directory');
+    it('should respond with a proper body', () => {
+      expect(response.body).toMatchSnapshot();
     });
   });
 

__tests__/manualREST/ollama.rest

@@ -48,7 +48,7 @@ Accept: application/json
 Content-Type: application/json
 
 {
-  "filter": "bai"
+  "filter": ""
 }
 
 ### show info of a specific model

controllers/AI.js

@@ -29,10 +29,6 @@ export const getStatus = async (req, res, next) => {
 export const getModels = async (req, res, next) => {
   try {
     const foundModels = await aiFilterModelsByName(req.body.filter);
-
-    console.log("🚀 ~ getModels ~ foundModels:", foundModels);
-
-
     return res.json(foundModels);
   } catch (error) {
     next(error);