org.openmrs.module.dhisreport.api.model.ReportTemplatesTest.java Source code

Introduction

Here is the source code for org.openmrs.module.dhisreport.api.model.ReportTemplatesTest.java

Source

/**
 *  Copyright 2012 Society for Health Information Systems Programmes, India (HISP India)
 *
 *  This file is part of DHIS2 Reporting module.
 *
 *  DHIS2 Reporting module 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.
    
 *  DHIS2 Reporting module 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 DHIS2 Reporting module.  If not, see <http://www.gnu.org/licenses/>.
 *
 **/
package org.openmrs.module.dhisreport.api.model;

import java.util.Collection;
import java.util.List;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.assertFalse;
import org.junit.Test;
import org.springframework.core.io.ClassPathResource;

/**
 * 
 * @author bobj
 */
public class ReportTemplatesTest {
    @Test
    public void unMarshallReportTemplates() throws Exception {
        ClassPathResource resource = new ClassPathResource("templates_ethiopia.xml");
        JAXBContext jaxbContext = JAXBContext.newInstance(ReportTemplates.class);

        Unmarshaller jaxbUnmarshaller = jaxbContext.createUnmarshaller();
        ReportTemplates reportTemplates = (ReportTemplates) jaxbUnmarshaller.unmarshal(resource.getInputStream());
        assertNotNull(reportTemplates);
        List<ReportDefinition> reportDefinitions = reportTemplates.getReportDefinitions();
        assertEquals(2, reportDefinitions.size());
        for (ReportDefinition rd : reportDefinitions) {
            for (DataValueTemplate dvt : rd.getDataValueTemplates()) {
                assertNotNull(dvt.getDataelement());
                assertNotNull(dvt.getDataelement().getCode());
                assertNotNull(dvt.getDataelement().getName());
                assertNotNull(dvt.getDataelement().getUid());
                assertNotNull(dvt.getDisaggregation());
                assertNotNull(dvt.getDisaggregation().getCode());
                assertNotNull(dvt.getDisaggregation().getName());
                assertNotNull(dvt.getDisaggregation().getUid());
            }
        }
    }

    @Test
    public void marshallReportTemplates() throws Exception {
        ClassPathResource resource = new ClassPathResource("templates_ethiopia.xml");
        JAXBContext jaxbContext = JAXBContext.newInstance(ReportTemplates.class);

        Unmarshaller jaxbUnmarshaller = jaxbContext.createUnmarshaller();
        ReportTemplates reportTemplates = (ReportTemplates) jaxbUnmarshaller.unmarshal(resource.getInputStream());
        Collection<DataValueTemplate> dvts = reportTemplates.getReportDefinitions().get(1).getDataValueTemplates();
        for (DataValueTemplate dvt : dvts) {
            dvt.setQuery("select count(*) from something & something_else");
        }
        Marshaller jaxbmarshaller = jaxbContext.createMarshaller();
        jaxbmarshaller.marshal(reportTemplates, System.out);
    }

    @Test
    public void testDodgyQueries() {
        String updateQuery = "Some Update query";
        String deleteQuery = "Some delete query";
        String safeQuery = "Some safe query";

        DataValueTemplate dvt = new DataValueTemplate();
        dvt.setQuery(updateQuery);
        assertTrue(dvt.potentialUpdateDelete());
        dvt.setQuery(deleteQuery);
        assertTrue(dvt.potentialUpdateDelete());
        dvt.setQuery(safeQuery);
        assertFalse(dvt.potentialUpdateDelete());
    }
}