def test_return_formatted_records(self):
self.base.repeated_query.return_value = [
MockTLSARecord(3, 0, 1, b"foo"),
MockTLSARecord(3, 2, 1, b"bar"),
]
self.assertSequenceEqual(
run_coroutine(network.lookup_tlsa(
b"foo.test",
5222,
executor=unittest.mock.sentinel.executor,
resolver=unittest.mock.sentinel.resolver,
)),
[
(3, 0, 1, b"foo"),
(3, 2, 1, b"bar"),
]
)
self.base.repeated_query.assert_called_with(
b"_5222._tcp.foo.test",
dns.rdatatype.TLSA,
require_ad=True,
executor=unittest.mock.sentinel.executor,
resolver=unittest.mock.sentinel.resolver,
)
def test_return_None_on_nxdomain(self):
self.base.repeated_query.return_value = None
self.assertIsNone(
run_coroutine(network.lookup_tlsa(
b"foo.test",
5222,
)), )
self.base.repeated_query.assert_called_with(
b"_5222._tcp.foo.test",
dns.rdatatype.TLSA,
require_ad=True,
)
def test_return_None_on_nxdomain(self):
self.base.repeated_query.return_value = None
self.assertIsNone(
run_coroutine(network.lookup_tlsa(
b"foo.test",
5222,
)),
)
self.base.repeated_query.assert_called_with(
b"_5222._tcp.foo.test",
dns.rdatatype.TLSA,
require_ad=True,
)
