Index: third_party/twisted_8_1/twisted/names/test/test_client.py |
diff --git a/third_party/twisted_8_1/twisted/names/test/test_client.py b/third_party/twisted_8_1/twisted/names/test/test_client.py |
deleted file mode 100644 |
index bc98dbdbecf5c780f08621d567002e5f6bcf5924..0000000000000000000000000000000000000000 |
--- a/third_party/twisted_8_1/twisted/names/test/test_client.py |
+++ /dev/null |
@@ -1,307 +0,0 @@ |
-# -*- test-case-name: twisted.names.test.test_client -*- |
-# Copyright (c) 2001-2007 Twisted Matrix Laboratories. |
-# See LICENSE for details. |
- |
-""" |
-Test cases for twisted.names.client |
-""" |
- |
-from twisted.names import client, dns |
-from twisted.names.error import DNSQueryTimeoutError |
-from twisted.trial import unittest |
-from twisted.names.common import ResolverBase |
-from twisted.internet import defer |
- |
-class FakeResolver(ResolverBase): |
- |
- def _lookup(self, name, cls, qtype, timeout): |
- """ |
- The getHostByNameTest does a different type of query that requires it |
- return an A record from an ALL_RECORDS lookup, so we accomodate that |
- here. |
- """ |
- if name == 'getHostByNameTest': |
- rr = dns.RRHeader(name=name, type=dns.A, cls=cls, ttl=60, |
- payload=dns.Record_A(address='127.0.0.1', ttl=60)) |
- else: |
- rr = dns.RRHeader(name=name, type=qtype, cls=cls, ttl=60) |
- |
- results = [rr] |
- authority = [] |
- addtional = [] |
- return defer.succeed((results, authority, addtional)) |
- |
- |
- |
-class StubDNSDatagramProtocol(object): |
- """ |
- L{dns.DNSDatagramProtocol}-alike. |
- |
- @ivar queries: A C{list} of tuples giving the arguments passed to |
- C{query} along with the L{defer.Deferred} which was returned from |
- the call. |
- """ |
- def __init__(self): |
- self.queries = [] |
- |
- |
- def query(self, address, queries, timeout=10, id=None): |
- """ |
- Record the given arguments and return a Deferred which will not be |
- called back by this code. |
- """ |
- result = defer.Deferred() |
- self.queries.append((address, queries, timeout, id, result)) |
- return result |
- |
- |
- |
-class ResolverTests(unittest.TestCase): |
- """ |
- Tests for L{client.Resolver}. |
- """ |
- def test_datagramQueryServerOrder(self): |
- """ |
- L{client.Resolver.queryUDP} should issue queries to its |
- L{dns.DNSDatagramProtocol} with server addresses taken from its own |
- C{servers} and C{dynServers} lists, proceeding through them in order |
- as L{DNSQueryTimeoutError}s occur. |
- """ |
- protocol = StubDNSDatagramProtocol() |
- protocol.transport = object() |
- |
- servers = [object(), object()] |
- dynServers = [object(), object()] |
- resolver = client.Resolver(servers=servers) |
- resolver.dynServers = dynServers |
- resolver.protocol = protocol |
- |
- expectedResult = object() |
- queryResult = resolver.queryUDP(None) |
- queryResult.addCallback(self.assertEqual, expectedResult) |
- |
- self.assertEqual(len(protocol.queries), 1) |
- self.assertIdentical(protocol.queries[0][0], servers[0]) |
- protocol.queries[0][-1].errback(DNSQueryTimeoutError(0)) |
- self.assertEqual(len(protocol.queries), 2) |
- self.assertIdentical(protocol.queries[1][0], servers[1]) |
- protocol.queries[1][-1].errback(DNSQueryTimeoutError(1)) |
- self.assertEqual(len(protocol.queries), 3) |
- self.assertIdentical(protocol.queries[2][0], dynServers[0]) |
- protocol.queries[2][-1].errback(DNSQueryTimeoutError(2)) |
- self.assertEqual(len(protocol.queries), 4) |
- self.assertIdentical(protocol.queries[3][0], dynServers[1]) |
- protocol.queries[3][-1].callback(expectedResult) |
- |
- return queryResult |
- |
- |
- |
-class ClientTestCase(unittest.TestCase): |
- |
- def setUp(self): |
- """ |
- Replace the resolver with a FakeResolver |
- """ |
- client.theResolver = FakeResolver() |
- self.hostname = 'example.com' |
- self.ghbntest = 'getHostByNameTest' |
- |
- def tearDown(self): |
- """ |
- By setting the resolver to None, it will be recreated next time a name |
- lookup is done. |
- """ |
- client.theResolver = None |
- |
- def checkResult(self, (results, authority, additional), qtype): |
- """ |
- Verify that the result is the same query type as what is expected. |
- """ |
- result = results[0] |
- self.assertEquals(str(result.name), self.hostname) |
- self.assertEquals(result.type, qtype) |
- |
- def checkGetHostByName(self, result): |
- """ |
- Test that the getHostByName query returns the 127.0.0.1 address. |
- """ |
- self.assertEquals(result, '127.0.0.1') |
- |
- def test_getHostByName(self): |
- """ |
- do a getHostByName of a value that should return 127.0.0.1. |
- """ |
- d = client.getHostByName(self.ghbntest) |
- d.addCallback(self.checkGetHostByName) |
- return d |
- |
- def test_lookupAddress(self): |
- """ |
- Do a lookup and test that the resolver will issue the correct type of |
- query type. We do this by checking that FakeResolver returns a result |
- record with the same query type as what we issued. |
- """ |
- d = client.lookupAddress(self.hostname) |
- d.addCallback(self.checkResult, dns.A) |
- return d |
- |
- def test_lookupIPV6Address(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupIPV6Address(self.hostname) |
- d.addCallback(self.checkResult, dns.AAAA) |
- return d |
- |
- def test_lookupAddress6(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupAddress6(self.hostname) |
- d.addCallback(self.checkResult, dns.A6) |
- return d |
- |
- def test_lookupNameservers(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupNameservers(self.hostname) |
- d.addCallback(self.checkResult, dns.NS) |
- return d |
- |
- def test_lookupCanonicalName(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupCanonicalName(self.hostname) |
- d.addCallback(self.checkResult, dns.CNAME) |
- return d |
- |
- def test_lookupAuthority(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupAuthority(self.hostname) |
- d.addCallback(self.checkResult, dns.SOA) |
- return d |
- |
- def test_lookupMailBox(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupMailBox(self.hostname) |
- d.addCallback(self.checkResult, dns.MB) |
- return d |
- |
- def test_lookupMailGroup(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupMailGroup(self.hostname) |
- d.addCallback(self.checkResult, dns.MG) |
- return d |
- |
- def test_lookupMailRename(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupMailRename(self.hostname) |
- d.addCallback(self.checkResult, dns.MR) |
- return d |
- |
- def test_lookupNull(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupNull(self.hostname) |
- d.addCallback(self.checkResult, dns.NULL) |
- return d |
- |
- def test_lookupWellKnownServices(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupWellKnownServices(self.hostname) |
- d.addCallback(self.checkResult, dns.WKS) |
- return d |
- |
- def test_lookupPointer(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupPointer(self.hostname) |
- d.addCallback(self.checkResult, dns.PTR) |
- return d |
- |
- def test_lookupHostInfo(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupHostInfo(self.hostname) |
- d.addCallback(self.checkResult, dns.HINFO) |
- return d |
- |
- def test_lookupMailboxInfo(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupMailboxInfo(self.hostname) |
- d.addCallback(self.checkResult, dns.MINFO) |
- return d |
- |
- def test_lookupMailExchange(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupMailExchange(self.hostname) |
- d.addCallback(self.checkResult, dns.MX) |
- return d |
- |
- def test_lookupText(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupText(self.hostname) |
- d.addCallback(self.checkResult, dns.TXT) |
- return d |
- |
- def test_lookupResponsibility(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupResponsibility(self.hostname) |
- d.addCallback(self.checkResult, dns.RP) |
- return d |
- |
- def test_lookupAFSDatabase(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupAFSDatabase(self.hostname) |
- d.addCallback(self.checkResult, dns.AFSDB) |
- return d |
- |
- def test_lookupService(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupService(self.hostname) |
- d.addCallback(self.checkResult, dns.SRV) |
- return d |
- |
- def test_lookupZone(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupZone(self.hostname) |
- d.addCallback(self.checkResult, dns.AXFR) |
- return d |
- |
- def test_lookupAllRecords(self): |
- """ |
- See L{test_lookupAddress} |
- """ |
- d = client.lookupAllRecords(self.hostname) |
- d.addCallback(self.checkResult, dns.ALL_RECORDS) |
- return d |