Fix unit tests

Author: jpggvilaca

web_ui/src/pages/annotator/annotation/layers/layers-factory.test.tsx

@@ -59,6 +59,7 @@ describe('LayersFactory', () => {
             selectedMediaItem,
             selectedMediaItemQuery: { isLoading: false },
             predictionsQuery: { data: undefined },
+            explanationsQuery: { data: undefined },
         } as SelectedMediaItemProps);
 
         const response = await annotatorRender(

web_ui/src/pages/annotator/providers/prediction-provider/use-prediction-roi-query.hook.test.tsx

@@ -180,11 +180,9 @@ describe('usePredictionsRoiQuery', () => {
                 expect.anything()
             );
         });
-
-        expect(mockedInferenceService.getExplanations).not.toHaveBeenCalled();
     });
 
-    describe('PredictionMode.ONLINE is sent as PredictionCache.NEVER, calls "getPredictions" and "getExplanations"', () => {
+    describe('PredictionMode.ONLINE is sent as PredictionCache.NEVER, calls "getPredictions"', () => {
         it('successful responses', async () => {
             jest.mocked(useAnnotatorMode).mockImplementation(() => ({
                 isActiveLearningMode: false,
@@ -209,15 +207,6 @@ describe('usePredictionsRoiQuery', () => {
                     expect.anything()
                 );
             });
-
-            expect(mockedInferenceService.getExplanations).toHaveBeenLastCalledWith(
-                datasetIdentifier,
-                selectedMediaItem,
-                taskId,
-                selectedInput,
-                // AbortController
-                expect.anything()
-            );
         });
 
         it('rejected requests are handle as empty', async () => {
@@ -226,7 +215,6 @@ describe('usePredictionsRoiQuery', () => {
                 currentMode: ANNOTATOR_MODE.PREDICTION,
             }));
             jest.mocked(mockedInferenceService.getPredictions).mockRejectedValue('test error');
-            jest.mocked(mockedInferenceService.getExplanations).mockRejectedValue('test error');
 
             const { result } = renderPredictionsRoiQuery({
                 selectedInput,

web_ui/src/pages/annotator/providers/selected-media-item-provider/use-predictions-query.hook.test.tsx

@@ -49,7 +49,6 @@ describe('usePredictionsQuery', (): void => {
 
     beforeEach(() => {
         mockedInferenceService.getPredictions = jest.fn();
-        mockedInferenceService.getExplanations = jest.fn();
     });
 
     it('task-chain projects call the service with taskId ', async (): Promise<void> => {
@@ -69,15 +68,6 @@ describe('usePredictionsQuery', (): void => {
             expect(result.current).toBeDefined();
         });
 
-        expect(mockedInferenceService.getExplanations).toHaveBeenCalledWith(
-            datasetIdentifier,
-            mediaItem,
-            taskId,
-            undefined,
-
-            // AbortController
-            expect.anything()
-        );
         expect(mockedInferenceService.getPredictions).toHaveBeenCalledWith(
             datasetIdentifier,
             coreLabels,
@@ -97,7 +87,6 @@ describe('usePredictionsQuery', (): void => {
         });
 
         await waitFor(() => {
-            expect(mockedInferenceService.getExplanations).not.toHaveBeenCalled();
             expect(mockedInferenceService.getPredictions).toHaveBeenCalledWith(
                 datasetIdentifier,
                 coreLabels,
@@ -111,7 +100,7 @@ describe('usePredictionsQuery', (): void => {
         });
     });
 
-    it('PredictionMode LATEST is handle as PredictionCache.ALWAYS', async (): Promise<void> => {
+    it('PredictionMode LATEST is handled as PredictionCache.ALWAYS', async (): Promise<void> => {
         const onSuccess = jest.fn();
 
         renderHookWithProviders(
@@ -126,7 +115,6 @@ describe('usePredictionsQuery', (): void => {
             expect(onSuccess).toHaveBeenCalled();
         });
 
-        expect(mockedInferenceService.getExplanations).not.toHaveBeenCalled();
         expect(mockedInferenceService.getPredictions).toHaveBeenCalledWith(
             datasetIdentifier,
             coreLabels,
@@ -139,24 +127,19 @@ describe('usePredictionsQuery', (): void => {
         );
     });
 
-    it('PredictionMode.ONLINE is sent as PredictionCache.NEVER, getExplanations is called', async (): Promise<void> => {
+    it('PredictionMode.ONLINE is handled as PredictionCache.NEVER', async (): Promise<void> => {
+        const onSuccess = jest.fn();
+
         renderHookWithProviders(
-            () => usePredictionsQuery({ ...predictionArguments, predictionId: PredictionMode.ONLINE }),
+            () => usePredictionsQuery({ ...predictionArguments, onSuccess, predictionId: PredictionMode.ONLINE }),
             {
                 wrapper,
                 providerProps: { ...initialProps },
             }
         );
 
         await waitFor(() => {
-            expect(mockedInferenceService.getExplanations).toHaveBeenCalledWith(
-                datasetIdentifier,
-                mediaItem,
-                undefined,
-                undefined,
-                // AbortController
-                expect.anything()
-            );
+            expect(onSuccess).toHaveBeenCalled();
         });
 
         expect(mockedInferenceService.getPredictions).toHaveBeenCalledWith(
@@ -170,25 +153,4 @@ describe('usePredictionsQuery', (): void => {
             expect.anything()
         );
     });
-
-    it('does not call "getExplanations" for keypoint detection projects', async (): Promise<void> => {
-        const mockedProjectService = createInMemoryProjectService();
-
-        mockedProjectService.getProject = async () =>
-            getMockedProject({ tasks: [getMockedTask({ domain: DOMAIN.KEYPOINT_DETECTION })] });
-
-        renderHookWithProviders(
-            () => usePredictionsQuery({ ...predictionArguments, predictionId: PredictionMode.ONLINE }),
-            {
-                wrapper,
-                providerProps: { ...initialProps, projectService: mockedProjectService },
-            }
-        );
-
-        await waitFor(() => {
-            expect(mockedInferenceService.getPredictions).toHaveBeenCalled();
-        });
-
-        expect(mockedInferenceService.getExplanations).not.toHaveBeenCalled();
-    });
 });

web_ui/src/pages/annotator/providers/selected-media-item-provider/use-predictions-query.hook.ts

@@ -84,7 +84,6 @@ export const usePredictionsQuery = ({
                 };
             }
 
-            // Only fetch predictions here
             const annotations = await inferenceService.getPredictions(
                 datasetIdentifier,
                 coreLabels,