@Test
public void testGetSusceptibilityDiseaseDataAssociatedWithGeneId() {

    Disease disease = Disease.builder()
            .diseaseId("OMIM:234567")
            .diseaseName("Test susceptibility disease association")
            .diseaseType(Disease.DiseaseType.SUSCEPTIBILITY)
            .inheritanceMode(InheritanceMode.UNKNOWN)
            .associatedGeneId(4444)
            .associatedGeneSymbol("GENE4")
            .phenotypeIds(ImmutableList.of("HP:0000002"))
            .build();
    List<Disease> expected = Lists.newArrayList(disease) ;
    assertThat(instance.getDiseaseDataAssociatedWithGeneId(4444), equalTo(expected));
}