MET-6199 update unit tests

europeana · Nov 4, 2024 · aa43971 · aa43971
1 parent 89d6f8b
commit aa43971
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/test/java/eu/europeana/metis/sandbox/service/debias/DeBiasStateServiceImplTest.java b/src/test/java/eu/europeana/metis/sandbox/service/debias/DeBiasStateServiceImplTest.java
@@ -185,10 +185,10 @@ void testGetDetectionInfo_ObjectWhenExistsProcessing_expectSuccess() {
     when(recordDeBiasMainRepository.findByRecordIdDatasetId(anyString())).thenReturn(List.of(recordDeBiasMainEntity));
     when(recordDeBiasDetailRepository.findByDebiasIdId(anyLong())).thenReturn(List.of(recordDeBiasDetailEntity));
 
-    when(recordLogRepository.findRecordLogByDatasetIdAndStepAndStatus(datasetId.toString(), Step.DEBIAS, Status.SUCCESS))
-        .thenReturn(Set.of(new RecordLogEntity()));
-    when(recordLogRepository.findRecordLogByDatasetIdAndStep(datasetId.toString(), Step.NORMALIZE))
-        .thenReturn(Set.of(new RecordLogEntity(), new RecordLogEntity()));
+    when(recordLogRepository.getProgressDeBiasCounterByDatasetId(datasetId.toString()))
+        .thenReturn(1);
+    when(recordLogRepository.getTotalDeBiasCounterByDatasetId(datasetId.toString()))
+        .thenReturn(2);
 
     DeBiasReportDto deBiasReportDto = debiasStateServiceImpl.getDeBiasReport(datasetId);