package org.infinispan.server.commons.subsystem;
import java.io.IOException;
import java.io.StringReader;
import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamConstants;
import javax.xml.stream.XMLStreamReader;
import org.jboss.as.controller.Extension;
import org.jboss.as.controller.RunningMode;
import org.jboss.as.subsystem.test.AbstractSubsystemBaseTest;
import org.jboss.as.subsystem.test.AdditionalInitialization;
import org.jboss.as.subsystem.test.ModelDescriptionValidator.ValidationConfiguration;
public abstract class ClusteringSubsystemTest extends AbstractSubsystemBaseTest {
private final String path;
protected ClusteringSubsystemTest(String name, Extension extension, String path) {
super(name, extension);
this.path = path;
}
@Override
protected String getSubsystemXml() throws IOException {
return readResource(path);
}
/**
* Need to override the XML comparison in the case where the input xsd and the output xsd differ.
*
* @param configId the id of the xml configuration
* @param original the original subsystem xml
* @param marshalled the marshalled subsystem xml
* @throws Exception
*/
@Override
protected void compareXml(String configId, final String original, final String marshalled) throws Exception {
final XMLStreamReader originalReader = XMLInputFactory.newInstance().createXMLStreamReader(new StringReader(original));
final XMLStreamReader marshalledReader = XMLInputFactory.newInstance().createXMLStreamReader(new StringReader(marshalled));
String originalNS = null;
if (originalReader.next() == XMLStreamConstants.START_ELEMENT) {
originalNS = originalReader.getNamespaceURI();
}
String marshalledNS = null;
if (marshalledReader.next() == XMLStreamConstants.START_ELEMENT) {
marshalledNS = marshalledReader.getNamespaceURI();
}
// only compare if namespace URIs are the same
if (originalNS.equals(marshalledNS)) {
compareXml(configId, original, marshalled, true);
}
}
@Override
protected AdditionalInitialization createAdditionalInitialization() {
return new AdditionalInitialization() {
@Override
protected RunningMode getRunningMode() {
return RunningMode.ADMIN_ONLY;
}
@Override
protected ValidationConfiguration getModelValidationConfiguration() {
return ClusteringSubsystemTest.this.getModelValidationConfiguration();
}
};
}
protected abstract ValidationConfiguration getModelValidationConfiguration();
}