source: src/test/java/negotiator/xml/XmlWriteStreamTest.java

package negotiator.xml;

import static org.junit.Assert.assertEquals;

import java.io.ByteArrayOutputStream;
import java.util.LinkedHashMap;
import java.util.Map;

import javax.xml.stream.XMLStreamException;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import genius.core.xml.Key;
import genius.core.xml.XmlWriteStream;

/**
 * Test (partial) XML stream writing
 */
public class XmlWriteStreamTest {
        private XmlWriteStream stream;
        private ByteArrayOutputStream output;

        @Before
        public void before() throws XMLStreamException {
                output = new ByteArrayOutputStream();
                stream = new XmlWriteStream(output, "test");
        }

        @After
        public void after() {
                System.out.println(output);
        }

        @Test
        public void TestBasicWrite() throws XMLStreamException {
                Map<Object, Object> result = basicResult();
                stream.write("Outcome", result);
                stream.close();

                assertEquals(
                                "<?xml version=\"1.0\" ?><test>\n<Outcome currentTime=\"2016-11-15 14:17:17\" timeOfAgreement=\"0.6666666666666666\" lastAction=\"Accept\" v=\"1\" v=\"2\">\n</Outcome></test>",
                                output.toString());
                System.out.println(output);
        }

        @Test
        public void TestFullWrite() throws XMLStreamException {
                Map<Object, Object> result = basicResult();

                result.put(new Key("resultOfAgent"), agentResult("A"));
                result.put(new Key("resultOfAgent"), agentResult("B"));

                stream.write("Outcome", result);
                stream.close();

                assertEquals(
                                "<?xml version=\"1.0\" ?><test>\n<Outcome currentTime=\"2016-11-15 14:17:17\" timeOfAgreement=\"0.6666666666666666\" lastAction=\"Accept\" v=\"1\" v=\"2\">\n<resultOfAgent agent=\"A\" agentName=\"QAgent\">\n</resultOfAgent>\n<resultOfAgent agent=\"B\" agentName=\"QAgent\">\n</resultOfAgent>\n</Outcome></test>",
                                output.toString());
        }

        private Map<Object, Object> basicResult() {
                // use linkedhashmap because the order is important for our exact string
                // assertEquals...
                Map<Object, Object> outcome = new LinkedHashMap<>();
                outcome.put("currentTime", "2016-11-15 14:17:17");
                outcome.put("timeOfAgreement", "0.6666666666666666");
                outcome.put("lastAction", "Accept");
                // check repeated values.
                outcome.put(new Key("v"), 1);
                outcome.put(new Key("v"), 2);
                return outcome;
        }

        private Map<Object, Object> agentResult(String name) {
                Map<Object, Object> outcome = new LinkedHashMap<>();
                outcome.put(new Key("agent"), name);
                outcome.put(new Key("agentName"), "QAgent");
                return outcome;
        }

}