package org.codehaus.xfire.aegis.stax;import java.io.ByteArrayOutputStream;import java.io.File;import java.io.StringReader;import org.codehaus.xfire.aegis.MessageWriter;import org.codehaus.xfire.aegis.jdom.JDOMWriter;import org.codehaus.xfire.test.AbstractXFireTest;import org.codehaus.xfire.util.jdom.StaxBuilder;import org.jdom.Document;import org.jdom.Element;/** * @author <a href="mailto:dan@envoisolutions.com">Dan Diephouse</a> * @since Nov 4, 2004 */public class WriterTest    extends AbstractXFireTest{    File output;        protected void setUp()        throws Exception    {        super.setUp();                output = File.createTempFile("writetest", ".xml");    }    protected void tearDown()        throws Exception    {        if (output.exists())            output.delete();                super.tearDown();    }    public void testLiteral()        throws Exception    {        ByteArrayOutputStream bos = new ByteArrayOutputStream();        ElementWriter writer = new ElementWriter(bos, "root", "urn:test");        write(writer);        writer.flush();        bos.close();                System.out.println(bos.toString());        StaxBuilder builder = new StaxBuilder();        Document doc = builder.build(new StringReader(bos.toString()));                testWrite(doc);    }        public void testYOM()        throws Exception    {        Document doc = new Document(new Element("root", "urn:test"));                write(new JDOMWriter(doc.getRootElement()));                testWrite(doc);    }        public void write(MessageWriter writer)    {        MessageWriter nons = writer.getElementWriter("nons");        nons.writeValue("nons");        nons.close();                MessageWriter intval = writer.getElementWriter("int");        intval.writeValueAsInt(new Integer(10000));        intval.close();        MessageWriter child1 = writer.getElementWriter("child1", "urn:child1");        MessageWriter att1 = child1.getAttributeWriter("att1");        att1.writeValue("att1");        att1.close();        MessageWriter att2 = child1.getAttributeWriter("att2", "");        att2.writeValue("att2");        att2.close();        MessageWriter att3 = child1.getAttributeWriter("att3", "urn:att3");        att3.writeValue("att3");        att3.close();        MessageWriter att4 = child1.getAttributeWriter("att4", null);        att4.writeValue("att4");        att4.close();                child1.close();                writer.close();    }        public void testWrite(Document doc) throws Exception    {        addNamespace("t", "urn:test");        addNamespace("c", "urn:child1");        addNamespace("a", "urn:att3");                assertValid("/t:root/t:nons[text()='nons']", doc);        assertValid("/t:root/t:int[text()='10000']", doc);        assertValid("/t:root/c:child1", doc);        assertValid("/t:root/c:child1[@c:att1]", doc);        assertValid("/t:root/c:child1[@att2]", doc);        assertValid("/t:root/c:child1[@a:att3]", doc);        assertValid("/t:root/c:child1[@att4]", doc);    }}