def _create_mock_object(): """ Create a mock object :return: """ mock_object = Mock() mock_object.id = ObjectId() return mock_object
def _create_mock_module(): """ Returns a mock module :return: """ mock_module = Mock(spec=Module) mock_module.name = "module" mock_module.url = "/module" mock_module.view = "Module.view" mock_module.id = ObjectId() return mock_module
def _create_mock_version_manager(title=""): """ Create a mock version manager :param title: :return: """ mock_vm = Mock(spec=VersionManager) mock_vm.title = title mock_vm.current = ObjectId() mock_vm.versions = [str(mock_vm.current)] mock_vm.id = ObjectId() return mock_vm
def _create_mock_template(mock_template_filename="", mock_template_content=""): """ Returns a mock template :param mock_template_filename: :param mock_template_content: :return: """ mock_template = Mock(spec=Template) mock_template.filename = mock_template_filename mock_template.content = mock_template_content mock_template.id = ObjectId() return mock_template
def _create_mock_oai_set(): """Mock an OaiSet. Returns: OaiSet mock. """ mock_oai_set = Mock(spec=OaiSet) mock_oai_set.set_spec = "oai_test" mock_oai_set.set_name = "test" mock_oai_set.id = ObjectId() return mock_oai_set
def test_get_record_with_xml_decl_use_raw( self, mock_get, mock_request, mock_check_identifier, mock_get_by_data, mock_get_by_metadata_prefix, mock_get_all_by_template_ids, ): # Arrange xml_decl = "<?xml version='1.0' encoding='UTF-8'?>" mock_oai_template = Mock(spec=Template) mock_oai_template.id = ObjectId() mock_metadata_format = Mock(spec=OaiProviderMetadataFormat) mock_metadata_format.is_template = True mock_metadata_format.template = mock_oai_template mock_oai_data = Mock(spec=OaiData) mock_oai_data.status = oai_status.ACTIVE mock_oai_data.template = mock_oai_template mock_oai_data.data.xml_content = (""" %s <body> <tag01>value_a</tag01> <tag02>value_b</tag02> </body> """ % xml_decl) mock_oai_data.oai_date_stamp = datetime(2019, 4, 1) mock_get.return_value = _create_mock_oai_settings() mock_request.return_value = "" mock_check_identifier.return_value = ObjectId() mock_get_by_data.return_value = mock_oai_data mock_get_by_metadata_prefix.return_value = mock_metadata_format mock_get_all_by_template_ids.return_value = "" data = { "verb": "GetRecord", "metadataPrefix": "dummy", "identifier": "dummy" } # Act response = RequestMock.do_request_get(OAIProviderView.as_view(), None, data=data) output_xml_data = response.context_data["xml"] # Assert self.assertNotIn(xml_decl, output_xml_data) self.assertEqual(response.status_code, HTTP_200_OK)
def _create_mock_bucket(): """Returns a mock bucket Args: Returns: """ mock_bucket = Mock(spec=Bucket) mock_bucket.label = "bucket" mock_bucket.color = "#000000" mock_bucket.types = [] mock_bucket.id = ObjectId() return mock_bucket
def _create_mock_oai_metadata_format(): """Mock an OaiMetadataFormat. Returns: OaiMetadataFormat mock. """ mock_oai_metadata_format = Mock(spec=OaiMetadataFormat) mock_oai_metadata_format.id = ObjectId() mock_oai_metadata_format.metadata_prefix = "oai_test" mock_oai_metadata_format.schema = "http://addressToMySchema/schema.xsd" mock_oai_metadata_format.xml_schema = "<root><test><test></root>" mock_oai_metadata_format.metadata_namespace = "testNamespace" return mock_oai_metadata_format
def _create_mock_type(filename="", content=""): """Returns a mock type Args: filename: content: Returns: """ mock_type = Mock(spec=Type) mock_type.filename = filename mock_type.content = content mock_type.id = ObjectId() return mock_type
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
def _create_mock_saved_query(user_id=None, template=None): """Returns a mock saved query Args: user_id: template: Returns: """ if user_id is None: user_id = ObjectId() if template is None: template = ObjectId() mock_saved_query = Mock(spec=SavedQuery) mock_saved_query.user_id = user_id mock_saved_query.template = template mock_saved_query.id = ObjectId() return mock_saved_query
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
def create_mock_user(user_id, is_staff=False, is_superuser=False, has_perm=False, is_anonymous=False): """ Mock a User. Args: user_id: is_staff: is_superuser: has_perm: is_anonymous: """ mock_user = Mock(spec=User) mock_user.id = user_id mock_user.is_staff = is_staff mock_user.is_superuser = is_superuser mock_user.has_perm.return_value = has_perm mock_user.is_anonymous = is_anonymous return mock_user