Added tests for retrieve_one

COS301-SE-2023 · Jul 25, 2023 · ca53bbd · ca53bbd
1 parent 5c7bb48
commit ca53bbd
Showing 1 changed file with 38 additions and 0 deletions.
diff --git a/backend/src/asset_manager/asset_manager.service.spec.ts b/backend/src/asset_manager/asset_manager.service.spec.ts
@@ -177,6 +177,44 @@ describe('AssetManagerService', () => {
       ).toBeCalledWith(retrieveAllDTO);
     });
 
+    describe('retrieve_one', () => {
+      it('should retrieve an image asset', async () => {
+        const assetDTO = new AssetDTO();
+        assetDTO.AssetID = 'test';
+        assetDTO.Format = 'image';
+        assetDTO.UserID = 1;
+
+        jest.spyOn(
+          imageManagerService,
+          'retrieveOne',
+        ).mockResolvedValue(assetDTO);
+
+        await service.retrieve_one(assetDTO);
+
+        expect(
+          imageManagerService.retrieveOne,
+        ).toBeCalledWith(assetDTO);
+      });
+
+      it('should retrieve a text asset', async () => {
+        const assetDTO = new AssetDTO();
+        assetDTO.AssetID = 'test';
+        assetDTO.Format = 'text';
+        assetDTO.UserID = 1;
+
+        jest.spyOn(
+          textManagerService,
+          'retrieveOne',
+        ).mockResolvedValue(assetDTO);
+
+        await service.retrieve_one(assetDTO);
+
+        expect(
+          textManagerService.retrieveOne,
+        ).toBeCalledWith(assetDTO);
+      });
+    });
+
     // it('should resize all image assets', async () => {
     //   const retrieveAllDTO = new RetrieveAllDTO();
     //   retrieveAllDTO.UserID = 1;