import com.thoughtworks.xstream.io.xml.Dom4JWriter;

import junit.framework.TestCase;

import org.dom4j.io.XMLWriter;

import java.io.StringWriter;


public class Dom4JWriterTest extends TestCase {
    public void testAddAttribute() {
        StringWriter stringWriter = new StringWriter();
        XMLWriter writer = new XMLWriter(stringWriter);
        Dom4JWriter dom4JWriter = new Dom4JWriter(writer);

        dom4JWriter.startNode("root");

        dom4JWriter.startNode("A");
        dom4JWriter.addAttribute("attrA", "valA");
        dom4JWriter.endNode();

        dom4JWriter.startNode("B");
        dom4JWriter.addAttribute("attrB", "valB");
        dom4JWriter.endNode();

        dom4JWriter.startNode("C");
        dom4JWriter.addAttribute("attrC", "valC");
        dom4JWriter.endNode();

        dom4JWriter.endNode();

        assertEquals("<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
            + "<root>"
              + "<A attrA=\"valA\"/>"
              + "<B attrB=\"valB\"/>"
              + "<C attrC=\"valC\"/>"
            + "</root>",
            stringWriter.toString());
    }
}