def _pods_in_shard():
raw_recs = srvlookup._query_srv_records(
'_%s._%s.%s' %
('xmpp-server', 'TCP', 'gmail.com' or srvlookup._get_domain()))
recs = sorted(srvlookup._build_result_set(raw_recs),
key=lambda r: (r.priority, -r.weight))
for rec in recs:
yield (rec.host, rec.port)
def test_resolver_query_should_be_invoked_with_fqdn(self):
with mock.patch('dns.resolver.query') as query:
query.return_value = mock.Mock('dns.resolver.Answer')
fqdn = 'foo.bar.baz'
srvlookup._query_srv_records(fqdn)
query.assert_called_once_with(fqdn, 'SRV')
def test_should_return_resolver_answer(self):
with mock.patch('dns.resolver.query') as query:
answer = mock.Mock('dns.resolver.Answer')
query.return_value = answer
self.assertEqual(srvlookup._query_srv_records('foo.bar.baz'),
answer)
def test_should_return_resolver_answer(self):
with mock.patch('dns.resolver.query') as query:
answer = mock.Mock('dns.resolver.Answer')
query.return_value = answer
self.assertEqual(srvlookup._query_srv_records('foo.bar.baz'),
answer)
def test_resolver_query_should_be_invoked_with_fqdn(self):
with mock.patch('dns.resolver.query') as query:
query.return_value = mock.Mock('dns.resolver.Answer')
fqdn = 'foo.bar.baz'
srvlookup._query_srv_records(fqdn)
query.assert_called_once_with(fqdn, 'SRV')
