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(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)
def setUp(self):
DigitalOceanNodeDriver.connectionCls.conn_classes = \
(None, DigitalOceanMockHttp)
DigitalOceanMockHttp.type = None
self.driver = DigitalOceanNodeDriver(*DIGITAL_OCEAN_PARAMS)
class DigitalOcean_v2_Tests(LibcloudTestCase):
def setUp(self):
DigitalOceanNodeDriver.connectionCls.conn_class = \
DigitalOceanMockHttp
DigitalOceanMockHttp.type = None
self.driver = DigitalOceanNodeDriver(*DIGITALOCEAN_v2_PARAMS)
def test_v1_Error(self):
self.assertRaises(DigitalOcean_v1_Error, DigitalOceanNodeDriver,
*DIGITALOCEAN_v1_PARAMS, api_version='v1')
def test_v2_uses_v1_key(self):
self.assertRaises(InvalidCredsError, DigitalOceanNodeDriver,
*DIGITALOCEAN_v1_PARAMS, api_version='v2')
def test_authentication(self):
DigitalOceanMockHttp.type = 'UNAUTHORIZED'
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[1]
self.assertTrue(size.id is not None)
self.assertEqual(size.name, '1gb')
self.assertEqual(size.ram, 1024)
def test_list_locations_success(self):
locations = self.driver.list_locations()
self.assertTrue(len(locations) >= 1)
location = locations[0]
self.assertEqual(location.id, 'nyc1')
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, 'example.com')
self.assertEqual(nodes[0].public_ips, ['104.236.32.182'])
self.assertEqual(nodes[0].extra['image']['id'], 6918990)
self.assertEqual(nodes[0].extra['size_slug'], '512mb')
def test_list_nodes_fills_created_datetime(self):
nodes = self.driver.list_nodes()
self.assertEqual(nodes[0].created_at, datetime(2014, 11, 14, 16, 29, 21, tzinfo=UTC))
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|HTTPStatus.NOT_FOUND)\)'
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]
DigitalOceanMockHttp.type = 'REBOOT'
result = self.driver.reboot_node(node)
self.assertTrue(result)
def test_create_image_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'SNAPSHOT'
result = self.driver.create_image(node, 'My snapshot')
self.assertTrue(result)
def test_get_image_success(self):
image = self.driver.get_image(12345)
self.assertEqual(image.name, 'My snapshot')
self.assertEqual(image.id, '12345')
self.assertEqual(image.extra['distribution'], 'Ubuntu')
def test_delete_image_success(self):
image = self.driver.get_image(12345)
DigitalOceanMockHttp.type = 'DESTROY'
result = self.driver.delete_image(image)
self.assertTrue(result)
def test_ex_power_on_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'POWERON'
result = self.driver.ex_power_on_node(node)
self.assertTrue(result)
def test_ex_shutdown_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'SHUTDOWN'
result = self.driver.ex_shutdown_node(node)
self.assertTrue(result)
def test_ex_hard_reboot_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'POWERCYCLE'
result = self.driver.ex_hard_reboot(node)
self.assertTrue(result)
def test_destroy_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'DESTROY'
result = self.driver.destroy_node(node)
self.assertTrue(result)
def test_ex_change_kernel_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'KERNELCHANGE'
result = self.driver.ex_change_kernel(node, 7515)
self.assertTrue(result)
def test_ex_rename_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'RENAME'
result = self.driver.ex_rename_node(node, 'fedora helios')
self.assertTrue(result)
def test_list_key_pairs(self):
keys = self.driver.list_key_pairs()
self.assertEqual(len(keys), 1)
self.assertEqual(keys[0].extra['id'], 7717)
self.assertEqual(keys[0].name, 'test1')
self.assertEqual(keys[0].public_key,
"ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAQQDGk5 example")
def test_create_key_pair(self):
DigitalOceanMockHttp.type = 'CREATE'
key = self.driver.create_key_pair(
name="test1",
public_key="ssh-rsa AAAAB3NzaC1yc2EAAAADAQsxRiUKn example"
)
self.assertEqual(key.name, "test1")
self.assertEqual(key.fingerprint,
"f5:d1:78:ed:28:72:5f:e1:ac:94:fd:1f:e0:a3:48:6d")
def test_delete_key_pair(self):
key = self.driver.list_key_pairs()[0]
result = self.driver.delete_key_pair(key)
self.assertTrue(result)
def test__paginated_request_single_page(self):
nodes = self.driver._paginated_request('/v2/droplets', 'droplets')
self.assertEqual(nodes[0]['name'], 'example.com')
self.assertEqual(nodes[0]['image']['id'], 6918990)
self.assertEqual(nodes[0]['size_slug'], '512mb')
def test__paginated_request_two_pages(self):
DigitalOceanMockHttp.type = 'PAGE_ONE'
nodes = self.driver._paginated_request('/v2/droplets', 'droplets')
self.assertEqual(len(nodes), 2)
def test_list_volumes(self):
volumes = self.driver.list_volumes()
self.assertEqual(len(volumes), 1)
volume = volumes[0]
self.assertEqual(volume.id, "62766883-2c28-11e6-b8e6-000f53306ae1")
self.assertEqual(volume.name, "example")
self.assertEqual(volume.size, 4)
self.assertEqual(volume.driver, self.driver)
def test_list_volumes_empty(self):
DigitalOceanMockHttp.type = 'EMPTY'
volumes = self.driver.list_volumes()
self.assertEqual(len(volumes), 0)
def test_create_volume(self):
nyc1 = [r for r in self.driver.list_locations() if r.id == 'nyc1'][0]
DigitalOceanMockHttp.type = 'CREATE'
volume = self.driver.create_volume(4, 'example', nyc1)
self.assertEqual(volume.id, "62766883-2c28-11e6-b8e6-000f53306ae1")
self.assertEqual(volume.name, "example")
self.assertEqual(volume.size, 4)
self.assertEqual(volume.driver, self.driver)
def test_attach_volume(self):
node = self.driver.list_nodes()[0]
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = 'ATTACH'
resp = self.driver.attach_volume(node, volume)
self.assertTrue(resp)
def test_detach_volume(self):
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = 'DETACH'
resp = self.driver.detach_volume(volume)
self.assertTrue(resp)
def test_destroy_volume(self):
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = 'DESTROY'
resp = self.driver.destroy_volume(volume)
self.assertTrue(resp)
def test_list_volume_snapshots(self):
volume = self.driver.list_volumes()[0]
snapshots = self.driver.list_volume_snapshots(volume)
self.assertEqual(len(snapshots), 3)
snapshot1, snapshot2, snapshot3 = snapshots
self.assertEqual(snapshot1.id, "c0def940-9324-11e6-9a56-000f533176b1")
self.assertEqual(snapshot2.id, "c2036724-9343-11e6-aef4-000f53315a41")
self.assertEqual(snapshot3.id, "d347e033-9343-11e6-9a56-000f533176b1")
def test_create_volume_snapshot(self):
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = 'CREATE'
snapshot = self.driver.create_volume_snapshot(volume, 'test-snapshot')
self.assertEqual(snapshot.id, "c0def940-9324-11e6-9a56-000f533176b1")
self.assertEqual(snapshot.name, 'test-snapshot')
self.assertEqual(volume.driver, self.driver)
def test_delete_volume_snapshot(self):
volume = self.driver.list_volumes()[0]
snapshot = self.driver.list_volume_snapshots(volume)[0]
DigitalOceanMockHttp.type = 'DELETE'
result = self.driver.delete_volume_snapshot(snapshot)
self.assertTrue(result)
class DigitalOcean_v2_Tests(LibcloudTestCase):
def setUp(self):
DigitalOceanNodeDriver.connectionCls.conn_class = \
DigitalOceanMockHttp
DigitalOceanMockHttp.type = None
self.driver = DigitalOceanNodeDriver(*DIGITALOCEAN_v2_PARAMS)
def test_v1_Error(self):
self.assertRaises(DigitalOcean_v1_Error,
DigitalOceanNodeDriver,
*DIGITALOCEAN_v1_PARAMS,
api_version='v1')
def test_v2_uses_v1_key(self):
self.assertRaises(InvalidCredsError,
DigitalOceanNodeDriver,
*DIGITALOCEAN_v1_PARAMS,
api_version='v2')
def test_authentication(self):
DigitalOceanMockHttp.type = 'UNAUTHORIZED'
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[1]
self.assertTrue(size.id is not None)
self.assertEqual(size.name, '1gb')
self.assertEqual(size.ram, 1024)
def test_list_locations_success(self):
locations = self.driver.list_locations()
self.assertTrue(len(locations) >= 1)
location = locations[0]
self.assertEqual(location.id, 'nyc1')
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, 'example.com')
self.assertEqual(nodes[0].public_ips, ['104.236.32.182'])
self.assertEqual(nodes[0].extra['image']['id'], 6918990)
self.assertEqual(nodes[0].extra['size_slug'], '512mb')
def test_list_nodes_fills_created_datetime(self):
nodes = self.driver.list_nodes()
self.assertEqual(nodes[0].created_at,
datetime(2014, 11, 14, 16, 29, 21, tzinfo=UTC))
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|HTTPStatus.NOT_FOUND)\)'
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]
DigitalOceanMockHttp.type = 'REBOOT'
result = self.driver.reboot_node(node)
self.assertTrue(result)
def test_create_image_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'SNAPSHOT'
result = self.driver.create_image(node, 'My snapshot')
self.assertTrue(result)
def test_get_image_success(self):
image = self.driver.get_image(12345)
self.assertEqual(image.name, 'My snapshot')
self.assertEqual(image.id, '12345')
self.assertEqual(image.extra['distribution'], 'Ubuntu')
def test_delete_image_success(self):
image = self.driver.get_image(12345)
DigitalOceanMockHttp.type = 'DESTROY'
result = self.driver.delete_image(image)
self.assertTrue(result)
def test_ex_power_on_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'POWERON'
result = self.driver.ex_power_on_node(node)
self.assertTrue(result)
def test_ex_shutdown_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'SHUTDOWN'
result = self.driver.ex_shutdown_node(node)
self.assertTrue(result)
def test_ex_hard_reboot_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'POWERCYCLE'
result = self.driver.ex_hard_reboot(node)
self.assertTrue(result)
def test_destroy_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'DESTROY'
result = self.driver.destroy_node(node)
self.assertTrue(result)
def test_ex_change_kernel_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'KERNELCHANGE'
result = self.driver.ex_change_kernel(node, 7515)
self.assertTrue(result)
def test_ex_enable_ipv6_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'ENABLEIPV6'
result = self.driver.ex_enable_ipv6(node)
self.assertTrue(result)
def test_ex_rename_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'RENAME'
result = self.driver.ex_rename_node(node, 'fedora helios')
self.assertTrue(result)
def test_list_key_pairs(self):
keys = self.driver.list_key_pairs()
self.assertEqual(len(keys), 1)
self.assertEqual(keys[0].extra['id'], 7717)
self.assertEqual(keys[0].name, 'test1')
self.assertEqual(keys[0].public_key,
"ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAQQDGk5 example")
def test_create_key_pair(self):
DigitalOceanMockHttp.type = 'CREATE'
key = self.driver.create_key_pair(
name="test1",
public_key="ssh-rsa AAAAB3NzaC1yc2EAAAADAQsxRiUKn example")
self.assertEqual(key.name, "test1")
self.assertEqual(key.fingerprint,
"f5:d1:78:ed:28:72:5f:e1:ac:94:fd:1f:e0:a3:48:6d")
def test_delete_key_pair(self):
key = self.driver.list_key_pairs()[0]
result = self.driver.delete_key_pair(key)
self.assertTrue(result)
def test__paginated_request_single_page(self):
nodes = self.driver._paginated_request('/v2/droplets', 'droplets')
self.assertEqual(nodes[0]['name'], 'example.com')
self.assertEqual(nodes[0]['image']['id'], 6918990)
self.assertEqual(nodes[0]['size_slug'], '512mb')
def test__paginated_request_two_pages(self):
DigitalOceanMockHttp.type = 'PAGE_ONE'
nodes = self.driver._paginated_request('/v2/droplets', 'droplets')
self.assertEqual(len(nodes), 2)
def test_list_volumes(self):
volumes = self.driver.list_volumes()
self.assertEqual(len(volumes), 1)
volume = volumes[0]
self.assertEqual(volume.id, "62766883-2c28-11e6-b8e6-000f53306ae1")
self.assertEqual(volume.name, "example")
self.assertEqual(volume.size, 4)
self.assertEqual(volume.driver, self.driver)
def test_list_volumes_empty(self):
DigitalOceanMockHttp.type = 'EMPTY'
volumes = self.driver.list_volumes()
self.assertEqual(len(volumes), 0)
def test_create_volume(self):
nyc1 = [r for r in self.driver.list_locations() if r.id == 'nyc1'][0]
DigitalOceanMockHttp.type = 'CREATE'
volume = self.driver.create_volume(4, 'example', nyc1)
self.assertEqual(volume.id, "62766883-2c28-11e6-b8e6-000f53306ae1")
self.assertEqual(volume.name, "example")
self.assertEqual(volume.size, 4)
self.assertEqual(volume.driver, self.driver)
def test_attach_volume(self):
node = self.driver.list_nodes()[0]
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = 'ATTACH'
resp = self.driver.attach_volume(node, volume)
self.assertTrue(resp)
def test_detach_volume(self):
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = 'DETACH'
resp = self.driver.detach_volume(volume)
self.assertTrue(resp)
def test_destroy_volume(self):
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = 'DESTROY'
resp = self.driver.destroy_volume(volume)
self.assertTrue(resp)
def test_list_volume_snapshots(self):
volume = self.driver.list_volumes()[0]
snapshots = self.driver.list_volume_snapshots(volume)
self.assertEqual(len(snapshots), 3)
snapshot1, snapshot2, snapshot3 = snapshots
self.assertEqual(snapshot1.id, "c0def940-9324-11e6-9a56-000f533176b1")
self.assertEqual(snapshot2.id, "c2036724-9343-11e6-aef4-000f53315a41")
self.assertEqual(snapshot3.id, "d347e033-9343-11e6-9a56-000f533176b1")
def test_create_volume_snapshot(self):
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = 'CREATE'
snapshot = self.driver.create_volume_snapshot(volume, 'test-snapshot')
self.assertEqual(snapshot.id, "c0def940-9324-11e6-9a56-000f533176b1")
self.assertEqual(snapshot.name, 'test-snapshot')
self.assertEqual(volume.driver, self.driver)
def test_delete_volume_snapshot(self):
volume = self.driver.list_volumes()[0]
snapshot = self.driver.list_volume_snapshots(volume)[0]
DigitalOceanMockHttp.type = 'DELETE'
result = self.driver.delete_volume_snapshot(snapshot)
self.assertTrue(result)
class DigitalOcean_v2_Tests(LibcloudTestCase):
def setUp(self):
DigitalOceanNodeDriver.connectionCls.conn_classes = \
(None, DigitalOceanMockHttp)
DigitalOceanMockHttp.type = None
self.driver = DigitalOceanNodeDriver(*DIGITALOCEAN_v2_PARAMS)
def test_authentication(self):
DigitalOceanMockHttp.type = 'UNAUTHORIZED'
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[1]
self.assertTrue(size.id is not None)
self.assertEqual(size.name, '1gb')
self.assertEqual(size.ram, 1024)
def test_list_locations_success(self):
locations = self.driver.list_locations()
self.assertTrue(len(locations) >= 1)
location = locations[0]
self.assertEqual(location.id, 'nyc1')
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, 'example.com')
self.assertEqual(nodes[0].public_ips, ['104.236.32.182'])
self.assertEqual(nodes[0].extra['image']['id'], 6918990)
self.assertEqual(nodes[0].extra['size_slug'], '512mb')
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]
DigitalOceanMockHttp.type = 'REBOOT'
result = self.driver.reboot_node(node)
self.assertTrue(result)
def test_create_image_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'SNAPSHOT'
result = self.driver.create_image(node, 'My snapshot')
self.assertTrue(result)
def test_get_image_success(self):
image = self.driver.get_image(12345)
self.assertEqual(image.name, 'My snapshot')
self.assertEqual(image.id, '12345')
self.assertEqual(image.extra['distribution'], 'Ubuntu')
def test_delete_image_success(self):
image = self.driver.get_image(12345)
DigitalOceanMockHttp.type = 'DESTROY'
result = self.driver.delete_image(image)
self.assertTrue(result)
def test_ex_power_on_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'POWERON'
result = self.driver.ex_power_on_node(node)
self.assertTrue(result)
def test_ex_shutdown_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'SHUTDOWN'
result = self.driver.ex_shutdown_node(node)
self.assertTrue(result)
def test_destroy_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'DESTROY'
result = self.driver.destroy_node(node)
self.assertTrue(result)
def test_ex_rename_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'RENAME'
result = self.driver.ex_rename_node(node, 'fedora helios')
self.assertTrue(result)
def test_list_key_pairs(self):
keys = self.driver.list_key_pairs()
self.assertEqual(len(keys), 1)
self.assertEqual(keys[0].extra['id'], 7717)
self.assertEqual(keys[0].name, 'test1')
self.assertEqual(keys[0].public_key,
"ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAQQDGk5 example")
def test_create_key_pair(self):
DigitalOceanMockHttp.type = 'CREATE'
key = self.driver.create_key_pair(
name="test1",
public_key="ssh-rsa AAAAB3NzaC1yc2EAAAADAQsxRiUKn example")
self.assertEqual(key.name, "test1")
self.assertEqual(key.fingerprint,
"f5:d1:78:ed:28:72:5f:e1:ac:94:fd:1f:e0:a3:48:6d")
def test_delete_key_pair(self):
key = self.driver.list_key_pairs()[0]
result = self.driver.delete_key_pair(key)
self.assertTrue(result)
def test__paginated_request_single_page(self):
nodes = self.driver._paginated_request('/v2/droplets', 'droplets')
self.assertEqual(nodes[0]['name'], 'example.com')
self.assertEqual(nodes[0]['image']['id'], 6918990)
self.assertEqual(nodes[0]['size_slug'], '512mb')
def test__paginated_request_two_pages(self):
DigitalOceanMockHttp.type = 'PAGE_ONE'
nodes = self.driver._paginated_request('/v2/droplets', 'droplets')
self.assertEqual(len(nodes), 2)
class DigitalOcean_v2_Tests(LibcloudTestCase):
def setUp(self):
DigitalOceanNodeDriver.connectionCls.conn_class = \
DigitalOceanMockHttp
DigitalOceanMockHttp.type = None
self.driver = DigitalOceanNodeDriver(*DIGITALOCEAN_v2_PARAMS)
def test_v1_Error(self):
self.assertRaises(DigitalOcean_v1_Error, DigitalOceanNodeDriver,
*DIGITALOCEAN_v1_PARAMS, api_version='v1')
def test_v2_uses_v1_key(self):
self.assertRaises(InvalidCredsError, DigitalOceanNodeDriver,
*DIGITALOCEAN_v1_PARAMS, api_version='v2')
def test_authentication(self):
DigitalOceanMockHttp.type = 'UNAUTHORIZED'
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[1]
self.assertTrue(size.id is not None)
self.assertEqual(size.name, '1gb')
self.assertEqual(size.ram, 1024)
def test_list_locations_success(self):
locations = self.driver.list_locations()
self.assertTrue(len(locations) >= 1)
location = locations[0]
self.assertEqual(location.id, 'nyc1')
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, 'example.com')
self.assertEqual(nodes[0].public_ips, ['104.236.32.182'])
self.assertEqual(nodes[0].extra['image']['id'], 6918990)
self.assertEqual(nodes[0].extra['size_slug'], '512mb')
self.assertEqual(len(nodes[0].extra['tags']), 2)
def test_list_nodes_fills_created_datetime(self):
nodes = self.driver.list_nodes()
self.assertEqual(nodes[0].created_at, datetime(2014, 11, 14, 16, 29, 21, tzinfo=UTC))
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|HTTPStatus.NOT_FOUND)\)'
assertRaisesRegex(self, 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]
DigitalOceanMockHttp.type = 'REBOOT'
result = self.driver.reboot_node(node)
self.assertTrue(result)
def test_create_image_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'SNAPSHOT'
result = self.driver.create_image(node, 'My snapshot')
self.assertTrue(result)
def test_get_image_success(self):
image = self.driver.get_image(12345)
self.assertEqual(image.name, 'My snapshot')
self.assertEqual(image.id, '12345')
self.assertEqual(image.extra['distribution'], 'Ubuntu')
def test_delete_image_success(self):
image = self.driver.get_image(12345)
DigitalOceanMockHttp.type = 'DESTROY'
result = self.driver.delete_image(image)
self.assertTrue(result)
def test_ex_power_on_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'POWERON'
result = self.driver.ex_power_on_node(node)
self.assertTrue(result)
def test_ex_shutdown_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'SHUTDOWN'
result = self.driver.ex_shutdown_node(node)
self.assertTrue(result)
def test_ex_hard_reboot_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'POWERCYCLE'
result = self.driver.ex_hard_reboot(node)
self.assertTrue(result)
def test_ex_rebuild_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'REBUILD'
result = self.driver.ex_rebuild_node(node)
self.assertTrue(result)
def test_ex_resize_node_success(self):
node = self.driver.list_nodes()[0]
size = self.driver.list_sizes()[0]
DigitalOceanMockHttp.type = 'RESIZE'
result = self.driver.ex_resize_node(node, size)
self.assertTrue(result)
def test_destroy_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'DESTROY'
result = self.driver.destroy_node(node)
self.assertTrue(result)
def test_ex_change_kernel_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'KERNELCHANGE'
result = self.driver.ex_change_kernel(node, 7515)
self.assertTrue(result)
def test_ex_enable_ipv6_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'ENABLEIPV6'
result = self.driver.ex_enable_ipv6(node)
self.assertTrue(result)
def test_ex_rename_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'RENAME'
result = self.driver.ex_rename_node(node, 'fedora helios')
self.assertTrue(result)
def test_list_key_pairs(self):
keys = self.driver.list_key_pairs()
self.assertEqual(len(keys), 1)
self.assertEqual(keys[0].extra['id'], 7717)
self.assertEqual(keys[0].name, 'test1')
self.assertEqual(keys[0].public_key,
"ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAQQDGk5 example")
def test_create_key_pair(self):
DigitalOceanMockHttp.type = 'CREATE'
key = self.driver.create_key_pair(
name="test1",
public_key="ssh-rsa AAAAB3NzaC1yc2EAAAADAQsxRiUKn example"
)
self.assertEqual(key.name, "test1")
self.assertEqual(key.fingerprint,
"f5:d1:78:ed:28:72:5f:e1:ac:94:fd:1f:e0:a3:48:6d")
def test_delete_key_pair(self):
key = self.driver.list_key_pairs()[0]
result = self.driver.delete_key_pair(key)
self.assertTrue(result)
def test__paginated_request_single_page(self):
nodes = self.driver._paginated_request('/v2/droplets', 'droplets')
self.assertEqual(nodes[0]['name'], 'example.com')
self.assertEqual(nodes[0]['image']['id'], 6918990)
self.assertEqual(nodes[0]['size_slug'], '512mb')
def test__paginated_request_two_pages(self):
DigitalOceanMockHttp.type = 'PAGE_ONE'
nodes = self.driver._paginated_request('/v2/droplets', 'droplets')
self.assertEqual(len(nodes), 2)
def test_list_volumes(self):
volumes = self.driver.list_volumes()
self.assertEqual(len(volumes), 1)
volume = volumes[0]
self.assertEqual(volume.id, "62766883-2c28-11e6-b8e6-000f53306ae1")
self.assertEqual(volume.name, "example")
self.assertEqual(volume.size, 4)
self.assertEqual(volume.driver, self.driver)
def test_list_volumes_empty(self):
DigitalOceanMockHttp.type = 'EMPTY'
volumes = self.driver.list_volumes()
self.assertEqual(len(volumes), 0)
def test_create_volume(self):
nyc1 = [r for r in self.driver.list_locations() if r.id == 'nyc1'][0]
DigitalOceanMockHttp.type = 'CREATE'
volume = self.driver.create_volume(4, 'example', nyc1)
self.assertEqual(volume.id, "62766883-2c28-11e6-b8e6-000f53306ae1")
self.assertEqual(volume.name, "example")
self.assertEqual(volume.size, 4)
self.assertEqual(volume.driver, self.driver)
def test_attach_volume(self):
node = self.driver.list_nodes()[0]
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = 'ATTACH'
resp = self.driver.attach_volume(node, volume)
self.assertTrue(resp)
def test_detach_volume(self):
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = 'DETACH'
resp = self.driver.detach_volume(volume)
self.assertTrue(resp)
def test_destroy_volume(self):
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = 'DESTROY'
resp = self.driver.destroy_volume(volume)
self.assertTrue(resp)
def test_list_volume_snapshots(self):
volume = self.driver.list_volumes()[0]
snapshots = self.driver.list_volume_snapshots(volume)
self.assertEqual(len(snapshots), 3)
snapshot1, snapshot2, snapshot3 = snapshots
self.assertEqual(snapshot1.id, "c0def940-9324-11e6-9a56-000f533176b1")
self.assertEqual(snapshot2.id, "c2036724-9343-11e6-aef4-000f53315a41")
self.assertEqual(snapshot3.id, "d347e033-9343-11e6-9a56-000f533176b1")
def test_create_volume_snapshot(self):
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = 'CREATE'
snapshot = self.driver.create_volume_snapshot(volume, 'test-snapshot')
self.assertEqual(snapshot.id, "c0def940-9324-11e6-9a56-000f533176b1")
self.assertEqual(snapshot.name, 'test-snapshot')
self.assertEqual(volume.driver, self.driver)
def test_delete_volume_snapshot(self):
volume = self.driver.list_volumes()[0]
snapshot = self.driver.list_volume_snapshots(volume)[0]
DigitalOceanMockHttp.type = 'DELETE'
result = self.driver.delete_volume_snapshot(snapshot)
self.assertTrue(result)
def test_ex_get_node_details(self):
node = self.driver.ex_get_node_details('3164444')
self.assertEqual(node.name, 'example.com')
self.assertEqual(node.public_ips, ['36.123.0.123'])
self.assertEqual(node.extra['image']['id'], 12089443)
self.assertEqual(node.extra['size_slug'], '8gb')
self.assertEqual(len(node.extra['tags']), 2)
def test_ex_create_floating_ip(self):
nyc1 = [r for r in self.driver.list_locations() if r.id == 'nyc1'][0]
floating_ip = self.driver.ex_create_floating_ip(nyc1)
# Note that this is the ID. There is no real ID for a floating IP at
# DigitalOcean, but the IP is unique so we can use that instead.
self.assertEqual(floating_ip.id, '167.138.123.111')
self.assertEqual(floating_ip.ip_address, '167.138.123.111')
self.assertEqual(floating_ip.extra['region']['slug'], 'nyc1')
# The newly created floating IP reserved to a region is not
# associated with any droplet. See the DigitalOcean API docs
# how to create a floating IP that is associated with an instance
# from the start. This API call creates an unattached IP.
self.assertIsNone(floating_ip.node_id)
def test_ex_delete_floating_ip(self):
nyc1 = [r for r in self.driver.list_locations() if r.id == 'nyc1'][0]
floating_ip = self.driver.ex_create_floating_ip(nyc1)
ret = self.driver.ex_delete_floating_ip(floating_ip)
# The API returns 204 NO CONTENT if all is well.
self.assertTrue(ret)
def test_floating_ip_can_be_deleted_by_calling_delete_on_floating_ip_object(self):
nyc1 = [r for r in self.driver.list_locations() if r.id == 'nyc1'][0]
floating_ip = self.driver.ex_create_floating_ip(nyc1)
ret = floating_ip.delete()
self.assertTrue(ret)
def test_list_floating_ips(self):
floating_ips = self.driver.ex_list_floating_ips()
self.assertEqual(len(floating_ips), 2, 'Wrong floating IPs count')
floating_ip = floating_ips[0]
self.assertEqual(floating_ip.id, '133.166.122.204')
self.assertEqual(floating_ip.ip_address, '133.166.122.204')
self.assertEqual(floating_ip.extra['region']['slug'], 'ams3')
self.assertEqual(84155775, floating_ip.node_id)
def test_get_floating_ip(self):
floating_ip = self.driver.ex_get_floating_ip('133.166.122.204')
self.assertEqual(floating_ip.id, '133.166.122.204')
self.assertEqual(floating_ip.ip_address, '133.166.122.204')
self.assertEqual(floating_ip.extra['region']['slug'], 'ams3')
self.assertEqual(84155775, floating_ip.node_id)
def test_ex_attach_floating_ip_to_node(self):
node = self.driver.list_nodes()[0]
floating_ip = self.driver.ex_get_floating_ip('133.166.122.204')
ret = self.driver.ex_attach_floating_ip_to_node(node, floating_ip)
self.assertTrue(ret)
def test_ex_detach_floating_ip_from_node(self):
node = self.driver.list_nodes()[0]
floating_ip = self.driver.ex_get_floating_ip('154.138.103.175')
ret = self.driver.ex_detach_floating_ip_from_node(node, floating_ip)
self.assertTrue(ret)
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)
class DigitalOcean_v2_Tests(LibcloudTestCase):
def setUp(self):
DigitalOceanNodeDriver.connectionCls.conn_classes = \
(None, DigitalOceanMockHttp)
DigitalOceanMockHttp.type = None
self.driver = DigitalOceanNodeDriver(*DIGITALOCEAN_v2_PARAMS)
def test_v2_uses_v1_key(self):
self.assertRaises(InvalidCredsError, DigitalOceanNodeDriver,
*DIGITALOCEAN_v1_PARAMS, api_version='v2')
def test_authentication(self):
DigitalOceanMockHttp.type = 'UNAUTHORIZED'
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[1]
self.assertTrue(size.id is not None)
self.assertEqual(size.name, '1gb')
self.assertEqual(size.ram, 1024)
def test_list_locations_success(self):
locations = self.driver.list_locations()
self.assertTrue(len(locations) >= 1)
location = locations[0]
self.assertEqual(location.id, 'nyc1')
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, 'example.com')
self.assertEqual(nodes[0].public_ips, ['104.236.32.182'])
self.assertEqual(nodes[0].extra['image']['id'], 6918990)
self.assertEqual(nodes[0].extra['size_slug'], '512mb')
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|HTTPStatus.NOT_FOUND)\)'
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]
DigitalOceanMockHttp.type = 'REBOOT'
result = self.driver.reboot_node(node)
self.assertTrue(result)
def test_create_image_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'SNAPSHOT'
result = self.driver.create_image(node, 'My snapshot')
self.assertTrue(result)
def test_get_image_success(self):
image = self.driver.get_image(12345)
self.assertEqual(image.name, 'My snapshot')
self.assertEqual(image.id, '12345')
self.assertEqual(image.extra['distribution'], 'Ubuntu')
def test_delete_image_success(self):
image = self.driver.get_image(12345)
DigitalOceanMockHttp.type = 'DESTROY'
result = self.driver.delete_image(image)
self.assertTrue(result)
def test_ex_power_on_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'POWERON'
result = self.driver.ex_power_on_node(node)
self.assertTrue(result)
def test_ex_shutdown_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'SHUTDOWN'
result = self.driver.ex_shutdown_node(node)
self.assertTrue(result)
def test_destroy_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'DESTROY'
result = self.driver.destroy_node(node)
self.assertTrue(result)
def test_ex_rename_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = 'RENAME'
result = self.driver.ex_rename_node(node, 'fedora helios')
self.assertTrue(result)
def test_list_key_pairs(self):
keys = self.driver.list_key_pairs()
self.assertEqual(len(keys), 1)
self.assertEqual(keys[0].extra['id'], 7717)
self.assertEqual(keys[0].name, 'test1')
self.assertEqual(keys[0].public_key,
"ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAQQDGk5 example")
def test_create_key_pair(self):
DigitalOceanMockHttp.type = 'CREATE'
key = self.driver.create_key_pair(
name="test1",
public_key="ssh-rsa AAAAB3NzaC1yc2EAAAADAQsxRiUKn example"
)
self.assertEqual(key.name, "test1")
self.assertEqual(key.fingerprint,
"f5:d1:78:ed:28:72:5f:e1:ac:94:fd:1f:e0:a3:48:6d")
def test_delete_key_pair(self):
key = self.driver.list_key_pairs()[0]
result = self.driver.delete_key_pair(key)
self.assertTrue(result)
def test__paginated_request_single_page(self):
nodes = self.driver._paginated_request('/v2/droplets', 'droplets')
self.assertEqual(nodes[0]['name'], 'example.com')
self.assertEqual(nodes[0]['image']['id'], 6918990)
self.assertEqual(nodes[0]['size_slug'], '512mb')
def test__paginated_request_two_pages(self):
DigitalOceanMockHttp.type = 'PAGE_ONE'
nodes = self.driver._paginated_request('/v2/droplets', 'droplets')
self.assertEqual(len(nodes), 2)
def createNode(provider, driverUno, driverDos, driverTres, driverCuatro, name,
size, image, location, ex_network):
pass
nodesProvider = ''
if provider == "Digital Ocean":
pass
accessKey = driverUno
sshKey = driverDos
driverTres = driverTres
driverCuatro = driverCuatro
driver = DigitalOceanNodeDriver(accessKey)
sshKeyID = list()
ssh = int(sshKey)
keyPairs = driver.list_key_pairs()
sizesNode = driver.list_sizes()
imagesNode = driver.list_images()
locationsNode = driver.list_locations()
for keyPair in keyPairs:
if keyPair.extra['id'] == ssh:
pass
sshKeyID.append(keyPair.extra['id'])
for sizeNode in sizesNode:
if sizeNode.name == size:
pass
sizeName = sizeNode
for imageNode in imagesNode:
if imageNode.id == image:
pass
imageId = imageNode
for locationNode in locationsNode:
if locationNode.id == location:
pass
locationId = locationNode
idNodes = driver.create_node(name, sizeName, imageId, locationId, None,
sshKeyID, None)
idNodes.extra = json.dumps(idNodes.extra)
attr = {
'id': idNodes.id,
'name': idNodes.name,
'state': idNodes.state,
'public_ips': idNodes.public_ips,
'private_ips': idNodes.private_ips,
'provider': 'DigitalOcean',
'extra': idNodes.extra
}
nodesProvider = json.dumps(attr)
if provider == "EC2":
pass
accessId = driverUno
secretKey = driverDos
region = driverTres
driverCuatro = driverCuatro
reg = region[0:len(region) - 1]
# driver = BaseEC2NodeDriver(accessId,secretKey,'eu-west-1')
cls = get_driver(Provider.EC2)
driver = cls(accessId, secretKey, region=reg)
sizesNode = driver.list_sizes()
imagesNode = driver.list_images()
sizeName = ''
imageId = ''
for sizeNode in sizesNode:
if sizeNode.id == size:
pass
sizeName = sizeNode
for imageNode in imagesNode:
if imageNode.name == image:
pass
imageId = imageNode
nodeCreate = driver.create_node(name=name,
image=imageId,
size=sizeName)
types = 'running'
node = checkStatus.checkStatus(driver, nodeCreate.id, types)
if node != 0:
pass
elastic = elasticIP.setElasticIP(driver, node.id)
if elastic != 0:
pass
v4 = []
ips = {
'ipaddress': elastic.ip,
'gateway': 'NULL',
'mask': 'NULL',
'private_ip': node.private_ips[0]
}
v4.append(ips)
network = {'v4': v4}
extra = {
'launch_time': node.extra['launch_time'],
'instance_type': node.extra['instance_type'],
'network': network
}
attr = {
'id': node.id,
'region': reg,
'name': node.name,
'state': node.state,
'public_ip': elastic.ip,
'provider': 'Amazon',
'extra': extra
}
print json.dumps(attr)
else:
v4 = []
ips = {
'ipaddress': node.public_ips[0],
'gateway': 'NULL',
'mask': 'NULL',
'private_ip': node.private_ips[0]
}
v4.append(ips)
network = {'v4': v4}
extra = {
'launch_time': node.extra['launch_time'],
'instance_type': node.extra['instance_type'],
'network': network
}
attr = {
'id': node.id,
'region': region,
'name': node.name,
'state': node.state,
'public_ip': node.public_ips[0],
'provider': 'Amazon',
'extra': extra,
'elastic': '0'
}
nodesProvider = json.dumps(attr)
if provider == "Azure":
pass
tenantId = driverUno
subscriptionId = driverDos
applicationId = driverTres
keyPaswd = driverCuatro
driver = AzureNodeDriver(tenantId, subscriptionId, applicationId,
keyPaswd)
locations = driver.list_locations()
networks = driver.ex_list_networks()
for network in networks:
pass
if network.name == ex_network:
pass
ex_network = network
for loc in locations:
pass
if loc.id == location:
pass
locationID = loc
sizesNode = driver.list_sizes(location=locationID)
imagesNode = driver.list_images(location=locationID,
ex_publisher=None,
ex_offer=None,
ex_sku=None,
ex_version=None)
for sizeNode in sizesNode:
if sizeNode.name == size:
pass
sizeName = sizeNode
for imageNode in imagesNode:
if imageNode.id == image:
pass
imageId = imageNode
nodeCreate = driver.create_node(name,
sizeName,
imageId,
None,
None,
None,
location=locationID,
ex_network=ex_network,
ex_subnet=None,
ex_nic=None)
types = 'running'
node = checkStatus.checkStatus(driver, nodeCreate.id, types)
if node != 0:
pass
v4 = []
ips = {
'ipaddress': node.public_ips,
'gateway': 'NULL',
'mask': 'NULL',
'private_ip': node.private_ips
}
v4.append(ips)
network = {'v4': v4}
extra = {
'instance_type':
node.extra['properties']['hardwareProfile']['vmSize'],
'network':
network
}
attr = {
'id': node.extra['properties']['vmId'],
'region': node.extra['location'],
'name': node.name,
'state': node.state,
'public_ip': node.public_ips,
'provider': 'Azure Virtual machines',
'extra': extra
}
nodesProvider = json.dumps(attr)
if provider == "Linode":
pass
apiKey = driverUno
driverDos = driverDos
driverTres = driverTres
driverCuatro = driverCuatro
nodeId = nodeId
driver = LinodeNodeDriver(apiKey)
node = driver.create_node(name, image, size, auth, location)
nodesProvider = json.dumps(node)
return nodesProvider
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)
def getNode(provider,driverUno,driverDos,driverTres,driverCuatro,nodeId):
pass
nodesProvider = ''
if provider == "Digital Ocean":
pass
accessKey = driverUno
driverDos = driverDos
driverTres = driverTres
driverCuatro = driverCuatro
nodeId = nodeId
driver = DigitalOceanNodeDriver(accessKey)
idsNodes = driver.list_nodes()
for idNodes in idsNodes:
#print idsNodes
if idNodes.id == nodeId:
pass
idNod = idNodes
# print idNod
idNod.extra = json.dumps(idNod.extra);
attr = {'id' : idNod.id, 'name' : idNod.name,
'state' : idNod.state, 'public_ips' : idNod.public_ips,
'private_ips' : idNod.private_ips, 'provider' : 'DigitalOcean',
'extra' : idNod.extra}
nodesProvider = json.dumps(attr)
if provider == "EC2":
pass
accessId = driverUno
secretKey = driverDos
region = driverTres
driverCuatro = driverCuatro
nodeId = nodeId
# driver = BaseEC2NodeDriver(accessId,secretKey,'eu-west-1')
cls = get_driver(Provider.EC2)
driver = cls(accessId, secretKey, region=region)
idsNodes = driver.list_nodes()
for idNodes in idsNodes:
#print idsNodes
if idNodes.id == nodeId:
pass
node = idNodes
if node.state == 'terminated':
v4 = []
ips = {'ipaddress' : node.public_ips, 'gateway' : 'NULL', 'mask' : 'NULL', 'private_ip' : node.private_ips}
v4.append(ips)
network = {'v4' : v4}
extra = {'launch_time' : node.extra['launch_time'], 'instance_type' : node.extra['instance_type'],
'network' : network}
attr = {'id' : node.id, 'region' : region, 'name': node.name, 'state' : node.state,
'public_ip' : node.public_ips, 'provider' : 'Amazon', 'extra' : extra}
if node.state != 'terminated':
v4 = []
ips = {'ipaddress' : node.public_ips[0], 'gateway' : 'NULL', 'mask' : 'NULL', 'private_ip' : node.private_ips[0]}
v4.append(ips)
network = {'v4' : v4}
extra = {'launch_time' : node.extra['launch_time'], 'instance_type' : node.extra['instance_type'],
'network' : network}
attr = {'id' : node.id, 'region' : region, 'name': node.name, 'state' : node.state,
'public_ip' : node.public_ips[0], 'provider' : 'Amazon', 'extra' : extra}
nodesProvider = json.dumps(attr)
if provider == "Azure":
pass
tenantId = driverUno
subscriptionId = driverDos
applicationId = driverTres
keyPaswd = driverCuatro
nodeId = nodeId
driver = AzureNodeDriver(tenantId,subscriptionId,applicationId,keyPaswd)
idsNodes = driver.list_nodes()
for idNodes in idsNodes:
#print idsNodes
if idNodes.id == nodeId:
pass
node = idNodes
# print idNod
v4 = []
ips = {'ipaddress' : node.public_ips, 'gateway' : 'NULL', 'mask' : 'NULL', 'private_ip' : node.private_ips}
v4.append(ips)
network = {'v4' : v4}
extra = {'instance_type' : node.extra['properties']['hardwareProfile']['vmSize'],
'network' : network}
attr = {'id' : node.extra['properties']['vmId'], 'region' : node.extra['location'], 'name': node.name, 'state' : node.state,
'public_ip' : node.public_ips, 'provider' : 'Azure Virtual machines', 'extra' : extra}
nodesProvider = json.dumps(node)
if provider == "Linode":
pass
apiKey = driverUno
driverDos = driverDos
driverTres = driverTres
driverCuatro = driverCuatro
nodeId = nodeId
driver = LinodeNodeDriver(apiKey)
for idNodes in idsNodes:
#print idsNodes
if idNodes.id == nodeId:
pass
idNod = idNodes
# print idNod
nodesProvider = json.dumps(idNod)
return nodesProvider
def getLocations(provider, driverUno, driverDos, driverTres, driverCuatro):
pass
nodesProvider = ''
if provider == "Digital Ocean":
pass
accessKey = driverUno
driverDos = driverDos
driverTres = driverTres
driverCuatro = driverCuatro
driver = DigitalOceanNodeDriver(accessKey)
listLocations = driver.list_locations()
nodelocations = []
for listLocation in listLocations:
data = []
flag = False
matches = re.search(r'[a-zA-Z]{0,}', listLocation.name)
match = re.search(r'[a-zA-Z]{0,}[ ][a-zA-Z]{0,}',
listLocation.name)
if match:
region = match.group()
else:
region = matches.group()
for nodelocation in nodelocations:
if nodelocation['city'] == region:
datacenter = {
'id': listLocation.id,
'name': listLocation.name
}
nodelocation['datacenter'].append(datacenter)
flag = True
if flag != True:
datacenter = {'id': listLocation.id, 'name': listLocation.name}
data.append(datacenter)
node = {'country': 'null', 'city': region, 'datacenter': data}
nodelocations.append(node)
nodesProvider = json.dumps(nodelocations)
if provider == "EC2":
pass
accessId = driverUno
secretKey = driverDos
region = driverTres
driverCuatro = driverCuatro
driver = EC2NodeDriver(accessId, secretKey)
listRegions = driver.list_regions()
# print listRegions
nodelocations = []
for listRegion in listRegions:
region = listRegion
if region == 'ap-south-1':
pass
region = 'ap-southeast-1'
if region == 'us-east-2':
pass
region = 'us-east-1'
if region == 'cn-north-1':
pass
region = 'us-east-1'
if region == 'ca-central-1':
pass
region = 'us-east-1'
if region == 'ap-northeast-2':
pass
region = 'ap-northeast-1'
if region == 'us-gov-west-1':
pass
region = 'us-east-1'
if region == 'eu-central-1':
pass
region = 'eu-west-1'
if region == 'eu-west-2':
pass
region = 'eu-west-1'
driverLocation = EC2NodeDriver(accessId, secretKey, None, None,
None, region, None)
listLocations = driverLocation.list_locations()
# print region
# print listLocations
datacenterArray = []
for listLocation in listLocations:
flag = False
for nodelocation in nodelocations:
if 'country' in nodelocation:
if nodelocation['country'] == listLocation.country:
datacenter = {
'id': listLocation.name,
'name': listLocation.name
}
datacenterArray.append(datacenter)
flag = True
if flag != True:
datacenter = {
'id': listLocation.name,
'name': listLocation.name
}
datacenterArray.append(datacenter)
regionName = getRegionName.regionsAmazon(listRegion)
if regionName != 'null':
regionID = {
'country': listLocation.country,
'city': regionName,
'datacenter': datacenterArray
}
nodelocations.append(regionID)
nodesProvider = json.dumps(nodelocations)
if provider == "Azure":
pass
tenantId = driverUno
subscriptionId = driverDos
applicationId = driverTres
keyPaswd = driverCuatro
driver = AzureNodeDriver(tenantId, subscriptionId, applicationId,
keyPaswd)
listLocations = driver.list_locations()
nodelocations = []
for listLocation in listLocations:
data = []
flag = False
country = listLocation.country
if country is not None:
pass
matches = country.partition(',')
if matches[1] == ',':
dos = matches[2]
country = dos[1:len(dos)]
city = matches[0]
for nodelocation in nodelocations:
if nodelocation['city'] == city:
datacenter = {
'id': listLocation.id,
'name': listLocation.name
}
nodelocation['datacenter'].append(datacenter)
flag = True
if flag != True:
datacenter = {
'id': listLocation.id,
'name': listLocation.name
}
data.append(datacenter)
node = {
'country': country,
'city': city,
'datacenter': data
}
nodelocations.append(node)
else:
city = 'null'
datacenter = {
'id': listLocation.id,
'name': listLocation.name
}
data.append(datacenter)
node = {
'country': country,
'city': city,
'datacenter': data
}
nodelocations.append(node)
else:
city = getRegionName.regionsAzure(listLocation.name)
country = 'null'
datacenter = {'id': listLocation.id, 'name': listLocation.name}
data.append(datacenter)
node = {'country': country, 'city': city, 'datacenter': data}
nodelocations.append(node)
nodesProvider = json.dumps(nodelocations)
if provider == "Linode":
pass
apiKey = driverUno
driverDos = driverDos
driverTres = driverTres
driverCuatro = driverCuatro
driver = DigitalOceanNodeDriver(apiKey)
listLocations = driver.list_locations()
nodesProvider = listLocations
return nodesProvider
def setUp(self):
DigitalOceanNodeDriver.connectionCls.conn_class = \
DigitalOceanMockHttp
DigitalOceanMockHttp.type = None
self.driver = DigitalOceanNodeDriver(*DIGITALOCEAN_v2_PARAMS)
def setUp(self):
DigitalOceanNodeDriver.connectionCls.conn_classes = \
(None, DigitalOceanMockHttp)
DigitalOceanMockHttp.type = None
self.driver = DigitalOceanNodeDriver(*DIGITAL_OCEAN_PARAMS)
def getAllNodes(provider, driverUno, driverDos, driverTres, driverCuatro):
pass
nodesProvider = ''
node = []
if provider == "Digital Ocean":
pass
accessKey = driverUno
driverDos = driverDos
driverTres = driverTres
driverCuatro = driverCuatro
driver = DigitalOceanNodeDriver(accessKey)
idsNodes = driver.list_nodes()
for idNodes in idsNodes:
#print idsNodes
pass
idNodes.extra = json.dumps(idNodes.extra)
attr = {
'id': idNodes.id,
'name': idNodes.name,
'state': idNodes.state,
'public_ips': idNodes.public_ips,
'private_ips': idNodes.private_ips,
'provider': 'DigitalOcean',
'extra': idNodes.extra
}
node.append(attr)
nodesProvider = json.dumps(attr)
if provider == "EC2":
pass
accessId = driverUno
secretKey = driverDos
region = driverTres
driverCuatro = driverCuatro
# driver = BaseEC2NodeDriver(accessId,secretKey,'eu-west-1')
cls = get_driver(Provider.EC2)
driver = cls(accessId, secretKey, region=region)
nodes = driver.list_nodes()
nodeFunction = []
for node in nodes:
pass
if node.state != 'terminated':
v4 = []
ips = {
'ipaddress': node.public_ips[0],
'gateway': 'NULL',
'mask': 'NULL',
'private_ip': node.private_ips[0]
}
v4.append(ips)
network = {'v4': v4}
extra = {
'launch_time': node.extra['launch_time'],
'instance_type': node.extra['instance_type'],
'network': network
}
attr = {
'id': node.id,
'region': region,
'name': node.name,
'state': node.state,
'public_ip': node.public_ips[0],
'provider': 'Amazon',
'extra': extra
}
nodeFunction.append(json.dumps(attr))
nodesProvider = nodeFunction
if provider == "Azure":
pass
tenantId = driverUno
subscriptionId = driverDos
applicationId = driverTres
keyPaswd = driverCuatro
driver = AzureNodeDriver(tenantId, subscriptionId, applicationId,
keyPaswd)
nodes = driver.list_nodes()
print nodes
if provider == "Linode":
pass
apiKey = driverUno
driverDos = driverDos
driverTres = driverTres
driverCuatro = driverCuatro
driver = LinodeNodeDriver(apiKey)
nodes = driver.list_nodes()
nodesProvider = nodes
return nodesProvider
def deleteNode(provider,driverUno,driverDos,driverTres,driverCuatro,nodeId): pass nodesProvider = '' if provider == "Digital Ocean": pass accessKey = driverUno driverDos = driverDos driverTres = driverTres driverCuatro = driverCuatro nodeId = nodeId driver = DigitalOceanNodeDriver(accessKey) idsNodes = driver.list_nodes() for idNodes in idsNodes: #print idsNodes if idNodes.id == nodeId: pass idNod = idNodes # print idNod nodeDelete = driver.destroy_node(idNod) types = 'terminated' node = checkStatus.checkStatus(driver, idNod.id, types) if node != 0: nodesProvider = nodeDelete if provider == "EC2": pass accessId = driverUno secretKey = driverDos region = driverTres driverCuatro = driverCuatro nodeId = nodeId # driver = BaseEC2NodeDriver(accessId,secretKey,'eu-west-1') cls = get_driver(Provider.EC2) driver = cls(accessId, secretKey, region=region) idsNodes = driver.list_nodes() for idNodes in idsNodes: #print idsNodes if idNodes.id == nodeId: pass idNod = idNodes # print idNod public_ips = driver.ex_describe_all_addresses(only_associated = True) for public_ip in public_ips: if public_ip.instance_id == idNod.id: ip = public_ip nodeDelete = driver.destroy_node(idNod) driver.ex_release_address(ip, domain = ip.domain) types = 'terminated' node = checkStatus.checkStatus(driver, idNod.id, types) if node != 0: nodesProvider = nodeDelete else: nodesProvider = 'null' if provider == "Azure": pass tenantId = driverUno subscriptionId = driverDos applicationId = driverTres keyPaswd = driverCuatro nodeId = nodeId driver = AzureNodeDriver(tenantId,subscriptionId,applicationId,keyPaswd) idsNodes = driver.list_nodes() for idNodes in idsNodes: #print idsNodes if idNodes.id == nodeId: pass idNod = idNodes # print idNod nodeDelete = driver.destroy_node(idNod) types = 'terminated' node = checkStatus.checkStatus(driver, idNod.id, types) if node != 0: nodesProvider = nodeDelete if provider == "Linode": pass apiKey = driverUno driverDos = driverDos driverTres = driverTres driverCuatro = driverCuatro nodeId = nodeId driver = LinodeNodeDriver(apiKey) idsNodes = driver.list_nodes() for idNodes in idsNodes: #print idsNodes if idNodes.id == nodeId: pass idNod = idNodes # print idNod nodeDelete = driver.destroy_node(idNod) types = 'terminated' node = checkStatus.checkStatus(driver, idNod.id, types) if node != 0: nodesProvider = nodeDelete return nodesProvider
def resizeNode(provider, driverUno, driverDos, driverTres, driverCuatro, size,
nodeId):
pass
nodesProvider = ''
if provider == "Digital Ocean":
pass
accessKey = driverUno
driverDos = driverDos
driverTres = driverTres
driverCuatro = driverCuatro
size = size
nodeId = nodeId
driver = DigitalOceanNodeDriver(accessKey)
idsNodes = driver.list_nodes()
sizesNode = driver.list_sizes()
for idNodes in idsNodes:
#print idsNodes
if idNodes.id == nodeId:
pass
idNod = idNodes
# print idNod
for sizeNode in sizesNode:
if sizeNode.name == size:
pass
sizeName = sizeNode
nodesProvider = driver.resize_node(sizeName, idNod)
if provider == "EC2":
pass
accessId = driverUno
secretKey = driverDos
region = driverTres
driverCuatro = driverCuatro
size = size
nodeId = nodeId
# driver = BaseEC2NodeDriver(accessId,secretKey,'eu-west-1')
cls = get_driver(Provider.EC2)
driver = cls(accessId, secretKey, region=region)
idsNodes = driver.list_nodes()
sizesNode = driver.list_sizes()
idNod = ''
sizeName = ''
for idNodes in idsNodes:
#print idsNodes
if idNodes.id == nodeId:
pass
idNod = idNodes
# print idNod
for sizeNode in sizesNode:
if sizeNode.id == size:
pass
sizeName = sizeNode
nodesProvider = driver.ex_change_node_size(idNod, sizeName)
if provider == "Azure":
pass
tenantId = driverUno
subscriptionId = driverDos
applicationId = driverTres
keyPaswd = driverCuatro
size = size
nodeId = nodeId
driver = AzureNodeDriver(tenantId, subscriptionId, applicationId,
keyPaswd)
idsNodes = driver.list_nodes()
sizesNode = driver.list_sizes()
for idNodes in idsNodes:
#print idsNodes
if idNodes.id == nodeId:
pass
idNod = idNodes
# print idNod
for sizeNode in sizesNode:
if sizeNode.name == size:
pass
sizeName = sizeNode
nodesProvider = driver.resize_node(sizeName, idNod)
if provider == "Linode":
pass
apiKey = driverUno
driverDos = driverDos
driverTres = driverTres
driverCuatro = driverCuatro
size = size
nodeId = nodeId
driver = LinodeNodeDriver(apiKey)
idsNodes = driver.list_nodes()
sizesNode = driver.list_sizes()
for idNodes in idsNodes:
#print idsNodes
if idNodes.id == nodeId:
pass
idNod = idNodes
# print idNod
for sizeNode in sizesNode:
if sizeNode.name == size:
pass
sizeName = sizeNode
nodesProvider = driver.ex_resize_node(sizeName, idNod)
return nodesProvider
def shutdownNode(provider, driverUno, driverDos, driverTres, driverCuatro,
nodeId):
pass
nodesProvider = ''
if provider == "Digital Ocean":
pass
accessKey = driverUno
driverDos = driverDos
driverTres = driverTres
driverCuatro = driverCuatro
nodeId = nodeId
driver = DigitalOceanNodeDriver(accessKey)
idsNodes = driver.list_nodes()
for idNodes in idsNodes:
#print idsNodes
if idNodes.id == nodeId:
pass
idNod = idNodes
# print idNod
nodeStopped = driver.ex_shutdown_node(idNod)
types = 'stopped'
node = checkStatus.checkStatus(driver, idNod.id, types)
if node != 0:
nodesProvider = nodeStopped
if provider == "EC2":
pass
accessId = driverUno
secretKey = driverDos
region = driverTres
driverCuatro = driverCuatro
nodeId = nodeId
# driver = BaseEC2NodeDriver(accessId,secretKey,'eu-west-1')
cls = get_driver(Provider.EC2)
driver = cls(accessId, secretKey, region=region)
idsNodes = driver.list_nodes()
for idNodes in idsNodes:
#print idsNodes
if idNodes.id == nodeId:
pass
idNod = idNodes
# print idNod
nodeStopped = driver.ex_stop_node(idNod)
types = 'stopped'
node = checkStatus.checkStatus(driver, idNod.id, types)
if node != 0:
nodesProvider = nodeStopped
if provider == "Azure":
pass
tenantId = driverUno
subscriptionId = driverDos
applicationId = driverTres
keyPaswd = driverCuatro
nodeId = nodeId
driver = AzureNodeDriver(tenantId, subscriptionId, applicationId,
keyPaswd)
idsNodes = driver.list_nodes()
for idNodes in idsNodes:
#print idsNodes
if idNodes.id == nodeId:
pass
idNod = idNodes
# print idNod
nodeStopped = driver.ex_stop_node(idNod)
types = 'stopped'
node = checkStatus.checkStatus(driver, idNod.id, types)
if node != 0:
nodesProvider = nodeStopped
if provider == "Linode":
pass
apiKey = driverUno
driverDos = driverDos
driverTres = driverTres
driverCuatro = driverCuatro
nodeId = nodeId
driver = LinodeNodeDriver(apiKey)
idsNodes = driver.list_nodes()
for idNodes in idsNodes:
#print idsNodes
if idNodes.id == nodeId:
pass
idNod = idNodes
# print idNod
nodeStopped = driver.ex_stop_node(idNod)
types = 'stopped'
node = checkStatus.checkStatus(driver, idNod.id, types)
if node != 0:
nodesProvider = nodeStopped
return nodesProvider
class DigitalOcean_v2_Tests(LibcloudTestCase):
def setUp(self):
DigitalOceanNodeDriver.connectionCls.conn_class = DigitalOceanMockHttp
DigitalOceanMockHttp.type = None
self.driver = DigitalOceanNodeDriver(*DIGITALOCEAN_v2_PARAMS)
def test_v1_Error(self):
self.assertRaises(
DigitalOcean_v1_Error,
DigitalOceanNodeDriver,
*DIGITALOCEAN_v1_PARAMS,
api_version="v1",
)
def test_v2_uses_v1_key(self):
self.assertRaises(
InvalidCredsError,
DigitalOceanNodeDriver,
*DIGITALOCEAN_v1_PARAMS,
api_version="v2",
)
def test_authentication(self):
DigitalOceanMockHttp.type = "UNAUTHORIZED"
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[1]
self.assertTrue(size.id is not None)
self.assertEqual(size.name, "1gb")
self.assertEqual(size.ram, 1024)
def test_list_sizes_filter_by_location_success(self):
location = self.driver.list_locations()[1]
sizes = self.driver.list_sizes(location=location)
self.assertTrue(len(sizes) >= 1)
size = sizes[0]
self.assertTrue(size.id is not None)
self.assertEqual(size.name, "512mb")
self.assertTrue(location.id in size.extra["regions"])
location = self.driver.list_locations()[1]
location.id = "doesntexist"
sizes = self.driver.list_sizes(location=location)
self.assertEqual(len(sizes), 0)
def test_list_locations_success(self):
locations = self.driver.list_locations()
self.assertTrue(len(locations) == 2)
location = locations[0]
self.assertEqual(location.id, "nyc1")
self.assertEqual(location.name, "New York 1")
locations = self.driver.list_locations(ex_available=True)
self.assertTrue(len(locations) == 2)
locations = self.driver.list_locations(ex_available=False)
self.assertTrue(len(locations) == 3)
def test_list_nodes_success(self):
nodes = self.driver.list_nodes()
self.assertEqual(len(nodes), 1)
self.assertEqual(nodes[0].name, "ubuntu-s-1vcpu-1gb-sfo3-01")
self.assertEqual(nodes[0].public_ips, ["128.199.13.158"])
self.assertEqual(nodes[0].extra["image"]["id"], 69463186)
self.assertEqual(nodes[0].extra["size_slug"], "s-1vcpu-1gb")
self.assertEqual(len(nodes[0].extra["tags"]), 0)
def test_list_nodes_fills_created_datetime(self):
nodes = self.driver.list_nodes()
self.assertEqual(nodes[0].created_at,
datetime(2020, 10, 15, 13, 58, 22, tzinfo=UTC))
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." +
r" \(code: (404|HTTPStatus.NOT_FOUND)\)")
assertRaisesRegex(
self,
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]
DigitalOceanMockHttp.type = "REBOOT"
result = self.driver.reboot_node(node)
self.assertTrue(result)
def test_create_image_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = "SNAPSHOT"
result = self.driver.create_image(node, "My snapshot")
self.assertTrue(result)
def test_get_image_success(self):
image = self.driver.get_image(12345)
self.assertEqual(image.name, "My snapshot")
self.assertEqual(image.id, "12345")
self.assertEqual(image.extra["distribution"], "Ubuntu")
def test_delete_image_success(self):
image = self.driver.get_image(12345)
DigitalOceanMockHttp.type = "DESTROY"
result = self.driver.delete_image(image)
self.assertTrue(result)
def test_ex_power_on_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = "POWERON"
result = self.driver.ex_power_on_node(node)
self.assertTrue(result)
def test_ex_shutdown_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = "SHUTDOWN"
result = self.driver.ex_shutdown_node(node)
self.assertTrue(result)
def test_ex_hard_reboot_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = "POWERCYCLE"
result = self.driver.ex_hard_reboot(node)
self.assertTrue(result)
def test_ex_rebuild_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = "REBUILD"
result = self.driver.ex_rebuild_node(node)
self.assertTrue(result)
def test_ex_resize_node_success(self):
node = self.driver.list_nodes()[0]
size = self.driver.list_sizes()[0]
DigitalOceanMockHttp.type = "RESIZE"
result = self.driver.ex_resize_node(node, size)
self.assertTrue(result)
def test_destroy_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = "DESTROY"
result = self.driver.destroy_node(node)
self.assertTrue(result)
def test_ex_change_kernel_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = "KERNELCHANGE"
result = self.driver.ex_change_kernel(node, 7515)
self.assertTrue(result)
def test_ex_enable_ipv6_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = "ENABLEIPV6"
result = self.driver.ex_enable_ipv6(node)
self.assertTrue(result)
def test_ex_rename_node_success(self):
node = self.driver.list_nodes()[0]
DigitalOceanMockHttp.type = "RENAME"
result = self.driver.ex_rename_node(node, "fedora helios")
self.assertTrue(result)
def test_list_key_pairs(self):
keys = self.driver.list_key_pairs()
self.assertEqual(len(keys), 1)
self.assertEqual(keys[0].extra["id"], 7717)
self.assertEqual(keys[0].name, "test1")
self.assertEqual(keys[0].public_key,
"ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAQQDGk5 example")
def test_create_key_pair(self):
DigitalOceanMockHttp.type = "CREATE"
key = self.driver.create_key_pair(
name="test1",
public_key="ssh-rsa AAAAB3NzaC1yc2EAAAADAQsxRiUKn example")
self.assertEqual(key.name, "test1")
self.assertEqual(key.fingerprint,
"f5:d1:78:ed:28:72:5f:e1:ac:94:fd:1f:e0:a3:48:6d")
def test_delete_key_pair(self):
key = self.driver.list_key_pairs()[0]
result = self.driver.delete_key_pair(key)
self.assertTrue(result)
def test__paginated_request_single_page(self):
nodes = self.driver._paginated_request("/v2/droplets", "droplets")
self.assertEqual(nodes[0]["name"], "ubuntu-s-1vcpu-1gb-sfo3-01")
self.assertEqual(nodes[0]["image"]["id"], 69463186)
self.assertEqual(nodes[0]["size_slug"], "s-1vcpu-1gb")
def test__paginated_request_two_pages(self):
DigitalOceanMockHttp.type = "PAGE_ONE"
nodes = self.driver._paginated_request("/v2/droplets", "droplets")
self.assertEqual(len(nodes), 2)
def test_list_volumes(self):
volumes = self.driver.list_volumes()
self.assertEqual(len(volumes), 1)
volume = volumes[0]
self.assertEqual(volume.id, "62766883-2c28-11e6-b8e6-000f53306ae1")
self.assertEqual(volume.name, "example")
self.assertEqual(volume.size, 4)
self.assertEqual(volume.driver, self.driver)
def test_list_volumes_empty(self):
DigitalOceanMockHttp.type = "EMPTY"
volumes = self.driver.list_volumes()
self.assertEqual(len(volumes), 0)
def test_create_volume(self):
nyc1 = [r for r in self.driver.list_locations() if r.id == "nyc1"][0]
DigitalOceanMockHttp.type = "CREATE"
volume = self.driver.create_volume(4, "example", nyc1)
self.assertEqual(volume.id, "62766883-2c28-11e6-b8e6-000f53306ae1")
self.assertEqual(volume.name, "example")
self.assertEqual(volume.size, 4)
self.assertEqual(volume.driver, self.driver)
def test_attach_volume(self):
node = self.driver.list_nodes()[0]
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = "ATTACH"
resp = self.driver.attach_volume(node, volume)
self.assertTrue(resp)
def test_detach_volume(self):
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = "DETACH"
resp = self.driver.detach_volume(volume)
self.assertTrue(resp)
def test_destroy_volume(self):
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = "DESTROY"
resp = self.driver.destroy_volume(volume)
self.assertTrue(resp)
def test_list_volume_snapshots(self):
volume = self.driver.list_volumes()[0]
snapshots = self.driver.list_volume_snapshots(volume)
self.assertEqual(len(snapshots), 3)
snapshot1, snapshot2, snapshot3 = snapshots
self.assertEqual(snapshot1.id, "c0def940-9324-11e6-9a56-000f533176b1")
self.assertEqual(snapshot2.id, "c2036724-9343-11e6-aef4-000f53315a41")
self.assertEqual(snapshot3.id, "d347e033-9343-11e6-9a56-000f533176b1")
def test_create_volume_snapshot(self):
volume = self.driver.list_volumes()[0]
DigitalOceanMockHttp.type = "CREATE"
snapshot = self.driver.create_volume_snapshot(volume, "test-snapshot")
self.assertEqual(snapshot.id, "c0def940-9324-11e6-9a56-000f533176b1")
self.assertEqual(snapshot.name, "test-snapshot")
self.assertEqual(volume.driver, self.driver)
def test_delete_volume_snapshot(self):
volume = self.driver.list_volumes()[0]
snapshot = self.driver.list_volume_snapshots(volume)[0]
DigitalOceanMockHttp.type = "DELETE"
result = self.driver.delete_volume_snapshot(snapshot)
self.assertTrue(result)
def test_ex_get_node_details(self):
node = self.driver.ex_get_node_details("3164444")
self.assertEqual(node.name, "example.com")
self.assertEqual(node.public_ips, ["36.123.0.123"])
self.assertEqual(node.extra["image"]["id"], 12089443)
self.assertEqual(node.extra["size_slug"], "8gb")
self.assertEqual(len(node.extra["tags"]), 2)
def test_ex_create_floating_ip(self):
nyc1 = [r for r in self.driver.list_locations() if r.id == "nyc1"][0]
floating_ip = self.driver.ex_create_floating_ip(nyc1)
# Note that this is the ID. There is no real ID for a floating IP at
# DigitalOcean, but the IP is unique so we can use that instead.
self.assertEqual(floating_ip.id, "167.138.123.111")
self.assertEqual(floating_ip.ip_address, "167.138.123.111")
self.assertEqual(floating_ip.extra["region"]["slug"], "nyc1")
# The newly created floating IP reserved to a region is not
# associated with any droplet. See the DigitalOcean API docs
# how to create a floating IP that is associated with an instance
# from the start. This API call creates an unattached IP.
self.assertIsNone(floating_ip.node_id)
def test_ex_delete_floating_ip(self):
nyc1 = [r for r in self.driver.list_locations() if r.id == "nyc1"][0]
floating_ip = self.driver.ex_create_floating_ip(nyc1)
ret = self.driver.ex_delete_floating_ip(floating_ip)
# The API returns 204 NO CONTENT if all is well.
self.assertTrue(ret)
def test_floating_ip_can_be_deleted_by_calling_delete_on_floating_ip_object(
self, ): # noqa: E501
nyc1 = [r for r in self.driver.list_locations() if r.id == "nyc1"][0]
floating_ip = self.driver.ex_create_floating_ip(nyc1)
ret = floating_ip.delete()
self.assertTrue(ret)
def test_list_floating_ips(self):
floating_ips = self.driver.ex_list_floating_ips()
self.assertEqual(len(floating_ips), 2, "Wrong floating IPs count")
floating_ip = floating_ips[0]
self.assertEqual(floating_ip.id, "133.166.122.204")
self.assertEqual(floating_ip.ip_address, "133.166.122.204")
self.assertEqual(floating_ip.extra["region"]["slug"], "ams3")
self.assertEqual(84155775, floating_ip.node_id)
def test_get_floating_ip(self):
floating_ip = self.driver.ex_get_floating_ip("133.166.122.204")
self.assertEqual(floating_ip.id, "133.166.122.204")
self.assertEqual(floating_ip.ip_address, "133.166.122.204")
self.assertEqual(floating_ip.extra["region"]["slug"], "ams3")
self.assertEqual(84155775, floating_ip.node_id)
def test_ex_attach_floating_ip_to_node(self):
node = self.driver.list_nodes()[0]
floating_ip = self.driver.ex_get_floating_ip("133.166.122.204")
ret = self.driver.ex_attach_floating_ip_to_node(node, floating_ip)
self.assertTrue(ret)
def test_ex_detach_floating_ip_from_node(self):
node = self.driver.list_nodes()[0]
floating_ip = self.driver.ex_get_floating_ip("154.138.103.175")
ret = self.driver.ex_detach_floating_ip_from_node(node, floating_ip)
self.assertTrue(ret)
def setUp(self):
DigitalOceanNodeDriver.connectionCls.conn_class = \
DigitalOceanMockHttp
DigitalOceanMockHttp.type = None
self.driver = DigitalOceanNodeDriver(*DIGITALOCEAN_v2_PARAMS)
import getRegionName
from libcloud.compute.types import Provider
from libcloud.compute.providers import get_driver
from libcloud.compute.drivers.digitalocean import DigitalOceanNodeDriver
from libcloud.compute.drivers.azure_arm import AzureNodeDriver
from libcloud.compute.drivers.ec2 import BaseEC2NodeDriver, EC2NodeDriver
from libcloud.compute.drivers.linode import LinodeNodeDriver
provider = sys.argv[1]
if provider == "Digital Ocean":
pass
accessKey = sys.argv[2]
sshKey = sys.argv[3]
driverTres = sys.argv[4]
driverCuatro = sys.argv[5]
driver = DigitalOceanNodeDriver(accessKey)
listLocations = driver.list_locations()
nodelocations = []
for listLocation in listLocations:
data = []
flag = False
matches = re.search(r'[a-zA-Z]{0,}', listLocation.name)
match = re.search(r'[a-zA-Z]{0,}[ ][a-zA-Z]{0,}', listLocation.name)
if match:
region = match.group()
else:
region = matches.group()
