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
