DefaultInitializrMetadataProviderTests.java

/*
 * 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.web.support;

import java.util.List;

import io.spring.initializr.metadata.DefaultMetadataElement;
import io.spring.initializr.metadata.InitializrMetadata;
import io.spring.initializr.test.metadata.InitializrMetadataTestBuilder;
import org.junit.Before;
import org.junit.Test;

import org.springframework.core.io.ClassPathResource;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.test.web.client.MockRestServiceServer;
import org.springframework.web.client.RestTemplate;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
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 DefaultInitializrMetadataProviderTests {

	private RestTemplate restTemplate;

	private MockRestServiceServer mockServer;

	@Before
	public void setUp() {
		restTemplate = new RestTemplate();
		mockServer = MockRestServiceServer.createServer(restTemplate);
	}

	@Test
	public void bootVersionsAreReplaced() {
		InitializrMetadata metadata = new InitializrMetadataTestBuilder()
				.addBootVersion("0.0.9.RELEASE", true)
				.addBootVersion("0.0.8.RELEASE", false)
				.build();
		assertEquals("0.0.9.RELEASE", metadata.getBootVersions().getDefault().getId());
		DefaultInitializrMetadataProvider provider =
				new DefaultInitializrMetadataProvider(metadata, restTemplate);
		expectJson(metadata.getConfiguration().getEnv().getSpringBootMetadataUrl(),
				"metadata/sagan/spring-boot.json");

		InitializrMetadata updatedMetadata = provider.get();
		assertNotNull(updatedMetadata.getBootVersions());
		List<DefaultMetadataElement> updatedBootVersions =
				updatedMetadata.getBootVersions().getContent();
		assertEquals(4, updatedBootVersions.size());
		assertBootVersion(updatedBootVersions.get(0), "1.4.1 (SNAPSHOT)", false);
		assertBootVersion(updatedBootVersions.get(1), "1.4.0", true);
		assertBootVersion(updatedBootVersions.get(2), "1.3.8 (SNAPSHOT)", false);
		assertBootVersion(updatedBootVersions.get(3), "1.3.7", false);
	}

	@Test
	public void defaultBootVersionIsAlwaysSet() {
		InitializrMetadata metadata = new InitializrMetadataTestBuilder()
				.addBootVersion("0.0.9.RELEASE", true)
				.addBootVersion("0.0.8.RELEASE", false).build();
		assertEquals("0.0.9.RELEASE", metadata.getBootVersions().getDefault().getId());
		DefaultInitializrMetadataProvider provider =
				new DefaultInitializrMetadataProvider(metadata, restTemplate);
		expectJson(metadata.getConfiguration().getEnv().getSpringBootMetadataUrl(),
				"metadata/sagan/spring-boot-no-default.json");

		InitializrMetadata updatedMetadata = provider.get();
		assertNotNull(updatedMetadata.getBootVersions());
		List<DefaultMetadataElement> updatedBootVersions =
				updatedMetadata.getBootVersions().getContent();
		assertEquals(4, updatedBootVersions.size());
		assertBootVersion(updatedBootVersions.get(0), "1.3.1 (SNAPSHOT)", true);
		assertBootVersion(updatedBootVersions.get(1), "1.3.0", false);
		assertBootVersion(updatedBootVersions.get(2), "1.2.6 (SNAPSHOT)", false);
		assertBootVersion(updatedBootVersions.get(3), "1.2.5", false);
	}

	private static void assertBootVersion(DefaultMetadataElement actual, String name,
			boolean defaultVersion) {
		assertEquals(name, actual.getName());
		assertEquals(defaultVersion, actual.isDefault());
	}

	private void expectJson(String url, String bodyPath) {
		HttpHeaders httpHeaders = new HttpHeaders();
		httpHeaders.setContentType(MediaType.APPLICATION_JSON);
		this.mockServer.expect(requestTo(url))
				.andExpect(method(HttpMethod.GET))
				.andRespond(withStatus(HttpStatus.OK)
						.body(new ClassPathResource(bodyPath))
						.headers(httpHeaders));
	}

}