def test_return_formatted_records(self):
self.base.repeated_query.return_value = [
MockSRVRecord(0, 1, "xmpp.foo.test.", 5222),
MockSRVRecord(2, 1, "xmpp.bar.test.", 5222),
]
self.assertSequenceEqual(
run_coroutine(network.lookup_srv(
b"foo.test",
"xmpp-client",
executor=unittest.mock.sentinel.executor,
resolver=unittest.mock.sentinel.resolver,
)),
[
(0, 1, ("xmpp.foo.test", 5222)),
(2, 1, ("xmpp.bar.test", 5222)),
]
)
self.base.repeated_query.assert_called_with(
b"_xmpp-client._tcp.foo.test",
dns.rdatatype.SRV,
executor=unittest.mock.sentinel.executor,
resolver=unittest.mock.sentinel.resolver,
)
def test_raise_ValueError_if_service_not_supported(self):
self.base.repeated_query.return_value = [
MockSRVRecord(0, 1, "xmpp.foo.test.", 5222),
MockSRVRecord(2, 1, ".", 5222),
]
with self.assertRaisesRegex(
ValueError,
r"'xmpp-client' over 'tcp' not supported at b'foo\.test'"):
run_coroutine(network.lookup_srv(
b"foo.test",
"xmpp-client",
))
def test_return_None_on_nxdomain(self):
self.base.repeated_query.return_value = None
self.assertIsNone(
run_coroutine(network.lookup_srv(
b"foo.test",
"xmpp-client",
)), )
self.base.repeated_query.assert_called_with(
b"_xmpp-client._tcp.foo.test",
dns.rdatatype.SRV,
)
def test_return_None_on_nxdomain(self):
self.base.repeated_query.return_value = None
self.assertIsNone(
run_coroutine(network.lookup_srv(
b"foo.test",
"xmpp-client",
)),
)
self.base.repeated_query.assert_called_with(
b"_xmpp-client._tcp.foo.test",
dns.rdatatype.SRV,
)
