class VultrTests(unittest.TestCase): def setUp(self): VultrMockHttp.type = None VultrDNSDriverV2.connectionCls.conn_class = VultrMockHttp self.driver = VultrDNSDriver('foo') def test_correct_class_is_used(self): self.assertIsInstance(self.driver, VultrDNSDriverV2) def test_unknown_api_version(self): self.assertRaises(NotImplementedError, VultrDNSDriver, 'foo', api_version='3') def test_list_zones(self): zones = self.driver.list_zones() self.assertEqual(len(zones), 2) zone = zones[0] self.assertEqual(zone.id, 'example.com') self.assertEqual(zone.domain, 'example.com') self.assertEqual(zone.extra['date_created'], '2021-09-07T09:52:18+00:00') def test_create_zone(self): zone = self.driver.create_zone('example.com') self.assertEqual(zone.id, 'example.com') self.assertEqual(zone.domain, 'example.com') self.assertEqual(zone.extra['date_created'], '2021-09-07T10:28:34+00:00') def test_get_zone(self): zone = self.driver.get_zone('example.com') self.assertEqual(zone.id, 'example.com') self.assertEqual(zone.domain, 'example.com') self.assertEqual(zone.extra['date_created'], '2021-09-07T09:52:18+00:00') def test_delete_zone(self): zone = self.driver.get_zone('example.com') response = self.driver.delete_zone(zone) self.assertTrue(response) def test_list_records(self): zone = self.driver.list_zones()[0] records = self.driver.list_records(zone) self.assertEqual(len(records), 5) record = records[0] self.assertEqual(record.id, '123') self.assertEqual(record.name, '') self.assertEqual(record.type, 'NS') self.assertEqual(record.data, 'ns1.vultr.com') def test_create_record(self): zone = self.driver.list_zones()[0] record = self.driver.create_record('test1', zone, 'A', '192.168.0.11') self.assertEqual(record.id, '123') self.assertEqual(record.zone.domain, zone.domain) self.assertEqual(record.type, 'A') self.assertEqual(record.name, 'test1') self.assertEqual(record.extra['priority'], 1) def test_update_record(self): zone = self.driver.list_zones()[0] record = self.driver.list_records(zone)[0] response = self.driver.update_record(record, name='test', data='192.168.0.0', extra=dict(ttl=300, priority=1)) self.assertTrue(response) def test_get_record(self): zone = self.driver.list_zones()[0] temp = self.driver.list_records(zone)[0] record = self.driver.get_record(zone.domain, temp.id) self.assertEqual(record.id, '123') self.assertEqual(record.zone.domain, zone.domain) self.assertEqual(record.type, 'NS') self.assertEqual(record.name, '') self.assertEqual(record.extra['priority'], -1) self.assertEqual(record.ttl, 300) def test_delete_record(self): zone = self.driver.list_zones()[0] record = self.driver.list_records(zone)[0] response = self.driver.delete_record(record) self.assertTrue(response)
class VultrTests(unittest.TestCase): def setUp(self): VultrMockHttp.type = None VultrDNSDriverV2.connectionCls.conn_class = VultrMockHttp self.driver = VultrDNSDriver("foo") def test_correct_class_is_used(self): self.assertIsInstance(self.driver, VultrDNSDriverV2) def test_unknown_api_version(self): self.assertRaises(NotImplementedError, VultrDNSDriver, "foo", api_version="3") def test_list_zones(self): zones = self.driver.list_zones() self.assertEqual(len(zones), 2) zone = zones[0] self.assertEqual(zone.id, "example.com") self.assertEqual(zone.domain, "example.com") self.assertEqual(zone.extra["date_created"], "2021-09-07T09:52:18+00:00") def test_create_zone(self): zone = self.driver.create_zone("example.com") self.assertEqual(zone.id, "example.com") self.assertEqual(zone.domain, "example.com") self.assertEqual(zone.extra["date_created"], "2021-09-07T10:28:34+00:00") def test_get_zone(self): zone = self.driver.get_zone("example.com") self.assertEqual(zone.id, "example.com") self.assertEqual(zone.domain, "example.com") self.assertEqual(zone.extra["date_created"], "2021-09-07T09:52:18+00:00") def test_delete_zone(self): zone = self.driver.get_zone("example.com") response = self.driver.delete_zone(zone) self.assertTrue(response) def test_list_records(self): zone = self.driver.list_zones()[0] records = self.driver.list_records(zone) self.assertEqual(len(records), 5) record = records[0] self.assertEqual(record.id, "123") self.assertEqual(record.name, "") self.assertEqual(record.type, "NS") self.assertEqual(record.data, "ns1.vultr.com") def test_create_record(self): zone = self.driver.list_zones()[0] record = self.driver.create_record("test1", zone, "A", "192.168.0.11") self.assertEqual(record.id, "123") self.assertEqual(record.zone.domain, zone.domain) self.assertEqual(record.type, "A") self.assertEqual(record.name, "test1") self.assertEqual(record.extra["priority"], 1) def test_update_record(self): zone = self.driver.list_zones()[0] record = self.driver.list_records(zone)[0] response = self.driver.update_record(record, name="test", data="192.168.0.0", extra=dict(ttl=300, priority=1)) self.assertTrue(response) def test_get_record(self): zone = self.driver.list_zones()[0] temp = self.driver.list_records(zone)[0] record = self.driver.get_record(zone.domain, temp.id) self.assertEqual(record.id, "123") self.assertEqual(record.zone.domain, zone.domain) self.assertEqual(record.type, "NS") self.assertEqual(record.name, "") self.assertEqual(record.extra["priority"], -1) self.assertEqual(record.ttl, 300) def test_delete_record(self): zone = self.driver.list_zones()[0] record = self.driver.list_records(zone)[0] response = self.driver.delete_record(record) self.assertTrue(response)