    }

    @Test
    public void shouldExtractEntityConfigsWithOptionalRefsFromChoice() {
        //change to the OptionalRef xsd
        didSchemaParser.setExtensionXsdLocation("classpath:test-schema/OptionalRefChoice-Extension.xsd");
        didSchemaParser.setup();

        Map<String, DidEntityConfig> entityConfigs = didSchemaParser.getEntityConfigs();

        Assert.assertEquals("Should extract 1 entity config for the 1 complexType containing a did ref (OptionalRefExample)", 1, entityConfigs.size());

        //check the entity configs extracted are for the correct types
        Assert.assertTrue(entityConfigs.containsKey(OPTIONAL_REF_TYPE));
