Implement RabbitMQ (#16)

* Implement RabbitMQ * Fix test suit
CUSportGo · Oct 23, 2023 · 9d0474a · 9d0474a
1 parent 8e0cc25
commit 9d0474a
Show file tree

Hide file tree

Showing 4 changed files with 44 additions and 17 deletions.
diff --git a/src/app.module.ts b/src/app.module.ts
@@ -4,10 +4,11 @@ import { AppService } from './app.service';
 import { PrismaModule } from './prisma/prisma.module';
 import { AuthModule } from './auth/auth.module';
 import { UserModule } from './user/user.module';
+import { ClientsModule, Transport } from '@nestjs/microservices';
 
 @Module({
   imports: [PrismaModule, AuthModule, UserModule],
   controllers: [AppController],
   providers: [AppService],
 })
-export class AppModule {}
+export class AppModule { }
diff --git a/src/auth/auth.module.ts b/src/auth/auth.module.ts
@@ -1,4 +1,4 @@
-import { Module } from '@nestjs/common';
+import { Inject, Module } from '@nestjs/common';
 import { PrismaModule } from '../prisma/prisma.module';
 import { UserRepository } from '../repository/user.repository';
 import { AuthService } from './auth.service';
@@ -7,9 +7,20 @@ import { JwtModule, JwtService } from '@nestjs/jwt';
 import { AccessTokenStrategy } from './strategies/accessToken.strategy';
 import { ConfigService } from '@nestjs/config';
 import { BlacklistRepository } from 'src/repository/blacklist.repository';
+import { ClientsModule, Transport } from '@nestjs/microservices';
 
 @Module({
-  imports: [PrismaModule, JwtModule.register({})],
+  imports: [PrismaModule, JwtModule.register({}), ClientsModule.register([{
+    name: 'EMAIL_SERVICE',
+    transport: Transport.RMQ,
+    options: {
+      urls: ['amqp://localhost:5672'],
+      queue: 'email_queue',
+      queueOptions: {
+        durable: false
+      }
+    }
+  }])],
   providers: [AuthService, UserRepository, AccessTokenStrategy, ConfigService, BlacklistRepository],
   controllers: [AuthController],
 })

diff --git a/src/auth/auth.service.spec.ts b/src/auth/auth.service.spec.ts
@@ -24,6 +24,10 @@ describe('AuthService', () => {
   const mockBlacklistRepository = {
     addOutdatedToken: jest.fn(),
   }
+
+  const mockEmailService = {
+    emit: jest.fn(),
+  }
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
       providers: [AuthService,
@@ -44,6 +48,10 @@ describe('AuthService', () => {
         {
           provide: BlacklistRepository,
           useValue: mockBlacklistRepository,
+        },
+        {
+          provide: 'EMAIL_SERVICE',
+          useValue: mockEmailService,
         }
       ],
     }).compile();

diff --git a/src/auth/auth.service.ts b/src/auth/auth.service.ts
@@ -1,4 +1,4 @@
-import { Injectable } from '@nestjs/common';
+import { Inject, Injectable } from '@nestjs/common';
 import { JwtService } from '@nestjs/jwt';
 import { UserRepository } from '../repository/user.repository';
 import * as bcrypt from 'bcrypt';
@@ -24,7 +24,7 @@ import {
   // ValidateGoogleResponse,
   ValidateOAuthRequest,
 } from './auth.pb';
-import { RpcException } from '@nestjs/microservices';
+import { ClientProxy, RpcException } from '@nestjs/microservices';
 import { status } from '@grpc/grpc-js';
 import { $Enums, Prisma } from '@prisma/client';
 import { v4 as uuidv4 } from 'uuid';
@@ -40,7 +40,8 @@ export class AuthService implements AuthServiceController {
     private blacklistRepo: BlacklistRepository,
     private jwtService: JwtService,
     private configService: ConfigService,
-  ) {}
+    @Inject('EMAIL_SERVICE') private client: ClientProxy,
+  ) { }
 
   public async login(request: LoginRequest): Promise<LoginResponse> {
     try {
@@ -433,15 +434,15 @@ export class AuthService implements AuthServiceController {
         'http://localhost:3000/reset-password/' + userToken;
 
       //set connection
-      const transporter = nodemailer.createTransport({
-        host: 'smtp.office365.com',
-        port: 587,
-        secure: false, // Set to true if using SSL
-        auth: {
-          user: process.env.EMAIL_USERNAME,
-          pass: process.env.EMAIL_PASSWORD,
-        },
-      });
+      // const transporter = nodemailer.createTransport({
+      //   host: 'smtp.office365.com',
+      //   port: 587,
+      //   secure: false, // Set to true if using SSL
+      //   auth: {
+      //     user: process.env.EMAIL_USERNAME,
+      //     pass: process.env.EMAIL_PASSWORD,
+      //   },
+      // });
 
       // Define email options
       const mailOptions = {
@@ -453,8 +454,14 @@ export class AuthService implements AuthServiceController {
           <a href="${linkToResetPassword}">Reset Password</a>
         `,
       };
-      await transporter.sendMail(mailOptions);
-      return { resetPasswordUrl: linkToResetPassword };
+
+
+      this.client.emit('forgot-password', { 'mailOptions': mailOptions });
+      // await transporter.sendMail(mailOptions);
+
+      return { resetPasswordUrl: linkToResetPassword }
+
+
     } catch (err: any) {
       console.log(err);
       if (!(err instanceof RpcException)) {