Ejemplo n.º 1
0
    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"})
Ejemplo n.º 2
0
    def test_disconnect(self):
        url, r = yield self.mk_relay()
        protocol = RelayProtocol({'self': {'id': 'the-user-id'}})
        protocol.transport = StringTransport()
        protocol.transport.loseConnection = Mock()
        protocol.transport.loseConnection.return_value = None

        r.connections['bot-id'] = protocol

        yield treq.post('%s/disconnect' % (url, ),
                        auth=('bot-id', 'bot-token'),
                        pool=self.pool)

        protocol.transport.loseConnection.assert_called_with()
Ejemplo n.º 3
0
    def test_disconnect(self):
        url, r = yield self.mk_relay()
        protocol = RelayProtocol({
            'self': {
                'id': 'the-user-id'
            }
        })
        protocol.transport = StringTransport()
        protocol.transport.loseConnection = Mock()
        protocol.transport.loseConnection.return_value = None

        r.connections['bot-id'] = protocol

        yield treq.post(
            '%s/disconnect' % (url,),
            auth=('bot-id', 'bot-token'),
            pool=self.pool)

        protocol.transport.loseConnection.assert_called_with()
Ejemplo n.º 4
0
    def test_get_protocol(self):
        _, r = yield self.mk_relay()
        mock_proto = RelayProtocol({
            'self': {
                'id': 'the-user-id'
            },
        })
        r.rtm_start = Mock()
        r.rtm_start.return_value = succeed(mock_proto)

        protocol = yield r.get_protocol('bot-id', 'bot-token')
        self.assertEqual(protocol, mock_proto)
        self.assertEqual(r.connections, {'bot-id': mock_proto})
        r.rtm_start.assert_called_with('bot-token')
Ejemplo n.º 5
0
    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"})
Ejemplo n.º 6
0
    def test_ping(self):
        protocol = RelayProtocol({
            'self': {
                'id': 'the-user-id',
            },
        })
        protocol.clock = Clock()
        protocol.factory = Mock()
        protocol.sendMessage = Mock()

        protocol.onOpen()
        protocol.clock.advance(3)
        protocol.sendMessage.assert_has_calls(
            [call('{"type": "ping"}'),
             call('{"type": "ping"}')])
Ejemplo n.º 7
0
    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, {})
Ejemplo n.º 8
0
    def test_connect_patched(self):
        url, r = yield self.mk_relay()
        r.get_protocol = Mock()
        r.get_protocol.return_value = succeed(
            RelayProtocol({'self': {
                'id': 'the-user-id'
            }}))

        response = yield treq.post('%s/connect' % (url, ),
                                   auth=('bot-id', 'bot-token'),
                                   pool=self.pool)
        data = yield response.json()
        self.assertEqual(data, {
            'self': {
                'id': 'the-user-id',
            },
        })
        r.get_protocol.assert_called_with(bot_id='bot-id',
                                          bot_token='bot-token')
Ejemplo n.º 9
0
    def test_ping(self):
        protocol = RelayProtocol({
            'self': {
                'id': 'the-user-id',
            },
        })
        protocol.clock = Clock()
        protocol.factory = Mock()
        protocol.sendMessage = Mock()

        protocol.onOpen()
        protocol.clock.advance(3)
        protocol.sendMessage.assert_has_calls([
            call('{"type": "ping"}'),
            call('{"type": "ping"}')])
Ejemplo n.º 10
0
    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, {})