Fix issue in last commit

This commit is contained in:
Rossen Stoyanchev 2014-03-25 09:59:09 -04:00
parent a247d5f2e8
commit bbf101eef6
2 changed files with 3 additions and 3 deletions

View File

@ -187,7 +187,7 @@ public class StompSubProtocolHandler implements SubProtocolHandler, ApplicationE
message = MessageBuilder.withPayload(message.getPayload()).setHeaders(headers).build();
if (this.eventPublisher != null && StompCommand.CONNECT.equals(headers.getMessageType())) {
if (this.eventPublisher != null && StompCommand.CONNECT.equals(headers.getCommand())) {
publishEvent(new SessionConnectEvent(this, message));
}
@ -265,7 +265,7 @@ public class StompSubProtocolHandler implements SubProtocolHandler, ApplicationE
try {
message = MessageBuilder.withPayload(message.getPayload()).setHeaders(headers).build();
if (this.eventPublisher != null && StompCommand.CONNECTED.equals(headers.getMessageType())) {
if (this.eventPublisher != null && StompCommand.CONNECTED.equals(headers.getCommand())) {
publishEvent(new SessionConnectedEvent(this, (Message<byte[]>) message));
}

View File

@ -183,7 +183,7 @@ public class StompSubProtocolHandlerTests {
this.protocolHandler.afterSessionEnded(this.session, CloseStatus.BAD_DATA, this.channel);
assertEquals(3, publisher.events.size());
assertEquals("Unexpected events " + publisher.events, 3, publisher.events.size());
assertEquals(SessionConnectEvent.class, publisher.events.get(0).getClass());
assertEquals(SessionConnectedEvent.class, publisher.events.get(1).getClass());
assertEquals(SessionDisconnectEvent.class, publisher.events.get(2).getClass());