def _create_mock_template_version_manager(title="", versions=None, is_disabled=False, user_id=""): """ Returns a mock template version manager :return: """ if versions is None: versions = [] mock_template_version_manager = Mock(spec=TemplateVersionManager) mock_template_version_manager.title = title mock_template_version_manager.id = ObjectId() mock_template_version_manager.versions = versions mock_template_version_manager.disabled_versions = [] mock_template_version_manager.is_disabled = is_disabled mock_template_version_manager.user = str(user_id) mock_template_version_manager._cls = TemplateVersionManager.class_name return mock_template_version_manager
def _create_mock_type_version_manager(title="", versions=None, user='******'): """Returns a mock type version manager Args: title: versions: Returns: """ mock_type_version_manager = Mock(spec=TypeVersionManager) mock_type_version_manager.title = title mock_type_version_manager.id = ObjectId() mock_type_version_manager.user = user if versions is not None: mock_type_version_manager.versions = versions else: mock_type_version_manager.versions = [] mock_type_version_manager.disabled_versions = [] return mock_type_version_manager