tests(): improvement

Author: Tony133

sample/07-sequelize/src/app.module.ts

@@ -1,5 +1,6 @@
 import { Module } from '@nestjs/common';
 import { SequelizeModule } from '@nestjs/sequelize';
+import { User } from './users/models/user.model';
 import { UsersModule } from './users/users.module';
 
 @Module({
@@ -13,6 +14,7 @@ import { UsersModule } from './users/users.module';
       database: 'test',
       autoLoadModels: true,
       synchronize: true,
+      models: [User],
     }),
     UsersModule,
   ],

…ple/07-sequelize/src/users/user.model.ts …sequelize/src/users/models/user.model.tssample/07-sequelize/src/users/user.model.ts renamed to sample/07-sequelize/src/users/models/user.model.ts sample/07-sequelize/src/users/user.model.ts renamed to sample/07-sequelize/src/users/models/user.model.ts

@@ -1,7 +1,7 @@
 import { Column, Model, Table } from 'sequelize-typescript';
 
 @Table
-export class User extends Model<User> {
+export class User extends Model {
   @Column
   firstName: string;
 

sample/07-sequelize/src/users/users.controller.ts

@@ -1,6 +1,6 @@
 import { Body, Controller, Delete, Get, Param, Post } from '@nestjs/common';
 import { CreateUserDto } from './dto/create-user.dto';
-import { User } from './user.model';
+import { User } from './models/user.model';
 import { UsersService } from './users.service';
 
 @Controller('users')

sample/07-sequelize/src/users/users.module.ts

@@ -1,12 +1,13 @@
 import { Module } from '@nestjs/common';
 import { SequelizeModule } from '@nestjs/sequelize';
-import { User } from './user.model';
+import { User } from './models/user.model';
 import { UsersController } from './users.controller';
 import { UsersService } from './users.service';
 
 @Module({
   imports: [SequelizeModule.forFeature([User])],
   providers: [UsersService],
   controllers: [UsersController],
+  exports: [SequelizeModule],
 })
 export class UsersModule {}

sample/07-sequelize/src/users/users.service.spec.ts

@@ -1,5 +1,5 @@
 import { Test, TestingModule } from '@nestjs/testing';
-import { User } from './user.model';
+import { User } from './models/user.model';
 import { UsersService } from './users.service';
 import { getModelToken } from '@nestjs/sequelize';
 
@@ -32,7 +32,7 @@ describe('UserService', () => {
           useValue: {
             findAll: jest.fn(() => userArray),
             findOne: jest.fn(),
-            save: jest.fn(),
+            create: jest.fn(() => oneUser),
             remove: jest.fn(),
             destroy: jest.fn(() => oneUser),
           },
@@ -50,11 +50,16 @@ describe('UserService', () => {
 
   describe('create()', () => {
     it('should successfully insert a user', () => {
-      const user = {
+      const oneUser = {
         firstName: 'firstName #1',
         lastName: 'lastName #1',
       };
-      expect(user).toEqual(user);
+      expect(
+        service.create({
+          firstName: 'firstName #1',
+          lastName: 'lastName #1',
+        }),
+      ).toEqual(oneUser);
     });
   });
 

sample/07-sequelize/src/users/users.service.ts

@@ -1,7 +1,7 @@
 import { Injectable } from '@nestjs/common';
 import { InjectModel } from '@nestjs/sequelize';
 import { CreateUserDto } from './dto/create-user.dto';
-import { User } from './user.model';
+import { User } from './models/user.model';
 
 @Injectable()
 export class UsersService {
@@ -11,11 +11,7 @@ export class UsersService {
   ) {}
 
   create(createUserDto: CreateUserDto): Promise<User> {
-    const user = new User();
-    user.firstName = createUserDto.firstName;
-    user.lastName = createUserDto.lastName;
-
-    return user.save();
+    return this.userModel.create(createUserDto);
   }
 
   async findAll(): Promise<User[]> {