AbstractClientApplicationTest.java

/*
 * Copyright 2014-2018 the original author or 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 de.codecentric.boot.admin.client;

import de.codecentric.boot.admin.client.registration.ApplicationRegistrator;

import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.concurrent.CountDownLatch;
import org.junit.Rule;
import org.junit.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.SpringBootConfiguration;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.boot.context.event.ApplicationReadyEvent;
import org.springframework.context.event.EventListener;
import com.github.tomakehurst.wiremock.client.ResponseDefinitionBuilder;
import com.github.tomakehurst.wiremock.common.ConsoleNotifier;
import com.github.tomakehurst.wiremock.junit.WireMockRule;
import com.github.tomakehurst.wiremock.matching.RequestPatternBuilder;

import static com.github.tomakehurst.wiremock.client.WireMock.created;
import static com.github.tomakehurst.wiremock.client.WireMock.equalTo;
import static com.github.tomakehurst.wiremock.client.WireMock.matching;
import static com.github.tomakehurst.wiremock.client.WireMock.matchingJsonPath;
import static com.github.tomakehurst.wiremock.client.WireMock.post;
import static com.github.tomakehurst.wiremock.client.WireMock.postRequestedFor;
import static com.github.tomakehurst.wiremock.client.WireMock.urlEqualTo;
import static com.github.tomakehurst.wiremock.client.WireMock.verify;
import static com.github.tomakehurst.wiremock.core.WireMockConfiguration.options;

public abstract class AbstractClientApplicationTest {

    @Rule
    public WireMockRule wireMock = new WireMockRule(options().dynamicPort().notifier(new ConsoleNotifier(true)));
    private static final CountDownLatch cdl = new CountDownLatch(1);

    public void setUp() throws Exception {
        ResponseDefinitionBuilder response = created().withHeader("Content-Type", "application/json")
                                                      .withHeader("Connection", "close")
                                                      .withHeader("Location", wireMock.url("/instances/abcdef"))
                                                      .withBody("{ \"id\" : \"abcdef\" }");
        wireMock.stubFor(post(urlEqualTo("/instances")).willReturn(response));
    }

    @Test
    public void test_context() throws InterruptedException, UnknownHostException {
        cdl.await();
        Thread.sleep(2500);
        String hostName = InetAddress.getLocalHost().getCanonicalHostName();
        String serviceHost = "http://" + hostName + ":" + getServerPort();
        String managementHost = "http://" + hostName + ":" + getManagementPort();
        RequestPatternBuilder request = postRequestedFor(urlEqualTo("/instances"));
        request.withHeader("Content-Type", equalTo("application/json"))
               .withRequestBody(matchingJsonPath("$.name", equalTo("Test-Client")))
               .withRequestBody(matchingJsonPath("$.healthUrl", equalTo(managementHost + "/mgmt/health")))
               .withRequestBody(matchingJsonPath("$.managementUrl", equalTo(managementHost + "/mgmt")))
               .withRequestBody(matchingJsonPath("$.serviceUrl", equalTo(serviceHost + "/")))
               .withRequestBody(matchingJsonPath("$.metadata.startup", matching(".+")));

        verify(request);
    }

    protected abstract int getServerPort();

    protected abstract int getManagementPort();

    @SpringBootConfiguration
    @EnableAutoConfiguration
    public static class TestClientApplication {
        @Autowired
        private ApplicationRegistrator registrator;

        @EventListener
        public void ping(ApplicationReadyEvent ev) {
            new Thread(() -> {
                try {
                    while (registrator.getRegisteredId() == null) {
                        Thread.sleep(500);
                    }
                } catch (InterruptedException e) {
                    Thread.interrupted();
                }
                cdl.countDown();
            }).start();
        }
    }
}