io.spring.initializr.actuate.stat.ProjectGenerationStatPublisherTests.java Source code

Introduction

Here is the source code for io.spring.initializr.actuate.stat.ProjectGenerationStatPublisherTests.java

Source

/*
 * Copyright 2012-2017 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 io.spring.initializr.actuate.stat;

import java.util.Collections;
import java.util.UUID;

import io.spring.initializr.actuate.stat.StatsProperties.Elastic;
import io.spring.initializr.generator.ProjectGeneratedEvent;
import io.spring.initializr.generator.ProjectRequest;
import org.junit.Before;
import org.junit.Test;

import org.springframework.boot.web.client.RestTemplateBuilder;
import org.springframework.http.HttpMethod;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.retry.policy.SimpleRetryPolicy;
import org.springframework.retry.support.RetryTemplate;
import org.springframework.test.web.client.MockRestServiceServer;

import static org.springframework.test.web.client.match.MockRestRequestMatchers.jsonPath;
import static org.springframework.test.web.client.match.MockRestRequestMatchers.method;
import static org.springframework.test.web.client.match.MockRestRequestMatchers.requestTo;
import static org.springframework.test.web.client.response.MockRestResponseCreators.withStatus;

/**
 * @author Stephane Nicoll
 */
public class ProjectGenerationStatPublisherTests extends AbstractInitializrStatTests {

    private RetryTemplate retryTemplate;

    private ProjectGenerationStatPublisher statPublisher;

    private MockRestServiceServer mockServer;

    @Before
    public void setUp() {
        StatsProperties properties = createProperties();
        ProjectRequestDocumentFactory documentFactory = new ProjectRequestDocumentFactory(
                createProvider(getMetadata()));
        this.retryTemplate = new RetryTemplate();
        this.statPublisher = new ProjectGenerationStatPublisher(documentFactory, properties,
                new RestTemplateBuilder(), this.retryTemplate);
        this.mockServer = MockRestServiceServer.createServer(this.statPublisher.getRestTemplate());
    }

    @Test
    public void publishSimpleDocument() {
        ProjectRequest request = createProjectRequest();
        request.setGroupId("com.example.foo");
        request.setArtifactId("my-project");

        this.mockServer.expect(requestTo("http://example.com/elastic/initializr/request"))
                .andExpect(method(HttpMethod.POST)).andExpect(jsonPath("$.groupId").value("com.example.foo"))
                .andExpect(jsonPath("$.artifactId").value("my-project"))
                .andRespond(withStatus(HttpStatus.CREATED).body(mockResponse(UUID.randomUUID().toString(), true))
                        .contentType(MediaType.APPLICATION_JSON));

        this.statPublisher.handleEvent(new ProjectGeneratedEvent(request));
        this.mockServer.verify();
    }

    @Test
    public void recoverFromError() {
        ProjectRequest request = createProjectRequest();

        this.mockServer.expect(requestTo("http://example.com/elastic/initializr/request"))
                .andExpect(method(HttpMethod.POST)).andRespond(withStatus(HttpStatus.INTERNAL_SERVER_ERROR));

        this.mockServer.expect(requestTo("http://example.com/elastic/initializr/request"))
                .andExpect(method(HttpMethod.POST)).andRespond(withStatus(HttpStatus.INTERNAL_SERVER_ERROR));

        this.mockServer.expect(requestTo("http://example.com/elastic/initializr/request"))
                .andExpect(method(HttpMethod.POST))
                .andRespond(withStatus(HttpStatus.CREATED).body(mockResponse(UUID.randomUUID().toString(), true))
                        .contentType(MediaType.APPLICATION_JSON));

        this.statPublisher.handleEvent(new ProjectGeneratedEvent(request));
        this.mockServer.verify();
    }

    @Test
    public void fatalErrorOnlyLogs() {
        ProjectRequest request = createProjectRequest();
        this.retryTemplate
                .setRetryPolicy(new SimpleRetryPolicy(2, Collections.singletonMap(Exception.class, true)));

        this.mockServer.expect(requestTo("http://example.com/elastic/initializr/request"))
                .andExpect(method(HttpMethod.POST)).andRespond(withStatus(HttpStatus.INTERNAL_SERVER_ERROR));

        this.mockServer.expect(requestTo("http://example.com/elastic/initializr/request"))
                .andExpect(method(HttpMethod.POST)).andRespond(withStatus(HttpStatus.INTERNAL_SERVER_ERROR));

        this.statPublisher.handleEvent(new ProjectGeneratedEvent(request));
        this.mockServer.verify();
    }

    private static String mockResponse(String id, boolean created) {
        return "{\"_index\":\"initializr\",\"_type\":\"request\",\"_id\":\"" + id + "\",\"_version\":1,\"_shards\""
                + ":{\"total\":1,\"successful\":1,\"failed\":0},\"created\":" + created + "}";
    }

    private static StatsProperties createProperties() {
        StatsProperties properties = new StatsProperties();
        Elastic elastic = properties.getElastic();
        elastic.setUri("http://example.com/elastic");
        elastic.setUsername("foo");
        elastic.setPassword("bar");
        return properties;
    }

}