def test_upsert_oai_harvester_raises_exception_if_save_failed(
self, mock_save):
# Arrange
mock_save.side_effect = Exception()
# Act + Assert
with self.assertRaises(Exception):
harvester_set_api.upsert(self.oai_harvester_set)
def test_upsert_oai_harvester_return_object(self, mock_create):
# Arrange
mock_create.return_value = self.oai_harvester_set
# Act
result = harvester_set_api.upsert(self.oai_harvester_set)
# Assert
self.assertIsInstance(result, OaiHarvesterSet)
def _upsert_set_for_registry(set_, registry):
""" Adds or updates an OaiHarvesterSet object for a registry.
Args:
set_: OaiHarvesterSet instance.
registry: OaiRegistry instance.
"""
try:
set_to_save = oai_harvester_set_api.get_by_set_spec_and_registry_id(
set_.set_spec, registry.id)
# Update current OaiHarvesterSet
set_to_save.set_name = set_.set_name
set_to_save.raw = set_.raw
except exceptions.DoesNotExist:
# Creation OaiHarvesterSet
set_to_save = set_
set_to_save.registry = registry
set_to_save.harvest = True
oai_harvester_set_api.upsert(set_to_save)
