001/*
002 * Copyright 2012-2016 the original author or authors.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 *      http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016
017package org.springframework.boot.cli.compiler.autoconfigure;
018
019import org.codehaus.groovy.ast.ClassNode;
020import org.codehaus.groovy.control.CompilationFailedException;
021import org.codehaus.groovy.control.customizers.ImportCustomizer;
022
023import org.springframework.boot.cli.compiler.AstUtils;
024import org.springframework.boot.cli.compiler.CompilerAutoConfiguration;
025import org.springframework.boot.cli.compiler.DependencyCustomizer;
026
027/**
028 * {@link CompilerAutoConfiguration} for JUnit.
029 *
030 * @author Greg Turnquist
031 */
032public class JUnitCompilerAutoConfiguration extends CompilerAutoConfiguration {
033
034        @Override
035        public boolean matches(ClassNode classNode) {
036                return AstUtils.hasAtLeastOneAnnotation(classNode, "Test");
037        }
038
039        @Override
040        public void applyDependencies(DependencyCustomizer dependencies)
041                        throws CompilationFailedException {
042                dependencies.add("spring-boot-starter-test");
043        }
044
045        @Override
046        public void applyImports(ImportCustomizer imports) throws CompilationFailedException {
047                imports.addStarImports("org.junit").addStaticStars("org.junit.Assert")
048                                .addStaticStars("org.hamcrest.MatcherAssert")
049                                .addStaticStars("org.hamcrest.Matchers");
050        }
051
052}