def test_check_origin(): ac = ActiveConnections() app = tornado.web.Application() app.active_connections = ac r = MockRequest() wh = WSHandler(app, r) assert wh.check_origin(None) assert wh.check_origin(False)
def test_open_connection(): ac = ActiveConnections() app = tornado.web.Application() app.active_connections = ac r = MockRequest() wh = WSHandler(app, r) wh.open() assert wh in ac.connections
def test_con_close(): ac = ActiveConnections() app = tornado.web.Application() app.active_connections = ac r = MockRequest() wh = WSHandler(app, r) ac.connections.append(wh) wh.on_close() assert wh not in ac.connections
def test_not_subscribing_on_faile_authencitacion(mock_auth): ac = ActiveConnections() app = tornado.web.Application() app.active_connections = ac r = MockRequest() wh = WSHandler(app, r) ac.connections.append(wh) message = json.dumps({ 'subscribe': ['newuser'] }) wh.on_message(message) assert 'newuser' not in wh.subscriptions, 'Didn\'t add subscription'
def test_on_message_autehticate(mock_auth, mock_get, mock_write): ac = ActiveConnections() app = tornado.web.Application() app.active_connections = ac r = MockRequest() wh = WSHandler(app, r) message = json.dumps({ 'id': 9, 'authenticate': { 'name': 'me', 'password': '******', }, }) wh.on_message(message) mock_auth.assert_called_once_with() mock_get.assert_called_once_with('me', 'secret')
def test_on_message_registration(mock_create, mock_write): ac = ActiveConnections() app = tornado.web.Application() app.active_connections = ac r = MockRequest() wh = WSHandler(app, r) message = json.dumps({ 'id': 9, 'register': { 'name': 'me', 'password': '******', }, }) wh.on_message(message) mock_write.assert_called_once_with(json.dumps({ 'id': 9, 'type': 'registration', 'user_id': '3', }))