OMNamespace ns = factory.createOMNamespace("urn://test", "tns");
ByteArrayInputStream bais1 = new ByteArrayInputStream(payload1.getBytes(encoding));
InputStreamDataSource isds1 = new InputStreamDataSource(bais1, encoding);
OMElement parent = factory.createOMElement("root", null);
OMSourcedElement omse = factory.createOMElement(isds1, localName, ns);
parent.addChild(omse);
OMNode firstChild = parent.getFirstOMChild();
assertTrue("Expected OMSourcedElement child", firstChild instanceof OMSourcedElement);
OMSourcedElement child = (OMSourcedElement) firstChild;
assertTrue("OMSourcedElement is expanded. This is unexpected", !child.isExpanded());
assertTrue("OMSourcedElement should be backed by a InputStreamDataSource",
child.getDataSource() instanceof InputStreamDataSource);
// A InputStreamDataSource consumes the backing object when read.
// Thus getting the XMLStreamReader of the ByteArrayDataSource should
// cause expansion of the OMSourcedElement.
XMLStreamReader reader = child.getXMLStreamReader();
reader.next();
assertTrue("OMSourcedElement is not expanded. This is unexpected",
child.isExpanded());
child.detach();
// Reset the tree
isds1 = new InputStreamDataSource(
new ByteArrayInputStream(payload1.getBytes(encoding)),
encoding);
omse = factory.createOMElement(isds1, localName, ns);
parent.addChild(omse);
firstChild = parent.getFirstOMChild();
child = (OMSourcedElement) firstChild;
// Likewise, an InputStreamDataSource consumes the backing object when
// written. Thus serializing the OMSourcedElement should cause the expansion
// of the OMSourcedElement.
ByteArrayOutputStream baos = new ByteArrayOutputStream();
parent.serialize(baos);
String output = baos.toString(encoding);
assertTrue("The payload was not present in the output",
output.indexOf(payload1) > 0);
assertTrue("OMSourcedElement is not expanded. This is unexpected", child.isExpanded());
// Reset the tree
child.detach();
isds1 = new InputStreamDataSource(
new ByteArrayInputStream(payload1.getBytes(encoding)),
encoding);
omse = factory.createOMElement(isds1, localName, ns);
parent.addChild(omse);
firstChild = parent.getFirstOMChild();
child = (OMSourcedElement) firstChild;
// Test getting the raw bytes from the ByteArrayDataSource.
OMDataSourceExt ds = (OMDataSourceExt) child.getDataSource();
byte[] bytes = ds.getXMLBytes(encoding); // Get the bytes as UTF-16
String payload = new String(bytes, encoding);
assertTrue("The obtained bytes did not match the payload",
payload1.equals(payload));
}