test if ml is disabled

immich-app · May 8, 2024 · 4c5bab8 · 4c5bab8
1 parent ea83cc9
commit 4c5bab8
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/server/src/services/search.service.spec.ts b/server/src/services/search.service.spec.ts
@@ -157,6 +157,26 @@ describe(SearchService.name, () => {
   });
 
   describe('handleSearchDuplicates', () => {
+    it('should skip if machine learning is disabled', async () => {
+      configMock.load.mockResolvedValue([{ key: SystemConfigKey.MACHINE_LEARNING_ENABLED, value: false }]);
+      const id = assetStub.livePhotoMotionAsset.id;
+      assetMock.getById.mockResolvedValue(assetStub.livePhotoMotionAsset);
+
+      const result = await sut.handleSearchDuplicates({ id });
+
+      expect(result).toBe(JobStatus.SKIPPED);
+    });
+
+    it('should skip if clip is disabled', async () => {
+      configMock.load.mockResolvedValue([{ key: SystemConfigKey.MACHINE_LEARNING_CLIP_ENABLED, value: false }]);
+      const id = assetStub.livePhotoMotionAsset.id;
+      assetMock.getById.mockResolvedValue(assetStub.livePhotoMotionAsset);
+
+      const result = await sut.handleSearchDuplicates({ id });
+
+      expect(result).toBe(JobStatus.SKIPPED);
+    });
+
     it('should fail if asset is not found', async () => {
       const result = await sut.handleSearchDuplicates({ id: assetStub.image.id });