LiveUpdateSessionHandler extends TextWebSocketHandler { @Override public void afterConnectionClosed( final WebSocketSession session, final CloseStatus status) throws Exception { manager.remove(new LiveUpdateSession(session)); super.afterConnectionClosed(session, status); }  @Override void afterConnectionEstablished(final WebSocketSession session); @Override void afterConnectionClosed(
      final WebSocketSession session, final CloseStatus status);  }
@Test public void testAfterConnectionClosed() throws Exception { final LiveUpdateService manager = mock(LiveUpdateService.class); final WebSocketSession session = mock(WebSocketSession.class); final LiveUpdateSessionHandler handler = new LiveUpdateSessionHandler(); handler.setManager(manager); handler.afterConnectionClosed(session, CloseStatus.NORMAL); verify(manager).remove(any()); }