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_get_creation_time(self):
node = self.driver.list_nodes()[0]
creation_time = self.driver.ex_get_creation_time(node)
self.assertEqual(creation_time, "2014-11-14T16:29:21Z")
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)
