fix study view service test

cBioPortal · Oct 6, 2023 · 4b9d713 · 4b9d713
1 parent 5f1ab78
commit 4b9d713
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/service/src/test/java/org/cbioportal/service/impl/StudyViewServiceImplTest.java b/service/src/test/java/org/cbioportal/service/impl/StudyViewServiceImplTest.java
@@ -21,6 +21,7 @@
 @RunWith(MockitoJUnitRunner.class)
 public class StudyViewServiceImplTest extends BaseServiceImplTest {
 
+    @Spy
     @InjectMocks
     private StudyViewServiceImpl studyViewService;
     @Mock
@@ -251,9 +252,8 @@ public void getMutationCountsByGeneSpecific() {
         genomicDataCounts.add(genomicDataCount2);
         genomicDataCountItem.setCounts(genomicDataCounts);
 
-        Mockito.when(studyViewService.getMutationTypeCountsByGeneSpecific(
-            anyList(), anyList(), anyList()
-        )).thenReturn(Collections.singletonList(genomicDataCountItem));
+        Mockito.doReturn(Collections.singletonList(genomicDataCountItem))
+            .when(studyViewService).getMutationTypeCountsByGeneSpecific(anyList(), anyList(), anyList());
 
         List<GenomicDataCountItem> result = studyViewService.getMutationCountsByGeneSpecific(
             studyIds, sampleIds, genomicDataFilters, alterationFilter, "SUMMARY");
@@ -263,7 +263,7 @@ public void getMutationCountsByGeneSpecific() {
         result = studyViewService.getMutationCountsByGeneSpecific(
             studyIds, sampleIds, genomicDataFilters, alterationFilter, "DETAILED");
         Assert.assertEquals(1, result.size());
-        Assert.assertTrue(3 <= result.get(0).getCounts().size());
+        Assert.assertEquals(4, result.get(0).getCounts().size());
     }
 
     @Test