def setUp(self):
super(BaseDriverTests, self).setUp()
self.mock_client = Mock()
self.mock_stdout = Mock()
self.driver = _DummyDriver(self.mock_client, out=self.mock_stdout)
self.store = TemplateStore(self.driver)
def _check_roles_exist(role_ids):
store = TemplateStore()
for i in role_ids:
try:
store.retrieve(i)
except UnknownUUID:
sys.stderr.write("No role with id %s " % i)
raise
def __init__(self):
super(PlansManager, self).__init__()
self.plan_store = DeploymentPlanStore()
self.seed_store = MasterSeedStore()
self.registry_store = ResourceRegistryStore()
self.template_store = TemplateStore()
self.master_template_store = MasterTemplateStore()
self.environment_store = EnvironmentFileStore()
def setUp(self):
super(PlansManagerTestCase, self).setUp()
self.plans_manager = PlansManager()
self.plan_store = DeploymentPlanStore()
self.template_store = TemplateStore()
self.seed_store = MasterSeedStore()
self.registry_store = ResourceRegistryStore()
def load_role(name, file_path, extra_data=None, relative_path=''):
name = role_name_from_path(file_path) if (name == '') else name
all_roles, created, updated = load_roles(roles=[],
seed_file=None,
resource_registry_path=None,
role_extra=extra_data)
process_role(file_path, name, TemplateStore(), all_roles, created, updated,
relative_path)
return created, updated
def _create_or_update(name, contents, store=None):
if store is None:
store = TemplateStore()
try:
role = store.retrieve_by_name(name)
if role.contents != contents:
role = store.update(role.uuid, contents)
return False, role
except UnknownName:
return True, store.create(name, contents)
def create_or_update(name,
contents,
store=None,
relative_path='',
registry_path=''):
if store is None:
store = TemplateStore()
try:
role = store.retrieve_by_name(name)
if role.contents != contents:
role = store.update(role.uuid, contents, relative_path,
registry_path)
return False, role
except UnknownName:
return True, store.create(name, contents, relative_path, registry_path)
def __init__(self):
super(RoleManager, self).__init__()
self.template_store = TemplateStore()
def setUp(self):
super(ModelsTests, self).setUp()
self.driver = Mock()
self.store = TemplateStore(self.driver)
def _delete_role(role_id):
TemplateStore().delete(role_id)
def setUp(self):
super(SQLAlchemyDriverTestCase, self).setUp()
self.driver = SQLAlchemyDriver()
self.store = TemplateStore(self.driver)
def setUp(self):
super(RoleManagerTests, self).setUp()
self.role_manager = RoleManager()
self.template_store = TemplateStore()