io.spring.initializr.generator.CustomProjectGeneratorTests.java Source code

Introduction

Here is the source code for io.spring.initializr.generator.CustomProjectGeneratorTests.java

Source

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

import java.io.File;
import java.util.Map;

import io.spring.initializr.test.generator.ProjectAssert;
import org.junit.Test;
import org.mockito.InOrder;
import org.mockito.Mockito;

import org.springframework.core.io.ClassPathResource;

import static org.mockito.ArgumentMatchers.argThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;

/**
 * Test for custom {@link ProjectGenerator}.
 *
 * @author Torsten Walter
 * @author Stephane Nicoll
 */
public class CustomProjectGeneratorTests extends AbstractProjectGeneratorTests {

    public CustomProjectGeneratorTests() {
        super(new MyProjectGenerator());
    }

    @Test
    public void generateCustomResource() {
        ProjectRequest request = createProjectRequest();
        request.setType("maven-project");
        request.setGroupId("com.example.custom");
        ProjectAssert project = generateProject(request);
        project.sourceCodeAssert("custom.txt").equalsTo(new ClassPathResource("project/custom/custom.txt"));
    }

    @Test
    public void generateCustomResourceDisabled() {
        ProjectRequest request = createProjectRequest();
        request.setType("gradle-build");
        request.setGroupId("com.example.custom");
        ProjectAssert project = generateProject(request);
        project.hasNoFile("custom.txt");
    }

    @Test
    public void projectGenerationEventFiredAfterCustomization() {
        ProjectRequest request = createProjectRequest();
        request.setType("maven-project");
        request.setGroupId("com.example.custom");
        generateProject(request);
        verifyProjectSuccessfulEventFor(request);

        Runnable customFileGenerated = ((MyProjectGenerator) this.projectGenerator).customFileGenerated;
        InOrder inOrder = Mockito.inOrder(this.eventPublisher, customFileGenerated);

        inOrder.verify(customFileGenerated, times(1)).run();
        inOrder.verify(this.eventPublisher, times(1))
                .publishEvent(argThat(new ProjectGeneratedEventMatcher(request)));
    }

    private static class MyProjectGenerator extends ProjectGenerator {

        private Runnable customFileGenerated = mock(Runnable.class);

        @Override
        protected File generateProjectStructure(ProjectRequest request, Map<String, Object> model) {
            model.put("customValue", 42);
            File dir = super.generateProjectStructure(request, model);
            if ("maven".equals(request.getBuild())) {
                write(new File(dir, "custom.txt"), "custom.txt", model);
                this.customFileGenerated.run();
            }
            return dir;
        }

    }

}