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}