Polishing

See gh-21798
This commit is contained in:
Rossen Stoyanchev
2023-10-10 10:51:33 +01:00
committed by rstoyanchev
parent a205eab618
commit 9eb39e182e
6 changed files with 68 additions and 49 deletions

View File

@@ -54,11 +54,11 @@ class StompBrokerRelayMessageHandlerTests {
private StompBrokerRelayMessageHandler brokerRelay;
private StubMessageChannel outboundChannel = new StubMessageChannel();
private final StubMessageChannel outboundChannel = new StubMessageChannel();
private StubTcpOperations tcpClient = new StubTcpOperations();
private final StubTcpOperations tcpClient = new StubTcpOperations();
private ArgumentCaptor<Runnable> messageCountTaskCaptor = ArgumentCaptor.forClass(Runnable.class);
private final ArgumentCaptor<Runnable> messageCountTaskCaptor = ArgumentCaptor.forClass(Runnable.class);
@BeforeEach

View File

@@ -23,6 +23,7 @@ import org.mockito.ArgumentCaptor;
import org.mockito.Mockito;
import org.springframework.core.testfixture.security.TestPrincipal;
import org.springframework.lang.Nullable;
import org.springframework.messaging.Message;
import org.springframework.messaging.StubMessageChannel;
import org.springframework.messaging.SubscribableChannel;
@@ -50,7 +51,8 @@ class UserDestinationMessageHandlerTests {
private final SubscribableChannel brokerChannel = mock();
private final UserDestinationMessageHandler handler = new UserDestinationMessageHandler(new StubMessageChannel(), this.brokerChannel, new DefaultUserDestinationResolver(this.registry));
private final UserDestinationMessageHandler handler = new UserDestinationMessageHandler(
new StubMessageChannel(), this.brokerChannel, new DefaultUserDestinationResolver(this.registry));
@Test
@@ -184,7 +186,9 @@ class UserDestinationMessageHandlerTests {
}
private Message<?> createWith(SimpMessageType type, String user, String sessionId, String destination) {
private Message<?> createWith(
SimpMessageType type, @Nullable String user, @Nullable String sessionId, @Nullable String destination) {
SimpMessageHeaderAccessor headers = SimpMessageHeaderAccessor.create(type);
if (destination != null) {
headers.setDestination(destination);