@Test
@SuppressWarnings({ "unchecked", "rawtypes" })
public void testHeadersPassedToMessagingTemplate() throws Exception {
	String sessionId = "sess1";
	String subscriptionId = "subs1";
	String destination = "/dest";
	Message<?> inputMessage = createInputMessage(sessionId, subscriptionId, destination, null);

	MessageSendingOperations messagingTemplate = Mockito.mock(MessageSendingOperations.class);
	SubscriptionMethodReturnValueHandler handler = new SubscriptionMethodReturnValueHandler(messagingTemplate);

	handler.handleReturnValue(PAYLOAD, this.subscribeEventReturnType, inputMessage);

	ArgumentCaptor<MessageHeaders> captor = ArgumentCaptor.forClass(MessageHeaders.class);
	verify(messagingTemplate).convertAndSend(eq("/dest"), eq(PAYLOAD), captor.capture());

	SimpMessageHeaderAccessor headerAccessor =
			MessageHeaderAccessor.getAccessor(captor.getValue(), SimpMessageHeaderAccessor.class);

	assertNotNull(headerAccessor);
	assertTrue(headerAccessor.isMutable());
	assertEquals(sessionId, headerAccessor.getSessionId());
	assertEquals(subscriptionId, headerAccessor.getSubscriptionId());
	assertEquals(this.subscribeEventReturnType, headerAccessor.getHeader(SimpMessagingTemplate.CONVERSION_HINT_HEADER));
}
 

@Test
@SuppressWarnings({ "unchecked", "rawtypes" })
public void testHeadersPassedToMessagingTemplate() throws Exception {
	String sessionId = "sess1";
	String subscriptionId = "subs1";
	String destination = "/dest";
	Message<?> inputMessage = createInputMessage(sessionId, subscriptionId, destination, null);

	MessageSendingOperations messagingTemplate = Mockito.mock(MessageSendingOperations.class);
	SubscriptionMethodReturnValueHandler handler = new SubscriptionMethodReturnValueHandler(messagingTemplate);

	handler.handleReturnValue(PAYLOAD, this.subscribeEventReturnType, inputMessage);

	ArgumentCaptor<MessageHeaders> captor = ArgumentCaptor.forClass(MessageHeaders.class);
	verify(messagingTemplate).convertAndSend(eq("/dest"), eq(PAYLOAD), captor.capture());

	SimpMessageHeaderAccessor headerAccessor =
			MessageHeaderAccessor.getAccessor(captor.getValue(), SimpMessageHeaderAccessor.class);

	assertNotNull(headerAccessor);
	assertTrue(headerAccessor.isMutable());
	assertEquals(sessionId, headerAccessor.getSessionId());
	assertEquals(subscriptionId, headerAccessor.getSubscriptionId());
	assertEquals(this.subscribeEventReturnType, headerAccessor.getHeader(SimpMessagingTemplate.CONVERSION_HINT_HEADER));
}
 

@Test
@SuppressWarnings({ "unchecked", "rawtypes" })
public void testHeadersPassedToMessagingTemplate() throws Exception {
	String sessionId = "sess1";
	String subscriptionId = "subs1";
	String destination = "/dest";
	Message<?> inputMessage = createInputMessage(sessionId, subscriptionId, destination, null);

	MessageSendingOperations messagingTemplate = Mockito.mock(MessageSendingOperations.class);
	SubscriptionMethodReturnValueHandler handler = new SubscriptionMethodReturnValueHandler(messagingTemplate);

	handler.handleReturnValue(PAYLOAD, this.subscribeEventReturnType, inputMessage);

	ArgumentCaptor<MessageHeaders> captor = ArgumentCaptor.forClass(MessageHeaders.class);
	verify(messagingTemplate).convertAndSend(eq("/dest"), eq(PAYLOAD), captor.capture());

	SimpMessageHeaderAccessor headerAccessor =
			MessageHeaderAccessor.getAccessor(captor.getValue(), SimpMessageHeaderAccessor.class);

	assertNotNull(headerAccessor);
	assertTrue(headerAccessor.isMutable());
	assertEquals(sessionId, headerAccessor.getSessionId());
	assertEquals(subscriptionId, headerAccessor.getSubscriptionId());
	assertEquals(this.subscribeEventReturnType, headerAccessor.getHeader(SimpMessagingTemplate.CONVERSION_HINT_HEADER));
}
 

public BroadcastHandler(MessageSendingOperations<String> template, String destination) {
	this.messagingTemplate = template;
	this.broadcastDestination = destination;
}
 

public BroadcastHandler(MessageSendingOperations<String> template, String destination) {
	this.messagingTemplate = template;
	this.broadcastDestination = destination;
}
 

public BroadcastHandler(MessageSendingOperations<String> template, String destination) {
	this.messagingTemplate = template;
	this.broadcastDestination = destination;
}
 

@Autowired
public QuoteServiceImpl(MessageSendingOperations<String> messagingTemplate) {
	this.messagingTemplate = messagingTemplate;
}
 

@Bean(name = "messageSendingOperations")
public MessageSendingOperations messageSendingOperations() {
    return mock(MessageSendingOperations.class);
}
 

@Autowired
public QuoteService(MessageSendingOperations<String> messagingTemplate) {
	this.messagingTemplate = messagingTemplate;
}
 

/**
 * Construct a new SubscriptionMethodReturnValueHandler.
 * @param template a messaging template to send messages to,
 * most likely the "clientOutboundChannel" (must not be {@code null})
 */
public SubscriptionMethodReturnValueHandler(MessageSendingOperations<String> template) {
	Assert.notNull(template, "messagingTemplate must not be null");
	this.messagingTemplate = template;
}
 

/**
 * Return the messaging template used to send resolved messages to the
 * broker channel.
 */
public MessageSendingOperations<String> getBrokerMessagingTemplate() {
	return this.messagingTemplate;
}
 

/**
 * Construct a new SubscriptionMethodReturnValueHandler.
 * @param template a messaging template to send messages to,
 * most likely the "clientOutboundChannel" (must not be {@code null})
 */
public SubscriptionMethodReturnValueHandler(MessageSendingOperations<String> template) {
	Assert.notNull(template, "messagingTemplate must not be null");
	this.messagingTemplate = template;
}
 

/**
 * Return the messaging template used to send resolved messages to the
 * broker channel.
 */
public MessageSendingOperations<String> getBrokerMessagingTemplate() {
	return this.messagingTemplate;
}
 

/**
 * Construct a new SubscriptionMethodReturnValueHandler.
 * @param messagingTemplate a messaging template to send messages to,
 * most likely the "clientOutboundChannel" (must not be {@code null})
 */
public SubscriptionMethodReturnValueHandler(MessageSendingOperations<String> messagingTemplate) {
	Assert.notNull(messagingTemplate, "messagingTemplate must not be null");
	this.messagingTemplate = messagingTemplate;
}
 

/**
 * Return the messaging template used to send resolved messages to the
 * broker channel.
 */
public MessageSendingOperations<String> getBrokerMessagingTemplate() {
	return this.messagingTemplate;
}