사내 단위 테스트 작성 문화 도입기 (2) DB Sociable Test 환경 구축

import { DataSource, EntityManager } from 'typeorm'; import { SnakeNamingStrategy } from 'typeorm-naming-strategies'; const PROJECT_ROOT = `${process.cwd()}`; export const integrationTestDataSource = new DataSource({ type: 'mysql', host: process.env.DB_HOST || 'localhost', port: Number(process.env.DB_PORT) || 3306, username: process.env.DB_USERNAME || 'test', password: process.env.DB_PASSWORD || 'test', database: process.env.DB_DATABASE || 'test', entities: [`${PROJECT_ROOT}/src/**/*.entity{.ts,.js}`], namingStrategy: new SnakeNamingStrategy(), charset: 'utf8mb4', logging: ['error', 'warn'], synchronize: false, }); export async function initIntegrationTestDataSource() { await integrationTestDataSource.initialize(); return integrationTestDataSource; } export async function startIntegrationTestTransaction(entityManager: EntityManager) { await entityManager.query(`SET FOREIGN_KEY_CHECKS = 0`); await entityManager.query('START TRANSACTION'); } export async function rollbackIntegrationTestTransaction(entityManager: EntityManager) { await entityManager.query('ROLLBACK'); }

describe('테스트 대상명', () => { let dataSource: DataSource; let entityManager: EntityManager; let testTarget: TestTarget; beforeAll(async () => { // 테스트 실행 전 한 번만 실행: DB 연결 초기화 dataSource = await initIntegrationTestDataSource(); entityManager = dataSource.manager; testTarget = new TestTarget(dataSource); }, 20000); beforeEach(async () => { // 각 테스트 케이스 실행 전마다 실행: 트랜잭션 시작 await startIntegrationTestTransaction(entityManager); }); afterEach(async () => { // 각 테스트 케이스 실행 후마다 실행: 트랜잭션 롤백 await rollbackIntegrationTestTransaction(entityManager); }); afterAll(async () => { // 테스트 실행 후 한 번만 실행: DB 연결 종료 await dataSource.destroy(); }); // 실제 테스트 케이스들... });

import { Injectable } from '@nestjs/common'; import { DataSource, EntityManager } from 'typeorm'; import { CourseRepository } from '@domain/course/repository/course.repository'; import { Course } from '@domain/course/entity/course.entity'; import * as R from 'remeda'; @Injectable() export class CourseRepositoryImpl implements CourseRepository { constructor(private dataSource: DataSource) {} async createCourse(entityManager: EntityManager, course: Course): Promise<Course> { // 연관 엔티티 먼저 저장 await this.upsertCourseVersions(entityManager, [course.latestCourseVersion]); // 주요 엔티티에서 중첩된 객체 제거 후 저장 const omittedCourse = R.omit(course, [ 'latestCourseVersion', ]); await entityManager.getRepository(Course).insert(omittedCourse); return course; } // 기타 구현 메서드... }

import { DataSource, EntityManager } from 'typeorm'; import { initIntegrationTestDataSource, rollbackIntegrationTestTransaction, startIntegrationTestTransaction, } from '../helpers/integration-test.helper'; import { CourseRepositoryImpl } from '@infrastructure/course/repository/course.repository.impl'; import { createCourseFactory } from '../factories/course.factory'; import { Course } from '@domain/course/entity/course.entity'; import { CourseVersion } from '@domain/course/entity/course-version.entity'; describe('CourseRepositoryImpl', () => { let repo: CourseRepositoryImpl; let dataSource: DataSource; let entityManager: EntityManager; beforeAll(async () => { dataSource = await initIntegrationTestDataSource(); entityManager = dataSource.manager; repo = new CourseRepositoryImpl(dataSource); }, 20000); beforeEach(async () => { await startIntegrationTestTransaction(entityManager); }); afterEach(async () => { await rollbackIntegrationTestTransaction(entityManager); }); afterAll(async () => { await dataSource.destroy(); }); describe('createCourse', () => { it('코스 객체가 생성되고 반환되어야 한다', async () => { // Given const course = createCourseFactory()[0]; // 테스트용 코스 데이터 생성 // When const res = await repo.createCourse(entityManager, course); // Then const createdCourse = await entityManager.getRepository(Course).findOne({ where: { id: course.id, }, }); expect(createdCourse).not.toBeNull(); const createdCourseVersion = await entityManager.getRepository(CourseVersion).findOne({ where: { id: course.latestCourseVersionId, }, }); expect(createdCourseVersion).not.toBeNull(); }); }); });

import { Inject, Injectable } from '@nestjs/common'; import { DataSource } from 'typeorm'; import { CourseRepository } from '@domain/course/repository/course.repository'; import { getCurrentDate } from '@common/utilities/time'; import { Course } from '@domain/course/entity/course.entity'; import { concurrent, map, pipe, toArray, toAsync } from '@fxts/core'; @Injectable() export class ArchiveCoursesByIdsUseCase { constructor( private dataSource: DataSource, @Inject('CourseRepository') private courseRepo: CourseRepository, ) {} async execute(courseIds: string[]): Promise<Course[]> { const entityManager = this.dataSource.manager; // 병렬로 코스 조회 const courses = await pipe( courseIds, toAsync, map(courseId => this.courseRepo.findCourseByIdWithProgressRelation(entityManager, courseId)), concurrent(5), toArray, ); // 코스 아카이브 처리 courses.map(course => course.archiveCourse(getCurrentDate())); // 변경사항 저장 await this.courseRepo.updateCourses(entityManager, courses); return courses; } }

import { DataSource, EntityManager } from 'typeorm'; import { initIntegrationTestDataSource, rollbackIntegrationTestTransaction, startIntegrationTestTransaction, } from '../helpers/integration-test.helper'; import { CourseRepositoryImpl } from '@infrastructure/course/repository/course.repository.impl'; import { createCourseFactory } from '../factories/course.factory'; import { Course } from '@domain/course/entity/course.entity'; import { CourseVersion } from '@domain/course/entity/course-version.entity'; import { ArchiveCoursesByIdsUseCase } from '@application/course/archive-courses-by-ids.use-case'; import { LearningProgress } from '@domain/learning/entities/learning-progress.entity'; import { CourseProgressMapping } from '@domain/course/entity/course-progress-mapping.entity'; import { CourseStatusEnum } from '@domain/course/course.enum'; import { ProgressStatus } from '@domain/learning/learning.enum'; describe('ArchiveCoursesByIdsUseCase', () => { let uc: ArchiveCoursesByIdsUseCase; let dataSource: DataSource; let entityManager: EntityManager; let courseRepo: CourseRepositoryImpl; beforeAll(async () => { dataSource = await initIntegrationTestDataSource(); entityManager = dataSource.manager; courseRepo = new CourseRepositoryImpl(dataSource); uc = new ArchiveCoursesByIdsUseCase(dataSource, courseRepo); }, 20000); beforeEach(async () => { await startIntegrationTestTransaction(entityManager); }); afterEach(async () => { await rollbackIntegrationTestTransaction(entityManager); }); afterAll(async () => { await dataSource.destroy(); }); describe('execute', () => { it('코스를 아카이브하고, 새 버전을 생성하며, 학습 진행 상태를 비활성화한다', async () => { // Given - 테스트 데이터 준비 const course = createCourseFactory()[0]; await entityManager.getRepository(Course).insert(course); await entityManager.getRepository(CourseProgressMapping).insert(course.learningProgressMapping); await entityManager.getRepository(LearningProgress).insert(course.learningProgressMapping.learningProgress); await entityManager.getRepository(CourseVersion).insert(course.latestCourseVersion); // When - 유스케이스 실행 await uc.execute([course.id]); // Then - 결과 검증 const archivedCourse = await entityManager.getRepository(Course).findOne({ where: { id: course.id }, relations: { courseVersions: true }, withDeleted: true, }); // 코스가 아카이브됨 expect(archivedCourse).toBeInstanceOf(Course); expect(archivedCourse.deletedAt).not.toBeNull(); // 아카이브 상태의 새 버전이 생성됨 const archivedVersion = archivedCourse.courseVersions.find(cv => cv.status === CourseStatusEnum.ARCHIVED ); expect(archivedVersion).toBeInstanceOf(CourseVersion); expect(archivedVersion.deletedAt).not.toBeNull(); // 연결된 학습 진행 상태가 비활성화됨 const learningProgress = await entityManager.getRepository(LearningProgress).findOne({ where: { id: course.learningProgressMapping.learningProgressId }, withDeleted: true, }); expect(learningProgress).toBeInstanceOf(LearningProgress); expect(learningProgress.progress).toEqual(ProgressStatus.INACTIVE); expect(learningProgress.isVisible).toEqual(0); }); }); });

export function createCourseFactory(override = {}) { const id = generateUuid(); const courseVersion = new CourseVersion({ id: generateUuid(), title: '테스트 강좌', description: '테스트 설명', difficulty: 'BEGINNER', status: 'PUBLISHED', ...override.version, }); return [ new Course({ id, latestCourseVersionId: courseVersion.id, latestCourseVersion: courseVersion, instructorId: generateUuid(), categoryId: generateUuid(), isPublic: true, ...override, }), ]; }