summaryrefslogtreecommitdiff
path: root/tests/test_dns.py
diff options
context:
space:
mode:
authorRichard van der Hoff <richard@matrix.org>2018-08-22 14:38:48 +0100
committerRichard van der Hoff <richard@matrix.org>2018-08-22 14:38:48 +0100
commitd18dd479ff799e77f326baa58052785255ad2d87 (patch)
tree61c89bfad10f58e3974214f7d336ef234daf5cad /tests/test_dns.py
parent9ac9a9ab1020ec4bf395ccf3e07ba6192f6ac558 (diff)
Imported Upstream version 0.33.3
Diffstat (limited to 'tests/test_dns.py')
-rw-r--r--tests/test_dns.py22
1 files changed, 6 insertions, 16 deletions
diff --git a/tests/test_dns.py b/tests/test_dns.py
index b647d926..90bd34be 100644
--- a/tests/test_dns.py
+++ b/tests/test_dns.py
@@ -27,7 +27,6 @@ from . import unittest
@unittest.DEBUG
class DnsTestCase(unittest.TestCase):
-
@defer.inlineCallbacks
def test_resolve(self):
dns_client_mock = Mock()
@@ -36,14 +35,11 @@ class DnsTestCase(unittest.TestCase):
host_name = "example.com"
answer_srv = dns.RRHeader(
- type=dns.SRV,
- payload=dns.Record_SRV(
- target=host_name,
- )
+ type=dns.SRV, payload=dns.Record_SRV(target=host_name)
)
dns_client_mock.lookupService.return_value = defer.succeed(
- ([answer_srv], None, None),
+ ([answer_srv], None, None)
)
cache = {}
@@ -68,9 +64,7 @@ class DnsTestCase(unittest.TestCase):
entry = Mock(spec_set=["expires"])
entry.expires = 0
- cache = {
- service_name: [entry]
- }
+ cache = {service_name: [entry]}
servers = yield resolve_service(
service_name, dns_client=dns_client_mock, cache=cache
@@ -93,12 +87,10 @@ class DnsTestCase(unittest.TestCase):
entry = Mock(spec_set=["expires"])
entry.expires = 999999999
- cache = {
- service_name: [entry]
- }
+ cache = {service_name: [entry]}
servers = yield resolve_service(
- service_name, dns_client=dns_client_mock, cache=cache, clock=clock,
+ service_name, dns_client=dns_client_mock, cache=cache, clock=clock
)
self.assertFalse(dns_client_mock.lookupService.called)
@@ -117,9 +109,7 @@ class DnsTestCase(unittest.TestCase):
cache = {}
with self.assertRaises(error.DNSServerError):
- yield resolve_service(
- service_name, dns_client=dns_client_mock, cache=cache
- )
+ yield resolve_service(service_name, dns_client=dns_client_mock, cache=cache)
@defer.inlineCallbacks
def test_name_error(self):