io.promagent.it.WildflyIT.java Source code

Introduction

Here is the source code for io.promagent.it.WildflyIT.java

Source

// Copyright 2017 The Promagent Authors
//
// 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 io.promagent.it;

import com.squareup.okhttp.OkHttpClient;
import com.squareup.okhttp.Request;
import com.squareup.okhttp.Response;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;

import java.util.Arrays;

import static org.junit.jupiter.api.Assertions.assertTrue;

public class WildflyIT {

    /**
     * Run some HTTP queries against a Docker container from image promagent/wildfly-kitchensink-promagent.
     * <p/>
     * The Docker container is started by the maven-docker-plugin when running <tt>mvn verify -Pwildfly</tt>.
     */
    @Test
    public void testWildfly() throws Exception {
        OkHttpClient client = new OkHttpClient();
        Request restRequest = new Request.Builder().url(System.getProperty("deployment.url") + "/rest/members")
                .build();

        // Execute REST call
        Response restResponse = client.newCall(restRequest).execute();
        Assertions.assertEquals(restResponse.code(), 200);
        Assertions.assertTrue(restResponse.body().string().contains("John Smith"));

        Thread.sleep(100); // metric is incremented after servlet has written the response, wait a little to get the updated metric
        assertMetrics(client, "1.0");

        // Execute REST call again
        restResponse = client.newCall(restRequest).execute();
        Assertions.assertEquals(restResponse.code(), 200);
        Assertions.assertTrue(restResponse.body().string().contains("John Smith"));

        Thread.sleep(100); // metric is incremented after servlet has written the response, wait a little to get the updated metric
        assertMetrics(client, "2.0");
    }

    private void assertMetrics(OkHttpClient client, String nCalls) throws Exception {

        Request metricsRequest = new Request.Builder().url(System.getProperty("promagent.url") + "/metrics")
                .build();
        Response metricsResponse = client.newCall(metricsRequest).execute();
        String[] metricsLines = metricsResponse.body().string().split("\n");

        String httpRequestsTotal = Arrays.stream(metricsLines).filter(m -> m.contains("http_requests_total"))
                .filter(m -> m.contains("method=\"GET\""))
                .filter(m -> m.contains("path=\"/wildfly-kitchensink/rest/members\""))
                .filter(m -> m.contains("status=\"200\"")).findFirst()
                .orElseThrow(() -> new Exception("http_requests_total metric not found."));

        assertTrue(httpRequestsTotal.endsWith(nCalls), "Value should be " + nCalls + " for " + httpRequestsTotal);

        String sqlQueriesTotal = Arrays.stream(metricsLines).filter(m -> m.contains("sql_queries_total")).filter(
                m -> m.matches(".*?query=\"select .*?id .*?email .*?name .*?phone_number .*? from Member .*?\".*?"))
                .filter(m -> m.contains("method=\"GET\""))
                .filter(m -> m.contains("path=\"/wildfly-kitchensink/rest/members\"")).findFirst()
                .orElseThrow(() -> new Exception("sql_queries_total metric not found."));

        assertTrue(sqlQueriesTotal.endsWith(nCalls), "Value should be " + nCalls + " for " + sqlQueriesTotal);
    }
}