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