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)