Move usermethods and models to data-schema

api/strategies/appleStrategy.test.js

@@ -3,11 +3,10 @@ const { MongoMemoryServer } = require('mongodb-memory-server');
 const jwt = require('jsonwebtoken');
 const { Strategy: AppleStrategy } = require('passport-apple');
 const socialLogin = require('./socialLogin');
-const User = require('~/models/User');
 const { logger } = require('~/config');
 const { createSocialUser, handleExistingUser } = require('./process');
 const { isEnabled } = require('~/server/utils');
-const { findUser } = require('~/models');
+const db = require('~/lib/db/connectDb');
 
 // Mocking external dependencies
 jest.mock('jsonwebtoken');
@@ -24,21 +23,20 @@ jest.mock('./process', () => ({
 jest.mock('~/server/utils', () => ({
   isEnabled: jest.fn(),
 }));
-jest.mock('~/models', () => ({
-  findUser: jest.fn(),
-}));
 
 describe('Apple Login Strategy', () => {
   let mongoServer;
   let appleStrategyInstance;
   const OLD_ENV = process.env;
   let getProfileDetails;
-
+  let User;
   // Start and stop in-memory MongoDB
   beforeAll(async () => {
     mongoServer = await MongoMemoryServer.create();
     const mongoUri = mongoServer.getUri();
-    await mongoose.connect(mongoUri);
+    await db.connectDb(mongoUri);
+
+    User = db.models.User;
   });
 
   afterAll(async () => {
@@ -64,7 +62,6 @@ describe('Apple Login Strategy', () => {
 
     // Define getProfileDetails within the test scope
     getProfileDetails = ({ idToken, profile }) => {
-      console.log('getProfileDetails called with idToken:', idToken);
       if (!idToken) {
         logger.error('idToken is missing');
         throw new Error('idToken is missing');
@@ -84,9 +81,7 @@ describe('Apple Login Strategy', () => {
         email: decoded.email,
         id: decoded.sub,
         avatarUrl: null, // Apple does not provide an avatar URL
-        username: decoded.email
-          ? decoded.email.split('@')[0].toLowerCase()
-          : `user_${decoded.sub}`,
+        username: decoded.email ? decoded.email.split('@')[0].toLowerCase() : `user_${decoded.sub}`,
         name: decoded.name
           ? `${decoded.name.firstName} ${decoded.name.lastName}`
           : profile.displayName || null,
@@ -96,8 +91,12 @@ describe('Apple Login Strategy', () => {
 
     // Mock isEnabled based on environment variable
     isEnabled.mockImplementation((flag) => {
-      if (flag === 'true') { return true; }
-      if (flag === 'false') { return false; }
+      if (flag === 'true') {
+        return true;
+      }
+      if (flag === 'false') {
+        return false;
+      }
       return false;
     });
 
@@ -154,9 +153,7 @@ describe('Apple Login Strategy', () => {
       });
 
       expect(jwt.decode).toHaveBeenCalledWith('fake_id_token');
-      expect(logger.debug).toHaveBeenCalledWith(
-        expect.stringContaining('Decoded Apple JWT'),
-      );
+      expect(logger.debug).toHaveBeenCalledWith(expect.stringContaining('Decoded Apple JWT'));
       expect(profileDetails).toEqual({
         email: 'john.doe@example.com',
         id: 'apple-sub-1234',
@@ -209,12 +206,13 @@ describe('Apple Login Strategy', () => {
 
     beforeEach(() => {
       jwt.decode.mockReturnValue(decodedToken);
-      findUser.mockImplementation(({ email }) => User.findOne({ email }));
+      User.findUser = jest.fn();
+      User.findUser.mockImplementation(({ email }) => User.findOne({ email }));
     });
 
     it('should create a new user if one does not exist and registration is allowed', async () => {
       // Mock findUser to return null (user does not exist)
-      findUser.mockResolvedValue(null);
+      User.findUser.mockResolvedValue(null);
 
       // Mock createSocialUser to create a user
       createSocialUser.mockImplementation(async (userData) => {
@@ -260,7 +258,7 @@ describe('Apple Login Strategy', () => {
       await existingUser.save();
 
       // Mock findUser to return the existing user
-      findUser.mockResolvedValue(existingUser);
+      User.findUser.mockResolvedValue(existingUser);
 
       // Mock handleExistingUser to update avatarUrl
       handleExistingUser.mockImplementation(async (user, avatarUrl) => {
@@ -297,7 +295,7 @@ describe('Apple Login Strategy', () => {
         appleStrategyInstance._verify(
           fakeAccessToken,
           fakeRefreshToken,
-          null,               // idToken is missing
+          null, // idToken is missing
           mockProfile,
           (err, user) => {
             mockVerifyCallback(err, user);
@@ -344,7 +342,7 @@ describe('Apple Login Strategy', () => {
 
     it('should handle errors during user creation', async () => {
       // Mock findUser to return null (user does not exist)
-      findUser.mockResolvedValue(null);
+      User.findUser.mockResolvedValue(null);
 
       // Mock createSocialUser to throw an error
       createSocialUser.mockImplementation(() => {