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(filename="", content="", is_disable=False): """Returns a mock type Args: filename: content: is_disable: Returns: """ mock_type = Mock(spec=Type) mock_type.filename = filename mock_type.content = content mock_type.id = ObjectId() mock_type.is_disabled = is_disable return mock_type