Polish "Handle null RSocketServer address when setting port property"

See gh-23084

spring-boot-project/spring-boot/src/main/java/org/springframework/boot/rsocket/context/RSocketPortInfoApplicationContextInitializer.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2019 the original author or authors.
+ * Copyright 2012-2020 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.

spring-boot-project/spring-boot/src/test/java/org/springframework/boot/rsocket/context/RSocketPortInfoApplicationContextInitializerTests.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2012-2019 the original author or authors.
+ * Copyright 2012-2020 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.
@@ -18,56 +18,56 @@ package org.springframework.boot.rsocket.context;
 import java.net.InetSocketAddress;
 
 import org.junit.jupiter.api.Test;
-import org.junit.jupiter.api.extension.ExtendWith;
 
-import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.rsocket.server.RSocketServer;
-import org.springframework.boot.rsocket.server.RSocketServerException;
-import org.springframework.context.ApplicationEventPublisher;
 import org.springframework.context.ConfigurableApplicationContext;
+import org.springframework.context.annotation.AnnotationConfigApplicationContext;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
-import org.springframework.test.context.junit.jupiter.SpringExtension;
 
-@ExtendWith(SpringExtension.class)
+import static org.assertj.core.api.Assertions.assertThat;
-public class RSocketPortInfoApplicationContextInitializerTests {
+import static org.mockito.BDDMockito.given;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
 
-	@Autowired
+/**
-	private RSocketPortInfoApplicationContextInitializer initializer;
+ * Tests for {@link RSocketPortInfoApplicationContextInitializer}.
-
+ *
-	@Autowired
+ * @author Spencer Gibb
-	private ConfigurableApplicationContext context;
+ * @author Andy Wilkinson
-
+ */
-	@Autowired
+class RSocketPortInfoApplicationContextInitializerTests {
-	private ApplicationEventPublisher publisher;
 
 	@Test
-	void nullAddressDoesNotThrow() {
+	void whenServerHasAddressThenInitializerSetsPortProperty() {
-		initializer.initialize(context);
+		try (ConfigurableApplicationContext context = new AnnotationConfigApplicationContext(Config.class)) {
-		publisher.publishEvent(new RSocketServerInitializedEvent(new RSocketServer() {
+			context.getBean(RSocketPortInfoApplicationContextInitializer.class).initialize(context);
-			@Override
+			RSocketServer server = mock(RSocketServer.class);
-			public void start() throws RSocketServerException {
+			given(server.address()).willReturn(new InetSocketAddress(65535));
+			context.publishEvent(new RSocketServerInitializedEvent(server));
+			assertThat(context.getEnvironment().getProperty("local.rsocket.server.port")).isEqualTo("65535");
+		}
+	}
 
-			}
+	@Test
-
+	void whenServerHasNoAddressThenInitializerDoesNotSetPortProperty() {
-			@Override
+		try (ConfigurableApplicationContext context = new AnnotationConfigApplicationContext(Config.class)) {
-			public void stop() throws RSocketServerException {
+			context.getBean(RSocketPortInfoApplicationContextInitializer.class).initialize(context);
-
+			RSocketServer server = mock(RSocketServer.class);
-			}
+			context.publishEvent(new RSocketServerInitializedEvent(server));
-
+			verify(server).address();
-			@Override
+			assertThat(context.getEnvironment().getProperty("local.rsocket.server.port")).isNull();
-			public InetSocketAddress address() {
+		}
-				return null;
-			}
-		}));
 	}
 
 	@Configuration(proxyBeanMethods = false)
 	static class Config {
 
 		@Bean
-		public RSocketPortInfoApplicationContextInitializer rSocketPortInfoApplicationContextInitializer() {
+		RSocketPortInfoApplicationContextInitializer rSocketPortInfoApplicationContextInitializer() {
 			return new RSocketPortInfoApplicationContextInitializer();
 		}
+
 	}
+
 }