Add @WebAppConfiguration to webapp tests

This commit is contained in:
Dave Syer
2014-03-18 10:21:31 +00:00
parent 59c009dba6
commit 4c292e60f2
2 changed files with 11 additions and 5 deletions

View File

@@ -79,6 +79,13 @@ class MainController {
File test = new File(new File(dir, "src/test/java"),request.packageName.replace(".", "/"))
test.mkdirs()
if (model.styles.contains("-web")) {
model.testAnnotations = "@WebAppConfiguration\n"
model.testImports = "import org.springframework.test.context.web.WebAppConfiguration;\n"
} else {
model.testAnnotations = ""
model.testImports = ""
}
write(test, "ApplicationTests.java", model)
File download = new File(tmpdir, dir.name + ".zip")
@@ -97,8 +104,8 @@ class MainController {
}
def write(File src, String name, def model) {
def body = template name, model
log.info("Creating: " + src + "/" + name)
def body = template name, model
new File(src, name).write(body)
}

View File

@@ -2,13 +2,12 @@ package ${packageName};
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.boot.test.SpringApplicationContextLoader;
import org.springframework.test.context.ContextConfiguration;
${testImports}import org.springframework.boot.test.SpringApplicationConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes = Application.class, loader=SpringApplicationContextLoader.class)
public class ApplicationTests {
@SpringApplicationConfiguration(classes = Application.class)
${testAnnotations}public class ApplicationTests {
@Test
public void contextLoads() {