class DigitalOceanTests(LibcloudTestCase): def setUp(self): DigitalOceanNodeDriver.connectionCls.conn_classes = \ (None, DigitalOceanMockHttp) DigitalOceanMockHttp.type = None self.driver = DigitalOceanNodeDriver(*DIGITAL_OCEAN_PARAMS) def test_authentication(self): DigitalOceanMockHttp.type = 'UNAUTHORIZED_CLIENT' self.assertRaises(InvalidCredsError, self.driver.list_nodes) def test_list_images_success(self): images = self.driver.list_images() self.assertTrue(len(images) >= 1) image = images[0] self.assertTrue(image.id is not None) self.assertTrue(image.name is not None) def test_list_sizes_success(self): sizes = self.driver.list_sizes() self.assertTrue(len(sizes) >= 1) size = sizes[0] self.assertTrue(size.id is not None) self.assertEqual(size.name, '512MB') self.assertEqual(size.ram, 512) size = sizes[4] self.assertTrue(size.id is not None) self.assertEqual(size.name, '8GB') self.assertEqual(size.ram, 8 * 1024) def test_list_locations_success(self): locations = self.driver.list_locations() self.assertTrue(len(locations) >= 1) location = locations[0] self.assertEqual(location.id, '1') self.assertEqual(location.name, 'New York 1') def test_list_nodes_success(self): nodes = self.driver.list_nodes() self.assertEqual(len(nodes), 1) self.assertEqual(nodes[0].name, 'test-2') self.assertEqual(nodes[0].public_ips, []) self.assertEqual(nodes[0].extra['image_id'], 1601) self.assertEqual(nodes[0].extra['size_id'], 66) def test_create_node_invalid_size(self): image = NodeImage(id='invalid', name=None, driver=self.driver) size = self.driver.list_sizes()[0] location = self.driver.list_locations()[0] DigitalOceanMockHttp.type = 'INVALID_IMAGE' expected_msg = r'You specified an invalid image for Droplet creation. \(code: 404\)' self.assertRaisesRegexp(Exception, expected_msg, self.driver.create_node, name='test', size=size, image=image, location=location) def test_reboot_node_success(self): node = self.driver.list_nodes()[0] result = self.driver.reboot_node(node) self.assertTrue(result) def test_destroy_node_success(self): node = self.driver.list_nodes()[0] result = self.driver.destroy_node(node) self.assertTrue(result) def test_ex_rename_node_success(self): node = self.driver.list_nodes()[0] result = self.driver.ex_rename_node(node, 'fedora helios') self.assertTrue(result) def test_ex_list_ssh_keys(self): keys = self.driver.ex_list_ssh_keys() self.assertEqual(len(keys), 1) self.assertEqual(keys[0].id, 7717) self.assertEqual(keys[0].name, 'test1') self.assertEqual(keys[0].pub_key, None) def test_ex_destroy_ssh_key(self): key = self.driver.ex_list_ssh_keys()[0] result = self.driver.ex_destroy_ssh_key(key.id) self.assertTrue(result)
class DigitalOceanTests(unittest.TestCase): def setUp(self): DigitalOceanNodeDriver.connectionCls.conn_classes = \ (None, DigitalOceanMockHttp) DigitalOceanMockHttp.type = None self.driver = DigitalOceanNodeDriver(*DIGITAL_OCEAN_PARAMS) def test_authentication(self): DigitalOceanMockHttp.type = 'UNAUTHORIZED_CLIENT' self.assertRaises(InvalidCredsError, self.driver.list_nodes) def test_list_images_success(self): images = self.driver.list_images() self.assertTrue(len(images) >= 1) image = images[0] self.assertTrue(image.id is not None) self.assertTrue(image.name is not None) def test_list_sizes_success(self): sizes = self.driver.list_sizes() self.assertTrue(len(sizes) >= 1) size = sizes[0] self.assertTrue(size.id is not None) self.assertEqual(size.name, '512MB') self.assertEqual(size.ram, 512) size = sizes[4] self.assertTrue(size.id is not None) self.assertEqual(size.name, '8GB') self.assertEqual(size.ram, 8 * 1024) def test_list_locations_success(self): locations = self.driver.list_locations() self.assertTrue(len(locations) >= 1) location = locations[0] self.assertEqual(location.id, '1') self.assertEqual(location.name, 'New York 1') def test_list_nodes_success(self): nodes = self.driver.list_nodes() self.assertEqual(len(nodes), 1) self.assertEqual(nodes[0].name, 'test-2') self.assertEqual(nodes[0].public_ips, []) def test_reboot_node_success(self): node = self.driver.list_nodes()[0] result = self.driver.reboot_node(node) self.assertTrue(result) def test_destroy_node_success(self): node = self.driver.list_nodes()[0] result = self.driver.destroy_node(node) self.assertTrue(result) def test_ex_list_ssh_keys(self): keys = self.driver.ex_list_ssh_keys() self.assertEqual(len(keys), 1) self.assertEqual(keys[0].id, 7717) self.assertEqual(keys[0].name, 'test1') self.assertEqual(keys[0].pub_key, None) def test_ex_destroy_ssh_key(self): key = self.driver.ex_list_ssh_keys()[0] result = self.driver.ex_destroy_ssh_key(key.id) self.assertTrue(result)