com.ibm.watson.developer_cloud.WatsonServiceUnitTest.java Source code

Introduction

Here is the source code for com.ibm.watson.developer_cloud.WatsonServiceUnitTest.java

Source

/**
 * Copyright 2015 IBM Corp. All Rights Reserved.
 *
 * Licensed under the Apache License, Version 2.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://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 */
package com.ibm.watson.developer_cloud;

import java.io.IOException;

import org.apache.commons.lang3.StringUtils;
import org.junit.After;

import com.google.gson.Gson;
import com.ibm.watson.developer_cloud.http.HttpMediaType;
import com.ibm.watson.developer_cloud.util.GsonSingleton;

import okhttp3.mockwebserver.MockResponse;
import okhttp3.mockwebserver.MockWebServer;

/**
 * Utility class to Mock the Watson Services.
 *
 */
public abstract class WatsonServiceUnitTest extends WatsonServiceTest {

    /** The Constant DELETE. */
    protected static final String DELETE = "DELETE";

    /** The Constant GET. */
    protected static final String GET = "GET";

    /** The Constant POST. */
    protected static final String POST = "POST";

    /** The Constant PUT. */
    protected static final String PUT = "PUT";

    private static final Gson GSON = GsonSingleton.getGson();

    /** The server. */
    protected MockWebServer server;

    /**
     * Setups and starts the mock server.
     *
     * @throws Exception the exception
     */
    @Override
    public void setUp() throws Exception {
        server = new MockWebServer();
        server.start();
    }

    /**
     * Tear down.
     *
     * @throws IOException Signals that an I/O exception has occurred.
     */
    @After
    public void tearDown() throws IOException {
        server.shutdown();
    }

    /**
     * Gets the mock web server url.
     *
     * @return the server url
     */
    protected String getMockWebServerUrl() {
        return StringUtils.chop(server.url("/").toString());
    }

    /**
     * Create a MockResponse with JSON content type and the object serialized to JSON as body.
     *
     * @param body the body
     * @return the mock response
     */
    protected static MockResponse jsonResponse(Object body) {
        return new MockResponse().addHeader(CONTENT_TYPE, HttpMediaType.APPLICATION_JSON)
                .setBody(GSON.toJson(body));
    }

}