Added half of asset service tests

COS301-SE-2023 · Jul 25, 2023 · b9aa1f3 · b9aa1f3
1 parent eca508b
commit b9aa1f3
Showing 1 changed file with 90 additions and 3 deletions.
diff --git a/backend/src/assets/assets.service.spec.ts b/backend/src/assets/assets.service.spec.ts
@@ -8,6 +8,8 @@ import { getRepositoryToken } from '@nestjs/typeorm';
 import { Asset } from './entities/asset.entity';
 import { AuthService } from '../auth/auth.service';
 import { JwtService } from '@nestjs/jwt';
+import { AssetDTO } from './dto/asset.dto';
+import { RetrieveAllDTO } from '../asset_manager/dto/retrieve_all.dto';
 
 describe('AssetsService', () => {
   let service: AssetsService;
@@ -31,9 +33,94 @@ describe('AssetsService', () => {
     );
   });
 
-  describe('root/config', () => {
-    it('should be defined', () => {
-      expect(service).toBeDefined();
+  describe('saveAsset', () => {
+    it('should initialize DateCreated and Size', async () => {
+      const assetDTO = new AssetDTO();
+      assetDTO.Content = 'test';
+
+      jest
+        .spyOn(Repository.prototype, 'create')
+        .mockReturnValue(assetDTO);
+
+      jest
+        .spyOn(Repository.prototype, 'save')
+        .mockResolvedValue(assetDTO);
+
+      const result = await service.saveAsset(
+        assetDTO,
+      );
+      expect(result.DateCreated).toBeDefined();
+      expect(result.Size).toBe(
+        assetDTO.Content.length,
+      );
+    });
+
+    it('should call create and save', async () => {
+      const assetDTO = new AssetDTO();
+      assetDTO.Content = 'test';
+
+      jest
+        .spyOn(Repository.prototype, 'create')
+        .mockReturnValue(assetDTO);
+
+      const saveSpy = jest
+        .spyOn(Repository.prototype, 'save')
+        .mockResolvedValue(assetDTO);
+
+      await service.saveAsset(assetDTO);
+      expect(saveSpy).toBeCalledWith(assetDTO);
+    });
+  });
+
+  describe('retrieveOne', () => {
+    it('should call findOne', async () => {
+      const assetDTO = new AssetDTO();
+      assetDTO.AssetID = 'test';
+      assetDTO.Format = 'test';
+
+      const findOneSpy = jest
+        .spyOn(Repository.prototype, 'findOne')
+        .mockResolvedValue(assetDTO);
+
+      const result = await service.retrieveOne(
+        assetDTO,
+      );
+
+      expect(result).toBe(assetDTO);
+      expect(findOneSpy).toBeCalledWith({
+        where: {
+          AssetID: assetDTO.AssetID,
+          Format: assetDTO.Format,
+        },
+      });
+    });
+  });
+
+  describe('retrieveAllAssets', () => {
+    it('should call createQueryBuilder', async () => {
+      const retrieveAllDTO = new RetrieveAllDTO();
+      retrieveAllDTO.UserID = 1;
+      retrieveAllDTO.ParentFolderID = 'test';
+
+      const createQueryBuilderSpy = jest
+        .spyOn(
+          Repository.prototype,
+          'createQueryBuilder',
+        )
+        .mockReturnValue({
+          where: jest.fn().mockReturnThis(),
+          andWhere: jest.fn().mockReturnThis(),
+          orderBy: jest.fn().mockReturnThis(),
+          getMany: jest
+            .fn()
+            .mockResolvedValue([]),
+        } as any);
+
+      await service.retrieveAllAssets(
+        retrieveAllDTO,
+      );
+
+      expect(createQueryBuilderSpy).toBeCalled();
     });
   });
 });