def test_protocol_relay(self): relay = Mock() relay.relay = Mock() protocol = RelayProtocol({'self': { 'id': 'the-user-id', }}) protocol.relay = relay protocol.bot_user_id = 'the-user-id' protocol.onMessage('{"foo": "bar"}', False) relay.relay.assert_called_with('the-user-id', {"foo": "bar"})
def test_protocol_relay(self): relay = Mock() relay.relay = Mock() protocol = RelayProtocol({ 'self': { 'id': 'the-user-id', } }) protocol.relay = relay protocol.bot_user_id = 'the-user-id' protocol.onMessage('{"foo": "bar"}', False) relay.relay.assert_called_with('the-user-id', {"foo": "bar"})
def test_protocol_close(self): _, r = yield self.mk_relay() protocol = RelayProtocol({ 'self': { 'id': 'the-user-id', }, }) protocol.factory = Mock() protocol.bot_user_id = 'bot-user-id' protocol.relay = r r.set_protocol('bot-user-id', protocol) protocol.onClose(True, None, None) self.assertEqual(r.connections, {})