mirror
/
miguelgrinberg.python-socketio
mirror of https://github.com/miguelgrinberg/python-socketio
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

v5 protocol: client unit tests

pull/599/head
Miguel Grinberg 4 years ago
parent
a4cc0b2c5b
commit
0c91e92eb1
No known key found for this signature in database GPG Key ID: 36848B262DF5F06C
4 changed files with 259 additions and 239 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 13
      socketio/asyncio_client.py
  2. 16
      socketio/client.py
  3. 221
      tests/asyncio/test_asyncio_client.py
  4. 248
      tests/common/test_client.py

13
socketio/asyncio_client.py
View File

@ -96,8 +96,10 @@ class AsyncClient(client.Client):
self.socketio_path = socketio_path self.socketio_path = socketio_path
if namespaces is None: if namespaces is None:
namespaces = set(self.handlers.keys()).union( namespaces = list(set(self.handlers.keys()).union(
set(self.namespace_handlers.keys())) set(self.namespace_handlers.keys())))
if len(namespaces) == 0:
namespaces = ['/']
elif isinstance(namespaces, six.string_types): elif isinstance(namespaces, six.string_types):
namespaces = [namespaces] namespaces = [namespaces]
self.connection_namespaces = namespaces self.connection_namespaces = namespaces
@ -293,10 +295,10 @@ class AsyncClient(client.Client):
async def _handle_connect(self, namespace, data): async def _handle_connect(self, namespace, data):
namespace = namespace or '/' namespace = namespace or '/'
self.logger.info('Namespace {} is connected'.format(namespace))
if namespace not in self.namespaces: if namespace not in self.namespaces:
self.logger.info('Namespace {} is connected'.format(namespace))
self.namespaces[namespace] = (data or {}).get('sid', self.sid) self.namespaces[namespace] = (data or {}).get('sid', self.sid)
await self._trigger_event('connect', namespace=namespace) await self._trigger_event('connect', namespace=namespace)
async def _handle_disconnect(self, namespace): async def _handle_disconnect(self, namespace):
if not self.connected: if not self.connected:
@ -306,6 +308,7 @@ class AsyncClient(client.Client):
if namespace in self.namespaces: if namespace in self.namespaces:
del self.namespaces[namespace] del self.namespaces[namespace]
if not self.namespaces: if not self.namespaces:
self.connected = False
await self.eio.disconnect(abort=True) await self.eio.disconnect(abort=True)
async def _handle_event(self, namespace, id, data): async def _handle_event(self, namespace, id, data):
@ -351,7 +354,7 @@ class AsyncClient(client.Client):
data = (data,) data = (data,)
await self._trigger_event('connect_error', namespace, *data) await self._trigger_event('connect_error', namespace, *data)
if namespace in self.namespaces: if namespace in self.namespaces:
self.namespaces.remove(namespace) del self.namespaces[namespace]
if namespace == '/': if namespace == '/':
self.namespaces = {} self.namespaces = {}
self.connected = False self.connected = False

16
socketio/client.py
View File

@ -108,8 +108,7 @@ class Client(object):
self.logger = logger self.logger = logger
else: else:
self.logger = default_logger self.logger = default_logger
if not logging.root.handlers and \ if self.logger.level == logging.NOTSET:
self.logger.level == logging.NOTSET:
if logger: if logger:
self.logger.setLevel(logging.INFO) self.logger.setLevel(logging.INFO)
else: else:
@ -119,7 +118,7 @@ class Client(object):
self.connection_url = None self.connection_url = None
self.connection_headers = None self.connection_headers = None
self.connection_transports = None self.connection_transports = None
self.connection_namespaces = None self.connection_namespaces = []
self.socketio_path = None self.socketio_path = None
self.sid = None self.sid = None
@ -265,8 +264,10 @@ class Client(object):
self.socketio_path = socketio_path self.socketio_path = socketio_path
if namespaces is None: if namespaces is None:
namespaces = set(self.handlers.keys()).union( namespaces = list(set(self.handlers.keys()).union(
set(self.namespace_handlers.keys())) set(self.namespace_handlers.keys())))
if len(namespaces) == 0:
namespaces = ['/']
elif isinstance(namespaces, six.string_types): elif isinstance(namespaces, six.string_types):
namespaces = [namespaces] namespaces = [namespaces]
self.connection_namespaces = namespaces self.connection_namespaces = namespaces
@ -476,10 +477,10 @@ class Client(object):
def _handle_connect(self, namespace, data): def _handle_connect(self, namespace, data):
namespace = namespace or '/' namespace = namespace or '/'
self.logger.info('Namespace {} is connected'.format(namespace))
if namespace not in self.namespaces: if namespace not in self.namespaces:
self.logger.info('Namespace {} is connected'.format(namespace))
self.namespaces[namespace] = (data or {}).get('sid', self.sid) self.namespaces[namespace] = (data or {}).get('sid', self.sid)
self._trigger_event('connect', namespace=namespace) self._trigger_event('connect', namespace=namespace)
def _handle_disconnect(self, namespace): def _handle_disconnect(self, namespace):
if not self.connected: if not self.connected:
@ -489,6 +490,7 @@ class Client(object):
if namespace in self.namespaces: if namespace in self.namespaces:
del self.namespaces[namespace] del self.namespaces[namespace]
if not self.namespaces: if not self.namespaces:
self.connected = False
self.eio.disconnect(abort=True) self.eio.disconnect(abort=True)
def _handle_event(self, namespace, id, data): def _handle_event(self, namespace, id, data):

221
tests/asyncio/test_asyncio_client.py
View File

