Examples with Message org.springframework.messaging.Message used on opensource projects

Example 31 with Message

use of org.springframework.messaging.Message in project spring-framework by spring-projects.

the class UserDestinationMessageHandlerTests method handleUnsubscribe.

@Test
public void handleUnsubscribe() {
    given(this.brokerChannel.send(Mockito.any(Message.class))).willReturn(true);
    this.handler.handleMessage(createWith(SimpMessageType.UNSUBSCRIBE, "joe", "123", "/user/queue/foo"));
    ArgumentCaptor<Message> captor = ArgumentCaptor.forClass(Message.class);
    Mockito.verify(this.brokerChannel).send(captor.capture());
    Message message = captor.getValue();
    assertEquals("/queue/foo-user123", SimpMessageHeaderAccessor.getDestination(message.getHeaders()));
}

Example 32 with Message

use of org.springframework.messaging.Message in project spring-framework by spring-projects.

the class UserDestinationMessageHandlerTests method handleMessageWithoutActiveSession.

@Test
public void handleMessageWithoutActiveSession() {
    this.handler.setBroadcastDestination("/topic/unresolved");
    given(this.brokerChannel.send(Mockito.any(Message.class))).willReturn(true);
    this.handler.handleMessage(createWith(SimpMessageType.MESSAGE, "joe", "123", "/user/joe/queue/foo"));
    ArgumentCaptor<Message> captor = ArgumentCaptor.forClass(Message.class);
    Mockito.verify(this.brokerChannel).send(captor.capture());
    Message message = captor.getValue();
    SimpMessageHeaderAccessor accessor = SimpMessageHeaderAccessor.wrap(message);
    assertEquals("/topic/unresolved", accessor.getDestination());
    assertEquals("/user/joe/queue/foo", accessor.getFirstNativeHeader(ORIGINAL_DESTINATION));
    // Should ignore our own broadcast to brokerChannel
    this.handler.handleMessage(message);
    Mockito.verifyNoMoreInteractions(this.brokerChannel);
}

Example 33 with Message

use of org.springframework.messaging.Message in project spring-framework by spring-projects.

the class UserDestinationMessageHandlerTests method handleMessageFromBrokerWithActiveSession.

@Test
public void handleMessageFromBrokerWithActiveSession() {
    TestSimpUser simpUser = new TestSimpUser("joe");
    simpUser.addSessions(new TestSimpSession("123"));
    when(this.registry.getUser("joe")).thenReturn(simpUser);
    this.handler.setBroadcastDestination("/topic/unresolved");
    given(this.brokerChannel.send(Mockito.any(Message.class))).willReturn(true);
    StompHeaderAccessor accessor = StompHeaderAccessor.create(StompCommand.MESSAGE);
    accessor.setSessionId("system123");
    accessor.setDestination("/topic/unresolved");
    accessor.setNativeHeader(ORIGINAL_DESTINATION, "/user/joe/queue/foo");
    accessor.setNativeHeader("customHeader", "customHeaderValue");
    accessor.setLeaveMutable(true);
    byte[] payload = "payload".getBytes(StandardCharsets.UTF_8);
    this.handler.handleMessage(MessageBuilder.createMessage(payload, accessor.getMessageHeaders()));
    ArgumentCaptor<Message> captor = ArgumentCaptor.forClass(Message.class);
    Mockito.verify(this.brokerChannel).send(captor.capture());
    assertNotNull(captor.getValue());
    SimpMessageHeaderAccessor headers = SimpMessageHeaderAccessor.wrap(captor.getValue());
    assertEquals("/queue/foo-user123", headers.getDestination());
    assertEquals("/user/queue/foo", headers.getFirstNativeHeader(ORIGINAL_DESTINATION));
    assertEquals("customHeaderValue", headers.getFirstNativeHeader("customHeader"));
    assertArrayEquals(payload, (byte[]) captor.getValue().getPayload());
}

Example 34 with Message

use of org.springframework.messaging.Message in project spring-framework by spring-projects.

the class UserDestinationMessageHandlerTests method handleMessage.

@Test
public void handleMessage() {
    TestSimpUser simpUser = new TestSimpUser("joe");
    simpUser.addSessions(new TestSimpSession("123"));
    when(this.registry.getUser("joe")).thenReturn(simpUser);
    given(this.brokerChannel.send(Mockito.any(Message.class))).willReturn(true);
    this.handler.handleMessage(createWith(SimpMessageType.MESSAGE, "joe", "123", "/user/joe/queue/foo"));
    ArgumentCaptor<Message> captor = ArgumentCaptor.forClass(Message.class);
    Mockito.verify(this.brokerChannel).send(captor.capture());
    SimpMessageHeaderAccessor accessor = SimpMessageHeaderAccessor.wrap(captor.getValue());
    assertEquals("/queue/foo-user123", accessor.getDestination());
    assertEquals("/user/queue/foo", accessor.getFirstNativeHeader(ORIGINAL_DESTINATION));
}

Example 35 with Message

use of org.springframework.messaging.Message in project spring-framework by spring-projects.

the class StompDecoderTests method decodeMultipleFramesFromSameBuffer.

@Test
public void decodeMultipleFramesFromSameBuffer() {
    String frame1 = "SEND\ndestination:test\n\nThe body of the message\0";
    String frame2 = "DISCONNECT\n\n\0";
    ByteBuffer buffer = ByteBuffer.wrap((frame1 + frame2).getBytes());
    final List<Message<byte[]>> messages = decoder.decode(buffer);
    assertEquals(2, messages.size());
    assertEquals(StompCommand.SEND, StompHeaderAccessor.wrap(messages.get(0)).getCommand());
    assertEquals(StompCommand.DISCONNECT, StompHeaderAccessor.wrap(messages.get(1)).getCommand());
}