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

Example 36 with Message

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

the class StompDecoderTests method decodeHeartbeat.

@Test
public void decodeHeartbeat() {
    String frame = "\n";
    ByteBuffer buffer = ByteBuffer.wrap(frame.getBytes());
    final List<Message<byte[]>> messages = decoder.decode(buffer);
    assertEquals(1, messages.size());
    assertEquals(SimpMessageType.HEARTBEAT, StompHeaderAccessor.wrap(messages.get(0)).getMessageType());
}

Example 37 with Message

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

the class SendToMethodReturnValueHandlerTests method sendToWithDestinationPlaceholders.

// SPR-12170
@Test
public void sendToWithDestinationPlaceholders() throws Exception {
    given(this.messageChannel.send(any(Message.class))).willReturn(true);
    Map<String, String> vars = new LinkedHashMap<>(1);
    vars.put("roomName", "roomA");
    String sessionId = "sess1";
    SimpMessageHeaderAccessor accessor = SimpMessageHeaderAccessor.create();
    accessor.setSessionId(sessionId);
    accessor.setSubscriptionId("sub1");
    accessor.setHeader(DestinationVariableMethodArgumentResolver.DESTINATION_TEMPLATE_VARIABLES_HEADER, vars);
    Message<?> message = MessageBuilder.createMessage(PAYLOAD, accessor.getMessageHeaders());
    this.handler.handleReturnValue(PAYLOAD, this.sendToWithPlaceholdersReturnType, message);
    verify(this.messageChannel, times(1)).send(this.messageCaptor.capture());
    SimpMessageHeaderAccessor actual = getCapturedAccessor(0);
    assertEquals(sessionId, actual.getSessionId());
    assertEquals("/topic/chat.message.filtered.roomA", actual.getDestination());
}

Example 38 with Message

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

the class SendToMethodReturnValueHandlerTests method sendToUserWithUserNameProvider.

@Test
public void sendToUserWithUserNameProvider() throws Exception {
    given(this.messageChannel.send(any(Message.class))).willReturn(true);
    String sessionId = "sess1";
    TestUser user = new UniqueUser();
    Message<?> inputMessage = createMessage(sessionId, "sub1", null, null, user);
    this.handler.handleReturnValue(PAYLOAD, this.sendToUserReturnType, inputMessage);
    verify(this.messageChannel, times(2)).send(this.messageCaptor.capture());
    SimpMessageHeaderAccessor accessor = getCapturedAccessor(0);
    assertEquals("/user/Me myself and I/dest1", accessor.getDestination());
    accessor = getCapturedAccessor(1);
    assertEquals("/user/Me myself and I/dest2", accessor.getDestination());
}

Example 39 with Message

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

the class SendToMethodReturnValueHandlerTests method sendToUserDefaultDestination.

@Test
public void sendToUserDefaultDestination() throws Exception {
    given(this.messageChannel.send(any(Message.class))).willReturn(true);
    String sessionId = "sess1";
    TestUser user = new TestUser();
    Message<?> inputMessage = createMessage(sessionId, "sub1", "/app", "/dest", user);
    this.handler.handleReturnValue(PAYLOAD, this.sendToUserDefaultDestReturnType, inputMessage);
    verify(this.messageChannel, times(1)).send(this.messageCaptor.capture());
    SimpMessageHeaderAccessor accessor = getCapturedAccessor(0);
    assertNull(accessor.getSessionId());
    assertNull(accessor.getSubscriptionId());
    assertEquals("/user/" + user.getName() + "/queue/dest", accessor.getDestination());
}

Example 40 with Message

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

the class SimpAnnotationMethodMessageHandlerTests method completableFutureFailure.

@Test
@SuppressWarnings("unchecked")
public void completableFutureFailure() {
    Message emptyMessage = (Message) MessageBuilder.withPayload(new byte[0]).build();
    given(this.channel.send(any(Message.class))).willReturn(true);
    given(this.converter.toMessage(any(), any(MessageHeaders.class))).willReturn(emptyMessage);
    CompletableFutureController controller = new CompletableFutureController();
    this.messageHandler.registerHandler(controller);
    this.messageHandler.setDestinationPrefixes(Arrays.asList("/app1", "/app2/"));
    Message<?> message = createMessage("/app1/completable-future");
    this.messageHandler.handleMessage(message);
    controller.future.completeExceptionally(new IllegalStateException());
    assertTrue(controller.exceptionCaught);
}