def setUp(self): self.iu = IPUpdater() self.api = DozensApi self.mc = MCache(server = Config.MEMCACHED_SERVER, username = Config.MEMCACHED_USERNAME, password = Config.MEMCACHED_PASSWORD) self.mc.refresh() self.dummy_A_record_0 = {u'name': u'hoge.me', u'prio': None, u'content': u'168.0.0.1', u'ttl': u'7200', u'type': u'A', u'id': u'1'} self.dummy_A_record_1 = {u'name': u'hoge.me', u'prio': None, u'content': u'168.0.0.2', u'ttl': u'7200', u'type': u'A', u'id': u'2'} self.dummy_A_record_2 = {u'name': u'hoge.me', u'prio': None, u'content': u'168.0.0.2', u'ttl': u'7200', u'type': u'C', u'id': u'3'}
class TestIPUpdater(unittest.TestCase): def setUp(self): self.iu = IPUpdater() self.api = DozensApi self.mc = MCache(server = Config.MEMCACHED_SERVER, username = Config.MEMCACHED_USERNAME, password = Config.MEMCACHED_PASSWORD) self.mc.refresh() self.dummy_A_record_0 = {u'name': u'hoge.me', u'prio': None, u'content': u'168.0.0.1', u'ttl': u'7200', u'type': u'A', u'id': u'1'} self.dummy_A_record_1 = {u'name': u'hoge.me', u'prio': None, u'content': u'168.0.0.2', u'ttl': u'7200', u'type': u'A', u'id': u'2'} self.dummy_A_record_2 = {u'name': u'hoge.me', u'prio': None, u'content': u'168.0.0.2', u'ttl': u'7200', u'type': u'C', u'id': u'3'} def test_001_extract_A_records(self): dummy_A_records = [self.dummy_A_record_0, self.dummy_A_record_1] with self.assertRaises(Exception): self.iu.extract_A_records(dummy_A_records) def test_002_extract_A_records(self): dummy_A_records = [self.dummy_A_record_2] res = self.iu.extract_A_records(dummy_A_records) self.assertIsNone(res) def test_003_extract_A_records(self): dummy_A_records = [] res = self.iu.extract_A_records(dummy_A_records) self.assertIsNone(res) def test_004_extract_A_records(self): dummy_A_records = [self.dummy_A_record_0] res = self.iu.extract_A_records(dummy_A_records) self.assertEqual(res['content'], self.dummy_A_record_0['content']) def test_004_get_dns_A_record(self): zone_name = "test.com" mc_key = Config.CACHE_KEY_PREFIX + zone_name self.mc.set(mc_key, self.dummy_A_record_0, Config.DEFAULT_TTL / 2) res = self.iu.get_dns_A_record(zone_name) self.assertEqual(res['content'], self.dummy_A_record_0['content']) def test_005_get_dns_A_record(self): pass