Fixing failed unit test for greendns. Making mock.py non-executable.

This commit is contained in:
Ryan Williams
2010-06-27 23:15:46 -07:00
parent 2783dcb5ce
commit 88e33c0d7b
2 changed files with 9 additions and 2 deletions

View File

@@ -91,6 +91,9 @@ class ResolverProxy(object):
self._resolver.cache = dns.resolver.Cache()
query = args[0]
if query is None:
args = list(args)
query = args[0] = '0.0.0.0'
if self._hosts and self._hosts.get(query):
answer = FakeAnswer()
record = FakeRecord()
@@ -199,7 +202,11 @@ def getnameinfo(sockaddr, flags):
Currently only supports IPv4.
"""
host, port = sockaddr
try:
host, port = sockaddr
except ValueError:
# must be ipv6 sockaddr, pretending we don't know how to resolve it
raise socket.gaierror(-2, 'name or service not known')
if (flags & socket.NI_NAMEREQD) and (flags & socket.NI_NUMERICHOST):
# Conflicting flags. Punt.
@@ -245,7 +252,7 @@ def is_ipv4_addr(host):
"""
try:
d1, d2, d3, d4 = map(int, host.split('.'))
except ValueError:
except (ValueError, AttributeError):
return False
if 0 <= d1 <= 255 and 0 <= d2 <= 255 and 0 <= d3 <= 255 and 0 <= d4 <= 255:

0
tests/mock.py Executable file → Normal file
View File