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