class UpcloudAuthenticationTests(LibcloudTestCase):
def setUp(self):
UpcloudDriver.connectionCls.conn_class = UpcloudMockHttp
self.driver = UpcloudDriver("nosuchuser", "nopwd")
def test_authentication_fails(self):
with self.assertRaises(InvalidCredsError):
self.driver.list_locations()
def setUp(self):
UpcloudDriver.connectionCls.conn_class = UpcloudMockHttp
# UpcloudDriver.connectionCls.responseCls = UpcloudPersistResponse
self.driver = UpcloudDriver(*UPCLOUD_PARAMS)
class UpcloudDriverTests(LibcloudTestCase):
def setUp(self):
UpcloudDriver.connectionCls.conn_class = UpcloudMockHttp
# UpcloudDriver.connectionCls.responseCls = UpcloudPersistResponse
self.driver = UpcloudDriver(*UPCLOUD_PARAMS)
def test_creating_driver(self):
cls = providers.get_driver(Provider.UPCLOUD)
self.assertIs(cls, UpcloudDriver)
def test_features(self):
features = self.driver.features['create_node']
self.assertIn('ssh_key', features)
self.assertIn('generates_password', features)
def test_list_locations(self):
locations = self.driver.list_locations()
self.assertTrue(len(locations) >= 1)
expected_node_location = NodeLocation(id='fi-hel1',
name='Helsinki #1',
country='FI',
driver=self.driver)
self.assert_object(expected_node_location, objects=locations)
def test_list_sizes(self):
location = NodeLocation(id='fi-hel1',
name='Helsinki #1',
country='FI',
driver=self.driver)
sizes = self.driver.list_sizes(location)
self.assertTrue(len(sizes) >= 1)
expected_node_size = NodeSize(id='1xCPU-1GB',
name='1xCPU-1GB',
ram=1024,
disk=30,
bandwidth=2048,
price=2.232,
driver=self.driver,
extra={
'core_number': 1,
'storage_tier': 'maxiops'
})
self.assert_object(expected_node_size, objects=sizes)
def test_list_images(self):
images = self.driver.list_images()
self.assertTrue(len(images) >= 1)
expected_node_image = NodeImage(
id='01000000-0000-4000-8000-000010010101',
name='Windows Server 2003 R2 Standard (CD 1)',
driver=self.driver,
extra={
'access': 'public',
'license': 0,
'size': 1,
'state': 'online',
'type': 'cdrom'
})
self.assert_object(expected_node_image, objects=images)
def test_create_node_from_template(self):
image = NodeImage(id='01000000-0000-4000-8000-000030060200',
name='Ubuntu Server 16.04 LTS (Xenial Xerus)',
extra={'type': 'template'},
driver=self.driver)
location = NodeLocation(id='fi-hel1',
name='Helsinki #1',
country='FI',
driver=self.driver)
size = NodeSize(id='1xCPU-1GB',
name='1xCPU-1GB',
ram=1024,
disk=30,
bandwidth=2048,
extra={'storage_tier': 'maxiops'},
price=None,
driver=self.driver)
node = self.driver.create_node(name='test_server',
size=size,
image=image,
location=location,
ex_hostname='myhost.somewhere')
self.assertTrue(
re.match(
'^[a-z0-9]{8}-[a-z0-9]{4}-[a-z0-9]{4}-[a-z0-9]{4}-[a-z0-9]{12}$',
node.id))
self.assertEquals(node.name, 'test_server')
self.assertEquals(node.state, NodeState.STARTING)
self.assertTrue(len(node.public_ips) > 0)
self.assertTrue(len(node.private_ips) > 0)
self.assertEquals(node.driver, self.driver)
self.assertTrue(len(node.extra['password']) > 0)
self.assertTrue(len(node.extra['vnc_password']) > 0)
def test_create_node_with_ssh_keys(self):
image = NodeImage(id='01000000-0000-4000-8000-000030060200',
name='Ubuntu Server 16.04 LTS (Xenial Xerus)',
extra={'type': 'template'},
driver=self.driver)
location = NodeLocation(id='fi-hel1',
name='Helsinki #1',
country='FI',
driver=self.driver)
size = NodeSize(id='1xCPU-1GB',
name='1xCPU-1GB',
ram=1024,
disk=30,
bandwidth=2048,
extra={'storage_tier': 'maxiops'},
price=None,
driver=self.driver)
auth = NodeAuthSSHKey('publikey')
node = self.driver.create_node(name='test_server',
size=size,
image=image,
location=location,
auth=auth)
self.assertTrue(
re.match(
'^[a-z0-9]{8}-[a-z0-9]{4}-[a-z0-9]{4}-[a-z0-9]{4}-[a-z0-9]{12}$',
node.id))
self.assertEquals(node.name, 'test_server')
self.assertEquals(node.state, NodeState.STARTING)
self.assertTrue(len(node.public_ips) > 0)
self.assertTrue(len(node.private_ips) > 0)
self.assertEquals(node.driver, self.driver)
def test_list_nodes(self):
nodes = self.driver.list_nodes()
self.assertTrue(len(nodes) >= 1)
node = nodes[0]
self.assertEquals(node.name, 'test_server')
self.assertEquals(node.state, NodeState.RUNNING)
self.assertTrue(len(node.public_ips) > 0)
self.assertTrue(len(node.private_ips) > 0)
self.assertEquals(node.driver, self.driver)
def test_reboot_node(self):
nodes = self.driver.list_nodes()
success = self.driver.reboot_node(nodes[0])
self.assertTrue(success)
def test_destroy_node(self):
if UpcloudDriver.connectionCls.conn_class == UpcloudMockHttp:
nodes = [
Node(id='00893c98_5d5a_4363_b177_88df518a2b60',
name='',
state='',
public_ips=[],
private_ips=[],
driver=self.driver)
]
else:
nodes = self.driver.list_nodes()
success = self.driver.destroy_node(nodes[0])
self.assertTrue(success)
def assert_object(self, expected_object, objects):
same_data = any(
[self.objects_equals(expected_object, obj) for obj in objects])
self.assertTrue(same_data, "Objects does not match")
def objects_equals(self, expected_obj, obj):
for name in vars(expected_obj):
expected_data = getattr(expected_obj, name)
actual_data = getattr(obj, name)
same_data = self.data_equals(expected_data, actual_data)
if not same_data:
break
return same_data
def data_equals(self, expected_data, actual_data):
if isinstance(expected_data, dict):
return self.dicts_equals(expected_data, actual_data)
else:
return expected_data == actual_data
def dicts_equals(self, d1, d2):
dict_keys_same = set(d1.keys()) == set(d2.keys())
if not dict_keys_same:
return False
for key in d1.keys():
if d1[key] != d2[key]:
return False
return True
def setUp(self):
UpcloudDriver.connectionCls.conn_class = UpcloudMockHttp
self.driver = UpcloudDriver("nosuchuser", "nopwd")
class UpcloudDriverTests(LibcloudTestCase):
def setUp(self):
UpcloudDriver.connectionCls.conn_class = UpcloudMockHttp
# UpcloudDriver.connectionCls.responseCls = UpcloudPersistResponse
self.driver = UpcloudDriver(*UPCLOUD_PARAMS)
def test_creating_driver(self):
cls = providers.get_driver(Provider.UPCLOUD)
self.assertIs(cls, UpcloudDriver)
def test_features(self):
features = self.driver.features["create_node"]
self.assertIn("ssh_key", features)
self.assertIn("generates_password", features)
def test_list_locations(self):
locations = self.driver.list_locations()
self.assertTrue(len(locations) >= 1)
expected_node_location = NodeLocation(id="fi-hel1",
name="Helsinki #1",
country="FI",
driver=self.driver)
self.assert_object(expected_node_location, objects=locations)
def test_list_sizes(self):
location = NodeLocation(id="fi-hel1",
name="Helsinki #1",
country="FI",
driver=self.driver)
sizes = self.driver.list_sizes(location)
self.assertTrue(len(sizes) >= 1)
expected_node_size = NodeSize(
id="1xCPU-1GB",
name="1xCPU-1GB",
ram=1024,
disk=30,
bandwidth=2048,
price=2.232,
driver=self.driver,
extra={
"core_number": 1,
"storage_tier": "maxiops"
},
)
self.assert_object(expected_node_size, objects=sizes)
def test_list_images(self):
images = self.driver.list_images()
self.assertTrue(len(images) >= 1)
expected_node_image = NodeImage(
id="01000000-0000-4000-8000-000010010101",
name="Windows Server 2003 R2 Standard (CD 1)",
driver=self.driver,
extra={
"access": "public",
"license": 0,
"size": 1,
"state": "online",
"type": "cdrom",
},
)
self.assert_object(expected_node_image, objects=images)
def test_create_node_from_template(self):
image = NodeImage(
id="01000000-0000-4000-8000-000030060200",
name="Ubuntu Server 16.04 LTS (Xenial Xerus)",
extra={"type": "template"},
driver=self.driver,
)
location = NodeLocation(id="fi-hel1",
name="Helsinki #1",
country="FI",
driver=self.driver)
size = NodeSize(
id="1xCPU-1GB",
name="1xCPU-1GB",
ram=1024,
disk=30,
bandwidth=2048,
extra={"storage_tier": "maxiops"},
price=None,
driver=self.driver,
)
node = self.driver.create_node(
name="test_server",
size=size,
image=image,
location=location,
ex_hostname="myhost.somewhere",
)
self.assertTrue(
re.match(
"^[a-z0-9]{8}-[a-z0-9]{4}-[a-z0-9]{4}-[a-z0-9]{4}-[a-z0-9]{12}$",
node.id,
))
self.assertEqual(node.name, "test_server")
self.assertEqual(node.state, NodeState.STARTING)
self.assertTrue(len(node.public_ips) > 0)
self.assertTrue(len(node.private_ips) > 0)
self.assertEqual(node.driver, self.driver)
self.assertTrue(len(node.extra["password"]) > 0)
self.assertTrue(len(node.extra["vnc_password"]) > 0)
def test_create_node_with_ssh_keys(self):
image = NodeImage(
id="01000000-0000-4000-8000-000030060200",
name="Ubuntu Server 16.04 LTS (Xenial Xerus)",
extra={"type": "template"},
driver=self.driver,
)
location = NodeLocation(id="fi-hel1",
name="Helsinki #1",
country="FI",
driver=self.driver)
size = NodeSize(
id="1xCPU-1GB",
name="1xCPU-1GB",
ram=1024,
disk=30,
bandwidth=2048,
extra={"storage_tier": "maxiops"},
price=None,
driver=self.driver,
)
auth = NodeAuthSSHKey("publikey")
node = self.driver.create_node(name="test_server",
size=size,
image=image,
location=location,
auth=auth)
self.assertTrue(
re.match(
"^[a-z0-9]{8}-[a-z0-9]{4}-[a-z0-9]{4}-[a-z0-9]{4}-[a-z0-9]{12}$",
node.id,
))
self.assertEqual(node.name, "test_server")
self.assertEqual(node.state, NodeState.STARTING)
self.assertTrue(len(node.public_ips) > 0)
self.assertTrue(len(node.private_ips) > 0)
self.assertEqual(node.driver, self.driver)
def test_list_nodes(self):
nodes = self.driver.list_nodes()
self.assertTrue(len(nodes) >= 1)
node = nodes[0]
self.assertEqual(node.name, "test_server")
self.assertEqual(node.state, NodeState.RUNNING)
self.assertTrue(len(node.public_ips) > 0)
self.assertTrue(len(node.private_ips) > 0)
self.assertEqual(node.driver, self.driver)
def test_reboot_node(self):
nodes = self.driver.list_nodes()
success = self.driver.reboot_node(nodes[0])
self.assertTrue(success)
def test_destroy_node(self):
if UpcloudDriver.connectionCls.conn_class == UpcloudMockHttp:
nodes = [
Node(
id="00893c98_5d5a_4363_b177_88df518a2b60",
name="",
state="",
public_ips=[],
private_ips=[],
driver=self.driver,
)
]
else:
nodes = self.driver.list_nodes()
success = self.driver.destroy_node(nodes[0])
self.assertTrue(success)
def assert_object(self, expected_object, objects):
same_data = any(
[self.objects_equals(expected_object, obj) for obj in objects])
self.assertTrue(same_data, "Objects does not match")
def objects_equals(self, expected_obj, obj):
for name in vars(expected_obj):
expected_data = getattr(expected_obj, name)
actual_data = getattr(obj, name)
same_data = self.data_equals(expected_data, actual_data)
if not same_data:
break
return same_data
def data_equals(self, expected_data, actual_data):
if isinstance(expected_data, dict):
return self.dicts_equals(expected_data, actual_data)
else:
return expected_data == actual_data
def dicts_equals(self, d1, d2):
dict_keys_same = set(d1.keys()) == set(d2.keys())
if not dict_keys_same:
return False
for key in d1.keys():
if d1[key] != d2[key]:
return False
return True