Commit 5d382b42 authored by JINMEI Tatuya's avatar JINMEI Tatuya
Browse files

[1372] simplified the code of MockDataSrcClient.find() and updated the

docstring accordingly.
parent d08c42ad
......@@ -110,21 +110,21 @@ class MockDataSrcClient:
def find(self, name, rrtype, target, options):
'''Mock ZoneFinder.find().
It returns the predefined SOA RRset to queries for SOA of the common
test zone name. It also emulates some unusual cases for special
(At the moment) this method only handles query for type SOA.
By default it returns a normal SOA RR(set) whose owner name is
the query name It also emulates some unusual cases for special
zone names.
'''
if name == TEST_ZONE_NAME and rrtype == RRType.SOA():
return (ZoneFinder.SUCCESS, self.__create_soa())
elif name == Name('nosoa.example.com') and rrtype == RRType.SOA():
if name == Name('nosoa.example.com') and rrtype == RRType.SOA():
return (ZoneFinder.NXDOMAIN, None)
elif name == Name('multisoa.example.com') and rrtype == RRType.SOA():
soa_rrset = self.__create_soa()
soa_rrset.add_rdata(soa_rrset.get_rdata()[0])
return (ZoneFinder.SUCCESS, soa_rrset)
else:
elif rrtype == RRType.SOA():
return (ZoneFinder.SUCCESS, self.__create_soa())
raise ValueError('Unexpected input to mock finder: bug in test case?')
def get_iterator(self, zone_name, adjust_ttl=False):
if zone_name == Name('notauth.example.com'):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment