org.openmrs.module.odkconnector.serialization.web.CohortWebConnectorTest.java Source code

Introduction

Here is the source code for org.openmrs.module.odkconnector.serialization.web.CohortWebConnectorTest.java

Source

/**
 * The contents of this file are subject to the OpenMRS Public License
 * Version 1.0 (the "License"); you may not use this file except in
 * compliance with the License. You may obtain a copy of the License at
 * http://license.openmrs.org
 *
 * Software distributed under the License is distributed on an "AS IS"
 * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
 * License for the specific language governing rights and limitations
 * under the License.
 *
 * Copyright (C) OpenMRS, LLC.  All Rights Reserved.
 */

package org.openmrs.module.odkconnector.serialization.web;

import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.junit.Test;
import org.openmrs.test.BaseModuleContextSensitiveTest;

public class CohortWebConnectorTest extends BaseModuleContextSensitiveTest {

    private static final Log log = LogFactory.getLog(CohortWebConnectorTest.class);

    private static final String SERVER_URL = "http://localhost:8081/openmrs";

    @Test
    public void serialize_shouldDisplayAllCohortInformation() throws Exception {

        // compose url
        URL u = new URL(SERVER_URL + "/module/odkconnector/download/cohort.form");

        // setup http url connection
        HttpURLConnection connection = (HttpURLConnection) u.openConnection();
        connection.setDoOutput(true);
        connection.setRequestMethod("POST");
        connection.setConnectTimeout(8000);
        connection.setReadTimeout(8000);
        connection.addRequestProperty("Content-type", "application/octet-stream");

        // write auth details to connection
        DataOutputStream outputStream = new DataOutputStream(new GZIPOutputStream(connection.getOutputStream()));
        outputStream.writeUTF("admin");
        outputStream.writeUTF("test");
        outputStream.writeBoolean(false);
        outputStream.close();

        DataInputStream inputStream = new DataInputStream(new GZIPInputStream(connection.getInputStream()));
        Integer responseStatus = inputStream.readInt();

        if (responseStatus == HttpURLConnection.HTTP_OK) {

            Integer cohortCounts = inputStream.readInt();
            System.out.println("Number of cohorts: " + cohortCounts);

            for (int i = 0; i < cohortCounts; i++) {
                System.out.println("Cohort ID: " + inputStream.readInt());
                System.out.println("Cohort Name: " + inputStream.readUTF());
            }
        }
        inputStream.close();
    }

}