@ -70,7 +70,6 @@ class TestAsyncClient(unittest.TestCase):
assert c.connection_transports == 'transports' assert c.connection_transports == 'transports'
assert c.connection_namespaces == ['/foo', '/', '/bar'] assert c.connection_namespaces == ['/foo', '/', '/bar']
assert c.socketio_path == 'path' assert c.socketio_path == 'path'
assert c.namespaces == ['/foo', '/bar']
c.eio.connect.mock.assert_called_once_with( c.eio.connect.mock.assert_called_once_with(
'url', 'url',
headers='headers', headers='headers',
@ -95,7 +94,6 @@ class TestAsyncClient(unittest.TestCase):
assert c.connection_transports == 'transports' assert c.connection_transports == 'transports'
assert c.connection_namespaces == ['/foo'] assert c.connection_namespaces == ['/foo']
assert c.socketio_path == 'path' assert c.socketio_path == 'path'
assert c.namespaces == ['/foo']
c.eio.connect.mock.assert_called_once_with( c.eio.connect.mock.assert_called_once_with(
'url', 'url',
headers='headers', headers='headers',
@ -119,9 +117,32 @@ class TestAsyncClient(unittest.TestCase):
assert c.connection_url == 'url' assert c.connection_url == 'url'
assert c.connection_headers == 'headers' assert c.connection_headers == 'headers'
assert c.connection_transports == 'transports' assert c.connection_transports == 'transports'
assert c.connection_namespaces is None assert c.connection_namespaces == ['/', '/foo'] or \
c.connection_namespaces == ['/foo', '/']
assert c.socketio_path == 'path'
c.eio.connect.mock.assert_called_once_with(
'url',
headers='headers',
transports='transports',
engineio_path='path',
)
def test_connect_no_namespaces(self):
c = asyncio_client.AsyncClient()
c.eio.connect = AsyncMock()
_run(
c.connect(
'url',
headers='headers',
transports='transports',
socketio_path='path',
)
)
assert c.connection_url == 'url'
assert c.connection_headers == 'headers'
assert c.connection_transports == 'transports'
assert c.connection_namespaces == ['/']
assert c.socketio_path == 'path' assert c.socketio_path == 'path'
assert c.namespaces == ['/foo']
c.eio.connect.mock.assert_called_once_with( c.eio.connect.mock.assert_called_once_with(
'url', 'url',
headers='headers', headers='headers',
@ -186,11 +207,11 @@ class TestAsyncClient(unittest.TestCase):
def test_emit_no_arguments(self): def test_emit_no_arguments(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = {'/': '1'}
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c.emit('foo')) _run(c.emit('foo'))
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.EVENT, namespace='/', data=['foo'], id=None, binary=False packet.EVENT, namespace='/', data=['foo'], id=None)
)
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
assert ( assert (
c._send_packet.mock.call_args_list[0][0][0].encode() c._send_packet.mock.call_args_list[0][0][0].encode()
@ -199,6 +220,7 @@ class TestAsyncClient(unittest.TestCase):
def test_emit_one_argument(self): def test_emit_one_argument(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = {'/': '1'}
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c.emit('foo', 'bar')) _run(c.emit('foo', 'bar'))
expected_packet = packet.Packet( expected_packet = packet.Packet(
@ -206,7 +228,6 @@ class TestAsyncClient(unittest.TestCase):
namespace='/', namespace='/',
data=['foo', 'bar'], data=['foo', 'bar'],
id=None, id=None,
binary=False,
) )
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
assert ( assert (
@ -216,6 +237,7 @@ class TestAsyncClient(unittest.TestCase):
def test_emit_one_argument_list(self): def test_emit_one_argument_list(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = {'/': '1'}
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c.emit('foo', ['bar', 'baz'])) _run(c.emit('foo', ['bar', 'baz']))
expected_packet = packet.Packet( expected_packet = packet.Packet(
@ -223,7 +245,6 @@ class TestAsyncClient(unittest.TestCase):
namespace='/', namespace='/',
data=['foo', ['bar', 'baz']], data=['foo', ['bar', 'baz']],
id=None, id=None,
binary=False,
) )
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
assert ( assert (
@ -233,6 +254,7 @@ class TestAsyncClient(unittest.TestCase):
def test_emit_two_arguments(self): def test_emit_two_arguments(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = {'/': '1'}
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c.emit('foo', ('bar', 'baz'))) _run(c.emit('foo', ('bar', 'baz')))
expected_packet = packet.Packet( expected_packet = packet.Packet(
@ -240,7 +262,6 @@ class TestAsyncClient(unittest.TestCase):
namespace='/', namespace='/',
data=['foo', 'bar', 'baz'], data=['foo', 'bar', 'baz'],
id=None, id=None,
binary=False,
) )
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
assert ( assert (
@ -250,12 +271,11 @@ class TestAsyncClient(unittest.TestCase):
def test_emit_namespace(self): def test_emit_namespace(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = ['/foo'] c.namespaces = {'/foo': '1'}
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c.emit('foo', namespace='/foo')) _run(c.emit('foo', namespace='/foo'))
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.EVENT, namespace='/foo', data=['foo'], id=None, binary=False packet.EVENT, namespace='/foo', data=['foo'], id=None)
)
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
assert ( assert (
c._send_packet.mock.call_args_list[0][0][0].encode() c._send_packet.mock.call_args_list[0][0][0].encode()
@ -264,7 +284,7 @@ class TestAsyncClient(unittest.TestCase):
def test_emit_unknown_namespace(self): def test_emit_unknown_namespace(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = ['/foo'] c.namespaces = {'/foo': '1'}
with pytest.raises(exceptions.BadNamespaceError): with pytest.raises(exceptions.BadNamespaceError):
_run(c.emit('foo', namespace='/bar')) _run(c.emit('foo', namespace='/bar'))
@ -272,10 +292,10 @@ class TestAsyncClient(unittest.TestCase):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
c._generate_ack_id = mock.MagicMock(return_value=123) c._generate_ack_id = mock.MagicMock(return_value=123)
c.namespaces = {'/': '1'}
_run(c.emit('foo', callback='cb')) _run(c.emit('foo', callback='cb'))
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.EVENT, namespace='/', data=['foo'], id=123, binary=False packet.EVENT, namespace='/', data=['foo'], id=123)
)
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
assert ( assert (
c._send_packet.mock.call_args_list[0][0][0].encode() c._send_packet.mock.call_args_list[0][0][0].encode()
@ -285,13 +305,12 @@ class TestAsyncClient(unittest.TestCase):
def test_emit_namespace_with_callback(self): def test_emit_namespace_with_callback(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = ['/foo'] c.namespaces = {'/foo': '1'}
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
c._generate_ack_id = mock.MagicMock(return_value=123) c._generate_ack_id = mock.MagicMock(return_value=123)
_run(c.emit('foo', namespace='/foo', callback='cb')) _run(c.emit('foo', namespace='/foo', callback='cb'))
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.EVENT, namespace='/foo', data=['foo'], id=123, binary=False packet.EVENT, namespace='/foo', data=['foo'], id=123)
)
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
assert ( assert (
c._send_packet.mock.call_args_list[0][0][0].encode() c._send_packet.mock.call_args_list[0][0][0].encode()
@ -300,7 +319,8 @@ class TestAsyncClient(unittest.TestCase):
c._generate_ack_id.assert_called_once_with('/foo', 'cb') c._generate_ack_id.assert_called_once_with('/foo', 'cb')
def test_emit_binary(self): def test_emit_binary(self):
c = asyncio_client.AsyncClient(binary=True) c = asyncio_client.AsyncClient()
c.namespaces = {'/': '1'}
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c.emit('foo', b'bar')) _run(c.emit('foo', b'bar'))
expected_packet = packet.Packet( expected_packet = packet.Packet(
@ -308,7 +328,6 @@ class TestAsyncClient(unittest.TestCase):
namespace='/', namespace='/',
data=['foo', b'bar'], data=['foo', b'bar'],
id=None, id=None,
binary=True,
) )
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
assert ( assert (
@ -317,7 +336,8 @@ class TestAsyncClient(unittest.TestCase):
) )
def test_emit_not_binary(self): def test_emit_not_binary(self):
c = asyncio_client.AsyncClient(binary=False) c = asyncio_client.AsyncClient()
c.namespaces = {'/': '1'}
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c.emit('foo', 'bar')) _run(c.emit('foo', 'bar'))
expected_packet = packet.Packet( expected_packet = packet.Packet(
@ -325,7 +345,6 @@ class TestAsyncClient(unittest.TestCase):
namespace='/', namespace='/',
data=['foo', 'bar'], data=['foo', 'bar'],
id=None, id=None,
binary=False,
) )
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
assert ( assert (
@ -351,7 +370,7 @@ class TestAsyncClient(unittest.TestCase):
def test_call(self): def test_call(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = {'/': '1'}
async def fake_event_wait(): async def fake_event_wait():
c._generate_ack_id.call_args_list[0][0][1]('foo', 321) c._generate_ack_id.call_args_list[0][0][1]('foo', 321)
@ -361,8 +380,7 @@ class TestAsyncClient(unittest.TestCase):
c.eio.create_event.return_value.wait = fake_event_wait c.eio.create_event.return_value.wait = fake_event_wait
assert _run(c.call('foo')) == ('foo', 321) assert _run(c.call('foo')) == ('foo', 321)
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.EVENT, namespace='/', data=['foo'], id=123, binary=False packet.EVENT, namespace='/', data=['foo'], id=123)
)
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
assert ( assert (
c._send_packet.mock.call_args_list[0][0][0].encode() c._send_packet.mock.call_args_list[0][0][0].encode()
@ -371,6 +389,7 @@ class TestAsyncClient(unittest.TestCase):
def test_call_with_timeout(self): def test_call_with_timeout(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = {'/': '1'}
async def fake_event_wait(): async def fake_event_wait():
await asyncio.sleep(1) await asyncio.sleep(1)
@ -382,8 +401,7 @@ class TestAsyncClient(unittest.TestCase):
with pytest.raises(exceptions.TimeoutError): with pytest.raises(exceptions.TimeoutError):
_run(c.call('foo', timeout=0.01)) _run(c.call('foo', timeout=0.01))
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.EVENT, namespace='/', data=['foo'], id=123, binary=False packet.EVENT, namespace='/', data=['foo'], id=123)
)
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
assert ( assert (
c._send_packet.mock.call_args_list[0][0][0].encode() c._send_packet.mock.call_args_list[0][0][0].encode()
@ -392,6 +410,7 @@ class TestAsyncClient(unittest.TestCase):
def test_disconnect(self): def test_disconnect(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = {'/': '1'}
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
c.eio = mock.MagicMock() c.eio = mock.MagicMock()
@ -409,7 +428,7 @@ class TestAsyncClient(unittest.TestCase):
def test_disconnect_namespaces(self): def test_disconnect_namespaces(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
c.eio = mock.MagicMock() c.eio = mock.MagicMock()
@ -417,7 +436,7 @@ class TestAsyncClient(unittest.TestCase):
c.eio.state = 'connected' c.eio.state = 'connected'
_run(c.disconnect()) _run(c.disconnect())
assert c._trigger_event.mock.call_count == 0 assert c._trigger_event.mock.call_count == 0
assert c._send_packet.mock.call_count == 3 assert c._send_packet.mock.call_count == 2
expected_packet = packet.Packet(packet.DISCONNECT, namespace='/foo') expected_packet = packet.Packet(packet.DISCONNECT, namespace='/foo')
assert ( assert (
c._send_packet.mock.call_args_list[0][0][0].encode() c._send_packet.mock.call_args_list[0][0][0].encode()
@ -428,11 +447,6 @@ class TestAsyncClient(unittest.TestCase):
c._send_packet.mock.call_args_list[1][0][0].encode() c._send_packet.mock.call_args_list[1][0][0].encode()
== expected_packet.encode() == expected_packet.encode()
) )
expected_packet = packet.Packet(packet.DISCONNECT, namespace='/')
assert (
c._send_packet.mock.call_args_list[2][0][0].encode()
== expected_packet.encode()
)
def test_start_background_task(self): def test_start_background_task(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
@ -451,67 +465,55 @@ class TestAsyncClient(unittest.TestCase):
def test_send_packet(self): def test_send_packet(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.eio.send = AsyncMock() c.eio.send = AsyncMock()
_run(c._send_packet(packet.Packet(packet.EVENT, 'foo', binary=False))) _run(c._send_packet(packet.Packet(packet.EVENT, 'foo')))
c.eio.send.mock.assert_called_once_with('2"foo"', binary=False) c.eio.send.mock.assert_called_once_with('2"foo"')
def test_send_packet_binary(self): def test_send_packet_binary(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.eio.send = AsyncMock() c.eio.send = AsyncMock()
_run(c._send_packet(packet.Packet(packet.EVENT, b'foo', binary=True))) _run(c._send_packet(packet.Packet(packet.EVENT, b'foo')))
assert c.eio.send.mock.call_args_list == [ assert c.eio.send.mock.call_args_list == [
mock.call('51-{"_placeholder":true,"num":0}', binary=False), mock.call('51-{"_placeholder":true,"num":0}'),
mock.call(b'foo', binary=True), mock.call(b'foo'),
] or c.eio.send.mock.call_args_list == [ ] or c.eio.send.mock.call_args_list == [
mock.call('51-{"num":0,"_placeholder":true}', binary=False), mock.call('51-{"num":0,"_placeholder":true}'),
mock.call(b'foo', binary=True), mock.call(b'foo'),
] ]
def test_send_packet_default_binary_py3(self): def test_send_packet_default_binary(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.eio.send = AsyncMock() c.eio.send = AsyncMock()
_run(c._send_packet(packet.Packet(packet.EVENT, 'foo'))) _run(c._send_packet(packet.Packet(packet.EVENT, 'foo')))
c.eio.send.mock.assert_called_once_with('2"foo"', binary=False) c.eio.send.mock.assert_called_once_with('2"foo"')
def test_handle_connect(self): def test_handle_connect(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c._handle_connect('/')) _run(c._handle_connect('/', {'sid': '123'}))
c._trigger_event.mock.assert_called_once_with('connect', namespace='/') c._trigger_event.mock.assert_called_once_with('connect', namespace='/')
c._send_packet.mock.assert_not_called() c._send_packet.mock.assert_not_called()
def test_handle_connect_with_namespaces(self): def test_handle_connect_with_namespaces(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c._handle_connect('/')) _run(c._handle_connect('/', {'sid': '3'}))
c._trigger_event.mock.assert_called_once_with('connect', namespace='/') c._trigger_event.mock.assert_called_once_with('connect', namespace='/')
assert c._send_packet.mock.call_count == 2 assert c.namespaces == {'/': '3', '/foo': '1', '/bar': '2'}
expected_packet = packet.Packet(packet.CONNECT, namespace='/foo')
assert (
c._send_packet.mock.call_args_list[0][0][0].encode()
== expected_packet.encode()
)
expected_packet = packet.Packet(packet.CONNECT, namespace='/bar')
assert (
c._send_packet.mock.call_args_list[1][0][0].encode()
== expected_packet.encode()
)
def test_handle_connect_namespace(self): def test_handle_connect_namespace(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = ['/foo'] c.namespaces = {'/foo': '1'}
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c._handle_connect('/foo')) _run(c._handle_connect('/foo', {'sid': '123'}))
_run(c._handle_connect('/bar')) _run(c._handle_connect('/bar', {'sid': '2'}))
assert c._trigger_event.mock.call_args_list == [ assert c._trigger_event.mock.call_count == 1
mock.call('connect', namespace='/foo'), c._trigger_event.mock.assert_called_once_with(
mock.call('connect', namespace='/bar'), 'connect', namespace='/bar')
] assert c.namespaces == {'/foo': '1', '/bar': '2'}
c._send_packet.mock.assert_not_called()
assert c.namespaces == ['/foo', '/bar']
def test_handle_disconnect(self): def test_handle_disconnect(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
@ -528,38 +530,36 @@ class TestAsyncClient(unittest.TestCase):
def test_handle_disconnect_namespace(self): def test_handle_disconnect_namespace(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
_run(c._handle_disconnect('/foo')) _run(c._handle_disconnect('/foo'))
c._trigger_event.mock.assert_called_once_with( c._trigger_event.mock.assert_called_once_with(
'disconnect', namespace='/foo' 'disconnect', namespace='/foo'
) )
assert c.namespaces == ['/bar'] assert c.namespaces == {'/bar': '2'}
assert c.connected assert c.connected
def test_handle_disconnect_unknown_namespace(self): def test_handle_disconnect_unknown_namespace(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
_run(c._handle_disconnect('/baz')) _run(c._handle_disconnect('/baz'))
c._trigger_event.mock.assert_called_once_with( c._trigger_event.mock.assert_called_once_with(
'disconnect', namespace='/baz' 'disconnect', namespace='/baz'
) )
assert c.namespaces == ['/foo', '/bar'] assert c.namespaces == {'/foo': '1', '/bar': '2'}
assert c.connected assert c.connected
def test_handle_disconnect_all_namespaces(self): def test_handle_disconnect_default_namespaces(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
_run(c._handle_disconnect('/')) _run(c._handle_disconnect('/'))
c._trigger_event.mock.assert_any_call('disconnect', namespace='/') c._trigger_event.mock.assert_called_with('disconnect', namespace='/')
c._trigger_event.mock.assert_any_call('disconnect', namespace='/foo') assert c.namespaces == {'/foo': '1', '/bar': '2'}
c._trigger_event.mock.assert_any_call('disconnect', namespace='/bar') assert c.connected
assert c.namespaces == []
assert not c.connected
def test_handle_event(self): def test_handle_event(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
@ -570,7 +570,7 @@ class TestAsyncClient(unittest.TestCase):
) )
def test_handle_event_with_id_no_arguments(self): def test_handle_event_with_id_no_arguments(self):
c = asyncio_client.AsyncClient(binary=True) c = asyncio_client.AsyncClient()
c._trigger_event = AsyncMock(return_value=None) c._trigger_event = AsyncMock(return_value=None)
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c._handle_event('/', 123, ['foo', ('bar', 'baz')])) _run(c._handle_event('/', 123, ['foo', ('bar', 'baz')]))
@ -579,15 +579,14 @@ class TestAsyncClient(unittest.TestCase):
) )
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.ACK, namespace='/', id=123, data=[], binary=None packet.ACK, namespace='/', id=123, data=[])
)
assert ( assert (
c._send_packet.mock.call_args_list[0][0][0].encode() c._send_packet.mock.call_args_list[0][0][0].encode()
== expected_packet.encode() == expected_packet.encode()
) )
def test_handle_event_with_id_one_argument(self): def test_handle_event_with_id_one_argument(self):
c = asyncio_client.AsyncClient(binary=True) c = asyncio_client.AsyncClient()
c._trigger_event = AsyncMock(return_value='ret') c._trigger_event = AsyncMock(return_value='ret')
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c._handle_event('/', 123, ['foo', ('bar', 'baz')])) _run(c._handle_event('/', 123, ['foo', ('bar', 'baz')]))
@ -596,15 +595,14 @@ class TestAsyncClient(unittest.TestCase):
) )
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.ACK, namespace='/', id=123, data=['ret'], binary=None packet.ACK, namespace='/', id=123, data=['ret'])
)
assert ( assert (
c._send_packet.mock.call_args_list[0][0][0].encode() c._send_packet.mock.call_args_list[0][0][0].encode()
== expected_packet.encode() == expected_packet.encode()
) )
def test_handle_event_with_id_one_list_argument(self): def test_handle_event_with_id_one_list_argument(self):
c = asyncio_client.AsyncClient(binary=True) c = asyncio_client.AsyncClient()
c._trigger_event = AsyncMock(return_value=['a', 'b']) c._trigger_event = AsyncMock(return_value=['a', 'b'])
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c._handle_event('/', 123, ['foo', ('bar', 'baz')])) _run(c._handle_event('/', 123, ['foo', ('bar', 'baz')]))
@ -613,15 +611,14 @@ class TestAsyncClient(unittest.TestCase):
) )
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.ACK, namespace='/', id=123, data=[['a', 'b']], binary=None packet.ACK, namespace='/', id=123, data=[['a', 'b']])
)
assert ( assert (
c._send_packet.mock.call_args_list[0][0][0].encode() c._send_packet.mock.call_args_list[0][0][0].encode()
== expected_packet.encode() == expected_packet.encode()
) )
def test_handle_event_with_id_two_arguments(self): def test_handle_event_with_id_two_arguments(self):
c = asyncio_client.AsyncClient(binary=True) c = asyncio_client.AsyncClient()
c._trigger_event = AsyncMock(return_value=('a', 'b')) c._trigger_event = AsyncMock(return_value=('a', 'b'))
c._send_packet = AsyncMock() c._send_packet = AsyncMock()
_run(c._handle_event('/', 123, ['foo', ('bar', 'baz')])) _run(c._handle_event('/', 123, ['foo', ('bar', 'baz')]))
@ -630,8 +627,7 @@ class TestAsyncClient(unittest.TestCase):
) )
assert c._send_packet.mock.call_count == 1 assert c._send_packet.mock.call_count == 1
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.ACK, namespace='/', id=123, data=['a', 'b'], binary=None packet.ACK, namespace='/', id=123, data=['a', 'b'])
)
assert ( assert (
c._send_packet.mock.call_args_list[0][0][0].encode() c._send_packet.mock.call_args_list[0][0][0].encode()
== expected_packet.encode() == expected_packet.encode()
@ -665,9 +661,9 @@ class TestAsyncClient(unittest.TestCase):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.connected = True c.connected = True
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
_run(c._handle_error('/', 'error')) _run(c._handle_error('/', 'error'))
assert c.namespaces == [] assert c.namespaces == {}
assert not c.connected assert not c.connected
c._trigger_event.mock.assert_called_once_with( c._trigger_event.mock.assert_called_once_with(
'connect_error', '/', 'error' 'connect_error', '/', 'error'
@ -677,19 +673,19 @@ class TestAsyncClient(unittest.TestCase):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.connected = True c.connected = True
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
_run(c._handle_error('/', None)) _run(c._handle_error('/', None))
assert c.namespaces == [] assert c.namespaces == {}
assert not c.connected assert not c.connected
c._trigger_event.mock.assert_called_once_with('connect_error', '/') c._trigger_event.mock.assert_called_once_with('connect_error', '/')
def test_handle_error_namespace(self): def test_handle_error_namespace(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
_run(c._handle_error('/bar', ['error', 'message'])) _run(c._handle_error('/bar', ['error', 'message']))
assert c.namespaces == ['/foo'] assert c.namespaces == {'/foo': '1'}
assert c.connected assert c.connected
c._trigger_event.mock.assert_called_once_with( c._trigger_event.mock.assert_called_once_with(
'connect_error', '/bar', 'error', 'message' 'connect_error', '/bar', 'error', 'message'
@ -698,19 +694,19 @@ class TestAsyncClient(unittest.TestCase):
def test_handle_error_namespace_with_no_arguments(self): def test_handle_error_namespace_with_no_arguments(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
_run(c._handle_error('/bar', None)) _run(c._handle_error('/bar', None))
assert c.namespaces == ['/foo'] assert c.namespaces == {'/foo': '1'}
assert c.connected assert c.connected
c._trigger_event.mock.assert_called_once_with('connect_error', '/bar') c._trigger_event.mock.assert_called_once_with('connect_error', '/bar')
def test_handle_error_unknown_namespace(self): def test_handle_error_unknown_namespace(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
_run(c._handle_error('/baz', 'error')) _run(c._handle_error('/baz', 'error'))
assert c.namespaces == ['/foo', '/bar'] assert c.namespaces == {'/foo': '1', '/bar': '2'}
assert c.connected assert c.connected
def test_trigger_event(self): def test_trigger_event(self):
@ -839,10 +835,23 @@ class TestAsyncClient(unittest.TestCase):
def test_eio_connect(self): def test_eio_connect(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.connection_namespaces = ['/', '/foo']
c._send_packet = AsyncMock()
c.eio.sid = 'foo' c.eio.sid = 'foo'
assert c.sid is None assert c.sid is None
c._handle_eio_connect() _run(c._handle_eio_connect())
assert c.sid == 'foo' assert c.sid == 'foo'
assert c._send_packet.mock.call_count == 2
expected_packet = packet.Packet(packet.CONNECT, namespace='/')
assert (
c._send_packet.mock.call_args_list[0][0][0].encode()
== expected_packet.encode()
)
expected_packet = packet.Packet(packet.CONNECT, namespace='/foo')
assert (
c._send_packet.mock.call_args_list[1][0][0].encode()
== expected_packet.encode()
)
def test_handle_eio_message(self): def test_handle_eio_message(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
@ -852,10 +861,10 @@ class TestAsyncClient(unittest.TestCase):
c._handle_ack = AsyncMock() c._handle_ack = AsyncMock()
c._handle_error = AsyncMock() c._handle_error = AsyncMock()
_run(c._handle_eio_message('0')) _run(c._handle_eio_message('0{"sid":"123"}'))
c._handle_connect.mock.assert_called_with(None) c._handle_connect.mock.assert_called_with(None, {'sid': '123'})
_run(c._handle_eio_message('0/foo')) _run(c._handle_eio_message('0/foo,{"sid":"123"}'))
c._handle_connect.mock.assert_called_with('/foo') c._handle_connect.mock.assert_called_with('/foo', {'sid': '123'})
_run(c._handle_eio_message('1')) _run(c._handle_eio_message('1'))
c._handle_disconnect.mock.assert_called_with(None) c._handle_disconnect.mock.assert_called_with(None)
_run(c._handle_eio_message('1/foo')) _run(c._handle_eio_message('1/foo'))
@ -895,6 +904,7 @@ class TestAsyncClient(unittest.TestCase):
def test_eio_disconnect(self): def test_eio_disconnect(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = {'/': '1'}
c.connected = True c.connected = True
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
c.sid = 'foo' c.sid = 'foo'
@ -908,15 +918,14 @@ class TestAsyncClient(unittest.TestCase):
def test_eio_disconnect_namespaces(self): def test_eio_disconnect_namespaces(self):
c = asyncio_client.AsyncClient() c = asyncio_client.AsyncClient()
c.namespaces = {'/foo': '1', '/bar': '2'}
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar']
c._trigger_event = AsyncMock() c._trigger_event = AsyncMock()
c.sid = 'foo' c.sid = 'foo'
c.eio.state = 'connected' c.eio.state = 'connected'
_run(c._handle_eio_disconnect()) _run(c._handle_eio_disconnect())
c._trigger_event.mock.assert_any_call('disconnect', namespace='/foo') c._trigger_event.mock.assert_any_call('disconnect', namespace='/foo')
c._trigger_event.mock.assert_any_call('disconnect', namespace='/bar') c._trigger_event.mock.assert_any_call('disconnect', namespace='/bar')
c._trigger_event.mock.assert_any_call('disconnect', namespace='/')
assert c.sid is None assert c.sid is None
assert not c.connected assert not c.connected

248
tests/common/test_client.py
View File

@ -37,7 +37,6 @@ class TestClient(unittest.TestCase):
reconnection_delay=5, reconnection_delay=5,
reconnection_delay_max=10, reconnection_delay_max=10,
randomization_factor=0.2, randomization_factor=0.2,
binary=True,
foo='bar', foo='bar',
) )
assert not c.reconnection assert not c.reconnection
@ -45,16 +44,15 @@ class TestClient(unittest.TestCase):
assert c.reconnection_delay == 5 assert c.reconnection_delay == 5
assert c.reconnection_delay_max == 10 assert c.reconnection_delay_max == 10
assert c.randomization_factor == 0.2 assert c.randomization_factor == 0.2
assert c.binary
engineio_client_class().assert_called_once_with(foo='bar') engineio_client_class().assert_called_once_with(foo='bar')
assert c.connection_url is None assert c.connection_url is None
assert c.connection_headers is None assert c.connection_headers is None
assert c.connection_transports is None assert c.connection_transports is None
assert c.connection_namespaces is None assert c.connection_namespaces == []
assert c.socketio_path is None assert c.socketio_path is None
assert c.sid is None assert c.sid is None
assert c.namespaces == [] assert c.namespaces == {}
assert c.handlers == {} assert c.handlers == {}
assert c.namespace_handlers == {} assert c.namespace_handlers == {}
assert c.callbacks == {} assert c.callbacks == {}
@ -174,7 +172,6 @@ class TestClient(unittest.TestCase):
assert c.connection_transports == 'transports' assert c.connection_transports == 'transports'
assert c.connection_namespaces == ['/foo', '/', '/bar'] assert c.connection_namespaces == ['/foo', '/', '/bar']
assert c.socketio_path == 'path' assert c.socketio_path == 'path'
assert c.namespaces == ['/foo', '/bar']
c.eio.connect.assert_called_once_with( c.eio.connect.assert_called_once_with(
'url', 'url',
headers='headers', headers='headers',
@ -197,7 +194,6 @@ class TestClient(unittest.TestCase):
assert c.connection_transports == 'transports' assert c.connection_transports == 'transports'
assert c.connection_namespaces == ['/foo'] assert c.connection_namespaces == ['/foo']
assert c.socketio_path == 'path' assert c.socketio_path == 'path'
assert c.namespaces == ['/foo']
c.eio.connect.assert_called_once_with( c.eio.connect.assert_called_once_with(
'url', 'url',
headers='headers', headers='headers',
@ -219,9 +215,30 @@ class TestClient(unittest.TestCase):
assert c.connection_url == 'url' assert c.connection_url == 'url'
assert c.connection_headers == 'headers' assert c.connection_headers == 'headers'
assert c.connection_transports == 'transports' assert c.connection_transports == 'transports'
assert c.connection_namespaces is None assert c.connection_namespaces == ['/foo', '/'] or \
c.connection_namespaces == ['/', '/foo']
assert c.socketio_path == 'path'
c.eio.connect.assert_called_once_with(
'url',
headers='headers',
transports='transports',
engineio_path='path',
)
def test_connect_no_namespaces(self):
c = client.Client()
c.eio.connect = mock.MagicMock()
c.connect(
'url',
headers='headers',
transports='transports',
socketio_path='path',
)
assert c.connection_url == 'url'
assert c.connection_headers == 'headers'
assert c.connection_transports == 'transports'
assert c.connection_namespaces == ['/']
assert c.socketio_path == 'path' assert c.socketio_path == 'path'
assert c.namespaces == ['/foo']
c.eio.connect.assert_called_once_with( c.eio.connect.assert_called_once_with(
'url', 'url',
headers='headers', headers='headers',
@ -283,13 +300,21 @@ class TestClient(unittest.TestCase):
assert c.eio.wait.call_count == 2 assert c.eio.wait.call_count == 2
assert c.sleep.call_count == 2 assert c.sleep.call_count == 2
def test_get_sid(self):
c = client.Client()
c.namespaces = {'/': '1', '/foo': '2'}
assert c.get_sid() == '1'
assert c.get_sid('/') == '1'
assert c.get_sid('/foo') == '2'
assert c.get_sid('/bar') is None
def test_emit_no_arguments(self): def test_emit_no_arguments(self):
c = client.Client() c = client.Client()
c.namespaces = {'/': '1'}
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c.emit('foo') c.emit('foo')
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.EVENT, namespace='/', data=['foo'], id=None, binary=False packet.EVENT, namespace='/', data=['foo'], id=None)
)
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
assert ( assert (
c._send_packet.call_args_list[0][0][0].encode() c._send_packet.call_args_list[0][0][0].encode()
@ -298,6 +323,7 @@ class TestClient(unittest.TestCase):
def test_emit_one_argument(self): def test_emit_one_argument(self):
c = client.Client() c = client.Client()
c.namespaces = {'/': '1'}
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c.emit('foo', 'bar') c.emit('foo', 'bar')
expected_packet = packet.Packet( expected_packet = packet.Packet(
@ -305,7 +331,6 @@ class TestClient(unittest.TestCase):
namespace='/', namespace='/',
data=['foo', 'bar'], data=['foo', 'bar'],
id=None, id=None,
binary=False,
) )
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
assert ( assert (
@ -315,6 +340,7 @@ class TestClient(unittest.TestCase):
def test_emit_one_argument_list(self): def test_emit_one_argument_list(self):
c = client.Client() c = client.Client()
c.namespaces = {'/': '1'}
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c.emit('foo', ['bar', 'baz']) c.emit('foo', ['bar', 'baz'])
expected_packet = packet.Packet( expected_packet = packet.Packet(
@ -322,7 +348,6 @@ class TestClient(unittest.TestCase):
namespace='/', namespace='/',
data=['foo', ['bar', 'baz']], data=['foo', ['bar', 'baz']],
id=None, id=None,
binary=False,
) )
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
assert ( assert (
@ -332,6 +357,7 @@ class TestClient(unittest.TestCase):
def test_emit_two_arguments(self): def test_emit_two_arguments(self):
c = client.Client() c = client.Client()
c.namespaces = {'/': '1'}
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c.emit('foo', ('bar', 'baz')) c.emit('foo', ('bar', 'baz'))
expected_packet = packet.Packet( expected_packet = packet.Packet(
@ -339,7 +365,6 @@ class TestClient(unittest.TestCase):
namespace='/', namespace='/',
data=['foo', 'bar', 'baz'], data=['foo', 'bar', 'baz'],
id=None, id=None,
binary=False,
) )
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
assert ( assert (
@ -353,8 +378,7 @@ class TestClient(unittest.TestCase):
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c.emit('foo', namespace='/foo') c.emit('foo', namespace='/foo')
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.EVENT, namespace='/foo', data=['foo'], id=None, binary=False packet.EVENT, namespace='/foo', data=['foo'], id=None)
)
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
assert ( assert (
c._send_packet.call_args_list[0][0][0].encode() c._send_packet.call_args_list[0][0][0].encode()
@ -369,12 +393,12 @@ class TestClient(unittest.TestCase):
def test_emit_with_callback(self): def test_emit_with_callback(self):
c = client.Client() c = client.Client()
c.namespaces = {'/': '1'}
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c._generate_ack_id = mock.MagicMock(return_value=123) c._generate_ack_id = mock.MagicMock(return_value=123)
c.emit('foo', callback='cb') c.emit('foo', callback='cb')
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.EVENT, namespace='/', data=['foo'], id=123, binary=False packet.EVENT, namespace='/', data=['foo'], id=123)
)
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
assert ( assert (
c._send_packet.call_args_list[0][0][0].encode() c._send_packet.call_args_list[0][0][0].encode()
@ -384,13 +408,12 @@ class TestClient(unittest.TestCase):
def test_emit_namespace_with_callback(self): def test_emit_namespace_with_callback(self):
c = client.Client() c = client.Client()
c.namespaces = ['/foo'] c.namespaces = {'/foo': '1'}
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c._generate_ack_id = mock.MagicMock(return_value=123) c._generate_ack_id = mock.MagicMock(return_value=123)
c.emit('foo', namespace='/foo', callback='cb') c.emit('foo', namespace='/foo', callback='cb')
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.EVENT, namespace='/foo', data=['foo'], id=123, binary=False packet.EVENT, namespace='/foo', data=['foo'], id=123)
)
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
assert ( assert (
c._send_packet.call_args_list[0][0][0].encode() c._send_packet.call_args_list[0][0][0].encode()
@ -399,7 +422,8 @@ class TestClient(unittest.TestCase):
c._generate_ack_id.assert_called_once_with('/foo', 'cb') c._generate_ack_id.assert_called_once_with('/foo', 'cb')
def test_emit_binary(self): def test_emit_binary(self):
c = client.Client(binary=True) c = client.Client()
c.namespaces = {'/': '1'}
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c.emit('foo', b'bar') c.emit('foo', b'bar')
expected_packet = packet.Packet( expected_packet = packet.Packet(
@ -407,7 +431,6 @@ class TestClient(unittest.TestCase):
namespace='/', namespace='/',
data=['foo', b'bar'], data=['foo', b'bar'],
id=None, id=None,
binary=True,
) )
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
assert ( assert (
@ -416,7 +439,8 @@ class TestClient(unittest.TestCase):
) )
def test_emit_not_binary(self): def test_emit_not_binary(self):
c = client.Client(binary=False) c = client.Client()
c.namespaces = {'/': '1'}
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c.emit('foo', 'bar') c.emit('foo', 'bar')
expected_packet = packet.Packet( expected_packet = packet.Packet(
@ -424,7 +448,6 @@ class TestClient(unittest.TestCase):
namespace='/', namespace='/',
data=['foo', 'bar'], data=['foo', 'bar'],
id=None, id=None,
binary=False,
) )
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
assert ( assert (
@ -450,7 +473,7 @@ class TestClient(unittest.TestCase):
def test_call(self): def test_call(self):
c = client.Client() c = client.Client()
c.namespaces = {'/': '1'}
def fake_event_wait(timeout=None): def fake_event_wait(timeout=None):
assert timeout == 60 assert timeout == 60
c._generate_ack_id.call_args_list[0][0][1]('foo', 321) c._generate_ack_id.call_args_list[0][0][1]('foo', 321)
@ -462,8 +485,7 @@ class TestClient(unittest.TestCase):
c.eio.create_event.return_value.wait = fake_event_wait c.eio.create_event.return_value.wait = fake_event_wait
assert c.call('foo') == ('foo', 321) assert c.call('foo') == ('foo', 321)
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.EVENT, namespace='/', data=['foo'], id=123, binary=False packet.EVENT, namespace='/', data=['foo'], id=123)
)
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
assert ( assert (
c._send_packet.call_args_list[0][0][0].encode() c._send_packet.call_args_list[0][0][0].encode()
@ -472,7 +494,7 @@ class TestClient(unittest.TestCase):
def test_call_with_timeout(self): def test_call_with_timeout(self):
c = client.Client() c = client.Client()
c.namespaces = {'/': '1'}
def fake_event_wait(timeout=None): def fake_event_wait(timeout=None):
assert timeout == 12 assert timeout == 12
return False return False
@ -484,8 +506,7 @@ class TestClient(unittest.TestCase):
with pytest.raises(exceptions.TimeoutError): with pytest.raises(exceptions.TimeoutError):
c.call('foo', timeout=12) c.call('foo', timeout=12)
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.EVENT, namespace='/', data=['foo'], id=123, binary=False packet.EVENT, namespace='/', data=['foo'], id=123)
)
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
assert ( assert (
c._send_packet.call_args_list[0][0][0].encode() c._send_packet.call_args_list[0][0][0].encode()
@ -494,6 +515,7 @@ class TestClient(unittest.TestCase):
def test_disconnect(self): def test_disconnect(self):
c = client.Client() c = client.Client()
c.namespaces = {'/': '1'}
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c.eio = mock.MagicMock() c.eio = mock.MagicMock()
@ -510,14 +532,14 @@ class TestClient(unittest.TestCase):
def test_disconnect_namespaces(self): def test_disconnect_namespaces(self):
c = client.Client() c = client.Client()
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c.eio = mock.MagicMock() c.eio = mock.MagicMock()
c.eio.state = 'connected' c.eio.state = 'connected'
c.disconnect() c.disconnect()
assert c._trigger_event.call_count == 0 assert c._trigger_event.call_count == 0
assert c._send_packet.call_count == 3 assert c._send_packet.call_count == 2
expected_packet = packet.Packet(packet.DISCONNECT, namespace='/foo') expected_packet = packet.Packet(packet.DISCONNECT, namespace='/foo')
assert ( assert (
c._send_packet.call_args_list[0][0][0].encode() c._send_packet.call_args_list[0][0][0].encode()
@ -528,11 +550,6 @@ class TestClient(unittest.TestCase):
c._send_packet.call_args_list[1][0][0].encode() c._send_packet.call_args_list[1][0][0].encode()
== expected_packet.encode() == expected_packet.encode()
) )
expected_packet = packet.Packet(packet.DISCONNECT, namespace='/')
assert (
c._send_packet.call_args_list[2][0][0].encode()
== expected_packet.encode()
)
def test_transport(self): def test_transport(self):
c = client.Client() c = client.Client()
@ -557,40 +574,26 @@ class TestClient(unittest.TestCase):
def test_send_packet(self): def test_send_packet(self):
c = client.Client() c = client.Client()
c.eio.send = mock.MagicMock() c.eio.send = mock.MagicMock()
c._send_packet(packet.Packet(packet.EVENT, 'foo', binary=False)) c._send_packet(packet.Packet(packet.EVENT, 'foo'))
c.eio.send.assert_called_once_with('2"foo"', binary=False) c.eio.send.assert_called_once_with('2"foo"')
def test_send_packet_binary(self): def test_send_packet_binary(self):
c = client.Client() c = client.Client()
c.eio.send = mock.MagicMock() c.eio.send = mock.MagicMock()
c._send_packet(packet.Packet(packet.EVENT, b'foo', binary=True)) c._send_packet(packet.Packet(packet.EVENT, b'foo'))
assert c.eio.send.call_args_list == [ assert c.eio.send.call_args_list == [
mock.call('51-{"_placeholder":true,"num":0}', binary=False), mock.call('51-{"_placeholder":true,"num":0}'),
mock.call(b'foo', binary=True), mock.call(b'foo'),
] or c.eio.send.call_args_list == [ ] or c.eio.send.call_args_list == [
mock.call('51-{"num":0,"_placeholder":true}', binary=False), mock.call('51-{"num":0,"_placeholder":true}'),
mock.call(b'foo', binary=True), mock.call(b'foo'),
] ]
@unittest.skipIf(sys.version_info < (3, 0), 'only for Python 3') def test_send_packet_default_binary(self):
def test_send_packet_default_binary_py3(self):
c = client.Client()
c.eio.send = mock.MagicMock()
c._send_packet(packet.Packet(packet.EVENT, 'foo'))
c.eio.send.assert_called_once_with('2"foo"', binary=False)
@unittest.skipIf(sys.version_info >= (3, 0), 'only for Python 2')
def test_send_packet_default_binary_py2(self):
c = client.Client() c = client.Client()
c.eio.send = mock.MagicMock() c.eio.send = mock.MagicMock()
c._send_packet(packet.Packet(packet.EVENT, 'foo')) c._send_packet(packet.Packet(packet.EVENT, 'foo'))
assert c.eio.send.call_args_list == [ c.eio.send.assert_called_once_with('2"foo"')
mock.call('51-{"_placeholder":true,"num":0}', binary=False),
mock.call(b'foo', binary=True),
] or c.eio.send.call_args_list == [
mock.call('51-{"num":0,"_placeholder":true}', binary=False),
mock.call(b'foo', binary=True),
]
def test_generate_ack_id(self): def test_generate_ack_id(self):
c = client.Client() c = client.Client()
@ -606,45 +609,34 @@ class TestClient(unittest.TestCase):
c = client.Client() c = client.Client()
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c._handle_connect('/') c._handle_connect('/', {'sid': '123'})
assert c.namespaces == {'/': '123'}
c._trigger_event.assert_called_once_with('connect', namespace='/') c._trigger_event.assert_called_once_with('connect', namespace='/')
c._send_packet.assert_not_called() c._send_packet.assert_not_called()
def test_handle_connect_with_namespaces(self): def test_handle_connect_with_namespaces(self):
c = client.Client() c = client.Client()
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c._handle_connect('/') c._handle_connect('/', {'sid': '3'})
c._trigger_event.assert_called_once_with('connect', namespace='/') c._trigger_event.assert_called_once_with('connect', namespace='/')
assert c._send_packet.call_count == 2 assert c.namespaces == {'/': '3', '/foo': '1', '/bar': '2'}
expected_packet = packet.Packet(packet.CONNECT, namespace='/foo')
assert (
c._send_packet.call_args_list[0][0][0].encode()
== expected_packet.encode()
)
expected_packet = packet.Packet(packet.CONNECT, namespace='/bar')
assert (
c._send_packet.call_args_list[1][0][0].encode()
== expected_packet.encode()
)
def test_handle_connect_namespace(self): def test_handle_connect_namespace(self):
c = client.Client() c = client.Client()
c.namespaces = ['/foo'] c.namespaces = {'/foo': '1'}
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c._handle_connect('/foo') c._handle_connect('/foo', {'sid': '123'})
c._handle_connect('/bar') c._handle_connect('/bar', {'sid': '2'})
assert c._trigger_event.call_args_list == [ assert c._trigger_event.call_count == 1
mock.call('connect', namespace='/foo'), c._trigger_event.assert_called_once_with('connect', namespace='/bar')
mock.call('connect', namespace='/bar'), assert c.namespaces == {'/foo': '1', '/bar': '2'}
]
c._send_packet.assert_not_called()
assert c.namespaces == ['/foo', '/bar']
def test_handle_disconnect(self): def test_handle_disconnect(self):
c = client.Client() c = client.Client()
c.namespace = {'/': '1'}
c.connected = True c.connected = True
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._handle_disconnect('/') c._handle_disconnect('/')
@ -656,38 +648,42 @@ class TestClient(unittest.TestCase):
def test_handle_disconnect_namespace(self): def test_handle_disconnect_namespace(self):
c = client.Client() c = client.Client()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._handle_disconnect('/foo') c._handle_disconnect('/foo')
c._trigger_event.assert_called_once_with( c._trigger_event.assert_called_once_with(
'disconnect', namespace='/foo' 'disconnect', namespace='/foo'
) )
assert c.namespaces == ['/bar'] assert c.namespaces == {'/bar': '2'}
assert c.connected assert c.connected
c._handle_disconnect('/bar')
c._trigger_event.assert_called_with(
'disconnect', namespace='/bar'
)
assert c.namespaces == {}
assert not c.connected
def test_handle_disconnect_unknown_namespace(self): def test_handle_disconnect_unknown_namespace(self):
c = client.Client() c = client.Client()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._handle_disconnect('/baz') c._handle_disconnect('/baz')
c._trigger_event.assert_called_once_with( c._trigger_event.assert_called_once_with(
'disconnect', namespace='/baz' 'disconnect', namespace='/baz'
) )
assert c.namespaces == ['/foo', '/bar'] assert c.namespaces == {'/foo': '1', '/bar': '2'}
assert c.connected assert c.connected
def test_handle_disconnect_all_namespaces(self): def test_handle_disconnect_default_namespace(self):
c = client.Client() c = client.Client()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._handle_disconnect('/') c._handle_disconnect('/')
c._trigger_event.assert_any_call('disconnect', namespace='/') c._trigger_event.assert_called_with('disconnect', namespace='/')
c._trigger_event.assert_any_call('disconnect', namespace='/foo') assert c.namespaces == {'/foo': '1', '/bar': '2'}
c._trigger_event.assert_any_call('disconnect', namespace='/bar') assert c.connected
assert c.namespaces == []
assert not c.connected
def test_handle_event(self): def test_handle_event(self):
c = client.Client() c = client.Client()
@ -696,60 +692,56 @@ class TestClient(unittest.TestCase):
c._trigger_event.assert_called_once_with('foo', '/', ('bar', 'baz')) c._trigger_event.assert_called_once_with('foo', '/', ('bar', 'baz'))
def test_handle_event_with_id_no_arguments(self): def test_handle_event_with_id_no_arguments(self):
c = client.Client(binary=True) c = client.Client()
c._trigger_event = mock.MagicMock(return_value=None) c._trigger_event = mock.MagicMock(return_value=None)
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c._handle_event('/', 123, ['foo', ('bar', 'baz')]) c._handle_event('/', 123, ['foo', ('bar', 'baz')])
c._trigger_event.assert_called_once_with('foo', '/', ('bar', 'baz')) c._trigger_event.assert_called_once_with('foo', '/', ('bar', 'baz'))
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.ACK, namespace='/', id=123, data=[], binary=None packet.ACK, namespace='/', id=123, data=[])
)
assert ( assert (
c._send_packet.call_args_list[0][0][0].encode() c._send_packet.call_args_list[0][0][0].encode()
== expected_packet.encode() == expected_packet.encode()
) )
def test_handle_event_with_id_one_argument(self): def test_handle_event_with_id_one_argument(self):
c = client.Client(binary=True) c = client.Client()
c._trigger_event = mock.MagicMock(return_value='ret') c._trigger_event = mock.MagicMock(return_value='ret')
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c._handle_event('/', 123, ['foo', ('bar', 'baz')]) c._handle_event('/', 123, ['foo', ('bar', 'baz')])
c._trigger_event.assert_called_once_with('foo', '/', ('bar', 'baz')) c._trigger_event.assert_called_once_with('foo', '/', ('bar', 'baz'))
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.ACK, namespace='/', id=123, data=['ret'], binary=None packet.ACK, namespace='/', id=123, data=['ret'])
)
assert ( assert (
c._send_packet.call_args_list[0][0][0].encode() c._send_packet.call_args_list[0][0][0].encode()
== expected_packet.encode() == expected_packet.encode()
) )
def test_handle_event_with_id_one_list_argument(self): def test_handle_event_with_id_one_list_argument(self):
c = client.Client(binary=True) c = client.Client()
c._trigger_event = mock.MagicMock(return_value=['a', 'b']) c._trigger_event = mock.MagicMock(return_value=['a', 'b'])
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c._handle_event('/', 123, ['foo', ('bar', 'baz')]) c._handle_event('/', 123, ['foo', ('bar', 'baz')])
c._trigger_event.assert_called_once_with('foo', '/', ('bar', 'baz')) c._trigger_event.assert_called_once_with('foo', '/', ('bar', 'baz'))
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.ACK, namespace='/', id=123, data=[['a', 'b']], binary=None packet.ACK, namespace='/', id=123, data=[['a', 'b']])
)
assert ( assert (
c._send_packet.call_args_list[0][0][0].encode() c._send_packet.call_args_list[0][0][0].encode()
== expected_packet.encode() == expected_packet.encode()
) )
def test_handle_event_with_id_two_arguments(self): def test_handle_event_with_id_two_arguments(self):
c = client.Client(binary=True) c = client.Client()
c._trigger_event = mock.MagicMock(return_value=('a', 'b')) c._trigger_event = mock.MagicMock(return_value=('a', 'b'))
c._send_packet = mock.MagicMock() c._send_packet = mock.MagicMock()
c._handle_event('/', 123, ['foo', ('bar', 'baz')]) c._handle_event('/', 123, ['foo', ('bar', 'baz')])
c._trigger_event.assert_called_once_with('foo', '/', ('bar', 'baz')) c._trigger_event.assert_called_once_with('foo', '/', ('bar', 'baz'))
assert c._send_packet.call_count == 1 assert c._send_packet.call_count == 1
expected_packet = packet.Packet( expected_packet = packet.Packet(
packet.ACK, namespace='/', id=123, data=['a', 'b'], binary=None packet.ACK, namespace='/', id=123, data=['a', 'b'])
)
assert ( assert (
c._send_packet.call_args_list[0][0][0].encode() c._send_packet.call_args_list[0][0][0].encode()
== expected_packet.encode() == expected_packet.encode()
@ -774,30 +766,30 @@ class TestClient(unittest.TestCase):
def test_handle_error(self): def test_handle_error(self):
c = client.Client() c = client.Client()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._handle_error('/', 'error') c._handle_error('/', 'error')
assert c.namespaces == [] assert c.namespaces == {}
assert not c.connected assert not c.connected
c._trigger_event.assert_called_once_with('connect_error', '/', 'error') c._trigger_event.assert_called_once_with('connect_error', '/', 'error')
def test_handle_error_with_no_arguments(self): def test_handle_error_with_no_arguments(self):
c = client.Client() c = client.Client()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._handle_error('/', None) c._handle_error('/', None)
assert c.namespaces == [] assert c.namespaces == {}
assert not c.connected assert not c.connected
c._trigger_event.assert_called_once_with('connect_error', '/') c._trigger_event.assert_called_once_with('connect_error', '/')
def test_handle_error_namespace(self): def test_handle_error_namespace(self):
c = client.Client() c = client.Client()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._handle_error('/bar', ['error', 'message']) c._handle_error('/bar', ['error', 'message'])
assert c.namespaces == ['/foo'] assert c.namespaces == {'/foo': '1'}
assert c.connected assert c.connected
c._trigger_event.assert_called_once_with( c._trigger_event.assert_called_once_with(
'connect_error', '/bar', 'error', 'message' 'connect_error', '/bar', 'error', 'message'
@ -806,19 +798,19 @@ class TestClient(unittest.TestCase):
def test_handle_error_namespace_with_no_arguments(self): def test_handle_error_namespace_with_no_arguments(self):
c = client.Client() c = client.Client()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c._handle_error('/bar', None) c._handle_error('/bar', None)
assert c.namespaces == ['/foo'] assert c.namespaces == {'/foo': '1'}
assert c.connected assert c.connected
c._trigger_event.assert_called_once_with('connect_error', '/bar') c._trigger_event.assert_called_once_with('connect_error', '/bar')
def test_handle_error_unknown_namespace(self): def test_handle_error_unknown_namespace(self):
c = client.Client() c = client.Client()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/foo': '1', '/bar': '2'}
c._handle_error('/baz', 'error') c._handle_error('/baz', 'error')
assert c.namespaces == ['/foo', '/bar'] assert c.namespaces == {'/foo': '1', '/bar': '2'}
assert c.connected assert c.connected
def test_trigger_event(self): def test_trigger_event(self):
@ -927,10 +919,23 @@ class TestClient(unittest.TestCase):
def test_handle_eio_connect(self): def test_handle_eio_connect(self):
c = client.Client() c = client.Client()
c.connection_namespaces = ['/', '/foo']
c._send_packet = mock.MagicMock()
c.eio.sid = 'foo' c.eio.sid = 'foo'
assert c.sid is None assert c.sid is None
c._handle_eio_connect() c._handle_eio_connect()
assert c.sid == 'foo' assert c.sid == 'foo'
assert c._send_packet.call_count == 2
expected_packet = packet.Packet(packet.CONNECT, namespace='/')
assert (
c._send_packet.call_args_list[0][0][0].encode()
== expected_packet.encode()
)
expected_packet = packet.Packet(packet.CONNECT, namespace='/foo')
assert (
c._send_packet.call_args_list[1][0][0].encode()
== expected_packet.encode()
)
def test_handle_eio_message(self): def test_handle_eio_message(self):
c = client.Client() c = client.Client()
@ -940,10 +945,10 @@ class TestClient(unittest.TestCase):
c._handle_ack = mock.MagicMock() c._handle_ack = mock.MagicMock()
c._handle_error = mock.MagicMock() c._handle_error = mock.MagicMock()
c._handle_eio_message('0') c._handle_eio_message('0{"sid":"123"}')
c._handle_connect.assert_called_with(None) c._handle_connect.assert_called_with(None, {'sid': '123'})
c._handle_eio_message('0/foo') c._handle_eio_message('0/foo,{"sid":"123"}')
c._handle_connect.assert_called_with('/foo') c._handle_connect.assert_called_with('/foo', {'sid': '123'})
c._handle_eio_message('1') c._handle_eio_message('1')
c._handle_disconnect.assert_called_with(None) c._handle_disconnect.assert_called_with(None)
c._handle_eio_message('1/foo') c._handle_eio_message('1/foo')
@ -981,6 +986,7 @@ class TestClient(unittest.TestCase):
def test_eio_disconnect(self): def test_eio_disconnect(self):
c = client.Client() c = client.Client()
c.namespaces = {'/': '1'}
c.connected = True c.connected = True
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c.start_background_task = mock.MagicMock() c.start_background_task = mock.MagicMock()
@ -994,7 +1000,7 @@ class TestClient(unittest.TestCase):
def test_eio_disconnect_namespaces(self): def test_eio_disconnect_namespaces(self):
c = client.Client() c = client.Client()
c.connected = True c.connected = True
c.namespaces = ['/foo', '/bar'] c.namespaces = {'/': '1', '/foo': '2', '/bar': '3'}
c._trigger_event = mock.MagicMock() c._trigger_event = mock.MagicMock()
c.start_background_task = mock.MagicMock() c.start_background_task = mock.MagicMock()
c.sid = 'foo' c.sid = 'foo'

Write Preview
Loading…
Cancel
Save