test fixes

EBIvariation · Sep 6, 2024 · 33d7664 · 33d7664
1 parent 0934216
commit 33d7664
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/eva-server/src/test/java/uk/ac/ebi/eva/server/ws/GeneWSServerV2Test.java b/eva-server/src/test/java/uk/ac/ebi/eva/server/ws/GeneWSServerV2Test.java
@@ -120,6 +120,8 @@ public void setUp() throws Exception {
                 .willReturn(Collections.singletonList(variantEntity));
         given(contigAliasService.getVariantsWithTranslatedContig(Arrays.asList(variantEntity, variantEntity), null))
                 .willReturn(Arrays.asList(variantEntity, variantEntity));
+        given(contigAliasService.translateContigFromInsdc(variantEntity.getChromosome(), null))
+                .willReturn(variantEntity.getChromosome());
     }
 
     @Test
@@ -132,11 +134,8 @@ public void testGetVariantsByExistingGene() throws URISyntaxException {
 
     @Test
     public void testGetVariantsByExistingGeneWithTranslatedContig() throws URISyntaxException {
-        VariantWithSamplesAndAnnotation tranlatedVariant = new VariantWithSamplesAndAnnotation("chr20", 1000, 1005,
-                "A", "C", MAIN_ID);
-        given(contigAliasService.getVariantsWithTranslatedContig(Collections.singletonList(variantEntity),
-                ContigNamingConvention.ENA_SEQUENCE_NAME))
-                .willReturn(Collections.singletonList(tranlatedVariant));
+        given(contigAliasService.translateContigFromInsdc(variantEntity.getChromosome(), ContigNamingConvention.ENA_SEQUENCE_NAME))
+                .willReturn("chr20");
 
         Variant variant = testGetVariantsByGeneHelper("ENSG00000227232", 1, HttpStatus.OK,
                 ContigNamingConvention.ENA_SEQUENCE_NAME).get(0);