org.drugis.addis.util.jaxb.JAXBHandlerTest.java Source code

Introduction

Here is the source code for org.drugis.addis.util.jaxb.JAXBHandlerTest.java

Source

/*
 * This file is part of ADDIS (Aggregate Data Drug Information System).
 * ADDIS is distributed from http://drugis.org/.
 * Copyright  2009 Gert van Valkenhoef, Tommi Tervonen.
 * Copyright  2010 Gert van Valkenhoef, Tommi Tervonen, Tijs Zwinkels,
 * Maarten Jacobs, Hanno Koeslag, Florin Schimbinschi, Ahmad Kamal, Daniel
 * Reid.
 * Copyright  2011 Gert van Valkenhoef, Ahmad Kamal, Daniel Reid, Florin
 * Schimbinschi.
 * Copyright  2012 Gert van Valkenhoef, Daniel Reid, Jol Kuiper, Wouter
 * Reckman.
 * Copyright  2013 Gert van Valkenhoef, Jol Kuiper.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.drugis.addis.util.jaxb;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.FilterOutputStream;
import java.io.IOException;
import java.util.List;

import javax.xml.bind.JAXBException;
import javax.xml.transform.TransformerException;

import org.apache.commons.lang.StringUtils;
import org.drugis.addis.entities.DomainImpl;
import org.drugis.addis.entities.Drug;
import org.drugis.addis.entities.data.AddisData;
import org.drugis.addis.gui.GUIFactory;
import org.drugis.addis.gui.Main;
import org.drugis.addis.util.jaxb.JAXBConvertor.ConversionException;
import org.drugis.addis.util.jaxb.JAXBHandler.XmlFormatType;
import org.junit.Test;
import org.xml.sax.SAXException;

public class JAXBHandlerTest {
    private static final String V3_XML = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>\n"
            + "<addis-data xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"\n"
            + "            xsi:noNamespaceSchemaLocation=\"http://drugis.org/files/addis-3.xsd\">\n"
            + "</addis-data>\n";

    private static final String V2_XML = "<?xml version=\"1.0\" encoding=\"utf-8\"?>\n"
            + "<addis-data xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:noNamespaceSchemaLocation=\"http://drugis.org/files/addis-2.xsd\">\n"
            + "</addis-data>\n";

    private static final String LEGACY_XML = "<?xml version=\"1.0\" ?>\n" + "<addis-data>\n" + "</addis-data>\n";

    private static final String OTHER_XML = "<?xml version=\"1.0\" ?>\n" + "<oranges>\n" + "</oranges>\n";

    @Test
    public void testUnmarshallMarshallXmlCompare()
            throws JAXBException, SAXException, TransformerException, IOException {
        // read xml file
        AddisData data = JAXBHandler.unmarshallAddisData(
                JAXBConvertor.transformToLatest(JAXBHandlerTest.class.getResourceAsStream("schema_test.xml"), 1));

        // write out
        String testFile = "jaxb_marshall_test.xml";
        JAXBHandler.marshallAddisData(data, new FileOutputStream(testFile));

        // read back generated xml
        AddisData data_clone = JAXBHandler.unmarshallAddisData(new FileInputStream(testFile));

        // compare
        assertEquals(data, data_clone);

        File temp = new File(testFile);
        temp.delete();
    }

    @Test
    public void doNotSerializeInvalidCharsTest() throws JAXBException, ConversionException, IOException {
        DomainImpl domainImpl = new DomainImpl();
        GUIFactory.suppressErrors(true);

        List<Character> invalidXMLChars = Main.XMLStreamFilter.getCharacters();
        Drug drug = new Drug(StringUtils.join(invalidXMLChars, " "), "#");
        domainImpl.getDrugs().add(drug);
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        FilterOutputStream os = new Main.XMLStreamFilter(out);

        JAXBHandler.marshallAddisData(JAXBConvertor.convertDomainToAddisData(domainImpl), os);

        os.close();

        JAXBHandler.unmarshallAddisData(new ByteArrayInputStream(out.toByteArray()));
        GUIFactory.suppressErrors(false);
    }

    @Test
    public void determineXmlVersionTest() throws IOException {
        ByteArrayInputStream emptyInput = new ByteArrayInputStream("".getBytes());
        XmlFormatType emptyVersion = JAXBHandler.determineXmlType(emptyInput);
        assertFalse(emptyVersion.isValid());
        assertFalse(emptyVersion.isLegacy());
        ByteArrayInputStream v3input = new ByteArrayInputStream(V3_XML.getBytes());
        XmlFormatType v3version = JAXBHandler.determineXmlType(v3input);
        assertTrue(v3version.isValid());
        assertFalse(v3version.isLegacy());
        assertEquals(3, v3version.getVersion());

        ByteArrayInputStream v2input = new ByteArrayInputStream(V2_XML.getBytes());
        XmlFormatType v2version = JAXBHandler.determineXmlType(v2input);
        assertTrue(v2version.isValid());
        assertFalse(v2version.isLegacy());
        assertEquals(2, v2version.getVersion());

        ByteArrayInputStream v0input = new ByteArrayInputStream(LEGACY_XML.getBytes());
        XmlFormatType v0version = JAXBHandler.determineXmlType(v0input);
        assertTrue(v0version.isValid());
        assertTrue(v0version.isLegacy());

        ByteArrayInputStream otherInput = new ByteArrayInputStream(OTHER_XML.getBytes());
        XmlFormatType otherVersion = JAXBHandler.determineXmlType(otherInput);
        assertFalse(otherVersion.isValid());
    }

}