Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Introduce SubProtocolHandler abstraction

Browse Source 
Add SubProtocolHandler to encapsulate the logic for using a
sub-protocol.

A SubProtocolWebSocketHandler is also provided to
delegate to the appropriate SubProtocolHandler based on the
negotiated sub-protocol value at handshake.

StompSubProtocolHandler provides handling for STOMP messages.

Issue: SPR-10786
This commit is contained in:
Andy Wilkinson
2013-07-30 11:29:57 +01:00
committed by Rossen Stoyanchev
parent e4d83bbefe
commit 9e20a25607
10 changed files with 532 additions and 341 deletions
Download Patch File Download Diff File Expand all files Collapse all files

106
spring-websocket/src/test/java/org/springframework/web/socket/support/MultiProtocolWebSocketHandlerTests.java
View File

@@ -1,106 +0,0 @@
/*
* Copyright 2002-2013 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.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIOsNS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.springframework.web.socket.support;
import java.util.HashMap;
import java.util.Map;
import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.springframework.web.socket.WebSocketHandler;
import static org.mockito.Mockito.*;
/**
* Test fixture for {@link MultiProtocolWebSocketHandler}.
*
* @author Rossen Stoyanchev
*/
public class MultiProtocolWebSocketHandlerTests {
private MultiProtocolWebSocketHandler multiProtocolHandler;
@Mock
WebSocketHandler stompHandler;
@Mock
WebSocketHandler mqttHandler;
@Mock
WebSocketHandler defaultHandler;
@Before
public void setup() {
MockitoAnnotations.initMocks(this);
Map<String, WebSocketHandler> handlers = new HashMap<String, WebSocketHandler>();
handlers.put("STOMP", this.stompHandler);
handlers.put("MQTT", this.mqttHandler);
this.multiProtocolHandler = new MultiProtocolWebSocketHandler();
this.multiProtocolHandler.setProtocolHandlers(handlers);
this.multiProtocolHandler.setDefaultProtocolHandler(this.defaultHandler);
}
@Test
public void subProtocol() throws Exception {
TestWebSocketSession session = new TestWebSocketSession();
session.setAcceptedProtocol("sToMp");
this.multiProtocolHandler.afterConnectionEstablished(session);
verify(this.stompHandler).afterConnectionEstablished(session);
verifyZeroInteractions(this.mqttHandler);
}
@Test(expected=IllegalStateException.class)
public void subProtocolNoMatch() throws Exception {
TestWebSocketSession session = new TestWebSocketSession();
session.setAcceptedProtocol("wamp");
this.multiProtocolHandler.afterConnectionEstablished(session);
}
@Test
public void noSubProtocol() throws Exception {
TestWebSocketSession session = new TestWebSocketSession();
this.multiProtocolHandler.afterConnectionEstablished(session);
verify(this.defaultHandler).afterConnectionEstablished(session);
verifyZeroInteractions(this.stompHandler, this.mqttHandler);
}
@Test(expected=IllegalStateException.class)
public void noSubProtocolNoDefaultHandler() throws Exception {
TestWebSocketSession session = new TestWebSocketSession();
this.multiProtocolHandler.setDefaultProtocolHandler(null);
this.multiProtocolHandler.afterConnectionEstablished(session);
}
}