class FakeFacility:
plumbery = FakePlumbery()
settings = fakeParameters
blueprints = fakeBlueprints
DimensionDataNodeDriver.connectionCls.conn_classes = (
None, DimensionDataMockHttp)
DimensionDataBackupDriver.connectionCls.conn_classes = (
None, DimensionDataMockHttp)
DimensionDataMockHttp.type = None
region = DimensionDataNodeDriver(*DIMENSIONDATA_PARAMS)
backup = DimensionDataBackupDriver(*DIMENSIONDATA_PARAMS)
location = FakeLocation()
_cache_network_domains = []
_cache_vlans = []
def get_location_id(self):
return 'EU6'
def get_setting(self, label):
if label in self.settings:
return self.settings[label]
return None
def main():
module = AnsibleModule(argument_spec=dict(
region=dict(default='na', choices=dd_regions),
state=dict(default='present', choices=['present', 'absent']),
node_ids=dict(required=True,
type='list',
aliases=['server_id', 'server_ids', 'node_id']),
scsi_id=dict(required=True, type='int'),
speed=dict(default='STANDARD',
choices=['STANDARD', 'ECONOMY', 'HIGHPERFORMANCE']),
size=dict(type='int'),
verify_ssl_cert=dict(required=False, default=True, type='bool'),
wait=dict(required=False, default=False, type='bool'),
wait_timeout=dict(required=False, default=300, type='int'),
wait_poll_interval=dict(required=False, default=2, type='int')))
if not HAS_LIBCLOUD:
module.fail_json(msg='libcloud is required for this module.')
# set short vars for readability
credentials = get_credentials()
if credentials is False:
module.fail_json("User credentials not found")
user_id = credentials['user_id']
key = credentials['key']
region = 'dd-%s' % module.params['region']
verify_ssl_cert = module.params['verify_ssl_cert']
# Instantiate driver
libcloud.security.VERIFY_SSL_CERT = verify_ssl_cert
client = DimensionDataNodeDriver(user_id, key, region=region)
handle_disk(module, client)
def test_as_library(self):
engine = PlumberyEngine(myEuropeanPlan, myAmericanBinding)
DimensionDataNodeDriver.connectionCls.conn_classes = (
None, DimensionDataMockHttp)
DimensionDataMockHttp.type = None
self.region = DimensionDataNodeDriver(*DIMENSIONDATA_PARAMS)
engine.set_shared_secret('fake_secret')
engine.set_user_name('fake_name')
engine.set_user_password('fake_password')
facilities = engine.list_facility('NA9')
self.assertEqual(len(facilities), 1)
facility = facilities[0]
self.assertEqual(facility.get_setting('regionId'), 'dd-na')
self.assertEqual(facility.get_setting('locationId'), 'NA9')
self.assertTrue(facility.get_blueprint('fake') is None)
blueprint = facility.get_blueprint('myBlueprint')
node = blueprint['nodes'][0]
self.assertEqual(list(node)[0], 'toto')
config = node['toto']['cloud-config']
self.assertEqual(config['hostname'], 'toto')
self.assertEqual(config['write_files'][0]['content'].count('toto'), 6)
engine.do('deploy')
engine.do('dispose')
def test_lifecycle(self):
engine = PlumberyEngine()
DimensionDataNodeDriver.connectionCls.conn_classes = (
None, DimensionDataMockHttp)
DimensionDataMockHttp.type = None
self.region = DimensionDataNodeDriver(*DIMENSIONDATA_PARAMS)
engine.set_shared_secret('fake_secret')
engine.set_user_name('fake_name')
engine.set_user_password('fake_password')
engine.do('build')
engine.build_all_blueprints()
engine.do('build', 'myBlueprint')
engine.build_blueprint('myBlueprint')
engine.do('deploy')
engine.do('deploy', 'myBlueprint')
engine.do('destroy')
engine.destroy_all_blueprints()
engine.do('destroy', 'myBlueprint')
engine.destroy_blueprint('myBlueprint')
engine.do('dispose')
engine.do('dispose', 'myBlueprint')
engine.do('polish')
engine.polish_all_blueprints()
engine.do('polish', 'myBlueprint')
engine.polish_blueprint('myBlueprint')
engine.do('refresh')
engine.do('refresh', 'myBlueprint')
engine.do('secrets')
engine.do('start')
engine.start_all_blueprints()
engine.do('start', 'myBlueprint')
engine.start_blueprint('myBlueprint')
engine.do('stop')
engine.stop_all_blueprints()
engine.do('stop', 'myBlueprint')
engine.stop_blueprint('myBlueprint')
engine.do('wipe')
engine.wipe_all_blueprints()
engine.do('wipe', 'myBlueprint')
engine.wipe_blueprint('myBlueprint')
banner = engine.document_elapsed()
self.assertEqual('Worked for you' in banner, True)
def setUp(self):
self.plumbery = PlumberyEngine()
self.plumbery.set_user_name('fake_user')
self.plumbery.set_user_password('fake_password')
DimensionDataNodeDriver.connectionCls.conn_classes = (
None, DimensionDataMockHttp)
DimensionDataMockHttp.type = None
self.plumbery.region = DimensionDataNodeDriver(*DIMENSIONDATA_PARAMS)
self.facility = PlumberyFacility(
plumbery=self.plumbery, fittings=fakeFittings)
self.facility.power_on()
class FakeFacility:
plumbery = FakePlumbery()
DimensionDataNodeDriver.connectionCls.conn_classes = (None, DimensionDataMockHttp)
DimensionDataMockHttp.type = None
region = DimensionDataNodeDriver(*DIMENSIONDATA_PARAMS)
location = FakeLocation()
_cache_network_domains = []
_cache_vlans = []
def test_process_blueprint(self):
engine = PlumberyEngine()
DimensionDataNodeDriver.connectionCls.conn_classes = (
None, DimensionDataMockHttp)
DimensionDataMockHttp.type = None
self.region = DimensionDataNodeDriver(*DIMENSIONDATA_PARAMS)
engine.set_shared_secret('fake_secret')
engine.set_user_name('fake_name')
engine.set_user_password('fake_password')
engine.set_fittings(myPrivatePlan)
engine.process_blueprint(action='dummy', names='fake')
action = FakeAction({})
engine.process_blueprint(action, names='fake')
self.assertEqual(action.count, 205)
def do_polish(polisher):
engine = PlumberyEngine(myInformation)
engine.set_shared_secret('fake_secret')
engine.set_user_name('fake_name')
engine.set_user_password('fake_password')
polisher.go(engine)
facility = engine.list_facility('NA9')[0]
DimensionDataNodeDriver.connectionCls.conn_classes = (
None, DimensionDataMockHttp)
DimensionDataMockHttp.type = None
facility.region = DimensionDataNodeDriver(*DIMENSIONDATA_PARAMS)
polisher.move_to(facility)
blueprint = facility.get_blueprint('test')
infrastructure = PlumberyInfrastructure(facility)
container = infrastructure.get_container(blueprint)
polisher.shine_container(container)
nodes = PlumberyNodes(facility)
node = nodes.get_node('stackstorm')
polisher.shine_node(node=node,
settings=fakeNodeSettings,
container=container)
node = nodes.get_node('node1')
polisher.shine_node(node=node,
settings=fakeNodeSettings,
container=container)
polisher.move_to(FakeFacility())
polisher.shine_container(FakeContainer())
polisher.shine_node(node=FakeNode(),
settings=fakeNodeSettings,
container=FakeContainer())
polisher.reap()
def test_set(self):
engine = PlumberyEngine()
DimensionDataNodeDriver.connectionCls.conn_classes = (
None, DimensionDataMockHttp)
DimensionDataMockHttp.type = None
self.region = DimensionDataNodeDriver(*DIMENSIONDATA_PARAMS)
settings = {
'safeMode': False,
'polishers': [
{
'ansible': {}
},
{
'configure': {}
},
]
}
engine.set_settings(settings)
self.assertEqual(engine.safeMode, False)
engine.add_facility(myFacility)
self.assertEqual(len(engine.facilities), 1)
self.assertEqual(engine.get_shared_user(), 'root')
engine.set_shared_user('ubuntu')
self.assertEqual(engine.get_shared_user(), 'ubuntu')
engine.set_shared_secret('fake_secret')
self.assertEqual(engine.get_shared_secret(), 'fake_secret')
random = engine.get_secret('random')
self.assertEqual(len(random), 9)
self.assertEqual(engine.get_secret('random'), random)
engine.set_user_name('fake_name')
self.assertEqual(engine.get_user_name(), 'fake_name')
engine.set_user_password('fake_password')
self.assertEqual(engine.get_user_password(), 'fake_password')
class FakeFacility:
_cache_network_domains = []
_cache_vlans = []
plumbery = FakePlumbery()
DimensionDataNodeDriver.connectionCls.conn_classes = (
None, DimensionDataMockHttp)
DimensionDataMockHttp.type = None
region = DimensionDataNodeDriver(*DIMENSIONDATA_PARAMS)
location = 1
def get_image(self, name):
return FakeImage()
def power_on(self):
pass
def get_location_id(self):
return 'EU6'
def init_client(self, user, password, region=DEFAULT_REGION):
self.node = DimensionDataNodeDriver(user, password, region)
self.backup = DimensionDataBackupDriver(user, password, region)
def _get_compute_driver(self, region):
api_user = self.config['api_user']
api_pass = self.config['api_password']
driver = DimensionDataNodeDriver(api_user, api_pass, region=region)
return driver
#!/usr/bin/env python
from libcloud.compute.drivers.dimensiondata import DimensionDataNodeDriver
from libcloud.common.dimensiondata import DEFAULT_REGION
import argparse
import json
parser = argparse.ArgumentParser(description='Dumppppp')
parser.add_argument('--user')
parser.add_argument('--password')
parser.add_argument('--region', default=DEFAULT_REGION)
args = parser.parse_args()
client = DimensionDataNodeDriver(args.user, args.password, args.region)
nodes = client.list_nodes()
node_list = []
#We need to convert a List of Nodes to a list of dictionarys. Silly I know,
#The other option is to add .to_json methods or write a json parser for node objects
for node in nodes:
temp_dict = {}
temp_dict['id'] = node.id
temp_dict['name'] = node.name
temp_dict['public_ips'] = node.public_ips
temp_dict['private_ips'] = node.private_ips
temp_dict['memoryMb'] = node.extra['memoryMb']
temp_dict['cpu'] = {'cpuCount': node.extra['cpu'].cpu_count,
'cores_per_socket': node.extra['cpu'].cores_per_socket,
'performace': node.extra['cpu'].performance
def test_set(self):
engine = PlumberyEngine()
DimensionDataNodeDriver.connectionCls.conn_classes = (
None, DimensionDataMockHttp)
DimensionDataMockHttp.type = None
self.region = DimensionDataNodeDriver(*DIMENSIONDATA_PARAMS)
file = os.path.abspath(
os.path.dirname(__file__)) + '/fixtures/dummy_rsa.pub'
settings = {
'keys': ["*hello-there*"],
}
with self.assertRaises(ValueError):
engine.set_settings(settings)
settings = {
'keys': [file],
}
engine.set_settings(settings)
self.assertTrue(isinstance(engine.get_shared_key_files(), list))
self.assertTrue(file in engine.get_shared_key_files())
settings = {
'safeMode': False,
'polishers': [
{
'ansible': {}
},
{
'configure': {}
},
],
'keys': [file, file],
}
engine.set_settings(settings)
self.assertEqual(engine.safeMode, False)
self.assertTrue(isinstance(engine.get_shared_key_files(), list))
self.assertTrue(file in engine.get_shared_key_files())
engine.add_facility(myFacility)
self.assertEqual(len(engine.facilities), 1)
self.assertEqual(engine.get_shared_user(), 'root')
engine.set_shared_user('ubuntu')
self.assertEqual(engine.get_shared_user(), 'ubuntu')
engine.set_shared_secret('fake_secret')
self.assertEqual(engine.get_shared_secret(), 'fake_secret')
random = engine.get_secret('random')
self.assertEqual(len(random), 9)
self.assertEqual(engine.get_secret('random'), random)
engine.set_user_name('fake_name')
self.assertEqual(engine.get_user_name(), 'fake_name')
engine.set_user_password('fake_password')
self.assertEqual(engine.get_user_password(), 'fake_password')
