org.openbravo.test.datasource.DatasourceTestUtil.java Source code

Introduction

Here is the source code for org.openbravo.test.datasource.DatasourceTestUtil.java

Source

/*
 *************************************************************************
 * The contents of this file are subject to the Openbravo  Public  License
 * Version  1.1  (the  "License"),  being   the  Mozilla   Public  License
 * Version 1.1  with a permitted attribution clause; you may not  use this
 * file except in compliance with the License. You  may  obtain  a copy of
 * the License at http://www.openbravo.com/legal/license.html 
 * 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. 
 * The Original Code is Openbravo ERP. 
 * The Initial Developer of the Original Code is Openbravo SLU 
 * All portions are Copyright (C) 2014-2015 Openbravo SLU 
 * All Rights Reserved. 
 * Contributor(s):  ______________________________________.
 ************************************************************************
 */

package org.openbravo.test.datasource;

import java.io.OutputStream;
import java.io.StringWriter;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.Map;
import java.util.Properties;

import junit.framework.TestCase;

import org.apache.commons.io.IOUtils;
import org.apache.commons.lang.StringUtils;
import org.codehaus.jettison.json.JSONObject;
import org.openbravo.base.exception.OBException;
import org.openbravo.base.session.OBPropertiesProvider;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Utility methods to deal with datasource calls.
 * 
 * @author alostale
 * 
 */
class DatasourceTestUtil {
    private static final Logger log = LoggerFactory.getLogger(DatasourceTestUtil.class);
    private static final String CONTEXT_PROPERTY = "context.url";

    static HttpURLConnection createConnection(String url, String wsPart, String method, String cookie)
            throws Exception {

        String completeUrl = url + wsPart;
        log.debug("Create conntection URL: {}, method {}", completeUrl, method);
        final URL connUrl = new URL(completeUrl);
        final HttpURLConnection hc = (HttpURLConnection) connUrl.openConnection();
        hc.setRequestMethod(method);
        hc.setAllowUserInteraction(false);
        hc.setDefaultUseCaches(false);
        hc.setDoOutput(true);
        hc.setDoInput(true);
        hc.setInstanceFollowRedirects(true);
        hc.setUseCaches(false);
        if (cookie != null) {
            hc.setRequestProperty("Cookie", cookie);
        }
        return hc;
    }

    static String authenticate(String openbravoURL, String user, String password) throws Exception {
        final HttpURLConnection hc = DatasourceTestUtil.createConnection(openbravoURL,
                "/secureApp/LoginHandler.html", "POST", null);
        final OutputStream os = hc.getOutputStream();
        String content = "user=" + user + "&password=" + password;
        os.write(content.getBytes("UTF-8"));
        os.flush();
        os.close();
        hc.connect();
        return hc.getHeaderField("Set-Cookie");
    }

    static String request(String openbravoURL, String wsPart, String method, String content, String cookie,
            int expectedResponse) throws Exception {
        return request(openbravoURL, wsPart, method, content, cookie, expectedResponse, null);
    }

    static String request(String openbravoURL, String wsPart, String method, String content, String cookie,
            int expectedResponse, String contentType) throws Exception {
        final HttpURLConnection hc = createConnection(openbravoURL, wsPart, method, cookie);
        if (contentType != null) {
            hc.setRequestProperty("Content-Type", contentType);
        }
        if (!"DELETE".equals(method)) {
            final OutputStream os = hc.getOutputStream();
            os.write(content.getBytes("UTF-8"));
            os.flush();
            os.close();
        }
        hc.connect();
        TestCase.assertEquals(expectedResponse, hc.getResponseCode());

        if (expectedResponse == 500) {
            // no content available anyway
            return "";
        }

        StringWriter writer = new StringWriter();
        IOUtils.copy(hc.getInputStream(), writer, "utf-8");
        return writer.toString();
    }

    static String getParamsContent(Map<String, String> params) {
        String content = "";
        for (String key : params.keySet()) {
            if (!content.isEmpty()) {
                content += "&";
            }
            content += key + "=" + params.get(key);
        }
        return content;
    }

    static String getOpenbravoURL() {
        Properties props = OBPropertiesProvider.getInstance().getOpenbravoProperties();
        String obURL = props.getProperty(CONTEXT_PROPERTY);
        if (StringUtils.isEmpty(obURL)) {
            throw new OBException(CONTEXT_PROPERTY + " is not set in Openbravo.properties");
        }
        log.debug("got OB context: " + obURL);

        return obURL;
    }

    static void changeProfile(String openbravoURL, String cookie, String roleId, String langId, String orgId,
            String warehouseId) throws Exception {
        JSONObject newProfile = new JSONObject();
        newProfile.put("language", langId);
        newProfile.put("organization", orgId);
        newProfile.put("role", roleId);
        newProfile.put("warehouse", warehouseId);
        request(openbravoURL,
                "/org.openbravo.client.kernel?command=save&_action=org.openbravo.client.application.navigationbarcomponents.UserInfoWidgetActionHandler",
                "POST", newProfile.toString(), cookie, 200, "application/json");
    }
}