def query(self, *args, **kwargs):
self.args = args
self.kwargs = kwargs
self.call_count += 1
if self.call_count < 2:
return greendns.HostsAnswer(args[0], 1, 5, [RR()], False)
else:
raise self.exc_type()
def query(self, *args, **kwargs):
self.args = args
self.kwargs = kwargs
if self.raises:
raise self.raises()
if hasattr(self, 'rrset'):
rrset = self.rrset
else:
rrset = [self.rr]
return greendns.HostsAnswer('foo', 1, 1, rrset, False)
