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

Use random id for WebSocket sessions

Browse Source 
Issue: SPR-17228
This commit is contained in:
Rossen Stoyanchev
2018-09-05 19:48:28 -04:00
parent d570f82456
commit 3302798e2f
4 changed files with 17 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
spring-websocket/src/test/java/org/springframework/web/socket/adapter/standard/StandardWebSocketHandlerAdapterTests.java
View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2013 the original author or authors.
* Copyright 2002-2018 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.
@@ -16,6 +16,7 @@
package org.springframework.web.socket.adapter.standard;
import java.net.URI;
import javax.websocket.CloseReason;
import javax.websocket.CloseReason.CloseCodes;
import javax.websocket.MessageHandler;
@@ -56,14 +57,15 @@ public class StandardWebSocketHandlerAdapterTests {
@Test
public void onOpen() throws Throwable {
given(this.session.getId()).willReturn("123");
URI uri = URI.create("http://example.org");
given(this.session.getRequestURI()).willReturn(uri);
this.adapter.onOpen(this.session, null);
verify(this.webSocketHandler).afterConnectionEstablished(this.webSocketSession);
verify(this.session, atLeast(2)).addMessageHandler(any(MessageHandler.Whole.class));
given(this.session.getId()).willReturn("123");
assertEquals("123", this.webSocketSession.getId());
given(this.session.getRequestURI()).willReturn(uri);
assertEquals(uri, this.webSocketSession.getUri());
}
@Test