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, {})
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, {})
