Add service tests

service/src/test/java/org/cbioportal/genome_nexus/service/internal/MutationAssessorServiceTest.java

@@ -0,0 +1,83 @@
+package org.cbioportal.genome_nexus.service.internal;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertThrows;
+import static org.mockito.Mockito.doReturn;
+
+import java.io.IOException;
+import java.util.Map;
+import java.util.Optional;
+
+import org.cbioportal.genome_nexus.model.MutationAssessor;
+import org.cbioportal.genome_nexus.model.VariantAnnotation;
+import org.cbioportal.genome_nexus.persistence.MutationAssessorRepository;
+import org.cbioportal.genome_nexus.service.exception.ResourceMappingException;
+import org.cbioportal.genome_nexus.service.exception.MutationAssessorNotFoundException;
+import org.cbioportal.genome_nexus.service.mock.MutationAssessorMockData;
+import org.cbioportal.genome_nexus.service.mock.VariantAnnotationMockData;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.mockito.Spy;
+import org.mockito.junit.MockitoJUnitRunner;
+
+@RunWith(MockitoJUnitRunner.class)
+public class MutationAssessorServiceTest
+{
+    @Spy
+    @InjectMocks
+    private MutationAssessorServiceImpl service;
+
+    @Mock
+    MutationAssessorRepository mutationAssessorRepository;
+
+    private MutationAssessorMockData mutationAssessorMockData = new MutationAssessorMockData();
+    private VariantAnnotationMockData variantAnnotationMockData = new VariantAnnotationMockData();
+
+    @Test
+    public void getMutationAssessorByMutationAssessorVariant()
+        throws ResourceMappingException, MutationAssessorNotFoundException
+    {
+        Map<String, MutationAssessor> mockData = this.mutationAssessorMockData.generateData();
+
+        // mock methods in order to prevent hitting the live mutation assessor web API
+        Mockito.when(mutationAssessorRepository.findById("P15056,p.V600E")).thenReturn(Optional.of(mockData.get("P15056,p.V600E")));
+        Mockito.when(mutationAssessorRepository.findById("P01116,p.G12C")).thenReturn(Optional.of(mockData.get("P01116,p.G12C")));
+        Mockito.when(mutationAssessorRepository.findById("INVALID")).thenReturn(Optional.empty());
+
+        MutationAssessor mutationAssessor1 = service.getMutationAssessorByMutationAssessorVariant("P15056,p.V600E", null);
+        assertEquals(mutationAssessor1.getF_impact(), mockData.get("P15056,p.V600E").getF_impact());
+
+        MutationAssessor mutationAssessor2 = service.getMutationAssessorByMutationAssessorVariant("P01116,p.G12C", null);
+        assertEquals(mutationAssessor2.getF_impact(), mockData.get("P01116,p.G12C").getF_impact());
+
+        VariantAnnotation invalidVariantAnnotation = new VariantAnnotation();
+        invalidVariantAnnotation.setVariant("INVALID");
+
+        Exception exception = assertThrows(Exception.class, () -> {
+            service.getMutationAssessorByMutationAssessorVariant("INVALID", invalidVariantAnnotation);
+        });
+        assertEquals("Mutation Assessor annotation not found for variant: INVALID", exception.getMessage());        
+    }
+
+    @Test
+    public void getMutationAssessorByVariantAnnotation()
+        throws ResourceMappingException, MutationAssessorNotFoundException,
+        IOException
+    {
+        Map<String, MutationAssessor> maMockData = this.mutationAssessorMockData.generateData();
+        Map<String, VariantAnnotation> variantMockData = this.variantAnnotationMockData.generateData();
+
+        // mock methods in order to prevent hitting the live mutation assessor web API
+        doReturn(maMockData.get("P15056,p.V600E")).when(service).getMutationAssessorByMutationAssessorVariant("P15056,p.V600E", null);
+        doReturn(maMockData.get("P01116,p.G12C")).when(service).getMutationAssessorByMutationAssessorVariant("P01116,p.G12C", null);
+
+        MutationAssessor mutationAssessor1 = service.getMutationAssessor(variantMockData.get("7:g.140453136A>T"));
+        assertEquals(mutationAssessor1.getF_score(), maMockData.get("P15056,p.V600E").getF_score());
+
+        MutationAssessor mutationAssessor2 = service.getMutationAssessor(variantMockData.get("12:g.25398285C>A"));
+        assertEquals(mutationAssessor2.getF_score(), maMockData.get("P01116,p.G12C").getF_score());
+    }
+}