6 min read
Original source

Writing unit tests

Testing our application can increase our confidence when it comes to creating a fully-functional API. In this article, we look into how we can test our…

Testing our application can increase our confidence when it comes to creating a fully-functional API. In this article, we look into how we can test our application by writing unit tests. We do so by using some of the utilities built into NestJS, as well as the Jest library. If you would like to get to know Jest better first, check out the first part of the JavaScript testing tutorial. Testing NestJS with unit tests The job of a unit test is to verify an individual piece of code. A tested unit can be a module, a class, or a function. Each of our tests should be isolated and independent of each other. By writing unit tests, we can make sure that individual parts of our application work as expected. Let’s write some tests for the  AuthenticationService. src/authentication/tests/authentication.service.spec.ts import { AuthenticationService } from '../authentication.service'; import { UsersService } from '../../users/users.service'; import { Repository } from 'typeorm'; import User from '../../users/user.entity'; import { JwtService } from '@nestjs/jwt'; import { ConfigService } from '@nestjs/config'; describe('The AuthenticationService', () => { const authenticationService = new AuthenticationService( new UsersService( new Repository() ), new JwtService({ secretOrPrivateKey: 'Secret key' }), new ConfigService() ); describe('when creating a cookie', () => { it('should return a string', () => { const userId = 1; expect( typeof authenticationService.getCookieWithJwtToken(userId) ).toEqual('string') }) }) }); PASS src/authentication/tests/authentication.service.spec.ts The AuthenticationService when creating a cookie ✓ should return a string (12ms) When we execute  npm run test, Jest looks for files ending with  .spec.ts and executes them. We can improve the above code. Each of our tests needs to be independent, and we need to ensure that. If we add more tests in the above file, all of them will use the same instance of the  AuthenticationService. It breaks the rule of all tests being independent. To deal with it, we can use the  beforeEach that runs before every test. src/authentication/tests/authentication.service.spec.ts import { AuthenticationService } from '../authentication.service'; import { UsersService } from '../../users/users.service'; import { Repository } from 'typeorm'; import User from '../../users/user.entity'; import { JwtService } from '@nestjs/jwt'; import { ConfigService } from '@nestjs/config'; describe('The AuthenticationService', () => { let authenticationService: AuthenticationService; beforeEach(() => { authenticationService = new AuthenticationService( new UsersService( new Repository() ), new JwtService({ secretOrPrivateKey: 'Secret key' }), new ConfigService() ); }) describe('when creating a cookie', () => { it('should return a string', () => { const userId = 1; expect( typeof authenticationService.getCookieWithJwtToken(userId) ).toEqual('string') }) }) });Now, we are sure that every test in the  authentication.service.spec.ts file gets a brand new instance of the  AuthenticationService. Unfortunately, the above code does not look very elegant. Because the constructor of the  AuthenticationService expects some dependencies, we provided them manually so far. Creating testing modules Fortunately, NestJS provides us with built-in utilities to deal with the above issue.npm install @nestjs/testingBy using  Test.createTestingModule().compile() we can create a module with its dependencies resolved. src/authentication/tests/authentication.service.spec.ts import { AuthenticationService } from '../authentication.service'; import { Test } from '@nestjs/testing'; import { UsersModule } from '../../users/users.module'; import { ConfigModule, ConfigService } from '@nestjs/config'; import { JwtModule } from '@nestjs/jwt'; import { DatabaseModule } from '../../database/database.module'; import * as Joi from '@hapi/joi'; describe('The AuthenticationService', () => { let authenticationService: AuthenticationService; beforeEach(async () => { const module = await Test.createTestingModule({ imports: [ UsersModule, ConfigModule.forRoot({ validationSchema: Joi.object({ POSTGRES_HOST: Joi.string().required(), POSTGRES_PORT: Joi.number().required(), POSTGRES_USER: Joi.string().required(), POSTGRES_PASSWORD: Joi.string().required(), POSTGRES_DB: Joi.string().required(), JWT_SECRET: Joi.string().required(), JWT_EXPIRATION_TIME: Joi.string().required(), PORT: Joi.number(), }) }), DatabaseModule, JwtModule.registerAsync({ imports: [ConfigModule], inject: [ConfigService], useFactory: async (configService: ConfigService) => ({ secret: configService.get('JWT_SECRET'), signOptions: { expiresIn: `${configService.get('JWT_EXPIRATION_TIME')}s`, }, }), }), ], providers: [ AuthenticationService ], }).compile(); authenticationService = await module.get(AuthenticationService); }) describe('when creating a cookie', () => { it('should return a string', () => { const userId = 1; expect( typeof authenticationService.getCookieWithJwtToken(userId) ).toEqual('string') }) }) });There are quite a few issues with the above code still. Let’s deal with them one by one. Mocking the database connection The biggest issue above is that we use the  DatabaseModule which means connecting to the real database. When doing unit tests, we want to avoid it. After removing the  DatabaseModule from our imports we can see an error: Error: Nest can’t resolve dependencies of the UserRepository (?). Please make sure that the argument Connection at index [0] is available in the TypeOrmModule context. To work around it, we need to provide a mocked User repository. To do so, we need to use the  getRepositoryToken from  '@nestjs/typeorm.import User from '../../users/user.entity';providers: [ AuthenticationService, { provide: getRepositoryToken(User), useValue: {}, } ],Unfortunately, the above error persists. This is because we imported  UsersModule that contains TypeOrmModule.forFeature([User]). We should avoid importing our modules when writing unit tests because we don’t want to test integration between classes just yet. We need to add  UsersService to our providers instead. src/authentication/tests/authentication.service.spec.ts import { AuthenticationService } from '../authentication.service'; import { Test } from '@nestjs/testing'; import { ConfigModule } from '@nestjs/config'; import { JwtModule } from '@nestjs/jwt'; import { getRepositoryToken } from '@nestjs/typeorm'; import User from '../../users/user.entity'; import { UsersService } from '../../users/users.service'; describe('The AuthenticationService', () => { let authenticationService: AuthenticationService; beforeEach(async () => { const module = await Test.createTestingModule({ imports: [ ConfigModule.forRoot({ // ... }), JwtModule.registerAsync({ // ... }), ], providers: [ UsersService, AuthenticationService, { provide: getRepositoryToken(User), useValue: {}, } ], }).compile(); authenticationService = await module.get(AuthenticationService); }) describe('when creating a cookie', () => { it('should return a string', () => { const userId = 1; expect( typeof authenticationService.getCookieWithJwtToken(userId) ).toEqual('string') }) }) });The object we put into  useValue above is our mocked repository. We add some methods to it later below. Mocking ConfigService and JwtService Since we want to avoid using modules, we also can replace ConfigModule and JwtModule with mocks. To be more precise, we need to provide mocked  ConfigService and  JwtService. A clean approach to that would be to create separate files for the above mocks. src/utils/mocks/config.service.ts const mockedConfigService = { get(key: string) { switch (key) { case 'JWT_EXPIRATION_TIME': return '3600' } } } src/utils/mocks/jwt.service.ts const mockedJwtService = { sign: () => '' }When we use the above, our test now looks like that: src/utils/mocks/config.service.ts import { AuthenticationService } from '../authentication.service'; import { Test } from '@nestjs/testing'; import { ConfigService } from '@nestjs/config'; import { JwtService } from '@nestjs/jwt'; import { getRepositoryToken } from '@nestjs/typeorm'; import User from '../../users/user.entity'; import { UsersService } from '../../users/users.service'; import mockedJwtService from '../../utils/mocks/jwt.service'; import mockedConfigService from '../../utils/mocks/config.service'; describe('The AuthenticationService', () => { let authenticationService: AuthenticationService; beforeEach(async () => { const module = await Test.createTestingModule({ providers: [ UsersService, AuthenticationService, { provide: ConfigService, useValue: mockedConfigService }, { provide: JwtService, useValue: mockedJwtService }, { provide: getRepositoryToken(User), useValue: {} } ], }) .compile(); authenticationService = await module.get(AuthenticationService); }) describe('when creating a

Writing unit tests | NestJS.io