com.google.cloud.tools.gradle.appengine.AssertConnection.java Source code

Introduction

Here is the source code for com.google.cloud.tools.gradle.appengine.AssertConnection.java

Source

/*
 * Copyright (c) 2016 Google Inc. All Right 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.google.cloud.tools.gradle.appengine;

import com.google.common.io.CharStreams;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.ConnectException;
import java.net.HttpURLConnection;
import java.net.URL;
import org.hamcrest.CoreMatchers;
import org.junit.Assert;

/** Assertions for checking connections web apps. */
public class AssertConnection {

    /** Connect and assert response is as expected. */
    public static void assertResponse(String url, int expectedCode, String expectedText) {
        try {
            HttpURLConnection urlConnection = (HttpURLConnection) new URL(url).openConnection();
            int responseCode = urlConnection.getResponseCode();
            Assert.assertEquals(expectedCode, responseCode);
            String response = CharStreams.toString(new InputStreamReader(urlConnection.getInputStream()));
            Assert.assertThat(response, CoreMatchers.equalTo(expectedText));
        } catch (IOException e) {
            Assert.fail("IOException while running test");
        }
    }

    /** Connect and assert that the target is unreachable. */
    public static void assertUnreachable(String url) throws IOException {
        try {
            HttpURLConnection urlConnection = (HttpURLConnection) new URL("http://localhost:8080").openConnection();
            urlConnection.getResponseCode();
            Assert.fail("ConnectException expected");
        } catch (IOException e) {
            if (!(e instanceof ConnectException)) {
                throw e;
            }
        }
    }
}