Mini Shell
# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements. See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from __future__ import with_statement
import sys
import json
from libcloud.utils.py3 import httplib
from libcloud.compute.drivers.kamatera import KamateraNodeDriver
from libcloud.compute.types import NodeState, Provider
from libcloud.compute.base import NodeImage, NodeLocation, NodeAuthSSHKey
from libcloud.common.exceptions import BaseHTTPError
from libcloud.compute import providers
from libcloud.test import LibcloudTestCase, unittest, MockHttp
from libcloud.test.file_fixtures import ComputeFileFixtures
from libcloud.test.secrets import KAMATERA_PARAMS
class KamateraAuthenticationTests(LibcloudTestCase):
def setUp(self):
KamateraNodeDriver.connectionCls.conn_class = KamateraMockHttp
self.driver = KamateraNodeDriver("nosuchuser", "nopwd")
def test_authentication_fails(self):
with self.assertRaises(BaseHTTPError):
self.driver.list_locations()
class KamateraNodeDriverTests(LibcloudTestCase):
def setUp(self):
KamateraTestDriver.connectionCls.conn_class = KamateraMockHttp
self.driver = KamateraTestDriver(*KAMATERA_PARAMS)
self.eu_node_location = NodeLocation(
id='EU',
name='Amsterdam',
country='The Netherlands',
driver=self.driver)
self.il_node_location = NodeLocation(
id='IL',
name='Rosh Haayin',
country='Israel',
driver=self.driver)
self.centos_8_EU_node_image = NodeImage(
id='EU:6000C2987c9641fd2619a149ba2ca01a',
name='CentOS 8.0 64-bit - Minimal Configuration',
driver=self.driver,
extra={'datacenter': 'EU',
'os': 'CentOS',
'code': '8.0 64bit_minimal',
'osDiskSizeGB': 5,
"ramMBMin": {"A": 256, "B": 256, "T": 256, "D": 256}})
self.small_node_size = self.driver.ex_get_size(
ramMB=4096,
diskSizeGB=30,
cpuType='B',
cpuCores=2,
monthlyTrafficPackage='t5000',
id='small',
name='small')
def test_creating_driver(self):
cls = providers.get_driver(Provider.KAMATERA)
self.assertIs(cls, KamateraNodeDriver)
def test_features(self):
features = self.driver.features['create_node']
self.assertIn('password', features)
self.assertIn('generates_password', features)
self.assertIn('ssh_key', features)
def test_list_locations(self):
locations = self.driver.list_locations()
self.assertTrue(len(locations) == 13)
self.assert_object(self.il_node_location, objects=locations)
def test_list_sizes(self):
sizes = self.driver.list_sizes(self.eu_node_location)
self.assertTrue(len(sizes) >= 1)
self.assert_object(self.small_node_size, objects=sizes)
def test_list_images(self):
images = self.driver.list_images(self.eu_node_location)
self.assertTrue(len(images) > 10)
self.assert_object(self.centos_8_EU_node_image, objects=images)
def test_ex_list_capabilities(self):
capabilities = self.driver.ex_list_capabilities(self.eu_node_location)
self.assertEqual(
set(['cpuTypes', 'defaultMonthlyTrafficPackage', 'diskSizeGB',
'monthlyTrafficPackage']), set(capabilities.keys()))
self.assertTrue(len(capabilities['cpuTypes']), 4)
self.assertEqual(
set(['id', 'description', 'name', 'ramMB', 'cpuCores']),
set(capabilities['cpuTypes'][0]))
def test_create_node(self):
node = self.driver.create_node(
name='test_server', size=self.small_node_size,
image=self.centos_8_EU_node_image, location=self.eu_node_location)
self.assertTrue(len(node.id) > 8)
self.assertEqual(node.name, 'my-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)
self.assertTrue(len(node.extra['generated_password']) > 0)
def test_create_node_with_ssh_keys(self):
node = self.driver.create_node(
name='test_server_pubkey', size=self.small_node_size,
image=self.centos_8_EU_node_image, location=self.eu_node_location,
auth=NodeAuthSSHKey('publickey'))
self.assertTrue(len(node.id) > 8)
self.assertEqual(node.name, 'my-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)
self.assertFalse('generated_password' in node.extra)
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.assertEqual(node.driver, self.driver)
def test_list_nodes_full(self):
nodes = self.driver.list_nodes(ex_full_details=True)
self.assertTrue(len(nodes) >= 1)
node = nodes[0]
self.assertEqual(node.name, 'my-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 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 (%s, %s)" % (
expected_object, objects[:2]))
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
class KamateraTestDriver(KamateraNodeDriver):
def ex_wait_command(self, *args, **kwargs):
kwargs['poll_interval_seconds'] = 0
return KamateraNodeDriver.ex_wait_command(self, *args, **kwargs)
class KamateraMockHttp(MockHttp):
fixtures = ComputeFileFixtures('kamatera')
def _service_server(self, method, url, body, headers):
client_id, secret = headers['AuthClientId'], headers['AuthSecret']
if client_id == 'nosuchuser' and secret == 'nopwd':
body = self.fixtures.load('failed_auth.json')
status = httplib.UNAUTHORIZED
else:
if url == '/service/server' and json.loads(body).get('ssh-key'):
body = self.fixtures.load('create_server_sshkey.json')
else:
body = self.fixtures.load({
'/service/server?datacenter=1': 'datacenters.json',
'/service/server?sizes=1&datacenter=EU':
'sizes_datacenter_EU.json',
'/service/server?images=1&datacenter=EU':
'images_datacenter_EU.json',
'/service/server?capabilities=1&datacenter=EU':
'capabilities_datacenter_EU.json',
'/service/server': 'create_server.json'
}[url])
status = httplib.OK
return status, body, {}, httplib.responses[status]
def _service_queue(self, method, url, body, headers):
if not hasattr(self, '_service_queue_call_count'):
self._service_queue_call_count = 0
self._service_queue_call_count += 1
body = self.fixtures.load({
'/service/queue?id=12345':
'queue_12345-%s.json' % self._service_queue_call_count
}[url])
status = httplib.OK
return status, body, {}, httplib.responses[status]
def _service_server_info(self, method, url, body, headers):
body = self.fixtures.load({
"/service/server/info": "server_info.json"
}[url])
status = httplib.OK
return status, body, {}, httplib.responses[status]
def _service_servers(self, method, url, body, headers):
body = self.fixtures.load({
'/service/servers': 'servers.json'
}[url])
status = httplib.OK
return status, body, {}, httplib.responses[status]
def _service_server_reboot(self, method, url, body, headers):
body = self.fixtures.load({
'/service/server/reboot': 'server_operation.json'
}[url])
status = httplib.OK
return status, body, {}, httplib.responses[status]
if __name__ == '__main__':
sys.exit(unittest.main())
Zerion Mini Shell 1.0