update StudyViewFilterHelper build method use in tests

cBioPortal · Jan 10, 2025 · e34a9d9 · e34a9d9
1 parent 98386ce
commit e34a9d9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/test/java/org/cbioportal/persistence/mybatisclickhouse/CNAGenesTest.java b/src/test/java/org/cbioportal/persistence/mybatisclickhouse/CNAGenesTest.java
@@ -79,7 +79,7 @@ public void getCnaGenesWithAlterationFilter() {
         onlyDriverFilter.setIncludeVUS(false);
         onlyDriverFilter.setIncludeUnknownOncogenicity(false);
 
-        var alterationCountByGenes1 = studyViewMapper.getCnaGenes(StudyViewFilterHelper.build(studyViewFilter, null, null),
+        var alterationCountByGenes1 = studyViewMapper.getCnaGenes(StudyViewFilterHelper.build(studyViewFilter, null, null, studyViewFilter.getStudyIds()),
             AlterationFilterHelper.build(onlyDriverFilter));
         assertEquals(0, alterationCountByGenes1.size());
 
@@ -88,7 +88,7 @@ public void getCnaGenesWithAlterationFilter() {
         onlyVUSFilter.setIncludeVUS(true);
         onlyVUSFilter.setIncludeUnknownOncogenicity(false);
 
-        var alterationCountByGenes2 = studyViewMapper.getCnaGenes(StudyViewFilterHelper.build(studyViewFilter, null, null),
+        var alterationCountByGenes2 = studyViewMapper.getCnaGenes(StudyViewFilterHelper.build(studyViewFilter, null, null, studyViewFilter.getStudyIds()),
             AlterationFilterHelper.build(onlyVUSFilter));
         assertEquals(0, alterationCountByGenes2.size());
 
@@ -97,7 +97,7 @@ public void getCnaGenesWithAlterationFilter() {
         onlyUnknownOncogenicityFilter.setIncludeVUS(false);
         onlyUnknownOncogenicityFilter.setIncludeUnknownOncogenicity(true);
 
-        var alterationCountByGenes3 = studyViewMapper.getCnaGenes(StudyViewFilterHelper.build(studyViewFilter, null, null),
+        var alterationCountByGenes3 = studyViewMapper.getCnaGenes(StudyViewFilterHelper.build(studyViewFilter, null, null, studyViewFilter.getStudyIds()),
             AlterationFilterHelper.build(onlyUnknownOncogenicityFilter));
         assertEquals(3, alterationCountByGenes3.size());
 

diff --git a/src/test/java/org/cbioportal/persistence/mybatisclickhouse/MutatedGenesTest.java b/src/test/java/org/cbioportal/persistence/mybatisclickhouse/MutatedGenesTest.java
@@ -92,7 +92,7 @@ public void getMutatedGenesWithAlterationFilter() {
         onlyDriverFilter.setIncludeVUS(false);
         onlyDriverFilter.setIncludeUnknownOncogenicity(false);
 
-        var alterationCountByGenes3 = studyViewMapper.getMutatedGenes(StudyViewFilterHelper.build(studyViewFilter, null, null),
+        var alterationCountByGenes3 = studyViewMapper.getMutatedGenes(StudyViewFilterHelper.build(studyViewFilter, null, null, studyViewFilter.getStudyIds()),
             AlterationFilterHelper.build(onlyDriverFilter));
         assertEquals(2, alterationCountByGenes3.size());
 
@@ -114,7 +114,7 @@ public void getMutatedGenesWithAlterationFilter() {
         onlyVUSFilter.setIncludeVUS(true);
         onlyVUSFilter.setIncludeUnknownOncogenicity(false);
 
-        var alterationCountByGenes4 = studyViewMapper.getMutatedGenes(StudyViewFilterHelper.build(studyViewFilter, null, null),
+        var alterationCountByGenes4 = studyViewMapper.getMutatedGenes(StudyViewFilterHelper.build(studyViewFilter, null, null, studyViewFilter.getStudyIds()),
             AlterationFilterHelper.build(onlyVUSFilter));
         assertEquals(3, alterationCountByGenes4.size());
 
@@ -136,7 +136,7 @@ public void getMutatedGenesWithAlterationFilter() {
         onlyUnknownOncogenicityFilter.setIncludeVUS(false);
         onlyUnknownOncogenicityFilter.setIncludeUnknownOncogenicity(true);
 
-        var alterationCountByGenes5 = studyViewMapper.getMutatedGenes(StudyViewFilterHelper.build(studyViewFilter, null, null),
+        var alterationCountByGenes5 = studyViewMapper.getMutatedGenes(StudyViewFilterHelper.build(studyViewFilter, null, null, studyViewFilter.getStudyIds()),
             AlterationFilterHelper.build(onlyUnknownOncogenicityFilter));
         assertEquals(0, alterationCountByGenes5.size());
     }