Exemplo n.º 1
0
    def test_creates_fake_srv_if_no_srvs_available(self):
        base = unittest.mock.Mock()

        nattempts = object()

        with unittest.mock.patch(
                "aioxmpp.network.lookup_srv",
                new=base.lookup_srv) as lookup_srv:
            lookup_srv.return_value = None

            result = run_coroutine(network.find_xmpp_host_addr(
                asyncio.get_event_loop(),
                base.domain,
                attempts=nattempts
            ))

        calls = list(base.mock_calls)
        self.assertSequenceEqual(
            calls,
            [
                unittest.mock.call.domain.encode("IDNA"),
                unittest.mock.call.lookup_srv(
                    service="xmpp-client",
                    domain=base.domain.encode(),
                    nattempts=nattempts
                )
            ]
        )

        self.assertSequenceEqual(
            result,
            [
                (0, 0, (base.domain.encode(), 5222)),
            ]
        )
    def test_returns_items_if_available(self):
        base = unittest.mock.Mock()
        base.lookup_srv = CoroutineMock()

        nattempts = object()
        items = object()

        with unittest.mock.patch("aioxmpp.network.lookup_srv",
                                 new=base.lookup_srv) as lookup_srv:
            lookup_srv.return_value = items

            result = run_coroutine(
                network.find_xmpp_host_addr(asyncio.get_event_loop(),
                                            base.domain,
                                            attempts=nattempts))

        calls = list(base.mock_calls)
        self.assertSequenceEqual(calls, [
            unittest.mock.call.domain.encode("IDNA"),
            unittest.mock.call.lookup_srv(service="xmpp-client",
                                          domain=base.domain.encode(),
                                          nattempts=nattempts)
        ])

        self.assertIs(result, items)
Exemplo n.º 3
0
    def test_propagates_ValueError_from_lookup_srv(self):
        base = unittest.mock.Mock()

        nattempts = object()

        with unittest.mock.patch("aioxmpp.network.lookup_srv",
                                 new=base.lookup_srv) as lookup_srv:
            lookup_srv.side_effect = ValueError()

            with self.assertRaises(ValueError):
                run_coroutine(
                    network.find_xmpp_host_addr(asyncio.get_event_loop(),
                                                base.domain,
                                                attempts=nattempts))
Exemplo n.º 4
0
    def test_propagates_ValueError_from_lookup_srv(self):
        base = unittest.mock.Mock()

        nattempts = object()

        with unittest.mock.patch(
                "aioxmpp.network.lookup_srv",
                new=base.lookup_srv) as lookup_srv:
            lookup_srv.side_effect = ValueError()

            with self.assertRaises(ValueError):
                run_coroutine(network.find_xmpp_host_addr(
                    asyncio.get_event_loop(),
                    base.domain,
                    attempts=nattempts
                ))