Polishing

This commit is contained in:
Rossen Stoyanchev
2019-11-08 17:47:59 +00:00
parent 08f1cb454a
commit 32f82c0ed0
2 changed files with 7 additions and 6 deletions

View File

@@ -117,8 +117,9 @@ public interface StompSession {
void disconnect();
/**
* Disconnect the session by sending a DISCONNECT frame.
* Variant of {@link #disconnect()} with headers.
* @param headers the headers for the disconnect message frame
* @since 5.2.2
*/
void disconnect(StompHeaders headers);

View File

@@ -662,7 +662,7 @@ public class DefaultStompSessionTests {
@Test
public void disconnectWithHeaders() {
this.session.afterConnected(this.connection);
assertTrue(this.session.isConnected());
assertThat(this.session.isConnected()).isTrue();
StompHeaders headers = new StompHeaders();
headers.add("foo", "bar");
@@ -672,11 +672,11 @@ public class DefaultStompSessionTests {
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");
assertThat(headers.size()).as(headers.toString()).isEqualTo(1);
assertThat(headers.get("foo").size()).isEqualTo(1);
assertThat(headers.get("foo").get(0)).isEqualTo("bar");
assertFalse(this.session.isConnected());
assertThat(this.session.isConnected()).isFalse();
verifyNoMoreInteractions(this.sessionHandler);
}