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_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_as_library(self):
engine = PlumberyEngine(myFacility)
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')
blueprint = facility.get_blueprint('fake')
self.assertEqual(blueprint.keys(),
['ethernet', 'domain', 'nodes', 'target'])
engine.do('deploy')
engine.do('refresh')
engine.do('dispose')
def test_set(self):
settings = {
'safeMode': False,
'polishers': [
{'ansible': {}},
{'spit': {}},
]
}
engine = PlumberyEngine()
DimensionDataNodeDriver.connectionCls.conn_classes = (
None, DimensionDataMockHttp)
DimensionDataMockHttp.type = None
self.region = DimensionDataNodeDriver(*DIMENSIONDATA_PARAMS)
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')
engine.set(settings)
self.assertEqual(engine.safeMode, False)
engine.add_facility(myFacility)
self.assertEqual(len(engine.facilities), 1)
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')
def test_settings_private(self):
engine = PlumberyEngine()
engine.set_shared_secret('fake_secret')
engine.set_user_name('fake_name')
engine.set_user_password('fake_password')
engine.set_fittings(myPrivatePlan)
facilities = engine.list_facility('quasimoto.com')
self.assertEqual(len(facilities), 1)
facilities[0].power_on()
def test_settings_private(self):
engine = PlumberyEngine()
engine.set_shared_secret('fake_secret')
engine.set_user_name('fake_name')
engine.set_user_password('fake_password')
engine.set_fittings(myPrivatePlan)
facilities = engine.list_facility('quasimoto.com')
self.assertEqual(len(facilities), 1)
facilities[0].power_on()
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 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 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 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='noop', 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 do_polish(polisher):
engine = PlumberyEngine()
engine.set_shared_secret('fake_secret')
engine.set_user_name('fake_name')
engine.set_user_password('fake_password')
engine.from_text(myInformation)
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 TestPlumberyEngine(unittest.TestCase):
def test_configure(self):
settings = {
'safeMode': False,
'polishers': [
{'ansible': {}},
{'spit': {}},
]
}
self.engine = PlumberyEngine()
self.engine.set_shared_secret('fake_secret')
self.assertEqual(self.engine.get_shared_secret(), 'fake_secret')
self.engine.set_user_name('fake_name')
self.assertEqual(self.engine.get_user_name(), 'fake_name')
self.engine.set_user_password('fake_password')
self.assertEqual(self.engine.get_user_password(), 'fake_password')
self.engine.configure(settings)
self.assertEqual(self.engine.safeMode, False)
try:
self.engine.setup(io.TextIOWrapper(io.BytesIO(myPlan)))
self.engine.add_facility(myFacility)
self.assertEqual(len(self.engine.facilities), 2)
except socket.gaierror:
pass
except InvalidCredsError:
pass
def test_lifecycle(self):
self.engine = PlumberyEngine()
self.engine.set_shared_secret('fake_secret')
self.assertEqual(self.engine.get_shared_secret(), 'fake_secret')
self.engine.set_user_name('fake_name')
self.assertEqual(self.engine.get_user_name(), 'fake_name')
self.engine.set_user_password('fake_password')
self.assertEqual(self.engine.get_user_password(), 'fake_password')
try:
self.engine.build_all_blueprints()
self.engine.build_blueprint('myBlueprint')
self.engine.start_all_nodes()
self.engine.start_nodes('myBlueprint')
self.engine.stop_all_nodes()
self.engine.stop_nodes('myBlueprint')
self.engine.destroy_all_nodes()
self.engine.destroy_nodes('myBlueprint')
except socket.gaierror:
pass
except InvalidCredsError:
pass
class TestPlumberyEngine(unittest.TestCase):
def test_set(self):
settings = {
'safeMode': False,
'polishers': [
{'ansible': {}},
{'spit': {}},
]
}
self.engine = PlumberyEngine()
self.engine.set_shared_secret('fake_secret')
self.assertEqual(self.engine.get_shared_secret(), 'fake_secret')
random = self.engine.get_random_secret()
self.assertEqual(len(random), 9)
self.assertEqual(self.engine.get_random_secret(), random)
self.engine.set_user_name('fake_name')
self.assertEqual(self.engine.get_user_name(), 'fake_name')
self.engine.set_user_password('fake_password')
self.assertEqual(self.engine.get_user_password(), 'fake_password')
self.engine.set(settings)
self.assertEqual(self.engine.safeMode, False)
try:
self.engine.from_text(myPlan)
self.engine.add_facility(myFacility)
self.assertEqual(len(self.engine.facilities), 2)
except socket.gaierror:
pass
except InvalidCredsError:
pass
def test_lifecycle(self):
self.engine = PlumberyEngine()
self.engine.set_shared_secret('fake_secret')
self.assertEqual(self.engine.get_shared_secret(), 'fake_secret')
self.engine.set_user_name('fake_name')
self.assertEqual(self.engine.get_user_name(), 'fake_name')
self.engine.set_user_password('fake_password')
self.assertEqual(self.engine.get_user_password(), 'fake_password')
try:
self.engine.do('build')
self.engine.build_all_blueprints()
self.engine.build_blueprint('myBlueprint')
self.engine.do('start')
self.engine.start_all_blueprints()
self.engine.start_blueprint('myBlueprint')
self.engine.do('polish')
self.engine.polish_all_blueprints()
self.engine.polish_blueprint('myBlueprint')
self.engine.do('stop')
self.engine.stop_all_blueprints()
self.engine.stop_blueprint('myBlueprint')
self.engine.wipe_all_blueprints()
self.engine.wipe_blueprint('myBlueprint')
self.engine.do('destroy')
self.engine.destroy_all_blueprints()
self.engine.destroy_blueprint('myBlueprint')
except socket.gaierror:
pass
except InvalidCredsError:
pass
def test_lookup(self):
self.engine = PlumberyEngine()
self.assertEqual(self.engine.lookup('plumbery.version'), __version__)
random = self.engine.lookup('random.secret')
self.assertEqual(len(random), 9)
self.assertEqual(self.engine.lookup('random.secret'), random)
def test_parser(self):
args = parse_args(['fittings.yaml', 'build', 'web'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'build')
self.assertEqual(args.blueprints, ['web'])
self.assertEqual(args.facilities, None)
args = parse_args(['fittings.yaml', 'build', 'web', '-d'])
self.assertEqual(
logging.getLogger().getEffectiveLevel(), logging.DEBUG)
args = parse_args(['fittings.yaml', 'build', 'web', '-q'])
self.assertEqual(
logging.getLogger().getEffectiveLevel(), logging.WARNING)
args = parse_args(['fittings.yaml', 'start', '@NA12'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'start')
self.assertEqual(args.blueprints, None)
self.assertEqual(args.facilities, ['NA12'])
args = parse_args([
'fittings.yaml', 'rub', 'web', 'sql', '@NA9', '@NA12'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'rub')
self.assertEqual(args.blueprints, ['web', 'sql'])
self.assertEqual(args.facilities, ['NA9', 'NA12'])
args = parse_args([
'fittings.yaml', 'rub', 'web', '@NA9', 'sql', '@NA12'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'rub')
self.assertEqual(args.blueprints, ['web', 'sql'])
self.assertEqual(args.facilities, ['NA9', 'NA12'])
args = parse_args(['fittings.yaml', 'polish'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'polish')
self.assertEqual(args.blueprints, None)
self.assertEqual(args.facilities, None)
def test_main(self):
engine = PlumberyEngine()
engine.from_text(myPlan)
engine.set_user_name('fake_name')
engine.set_user_password('fake_password')
with self.assertRaises(SystemExit):
main(['bad args'], engine)
with self.assertRaises(SystemExit):
main(['fittings.yaml'], engine)
with self.assertRaises(SystemExit):
main(['fittings.yaml', 'xyz123', 'web'], engine)
with self.assertRaises(SystemExit):
main(['-v'], engine)
with self.assertRaises(SystemExit):
main(['fittings.yaml', 'build', 'web', '-v'], engine)
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')
class TestPlumberyEngine(unittest.TestCase):
def test_set(self):
settings = {
'safeMode': False,
'polishers': [
{
'ansible': {}
},
{
'spit': {}
},
]
}
self.engine = PlumberyEngine()
self.engine.set_shared_secret('fake_secret')
self.assertEqual(self.engine.get_shared_secret(), 'fake_secret')
self.engine.set_user_name('fake_name')
self.assertEqual(self.engine.get_user_name(), 'fake_name')
self.engine.set_user_password('fake_password')
self.assertEqual(self.engine.get_user_password(), 'fake_password')
self.engine.set(settings)
self.assertEqual(self.engine.safeMode, False)
try:
self.engine.from_text(myPlan)
self.engine.add_facility(myFacility)
self.assertEqual(len(self.engine.facilities), 2)
except socket.gaierror:
pass
except InvalidCredsError:
pass
def test_lifecycle(self):
self.engine = PlumberyEngine()
self.engine.set_shared_secret('fake_secret')
self.assertEqual(self.engine.get_shared_secret(), 'fake_secret')
self.engine.set_user_name('fake_name')
self.assertEqual(self.engine.get_user_name(), 'fake_name')
self.engine.set_user_password('fake_password')
self.assertEqual(self.engine.get_user_password(), 'fake_password')
try:
self.engine.build_all_blueprints()
self.engine.build_blueprint('myBlueprint')
self.engine.start_all_nodes()
self.engine.start_nodes('myBlueprint')
self.engine.polish_all_blueprints()
self.engine.polish_blueprint('myBlueprint')
self.engine.stop_all_nodes()
self.engine.stop_nodes('myBlueprint')
self.engine.destroy_all_nodes()
self.engine.destroy_nodes('myBlueprint')
self.engine.destroy_all_blueprints()
self.engine.destroy_blueprint('myBlueprint')
except socket.gaierror:
pass
except InvalidCredsError:
pass
def test_parser(self):
with self.assertRaises(SystemExit):
args = parse_args(['bad args'])
with self.assertRaises(SystemExit):
args = parse_args(['fittings.yaml'])
args = parse_args(['fittings.yaml', 'build', 'web'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'build')
self.assertEqual(args.blueprints, ['web'])
self.assertEqual(args.facilities, None)
args = parse_args(['fittings.yaml', 'start', '@NA12'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'start')
self.assertEqual(args.blueprints, None)
self.assertEqual(args.facilities, ['NA12'])
args = parse_args(
['fittings.yaml', 'rub', 'web', 'sql', '@NA9', '@NA12'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'rub')
self.assertEqual(args.blueprints, ['web', 'sql'])
self.assertEqual(args.facilities, ['NA9', 'NA12'])
args = parse_args(['fittings.yaml', 'polish'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'polish')
self.assertEqual(args.blueprints, None)
self.assertEqual(args.facilities, None)
def test_main(self):
engine = PlumberyEngine()
engine.from_text(myPlan)
engine.set_user_name('fake_name')
engine.set_user_password('fake_password')
class TestPlumberyEngine(unittest.TestCase):
def test_set(self):
settings = {
'safeMode': False,
'polishers': [
{'ansible': {}},
{'spit': {}},
]
}
self.engine = PlumberyEngine()
self.engine.set_shared_secret('fake_secret')
self.assertEqual(self.engine.get_shared_secret(), 'fake_secret')
random = self.engine.get_secret('random')
self.assertEqual(len(random), 9)
self.assertEqual(self.engine.get_secret('random'), random)
self.engine.set_user_name('fake_name')
self.assertEqual(self.engine.get_user_name(), 'fake_name')
self.engine.set_user_password('fake_password')
self.assertEqual(self.engine.get_user_password(), 'fake_password')
self.engine.set(settings)
self.assertEqual(self.engine.safeMode, False)
try:
self.engine.from_text(myPlan)
cloudConfig = self.engine.get_cloud_config()
self.assertEqual(len(cloudConfig.keys()), 3)
self.engine.add_facility(myFacility)
self.assertEqual(len(self.engine.facilities), 2)
except socket.gaierror:
pass
except InvalidCredsError:
pass
def test_lifecycle(self):
self.engine = PlumberyEngine()
self.engine.set_shared_secret('fake_secret')
self.assertEqual(self.engine.get_shared_secret(), 'fake_secret')
self.engine.set_user_name('fake_name')
self.assertEqual(self.engine.get_user_name(), 'fake_name')
self.engine.set_user_password('fake_password')
self.assertEqual(self.engine.get_user_password(), 'fake_password')
try:
self.engine.do('build')
self.engine.build_all_blueprints()
self.engine.build_blueprint('myBlueprint')
self.engine.do('start')
self.engine.start_all_blueprints()
self.engine.start_blueprint('myBlueprint')
self.engine.do('polish')
self.engine.polish_all_blueprints()
self.engine.polish_blueprint('myBlueprint')
self.engine.do('stop')
self.engine.stop_all_blueprints()
self.engine.stop_blueprint('myBlueprint')
self.engine.wipe_all_blueprints()
self.engine.wipe_blueprint('myBlueprint')
self.engine.do('destroy')
self.engine.destroy_all_blueprints()
self.engine.destroy_blueprint('myBlueprint')
except socket.gaierror:
pass
except InvalidCredsError:
pass
def test_lookup(self):
self.engine = PlumberyEngine()
self.assertEqual(self.engine.lookup('plumbery.version'), __version__)
self.engine.secrets = {}
random = self.engine.lookup('random.secret')
self.assertEqual(len(random), 9)
self.assertEqual(self.engine.lookup('random.secret'), random)
md5 = self.engine.lookup('random.md5.secret')
self.assertEqual(len(md5), 32)
self.assertNotEqual(md5, random)
sha = self.engine.lookup('random.sha1.secret')
self.assertEqual(len(sha), 40)
self.assertNotEqual(sha, random)
sha = self.engine.lookup('random.sha256.secret')
self.assertEqual(len(sha), 64)
self.assertNotEqual(sha, random)
id1 = self.engine.lookup('id1.uuid')
self.assertEqual(len(id1), 36)
self.assertEqual(self.engine.lookup('id1.uuid'), id1)
id2 = self.engine.lookup('id2.uuid')
self.assertEqual(len(id2), 36)
self.assertNotEqual(id1, id2)
self.engine.lookup('application.secret')
self.engine.lookup('database.secret')
self.engine.lookup('master.secret')
self.engine.lookup('slave.secret')
original = 'hello world'
text = self.engine.lookup('pair1.rsa_public')
self.assertEqual(text.startswith('ssh-rsa '), True)
key = RSA.importKey(text)
encrypted = key.publickey().encrypt(original, 32)
privateKey = self.engine.lookup('pair1.rsa_private')
self.assertEqual(privateKey.startswith(
'-----BEGIN RSA PRIVATE KEY-----'), True)
key = RSA.importKey(self.engine.lookup('pair1.rsa_private'))
decrypted = key.decrypt(ast.literal_eval(str(encrypted)))
self.assertEqual(decrypted, original)
self.assertEqual(len(self.engine.secrets), 12)
with self.assertRaises(LookupError):
localKey = self.engine.lookup('local.rsa_private')
localKey = self.engine.lookup('local.rsa_public')
try:
path = '~/.ssh/id_rsa.pub'
with open(os.path.expanduser(path)) as stream:
text = stream.read()
stream.close()
self.assertEqual(localKey.strip(), text.strip())
logging.info("Successful lookup of local public key")
except IOError:
pass
def test_secrets(self):
engine = PlumberyEngine()
engine.secrets = {'hello': 'world'}
engine.save_secrets(plan='test_engine.yaml')
self.assertEqual(os.path.isfile('.test_engine.secrets'), True)
engine.secrets = {}
engine.load_secrets(plan='test_engine.yaml')
self.assertEqual(engine.secrets['hello'], 'world')
engine.forget_secrets(plan='test_engine.yaml')
self.assertEqual(os.path.isfile('.test_engine.secrets'), False)
def test_defaults(self):
engine = PlumberyEngine()
engine.from_text(defaultsPlan)
self.assertEqual(engine.get_default('locationId'), 'EU6')
self.assertEqual(engine.get_default('regionId'), 'dd-eu')
self.assertEqual(engine.get_default('ipv4'), 'auto')
def test_parser(self):
args = parse_args(['fittings.yaml', 'build', 'web'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'build')
self.assertEqual(args.blueprints, ['web'])
self.assertEqual(args.facilities, None)
args = parse_args(['fittings.yaml', 'build', 'web', '-d'])
self.assertEqual(
logging.getLogger().getEffectiveLevel(), logging.DEBUG)
args = parse_args(['fittings.yaml', 'build', 'web', '-q'])
self.assertEqual(
logging.getLogger().getEffectiveLevel(), logging.WARNING)
args = parse_args(['fittings.yaml', 'start', '@NA12'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'start')
self.assertEqual(args.blueprints, None)
self.assertEqual(args.facilities, ['NA12'])
args = parse_args([
'fittings.yaml', 'rub', 'web', 'sql', '@NA9', '@NA12'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'rub')
self.assertEqual(args.blueprints, ['web', 'sql'])
self.assertEqual(args.facilities, ['NA9', 'NA12'])
args = parse_args([
'fittings.yaml', 'rub', 'web', '@NA9', 'sql', '@NA12'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'rub')
self.assertEqual(args.blueprints, ['web', 'sql'])
self.assertEqual(args.facilities, ['NA9', 'NA12'])
args = parse_args(['fittings.yaml', 'polish'])
self.assertEqual(args.fittings, 'fittings.yaml')
self.assertEqual(args.action, 'polish')
self.assertEqual(args.blueprints, None)
self.assertEqual(args.facilities, None)
def test_main(self):
engine = PlumberyEngine()
engine.from_text(myPlan)
engine.set_user_name('fake_name')
engine.set_user_password('fake_password')
with self.assertRaises(SystemExit):
main(['bad args'], engine)
with self.assertRaises(SystemExit):
main(['fittings.yaml'], engine)
with self.assertRaises(SystemExit):
main(['fittings.yaml', 'xyz123', 'web'], engine)
with self.assertRaises(SystemExit):
main(['-v'], engine)
with self.assertRaises(SystemExit):
main(['fittings.yaml', 'build', 'web', '-v'], engine)
