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
