@Before public void setUp() throws Exception { wrappedWriter = mock(XMLEventWriter.class); writer = new UnclosedElementCollectingEventWriter(wrappedWriter); }
@Test public void testMultipleUnclosedElements() throws Exception { writer.add(eventFactory.createStartElement(elementA, null, null)); writer.add(eventFactory.createStartElement(elementB, null, null)); writer.add(eventFactory.createStartElement(elementC, null, null)); writer.add(eventFactory.createEndElement(elementC, null)); assertEquals(2, writer.getUnclosedElements().size()); assertEquals(elementA, writer.getUnclosedElements().get(0)); assertEquals(elementB, writer.getUnclosedElements().get(1)); verify(wrappedWriter, Mockito.times(4)).add(Mockito.any(XMLEvent.class)); }
try { if (headerCallback != null) { UnclosedElementCollectingEventWriter headerCallbackWriter = new UnclosedElementCollectingEventWriter(delegateEventWriter); headerCallback.write(headerCallbackWriter); unclosedHeaderCallbackElements = headerCallbackWriter.getUnclosedElements();
try { if (headerCallback != null) { UnclosedElementCollectingEventWriter headerCallbackWriter = new UnclosedElementCollectingEventWriter(delegateEventWriter); headerCallback.write(headerCallbackWriter); unclosedHeaderCallbackElements = headerCallbackWriter.getUnclosedElements();
@Test public void testMultipleIdenticalUnclosedElement() throws Exception { writer.add(eventFactory.createStartElement(elementA, null, null)); writer.add(eventFactory.createStartElement(elementA, null, null)); assertEquals(2, writer.getUnclosedElements().size()); assertEquals(elementA, writer.getUnclosedElements().get(0)); assertEquals(elementA, writer.getUnclosedElements().get(1)); verify(wrappedWriter, Mockito.times(2)).add(Mockito.any(XMLEvent.class)); }
@Test public void testSingleUnclosedElement() throws Exception { writer.add(eventFactory.createStartElement(elementA, null, null)); writer.add(eventFactory.createEndElement(elementA, null)); writer.add(eventFactory.createStartElement(elementB, null, null)); assertEquals(1, writer.getUnclosedElements().size()); assertEquals(elementB, writer.getUnclosedElements().get(0)); verify(wrappedWriter, Mockito.times(3)).add(Mockito.any(XMLEvent.class)); }
@Test public void testNoUnclosedElements() throws Exception { writer.add(eventFactory.createStartElement(elementA, null, null)); writer.add(eventFactory.createEndElement(elementA, null)); assertEquals(0, writer.getUnclosedElements().size()); verify(wrappedWriter, Mockito.times(2)).add(Mockito.any(XMLEvent.class)); }