def test_get_cnames_for(self): def _load(): return yaml.load(TEST_NET) with patch.dict(self._salt, { 'mc_utils.uniquify': mc_utils.uniquify, 'mc_utils.dictupdate': mc_utils.dictupdate, 'mc_pillar.loaddb_do': _load, 'mc_pillar.query': mc_pillar.query, 'mc_pillar.load_db': mc_pillar.load_db, 'mc_pillar.get_sysadmins_keys': mc_pillar.get_sysadmins_keys, 'mc_pillar.ips_for': mc_pillar.ips_for, 'mc_pillar.ip_for': mc_pillar.ip_for, }): res = mc_pillar.rrs_cnames_for('makina-corpus.net') self.assertEqual(res, ' boo.moo.makina-corpus.net.' ' CNAME www.foo.fr.\n' ' foo.moo.makina-corpus.net.' ' CNAME a.makina-corpus.net.\n' ' testb.makina-corpus.net.' ' CNAME testa.makina-corpus.net.\n' ' testc.makina-corpus.net.' ' CNAME testb.makina-corpus.net.') self.assertEqual(mc_pillar.ip_for('testc.makina-corpus.net'), '1.2.3.10')
def test_get_ns_rr(self): def _load(): return yaml.load(TEST_NET) with patch.dict(self._salt, { 'mc_utils.uniquify': mc_utils.uniquify, 'mc_pillar.loaddb_do': _load, 'mc_pillar.query': mc_pillar.query, 'mc_pillar.load_db': mc_pillar.load_db, 'mc_utils.dictupdate': mc_utils.dictupdate, 'mc_pillar.get_sysadmins_keys': mc_pillar.get_sysadmins_keys, 'mc_pillar.ips_for': mc_pillar.ips_for, 'mc_pillar.ip_for': mc_pillar.ip_for, }): res = mc_pillar.rrs_ns_for('makina-corpus.net') self.assertEqual(res, ' @ IN NS ns1.makina-corpus.net.') res = mc_pillar.rrs_ns_for('makina-corpus.com') self.assertEqual(res, ' @ IN NS ns1.makina-corpus.com.') ip1 = mc_pillar.ip_for('ns1.makina-corpus.net') ip3 = mc_pillar.ip_for('ns1.makina-corpus.fr') ip2 = mc_pillar.ip_for('ns2.makina-corpus.fr') self.assertEqual(ip1, '1.2.3.6') self.assertEqual(ip2, '1.2.3.6') self.assertEqual(ip3, '1.2.3.5')