|
|
|
|
@ -1,5 +1,5 @@
@@ -1,5 +1,5 @@
|
|
|
|
|
/* |
|
|
|
|
* Copyright 2002-2014 the original author or authors. |
|
|
|
|
* Copyright 2002-2017 the original author or authors. |
|
|
|
|
* |
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
|
|
|
* you may not use this file except in compliance with the License. |
|
|
|
|
@ -33,6 +33,7 @@ import org.springframework.messaging.support.MessageBuilder;
@@ -33,6 +33,7 @@ import org.springframework.messaging.support.MessageBuilder;
|
|
|
|
|
import org.springframework.messaging.support.MessageHeaderAccessor; |
|
|
|
|
import org.springframework.util.AlternativeJdkIdGenerator; |
|
|
|
|
import org.springframework.util.LinkedMultiValueMap; |
|
|
|
|
import org.springframework.util.MimeType; |
|
|
|
|
import org.springframework.util.MimeTypeUtils; |
|
|
|
|
import org.springframework.util.MultiValueMap; |
|
|
|
|
|
|
|
|
|
@ -48,9 +49,9 @@ public class StompHeaderAccessorTests {
@@ -48,9 +49,9 @@ public class StompHeaderAccessorTests {
|
|
|
|
|
|
|
|
|
|
private static final Charset UTF_8 = Charset.forName("UTF-8"); |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
|
public void createWithCommand() { |
|
|
|
|
|
|
|
|
|
StompHeaderAccessor accessor = StompHeaderAccessor.create(StompCommand.CONNECTED); |
|
|
|
|
assertEquals(StompCommand.CONNECTED, accessor.getCommand()); |
|
|
|
|
|
|
|
|
|
@ -60,7 +61,6 @@ public class StompHeaderAccessorTests {
@@ -60,7 +61,6 @@ public class StompHeaderAccessorTests {
|
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
|
public void createWithSubscribeNativeHeaders() { |
|
|
|
|
|
|
|
|
|
MultiValueMap<String, String> extHeaders = new LinkedMultiValueMap<>(); |
|
|
|
|
extHeaders.add(StompHeaderAccessor.STOMP_ID_HEADER, "s1"); |
|
|
|
|
extHeaders.add(StompHeaderAccessor.STOMP_DESTINATION_HEADER, "/d"); |
|
|
|
|
@ -75,7 +75,6 @@ public class StompHeaderAccessorTests {
@@ -75,7 +75,6 @@ public class StompHeaderAccessorTests {
|
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
|
public void createWithUnubscribeNativeHeaders() { |
|
|
|
|
|
|
|
|
|
MultiValueMap<String, String> extHeaders = new LinkedMultiValueMap<>(); |
|
|
|
|
extHeaders.add(StompHeaderAccessor.STOMP_ID_HEADER, "s1"); |
|
|
|
|
|
|
|
|
|
@ -88,7 +87,6 @@ public class StompHeaderAccessorTests {
@@ -88,7 +87,6 @@ public class StompHeaderAccessorTests {
|
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
|
public void createWithMessageFrameNativeHeaders() { |
|
|
|
|
|
|
|
|
|
MultiValueMap<String, String> extHeaders = new LinkedMultiValueMap<>(); |
|
|
|
|
extHeaders.add(StompHeaderAccessor.DESTINATION_HEADER, "/d"); |
|
|
|
|
extHeaders.add(StompHeaderAccessor.STOMP_SUBSCRIPTION_HEADER, "s1"); |
|
|
|
|
@ -103,7 +101,6 @@ public class StompHeaderAccessorTests {
@@ -103,7 +101,6 @@ public class StompHeaderAccessorTests {
|
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
|
public void createWithConnectNativeHeaders() { |
|
|
|
|
|
|
|
|
|
MultiValueMap<String, String> extHeaders = new LinkedMultiValueMap<>(); |
|
|
|
|
extHeaders.add(StompHeaderAccessor.STOMP_LOGIN_HEADER, "joe"); |
|
|
|
|
extHeaders.add(StompHeaderAccessor.STOMP_PASSCODE_HEADER, "joe123"); |
|
|
|
|
@ -124,7 +121,6 @@ public class StompHeaderAccessorTests {
@@ -124,7 +121,6 @@ public class StompHeaderAccessorTests {
|
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
|
public void toNativeHeadersSubscribe() { |
|
|
|
|
|
|
|
|
|
StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.SUBSCRIBE); |
|
|
|
|
headers.setSubscriptionId("s1"); |
|
|
|
|
headers.setDestination("/d"); |
|
|
|
|
@ -138,7 +134,6 @@ public class StompHeaderAccessorTests {
@@ -138,7 +134,6 @@ public class StompHeaderAccessorTests {
|
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
|
public void toNativeHeadersUnsubscribe() { |
|
|
|
|
|
|
|
|
|
StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.UNSUBSCRIBE); |
|
|
|
|
headers.setSubscriptionId("s1"); |
|
|
|
|
|
|
|
|
|
@ -150,7 +145,6 @@ public class StompHeaderAccessorTests {
@@ -150,7 +145,6 @@ public class StompHeaderAccessorTests {
|
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
|
public void toNativeHeadersMessageFrame() { |
|
|
|
|
|
|
|
|
|
StompHeaderAccessor headers = StompHeaderAccessor.create(StompCommand.MESSAGE); |
|
|
|
|
headers.setSubscriptionId("s1"); |
|
|
|
|
headers.setDestination("/d"); |
|
|
|
|
@ -168,9 +162,8 @@ public class StompHeaderAccessorTests {
@@ -168,9 +162,8 @@ public class StompHeaderAccessorTests {
|
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
|
public void toNativeHeadersContentType() { |
|
|
|
|
|
|
|
|
|
SimpMessageHeaderAccessor simpHeaderAccessor = SimpMessageHeaderAccessor.create(); |
|
|
|
|
simpHeaderAccessor.setContentType(MimeTypeUtils.APPLICATION_ATOM_XML); |
|
|
|
|
simpHeaderAccessor.setContentType(MimeType.valueOf("application/atom+xml")); |
|
|
|
|
Message<byte[]> message = MessageBuilder.createMessage(new byte[0], simpHeaderAccessor.getMessageHeaders()); |
|
|
|
|
|
|
|
|
|
StompHeaderAccessor stompHeaderAccessor = StompHeaderAccessor.wrap(message); |
|
|
|
|
@ -181,7 +174,6 @@ public class StompHeaderAccessorTests {
@@ -181,7 +174,6 @@ public class StompHeaderAccessorTests {
|
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
|
public void encodeConnectWithLoginAndPasscode() throws UnsupportedEncodingException { |
|
|
|
|
|
|
|
|
|
MultiValueMap<String, String> extHeaders = new LinkedMultiValueMap<>(); |
|
|
|
|
extHeaders.add(StompHeaderAccessor.STOMP_LOGIN_HEADER, "joe"); |
|
|
|
|
extHeaders.add(StompHeaderAccessor.STOMP_PASSCODE_HEADER, "joe123"); |
|
|
|
|
@ -195,7 +187,6 @@ public class StompHeaderAccessorTests {
@@ -195,7 +187,6 @@ public class StompHeaderAccessorTests {
|
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
|
public void modifyCustomNativeHeader() { |
|
|
|
|
|
|
|
|
|
MultiValueMap<String, String> extHeaders = new LinkedMultiValueMap<>(); |
|
|
|
|
extHeaders.add(StompHeaderAccessor.STOMP_ID_HEADER, "s1"); |
|
|
|
|
extHeaders.add(StompHeaderAccessor.STOMP_DESTINATION_HEADER, "/d"); |
|
|
|
|
|