/* * Copyright © 2024 Hexastack. All rights reserved. * * Licensed under the GNU Affero General Public License v3.0 (AGPLv3) with the following additional terms: * 1. The name "Hexabot" is a trademark of Hexastack. You may not use this name in derivative works without express written permission. * 2. All derivative works must include clear attribution to the original creator and software, Hexastack and Hexabot, in a prominent location (e.g., in the software's "About" section, documentation, and README file). */ import { EventEmitter2 } from '@nestjs/event-emitter'; import { MongooseModule } from '@nestjs/mongoose'; import { Test } from '@nestjs/testing'; import { AttachmentRepository } from '@/attachment/repositories/attachment.repository'; import { AttachmentModel } from '@/attachment/schemas/attachment.schema'; import { AttachmentService } from '@/attachment/services/attachment.service'; import { LoggerService } from '@/logger/logger.service'; import { RoleRepository } from '@/user/repositories/role.repository'; import { UserRepository } from '@/user/repositories/user.repository'; import { PermissionModel } from '@/user/schemas/permission.schema'; import { RoleModel } from '@/user/schemas/role.schema'; import { User, UserModel } from '@/user/schemas/user.schema'; import { RoleService } from '@/user/services/role.service'; import { UserService } from '@/user/services/user.service'; import { installMessageFixtures, messageFixtures, } from '@/utils/test/fixtures/message'; import { getPageQuery } from '@/utils/test/pagination'; import { sortRowsBy } from '@/utils/test/sort'; import { closeInMongodConnection, rootMongooseTestModule, } from '@/utils/test/test'; import { MessageRepository } from '../repositories/message.repository'; import { Message, MessageModel } from '../schemas/message.schema'; import { Subscriber, SubscriberModel } from '../schemas/subscriber.schema'; import { SubscriberRepository } from './../repositories/subscriber.repository'; import { MessageService } from './message.service'; import { SubscriberService } from './subscriber.service'; describe('MessageService', () => { let messageRepository: MessageRepository; let messageService: MessageService; let subscriberRepository: SubscriberRepository; let userRepository: UserRepository; let allMessages: Message[]; let allSubscribers: Subscriber[]; let allUsers: User[]; let message: Message; let sender: Subscriber; let recipient: Subscriber; let messagesWithSenderAndRecipient: Message[]; let user: User; beforeAll(async () => { const module = await Test.createTestingModule({ imports: [ rootMongooseTestModule(installMessageFixtures), MongooseModule.forFeature([ UserModel, RoleModel, PermissionModel, SubscriberModel, MessageModel, AttachmentModel, ]), ], providers: [ LoggerService, AttachmentService, AttachmentRepository, UserService, UserRepository, RoleService, RoleRepository, SubscriberService, SubscriberRepository, MessageService, MessageRepository, EventEmitter2, ], }).compile(); messageService = module.get(MessageService); messageRepository = module.get(MessageRepository); subscriberRepository = module.get(SubscriberRepository); userRepository = module.get(UserRepository); allSubscribers = await subscriberRepository.findAll(); allUsers = await userRepository.findAll(); allMessages = await messageRepository.findAll(); message = await messageRepository.findOne({ mid: 'mid-1' }); sender = await subscriberRepository.findOne(message['sender']); recipient = await subscriberRepository.findOne(message['recipient']); user = await userRepository.findOne(message['sentBy']); messagesWithSenderAndRecipient = allMessages.map((message) => ({ ...message, sender: allSubscribers.find(({ id }) => id === message['sender']).id, recipient: allSubscribers.find(({ id }) => id === message['recipient']) .id, sentBy: allUsers.find(({ id }) => id === message['sentBy']).id, })); }); afterEach(jest.clearAllMocks); afterAll(closeInMongodConnection); describe('findOneAndPopulate', () => { it('should find message by id, and populate its corresponding sender and recipient', async () => { jest.spyOn(messageRepository, 'findOneAndPopulate'); const result = await messageService.findOneAndPopulate(message.id); expect(messageRepository.findOneAndPopulate).toHaveBeenCalledWith( message.id, ); expect(result).toEqualPayload({ ...messageFixtures.find(({ mid }) => mid === message.mid), sender, recipient, sentBy: user.id, }); }); }); describe('findPageAndPopulate', () => { const pageQuery = getPageQuery(); it('should find messages, and foreach message populate the corresponding sender and recipient', async () => { jest.spyOn(messageRepository, 'findPageAndPopulate'); const result = await messageService.findPageAndPopulate({}, pageQuery); const messagesWithSenderAndRecipient = allMessages.map((message) => ({ ...message, sender: allSubscribers.find(({ id }) => id === message['sender']), recipient: allSubscribers.find(({ id }) => id === message['recipient']), sentBy: allUsers.find(({ id }) => id === message['sentBy']).id, })); expect(messageRepository.findPageAndPopulate).toHaveBeenCalledWith( {}, pageQuery, ); expect(result).toEqualPayload(messagesWithSenderAndRecipient); }); }); describe('findHistoryUntilDate', () => { it('should return history until given date', async () => { const until: Date = new Date( new Date().setMonth(new Date().getMonth() + 1), ); const result = await messageService.findHistoryUntilDate( sender, until, 30, ); const historyMessages = messagesWithSenderAndRecipient.filter( (message) => message.createdAt <= until, ); expect(result).toEqualPayload(historyMessages); }); }); describe('findHistorySinceDate', () => { it('should return history since given date', async () => { const since: Date = new Date(); const result = await messageService.findHistorySinceDate( sender, since, 30, ); const messagesWithSenderAndRecipient = allMessages.map((message) => ({ ...message, sender: allSubscribers.find(({ id }) => id === message['sender']).id, recipient: allSubscribers.find(({ id }) => id === message['recipient']) .id, sentBy: allUsers.find(({ id }) => id === message['sentBy']).id, })); const historyMessages = messagesWithSenderAndRecipient.filter( (message) => message.createdAt > since, ); expect(result).toEqual( historyMessages.sort((message1, message2) => sortRowsBy(message1, message2, 'createdAt', 'asc'), ), ); }); }); });