Polish BannerTests

Closes spring-projectsgh-12029
adamduan · Feb 13, 2018 · cd0c5ba · cd0c5ba
1 parent a9eefae
commit cd0c5ba
Showing 1 changed file with 13 additions and 13 deletions.
diff --git a/spring-boot-project/spring-boot/src/test/java/org/springframework/boot/BannerTests.java b/spring-boot-project/spring-boot/src/test/java/org/springframework/boot/BannerTests.java
@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2017 the original author or authors.
+ * 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.
@@ -71,41 +71,36 @@ public void setup() {
 
 	@Test
 	public void testDefaultBanner() {
-		SpringApplication application = new SpringApplication(Config.class);
-		application.setWebApplicationType(WebApplicationType.NONE);
+		SpringApplication application = createSpringApplication();
 		this.context = application.run();
 		assertThat(this.out.toString()).contains(":: Spring Boot ::");
 	}
 
 	@Test
 	public void testDefaultBannerInLog() {
-		SpringApplication application = new SpringApplication(Config.class);
-		application.setWebApplicationType(WebApplicationType.NONE);
+		SpringApplication application = createSpringApplication();
 		this.context = application.run();
 		assertThat(this.out.toString()).contains(":: Spring Boot ::");
 	}
 
 	@Test
 	public void testCustomBanner() {
-		SpringApplication application = new SpringApplication(Config.class);
-		application.setWebApplicationType(WebApplicationType.NONE);
+		SpringApplication application = createSpringApplication();
 		application.setBanner(new DummyBanner());
 		this.context = application.run();
 		assertThat(this.out.toString()).contains("My Banner");
 	}
 
 	@Test
 	public void testBannerInContext() {
-		SpringApplication application = new SpringApplication(Config.class);
-		application.setWebApplicationType(WebApplicationType.NONE);
+		SpringApplication application = createSpringApplication();
 		this.context = application.run();
 		assertThat(this.context.containsBean("springBootBanner")).isTrue();
 	}
 
 	@Test
 	public void testCustomBannerInContext() {
-		SpringApplication application = new SpringApplication(Config.class);
-		application.setWebApplicationType(WebApplicationType.NONE);
+		SpringApplication application = createSpringApplication();
 		Banner banner = mock(Banner.class);
 		application.setBanner(banner);
 		this.context = application.run();
@@ -122,13 +117,18 @@ public void testCustomBannerInContext() {
 
 	@Test
 	public void testDisableBannerInContext() {
-		SpringApplication application = new SpringApplication(Config.class);
+		SpringApplication application = createSpringApplication();
 		application.setBannerMode(Mode.OFF);
-		application.setWebApplicationType(WebApplicationType.NONE);
 		this.context = application.run();
 		assertThat(this.context.containsBean("springBootBanner")).isFalse();
 	}
 
+	private SpringApplication createSpringApplication() {
+		SpringApplication application = new SpringApplication(Config.class);
+		application.setWebApplicationType(WebApplicationType.NONE);
+		return application;
+	}
+
 	static class DummyBanner implements Banner {
 
 		@Override