def test_harvest_by_metadata_formats_and_sets_saves_record(
self, mock_convert_file, mock_get):
""" Test harvest by metadata formats and sets save
Args:
mock_get:
Returns:
"""
# Arrange
mock_get.return_value.status_code = status.HTTP_200_OK
mock_get.return_value.text = OaiPmhMock.\
mock_oai_response_list_records(with_resumption_token=False)
metadata_format = [self.fixture.oai_metadata_formats[0]]
set_ = [self.fixture.oai_sets[0]]
mock_convert_file.return_value = None
# Act
result = oai_registry_api._harvest_by_metadata_formats_and_sets(
self.fixture.registry, metadata_format, set_,
self.fixture.oai_sets)
# Assert
record_in_database = oai_record_api.get_all_by_registry_id(
self.fixture.registry.id)
self.assertEquals(result, [])
self.assertTrue(len(record_in_database) > 0)
def test_harvest_registry_saves_record(self, mock_convert_file, mock_get):
""" Test harvest save
Args:
mock_get:
mock_convert_file:
Returns:
"""
# Arrange
mock_get.return_value.status_code = status.HTTP_200_OK
mock_get.return_value.text = OaiPmhMock.\
mock_oai_response_list_records(with_resumption_token=False)
mock_convert_file.return_value = None
# Act
result = oai_registry_api.harvest_registry(self.fixture.registry)
# Assert
record_in_database = oai_record_api.get_all_by_registry_id(self.fixture.registry.id)
self.assertEquals(result, [])
self.assertTrue(len(record_in_database) > 0)
def test_get_all_by_registry_id_return_object(self, mock_get_all):
_generic_get_all_test(
self, mock_get_all,
oai_record_api.get_all_by_registry_id(ObjectId()))