def test_upsert_oai_identifier_raises_exception_if_save_failed( self, mock_save): # Arrange mock_save.side_effect = Exception() # Act + Assert with self.assertRaises(Exception): oai_identify_api.upsert(self.oai_identify)
def _upsert_identify_for_registry(identify, registry): """ Adds or updates an identify object for a registry. Args: identify: OaiIdentify instance. registry: OaiRegistry instance. """ try: identify_db = oai_identify_api.get_by_registry_id(registry.id) identify.id = identify_db.id except exceptions.DoesNotExist: pass identify.registry = registry api_oai_identify.upsert(identify)
def test_upsert_oai_identify_with_raw_return_object(self, mock_create): # Arrange mock_create.return_value = self.oai_identify # Act result = oai_identify_api.upsert(self.oai_identify) # Assert self.assertIsInstance(result, OaiIdentify)
def test_upsert_oai_identify_invalid_raw_return_object(self, mock_create): # Arrange self.oai_identify.raw = '<root?</root>' mock_create.return_value = self.oai_identify # Act result = oai_identify_api.upsert(self.oai_identify) # Act + Assert self.assertIsInstance(result, OaiIdentify)
def test_upsert_oai_identify_no_exception_if_raw_not_string( self, mock_create): # Arrange self.oai_identify.raw = OrderedDict([(u'test', u'Hello')]) mock_create.return_value = self.oai_identify # Act result = oai_identify_api.upsert(self.oai_identify) # Assert self.assertIsInstance(result, OaiIdentify)
def test_upsert_oai_identify_invalid_raw_return_empty_raw( self, mock_create): # Arrange self.oai_identify.raw = '<root?</root>' mock_create.return_value = self.oai_identify # Act result = oai_identify_api.upsert(self.oai_identify) # Act + Assert self.assertEquals(result.raw, {})