fetch-document-by-id-review-changes-3

src/document/document.controller.spec.ts

+/**
+ @file        <file name>
+ @description  <description>
+ @author      <Your Name>
+ @created     <YYYY-MM-DD>
+**/
 import { Test, TestingModule } from '@nestjs/testing';
 import { DocumentController } from './document.controller';
 import { DocumentService } from './document.service';
@@ -82,7 +88,7 @@ describe('DocumentController', () => {
     it('should throw NotFoundException if document is not found', async () => {
       const dto = { id: 'not-found-id' };
       mockDocumentService.fetchDocument.mockResolvedValue(undefined);
-      await expect(controller.download(dto)).rejects.toThrow('Document not found');
+      await expect(controller.download(dto)).rejects.toThrow('404');
     });
   });
 });

src/document/document.service.spec.ts

@@ -178,7 +178,7 @@ describe('DocumentService', () => {
 
     it('should throw NotFoundException if document is not found', async () => {
       jest.spyOn(service['prisma'].documents, 'findUnique').mockResolvedValue(null);
-      await expect(service.fetchDocument('not-found-id')).rejects.toThrow('Document not found');
+      await expect(service.fetchDocument('not-found-id')).rejects.toThrow('404');
     });
   });
 });