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