async def test_remote_messages(self, make_session):
messages = []
session = make_session(result=messages)
await session._remote_messages(("msg1", "msg2"))
assert messages == [
(protocol.SockjsMessage(protocol.MSG_MESSAGE, "msg1"), session),
(protocol.SockjsMessage(protocol.MSG_MESSAGE, "msg2"), session),
]
async def test_remote_messages(self, make_session):
messages = []
session = make_session(result=messages)
await session._remote_messages(('msg1', 'msg2'))
assert messages == \
[(protocol.SockjsMessage(protocol.MSG_MESSAGE, 'msg1'),
session),
(protocol.SockjsMessage(protocol.MSG_MESSAGE, 'msg2'),
session)]
def test_remote_messages(self, make_session):
messages = []
session = make_session(result=messages)
yield from session._remote_messages(('msg1', 'msg2'))
assert messages == \
[(protocol.SockjsMessage(tp=protocol.MSG_MESSAGE, data='msg1'),
session),
(protocol.SockjsMessage(tp=protocol.MSG_MESSAGE, data='msg2'),
session)]
async def test_remote_message(self, make_session):
messages = []
session = make_session(result=messages)
await session._remote_message('msg')
assert messages == \
[(protocol.SockjsMessage(tp=protocol.MSG_MESSAGE, data='msg'),
session)]
async def test_remote_close(self, make_session):
messages = []
session = make_session(result=messages)
await session._remote_close()
assert not session.interrupted
assert session.state == protocol.STATE_CLOSING
assert messages == [(protocol.SockjsMessage(protocol.MSG_CLOSE,
None), session)]
async def test_remote_close_with_exc(self, make_session):
messages = []
session = make_session(result=messages)
exc = ValueError()
await session._remote_close(exc=exc)
assert session.interrupted
assert session.state == protocol.STATE_CLOSING
assert messages == [(protocol.SockjsMessage(protocol.MSG_CLOSE,
exc), session)]
def test_remote_close(self, make_session):
messages = []
session = make_session(result=messages)
yield from session._remote_close()
assert not session.interrupted
assert session.state == protocol.STATE_CLOSING
assert messages == \
[(protocol.SockjsMessage(
tp=protocol.MSG_CLOSE, data=None), session)]
