test: edge controller 테스트코드

Author: summersummerwhy

apps/backend/src/edge/edge.controller.spec.ts

@@ -1,25 +1,120 @@
 import { Test, TestingModule } from '@nestjs/testing';
 import { EdgeController } from './edge.controller';
 import { EdgeService } from './edge.service';
+import { CreateEdgeDto } from './dtos/createEdge.dto';
+import { EdgeResponseMessage } from './edge.controller';
+import { EdgeNotFoundException } from '../exception/edge.exception';
+import { Edge } from './edge.entity';
+import { Node } from '../node/node.entity';
 
 describe('EdgeController', () => {
   let controller: EdgeController;
+  let edgeService: EdgeService;
 
   beforeEach(async () => {
     const module: TestingModule = await Test.createTestingModule({
       controllers: [EdgeController],
       providers: [
         {
           provide: EdgeService,
-          useValue: {},
+          useValue: {
+            createEdge: jest.fn(),
+            deleteEdge: jest.fn(),
+            findEdges: jest.fn(),
+          },
         },
       ],
     }).compile();
 
     controller = module.get<EdgeController>(EdgeController);
+    edgeService = module.get<EdgeService>(EdgeService);
   });
 
   it('컨트롤러 클래스가 정상적으로 인스턴스화된다.', () => {
     expect(controller).toBeDefined();
   });
+
+  describe('createEdge', () => {
+    it('엣지가 성공적으로 만들어진다', async () => {
+      const dto: CreateEdgeDto = { fromNode: 1, toNode: 3 };
+      const expectedResponse = {
+        message: EdgeResponseMessage.EDGE_CREATED,
+      };
+
+      jest.spyOn(edgeService, 'createEdge').mockResolvedValue(undefined);
+      const result = await controller.createEdge(dto);
+
+      expect(edgeService.createEdge).toHaveBeenCalledWith(dto);
+      expect(result).toEqual(expectedResponse);
+    });
+  });
+
+  describe('deleteEdge', () => {
+    it('id에 해당하는 엣지를 찾아 삭제한다.', async () => {
+      const id = 2;
+      const expectedResponse = {
+        message: EdgeResponseMessage.EDGE_DELETED,
+      };
+
+      const result = await controller.deleteEdge(id);
+
+      expect(edgeService.deleteEdge).toHaveBeenCalledWith(id);
+      expect(result).toEqual(expectedResponse);
+    });
+
+    it('id에 해당하는 엣지가 존재하지 않으면 NodeNotFoundException을 throw한다.', async () => {
+      jest
+        .spyOn(edgeService, 'deleteEdge')
+        .mockRejectedValue(new EdgeNotFoundException());
+
+      await expect(controller.deleteEdge(1)).rejects.toThrow(
+        EdgeNotFoundException,
+      );
+    });
+  });
+
+  describe('findEdges', () => {
+    it('모든 엣지 목록을 반환한다.', async () => {
+      const node3 = {
+        id: 3,
+        x: 0,
+        y: 0,
+        title: 'Node Title',
+        page: null,
+        outgoingEdges: [],
+        incomingEdges: [],
+      } as Node;
+      const node4 = {
+        id: 4,
+        x: 0,
+        y: 0,
+        title: 'Node Title',
+        page: null,
+        outgoingEdges: [],
+        incomingEdges: [],
+      } as Node;
+      const node5 = {
+        id: 5,
+        x: 0,
+        y: 0,
+        title: 'Node Title',
+        page: null,
+        outgoingEdges: [],
+        incomingEdges: [],
+      } as Node;
+
+      const expectedEdges = [
+        { id: 1, fromNode: node3, toNode: node5 },
+        { id: 2, fromNode: node3, toNode: node4 },
+      ] as Edge[];
+      node3.outgoingEdges = [];
+
+      jest.spyOn(edgeService, 'findEdges').mockResolvedValue(expectedEdges);
+
+      await expect(controller.findEdges()).resolves.toEqual({
+        message: EdgeResponseMessage.EDGE_ALL_RETURNED,
+        edges: expectedEdges,
+      });
+    });
+  });
 });

apps/backend/src/edge/edge.controller.ts

@@ -33,11 +33,11 @@ export class EdgeController {
   })
   @Get('/')
   @HttpCode(HttpStatus.OK)
-  async getNodes() {
-    const nodes = await this.edgeService.findEdges();
+  async findEdges() {
+    const edges = await this.edgeService.findEdges();
     return {
       message: EdgeResponseMessage.EDGE_ALL_RETURNED,
-      nodes: nodes,
+      edges: edges,
     };
   }
 
@@ -56,7 +56,7 @@ export class EdgeController {
   @ApiOperation({ summary: '엣지를 삭제합니다.' })
   @Delete('/:id')
   @HttpCode(HttpStatus.OK)
-  async deleteNode(
+  async deleteEdge(
     @Param('id', ParseIntPipe) id: number,
   ): Promise<{ message: string }> {
     await this.edgeService.deleteEdge(id);