customSchemaInfo.setNamespace("urn:custom");
final Map<String, SchemaInfo> ns2schema = new HashMap<String, SchemaInfo>();
ns2schema.put(defaultSchemaInfo.getNamespace(), defaultSchemaInfo);
ns2schema.put(customSchemaInfo.getNamespace(), customSchemaInfo);
WsdlInfo defaultWsdlInfo = new WsdlInfo();
defaultWsdlInfo.setTargetNamespace("urn:default");
WsdlInfo customWsdlInfo = new WsdlInfo();
customWsdlInfo.setTargetNamespace("urn:custom");
final Map<String, WsdlInfo> ns2wsdl = new HashMap<String, WsdlInfo>();
ns2wsdl.put(defaultWsdlInfo.getTargetNamespace(), defaultWsdlInfo);
ns2wsdl.put(customWsdlInfo.getTargetNamespace(), customWsdlInfo);
final Map<String, String> ns2prefix = new HashMap<String, String>();
ns2prefix.put("urn:default", "default1");
ns2prefix.put("urn:custom", "default2");
final EnunciateFreemarkerModel model = new EnunciateFreemarkerModel() {
@Override
public Map<String, String> getNamespacesToPrefixes() {
return ns2prefix;
}
@Override
public Map<String, SchemaInfo> getNamespacesToSchemas() {
return ns2schema;
}
@Override
public Map<String, WsdlInfo> getNamespacesToWSDLs() {
return ns2wsdl;
}
};
final ArrayList<URL> processList = new ArrayList<URL>(Arrays.asList(XMLDeploymentModule.class.getResource("xml.fmt")));
XMLDeploymentModule module = new XMLDeploymentModule() {
@Override
public EnunciateFreemarkerModel getModel() {
return model;
}
@Override
protected boolean isUpToDate(File artifactDir) {
return false;
}
@Override
public void processTemplate(URL templateURL, Object model) {
processList.remove(templateURL);
}
};
SchemaConfig customSchemaConfig = new SchemaConfig();
customSchemaConfig.setFile("custom.xsd");
customSchemaConfig.setLocation("urn:custom.xsd");
customSchemaConfig.setNamespace("urn:custom");
module.addSchemaConfig(customSchemaConfig);
WsdlConfig wsdlConfig = new WsdlConfig();
wsdlConfig.setNamespace("urn:custom");
wsdlConfig.setFile("custom.wsdl");
module.addWsdlConfig(wsdlConfig);
Enunciate enunciate = new Enunciate(new String[0]);
File generateDir = enunciate.createTempDir();
enunciate.setGenerateDir(generateDir);
module.init(enunciate);
module.initModel(model);
module.doFreemarkerGenerate();
assertTrue("Not all templates were processed. Unprocessed templates: " + processList.toString(), processList.isEmpty());
assertEquals("default1.xsd", defaultSchemaInfo.getProperty("filename"));
assertEquals("default1.xsd", defaultSchemaInfo.getProperty("location"));
assertEquals("custom.xsd", customSchemaInfo.getProperty("filename"));
assertEquals("urn:custom.xsd", customSchemaInfo.getProperty("location"));
assertEquals("default1.wsdl", defaultWsdlInfo.getProperty("filename"));
assertEquals("custom.wsdl", customWsdlInfo.getProperty("filename"));
File artifactDir = new File(generateDir, "xml");
assertEquals(new File(artifactDir, "default1.xsd"), defaultSchemaInfo.getProperty("file"));
assertEquals(new File(artifactDir, "default1.wsdl"), defaultWsdlInfo.getProperty("file"));
assertEquals(new File(artifactDir, "custom.xsd"), customSchemaInfo.getProperty("file"));
assertEquals(new File(artifactDir, "custom.wsdl"), customWsdlInfo.getProperty("file"));
}