def test_emit(self): ns = asyncio_namespace.AsyncNamespace('/foo') mock_server = mock.MagicMock() mock_server.emit = AsyncMock() ns._set_server(mock_server) _run( ns.emit('ev', data='data', room='room', skip_sid='skip', callback='cb')) ns.server.emit.mock.assert_called_with('ev', data='data', room='room', skip_sid='skip', namespace='/foo', callback='cb') _run( ns.emit('ev', data='data', room='room', skip_sid='skip', namespace='/bar', callback='cb')) ns.server.emit.mock.assert_called_with('ev', data='data', room='room', skip_sid='skip', namespace='/bar', callback='cb')
def test_rooms(self): ns = asyncio_namespace.AsyncNamespace('/foo') ns._set_server(mock.MagicMock()) ns.rooms('sid') ns.server.rooms.assert_called_with('sid', namespace='/foo') ns.rooms('sid', namespace='/bar') ns.server.rooms.assert_called_with('sid', namespace='/bar')
def test_disconnect(self): ns = asyncio_namespace.AsyncNamespace('/foo') mock_server = mock.MagicMock() mock_server.disconnect = AsyncMock() ns._set_server(mock_server) _run(ns.disconnect('sid')) ns.server.disconnect.mock.assert_called_with('sid', namespace='/foo') _run(ns.disconnect('sid', namespace='/bar')) ns.server.disconnect.mock.assert_called_with('sid', namespace='/bar')
def test_close_room(self): ns = asyncio_namespace.AsyncNamespace('/foo') mock_server = mock.MagicMock() mock_server.close_room = AsyncMock() ns._set_server(mock_server) _run(ns.close_room('room')) ns.server.close_room.mock.assert_called_with('room', namespace='/foo') _run(ns.close_room('room', namespace='/bar')) ns.server.close_room.mock.assert_called_with('room', namespace='/bar')
def test_leave_room(self): ns = asyncio_namespace.AsyncNamespace('/foo') ns._set_server(mock.MagicMock()) ns.leave_room('sid', 'room') ns.server.leave_room.assert_called_with('sid', 'room', namespace='/foo') ns.leave_room('sid', 'room', namespace='/bar') ns.server.leave_room.assert_called_with('sid', 'room', namespace='/bar')
def test_session(self): ns = asyncio_namespace.AsyncNamespace('/foo') mock_server = mock.MagicMock() mock_server.get_session = AsyncMock() mock_server.save_session = AsyncMock() ns._set_server(mock_server) _run(ns.get_session('sid')) ns.server.get_session.mock.assert_called_with('sid', namespace='/foo') _run(ns.get_session('sid', namespace='/bar')) ns.server.get_session.mock.assert_called_with('sid', namespace='/bar') _run(ns.save_session('sid', {'a': 'b'})) ns.server.save_session.mock.assert_called_with('sid', {'a': 'b'}, namespace='/foo') _run(ns.save_session('sid', {'a': 'b'}, namespace='/bar')) ns.server.save_session.mock.assert_called_with('sid', {'a': 'b'}, namespace='/bar') ns.session('sid') ns.server.session.assert_called_with('sid', namespace='/foo') ns.session('sid', namespace='/bar') ns.server.session.assert_called_with('sid', namespace='/bar')