Skip to content

Commit 5c80d6d

Browse files
committed
Merge pull request #12029 from dreis2211
* pr/12029: Polish BannerTests
2 parents a9eefae + cd0c5ba commit 5c80d6d

File tree

1 file changed

+13
-13
lines changed
  • spring-boot-project/spring-boot/src/test/java/org/springframework/boot

1 file changed

+13
-13
lines changed

spring-boot-project/spring-boot/src/test/java/org/springframework/boot/BannerTests.java

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
/*
2-
* Copyright 2012-2017 the original author or authors.
2+
* Copyright 2012-2018 the original author or authors.
33
*
44
* Licensed under the Apache License, Version 2.0 (the "License");
55
* you may not use this file except in compliance with the License.
@@ -71,41 +71,36 @@ public void setup() {
7171

7272
@Test
7373
public void testDefaultBanner() {
74-
SpringApplication application = new SpringApplication(Config.class);
75-
application.setWebApplicationType(WebApplicationType.NONE);
74+
SpringApplication application = createSpringApplication();
7675
this.context = application.run();
7776
assertThat(this.out.toString()).contains(":: Spring Boot ::");
7877
}
7978

8079
@Test
8180
public void testDefaultBannerInLog() {
82-
SpringApplication application = new SpringApplication(Config.class);
83-
application.setWebApplicationType(WebApplicationType.NONE);
81+
SpringApplication application = createSpringApplication();
8482
this.context = application.run();
8583
assertThat(this.out.toString()).contains(":: Spring Boot ::");
8684
}
8785

8886
@Test
8987
public void testCustomBanner() {
90-
SpringApplication application = new SpringApplication(Config.class);
91-
application.setWebApplicationType(WebApplicationType.NONE);
88+
SpringApplication application = createSpringApplication();
9289
application.setBanner(new DummyBanner());
9390
this.context = application.run();
9491
assertThat(this.out.toString()).contains("My Banner");
9592
}
9693

9794
@Test
9895
public void testBannerInContext() {
99-
SpringApplication application = new SpringApplication(Config.class);
100-
application.setWebApplicationType(WebApplicationType.NONE);
96+
SpringApplication application = createSpringApplication();
10197
this.context = application.run();
10298
assertThat(this.context.containsBean("springBootBanner")).isTrue();
10399
}
104100

105101
@Test
106102
public void testCustomBannerInContext() {
107-
SpringApplication application = new SpringApplication(Config.class);
108-
application.setWebApplicationType(WebApplicationType.NONE);
103+
SpringApplication application = createSpringApplication();
109104
Banner banner = mock(Banner.class);
110105
application.setBanner(banner);
111106
this.context = application.run();
@@ -122,13 +117,18 @@ public void testCustomBannerInContext() {
122117

123118
@Test
124119
public void testDisableBannerInContext() {
125-
SpringApplication application = new SpringApplication(Config.class);
120+
SpringApplication application = createSpringApplication();
126121
application.setBannerMode(Mode.OFF);
127-
application.setWebApplicationType(WebApplicationType.NONE);
128122
this.context = application.run();
129123
assertThat(this.context.containsBean("springBootBanner")).isFalse();
130124
}
131125

126+
private SpringApplication createSpringApplication() {
127+
SpringApplication application = new SpringApplication(Config.class);
128+
application.setWebApplicationType(WebApplicationType.NONE);
129+
return application;
130+
}
131+
132132
static class DummyBanner implements Banner {
133133

134134
@Override

0 commit comments

Comments
 (0)