AbstractProjectGeneratorTests.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.generator;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;

import io.spring.initializr.metadata.Dependency;
import io.spring.initializr.metadata.InitializrMetadata;
import io.spring.initializr.metadata.SimpleInitializrMetadataProvider;
import io.spring.initializr.test.generator.GradleBuildAssert;
import io.spring.initializr.test.generator.PomAssert;
import io.spring.initializr.test.generator.ProjectAssert;
import io.spring.initializr.test.metadata.InitializrMetadataTestBuilder;
import org.junit.Before;
import org.junit.Rule;
import org.junit.rules.TemporaryFolder;
import org.mockito.ArgumentMatcher;

import org.springframework.context.ApplicationEventPublisher;

import static org.mockito.Matchers.argThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;

/**
 * @author Stephane Nicoll
 */
public abstract class AbstractProjectGeneratorTests {

	@Rule
	public final TemporaryFolder folder = new TemporaryFolder();

	protected final ProjectGenerator projectGenerator = new ProjectGenerator();

	private final ApplicationEventPublisher eventPublisher = mock(
			ApplicationEventPublisher.class);

	@Before
	public void setup() throws IOException {
		Dependency web = Dependency.withId("web");
		web.getFacets().add("web");
		InitializrMetadata metadata = InitializrMetadataTestBuilder.withDefaults()
				.addDependencyGroup("web", web).addDependencyGroup("test", "security",
						"data-jpa", "aop", "batch", "integration")
				.build();
		applyMetadata(metadata);
		projectGenerator.setEventPublisher(eventPublisher);
		projectGenerator
				.setRequestResolver(new ProjectRequestResolver(new ArrayList<>()));
		projectGenerator.setTmpdir(folder.newFolder().getAbsolutePath());
	}

	protected PomAssert generateMavenPom(ProjectRequest request) {
		request.setType("maven-build");
		String content = new String(projectGenerator.generateMavenPom(request));
		return new PomAssert(content).validateProjectRequest(request);
	}

	protected GradleBuildAssert generateGradleBuild(ProjectRequest request) {
		request.setType("gradle-build");
		String content = new String(projectGenerator.generateGradleBuild(request));
		return new GradleBuildAssert(content).validateProjectRequest(request);
	}

	protected ProjectAssert generateProject(ProjectRequest request) {
		File dir = projectGenerator.generateProjectStructure(request);
		return new ProjectAssert(dir);
	}

	public ProjectRequest createProjectRequest(String... styles) {
		ProjectRequest request = new ProjectRequest();
		request.initialize(projectGenerator.getMetadataProvider().get());
		request.getStyle().addAll(Arrays.asList(styles));
		return request;
	}

	protected void applyMetadata(InitializrMetadata metadata) {
		projectGenerator.setMetadataProvider(new SimpleInitializrMetadataProvider(metadata));
	}

	protected void verifyProjectSuccessfulEventFor(ProjectRequest request) {
		verify(eventPublisher, times(1)).publishEvent(argThat(new ProjectGeneratedEventMatcher(request)));
	}

	protected void verifyProjectFailedEventFor(ProjectRequest request, Exception ex) {
		verify(eventPublisher, times(1)).publishEvent(argThat(new ProjectFailedEventMatcher(request, ex)));
	}

	private static class ProjectGeneratedEventMatcher
			extends ArgumentMatcher<ProjectGeneratedEvent> {

		private final ProjectRequest request;

		ProjectGeneratedEventMatcher(ProjectRequest request) {
			this.request = request;
		}

		@Override
		public boolean matches(Object argument) {
			ProjectGeneratedEvent event = (ProjectGeneratedEvent) argument;
			return request.equals(event.getProjectRequest());
		}
	}

	private static class ProjectFailedEventMatcher
			extends ArgumentMatcher<ProjectFailedEvent> {

		private final ProjectRequest request;
		private final Exception cause;

		ProjectFailedEventMatcher(ProjectRequest request, Exception cause) {
			this.request = request;
			this.cause = cause;
		}

		@Override
		public boolean matches(Object argument) {
			ProjectFailedEvent event = (ProjectFailedEvent) argument;
			return request.equals(event.getProjectRequest())
					&& cause.equals(event.getCause());
		}
	}

}