Examples with StompHeaderAccessor org.springframework.messaging.simp.stomp.StompHeaderAccessor used on opensource projects

Example 36 with StompHeaderAccessor

use of org.springframework.messaging.simp.stomp.StompHeaderAccessor in project spring-framework by spring-projects.

the class MessageBrokerConfigurationTests method clientOutboundChannelUsedByAnnotatedMethod.

@Test
public void clientOutboundChannelUsedByAnnotatedMethod() {
    TestChannel channel = this.simpleBrokerContext.getBean("clientOutboundChannel", TestChannel.class);
    SimpAnnotationMethodMessageHandler messageHandler = this.simpleBrokerContext.getBean(SimpAnnotationMethodMessageHandler.class);
    StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.SUBSCRIBE);
    headers.setSessionId("sess1");
    headers.setSessionAttributes(new ConcurrentHashMap<>());
    headers.setSubscriptionId("subs1");
    headers.setDestination("/foo");
    Message<?> message = MessageBuilder.withPayload(new byte[0]).setHeaders(headers).build();
    messageHandler.handleMessage(message);
    message = channel.messages.get(0);
    headers = StompHeaderAccessor.wrap(message);
    assertEquals(SimpMessageType.MESSAGE, headers.getMessageType());
    assertEquals("/foo", headers.getDestination());
    assertEquals("bar", new String((byte[]) message.getPayload()));
}

Example 37 with StompHeaderAccessor

use of org.springframework.messaging.simp.stomp.StompHeaderAccessor in project spring-framework by spring-projects.

the class MessageBrokerConfigurationTests method brokerChannelUsedByAnnotatedMethod.

@Test
public void brokerChannelUsedByAnnotatedMethod() {
    TestChannel channel = this.simpleBrokerContext.getBean("brokerChannel", TestChannel.class);
    SimpAnnotationMethodMessageHandler messageHandler = this.simpleBrokerContext.getBean(SimpAnnotationMethodMessageHandler.class);
    StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.SEND);
    headers.setSessionId("sess1");
    headers.setSessionAttributes(new ConcurrentHashMap<>());
    headers.setDestination("/foo");
    Message<?> message = MessageBuilder.createMessage(new byte[0], headers.getMessageHeaders());
    messageHandler.handleMessage(message);
    message = channel.messages.get(0);
    headers = StompHeaderAccessor.wrap(message);
    assertEquals(SimpMessageType.MESSAGE, headers.getMessageType());
    assertEquals("/bar", headers.getDestination());
    assertEquals("bar", new String((byte[]) message.getPayload()));
}

Example 38 with StompHeaderAccessor

use of org.springframework.messaging.simp.stomp.StompHeaderAccessor in project spring-framework by spring-projects.

the class MessageBrokerConfigurationTests method clientOutboundChannelUsedBySimpleBroker.

@Test
public void clientOutboundChannelUsedBySimpleBroker() {
    TestChannel channel = this.simpleBrokerContext.getBean("clientOutboundChannel", TestChannel.class);
    SimpleBrokerMessageHandler broker = this.simpleBrokerContext.getBean(SimpleBrokerMessageHandler.class);
    StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.SUBSCRIBE);
    headers.setSessionId("sess1");
    headers.setSubscriptionId("subs1");
    headers.setDestination("/foo");
    Message<?> message = MessageBuilder.createMessage(new byte[0], headers.getMessageHeaders());
    // subscribe
    broker.handleMessage(message);
    headers = StompHeaderAccessor.create(StompCommand.SEND);
    headers.setSessionId("sess1");
    headers.setDestination("/foo");
    message = MessageBuilder.createMessage("bar".getBytes(), headers.getMessageHeaders());
    // message
    broker.handleMessage(message);
    message = channel.messages.get(0);
    headers = StompHeaderAccessor.wrap(message);
    assertEquals(SimpMessageType.MESSAGE, headers.getMessageType());
    assertEquals("/foo", headers.getDestination());
    assertEquals("bar", new String((byte[]) message.getPayload()));
}