Allow sending headers with the disconnect frame

This commit is contained in:
Ilja
2019-04-01 10:14:53 +02:00
committed by Rossen Stoyanchev
parent b0b6423714
commit 08f1cb454a
3 changed files with 35 additions and 0 deletions

View File

@@ -659,4 +659,25 @@ public class DefaultStompSessionTests {
verifyNoMoreInteractions(this.sessionHandler);
}
@Test
public void disconnectWithHeaders() {
this.session.afterConnected(this.connection);
assertTrue(this.session.isConnected());
StompHeaders headers = new StompHeaders();
headers.add("foo", "bar");
this.session.disconnect(headers);
Message<byte[]> message = this.messageCaptor.getValue();
StompHeaderAccessor accessor = MessageHeaderAccessor.getAccessor(message, StompHeaderAccessor.class);
headers = StompHeaders.readOnlyStompHeaders(accessor.getNativeHeaders());
assertEquals(headers.toString(), 1, headers.size());
assertEquals(headers.get("foo").size(), 1);
assertEquals(headers.get("foo").get(0), "bar");
assertFalse(this.session.isConnected());
verifyNoMoreInteractions(this.sessionHandler);
}
}