4747 * JUnit5 condition for an embedded broker.
4848 *
4949 * @author Gary Russell
50+ * @author Artem Bilan
51+ *
5052 * @since 2.3
5153 *
5254 */
@@ -84,20 +86,17 @@ public void afterAll(ExtensionContext context) {
8486 @ Override
8587 public ConditionEvaluationResult evaluateExecutionCondition (ExtensionContext context ) {
8688 Optional <AnnotatedElement > element = context .getElement ();
87- if (element .isPresent ()) {
88- /*
89- * When running in a spring test context, the EmbeddedKafkaContextCustomizer will
90- * create the broker.
91- */
92- if (AnnotatedElementUtils .findMergedAnnotation (element .get (), SpringJUnitConfig .class ) == null ) {
93- EmbeddedKafka embedded = AnnotatedElementUtils .findMergedAnnotation (element .get (), EmbeddedKafka .class );
94- if (embedded != null ) {
95- EmbeddedKafkaBroker broker = getBrokerFromStore (context );
96- if (broker == null ) {
97- broker = createBroker (embedded );
98- BROKERS .set (broker );
99- getStore (context ).put (EMBEDDED_BROKER , broker );
100- }
89+ if (element .isPresent () &&
90+ AnnotatedElementUtils .findMergedAnnotation (element .get (), SpringJUnitConfig .class ) == null ) {
91+
92+ EmbeddedKafka embedded = AnnotatedElementUtils .findMergedAnnotation (element .get (), EmbeddedKafka .class );
93+ // When running in a spring test context, the EmbeddedKafkaContextCustomizer will create the broker.
94+ if (embedded != null ) {
95+ EmbeddedKafkaBroker broker = getBrokerFromStore (context );
96+ if (broker == null ) {
97+ broker = createBroker (embedded );
98+ BROKERS .set (broker );
99+ getStore (context ).put (EMBEDDED_BROKER , broker );
101100 }
102101 }
103102 }
@@ -108,7 +107,7 @@ public ConditionEvaluationResult evaluateExecutionCondition(ExtensionContext con
108107 private EmbeddedKafkaBroker createBroker (EmbeddedKafka embedded ) {
109108 EmbeddedKafkaBroker broker ;
110109 broker = new EmbeddedKafkaBroker (embedded .count (),
111- embedded .controlledShutdown (), embedded .topics ());
110+ embedded .controlledShutdown (), embedded .topics ());
112111 broker .kafkaPorts (embedded .ports ());
113112 Properties properties = new Properties ();
114113
@@ -135,7 +134,7 @@ private EmbeddedKafkaBroker createBroker(EmbeddedKafka embedded) {
135134 try (InputStream in = propertiesResource .getInputStream ()) {
136135 Properties p = new Properties ();
137136 p .load (in );
138- p .forEach (( key , value ) -> properties . putIfAbsent ( key , value ) );
137+ p .forEach (properties :: putIfAbsent );
139138 }
140139 catch (IOException ex ) {
141140 throw new IllegalStateException (
@@ -148,10 +147,9 @@ private EmbeddedKafkaBroker createBroker(EmbeddedKafka embedded) {
148147 }
149148
150149 private EmbeddedKafkaBroker getBrokerFromStore (ExtensionContext context ) {
151- EmbeddedKafkaBroker broker = getParentStore (context ).get (EMBEDDED_BROKER , EmbeddedKafkaBroker .class ) == null
150+ return getParentStore (context ).get (EMBEDDED_BROKER , EmbeddedKafkaBroker .class ) == null
152151 ? getStore (context ).get (EMBEDDED_BROKER , EmbeddedKafkaBroker .class )
153152 : getParentStore (context ).get (EMBEDDED_BROKER , EmbeddedKafkaBroker .class );
154- return broker ;
155153 }
156154
157155 private Store getStore (ExtensionContext context ) {
0 commit comments