def test_oai_provider_set_upsert_raises_error_if_save_failed( self, mock_save): # Arrange mock_save.side_effect = Exception() # Act # Assert with self.assertRaises(Exception): provider_set_api.upsert(self.mock_oai_provider_set)
def _save(self, form): # Save treatment. try: self.object.templates_manager = [ x.id for x in self.object.templates_manager ] oai_provider_set_api.upsert(self.object) except Exception as e: form.add_error(None, str(e))
def test_oai_provider_set_upsert_return_object(self, mock_save): # Arrange mock_save.return_value = self.mock_oai_provider_set # Act result = provider_set_api.upsert(self.mock_oai_provider_set) # Assert self.assertIsInstance(result, OaiProviderSet)
def update(self, instance, validated_data): instance.set_spec = validated_data.get('set_spec', instance.set_spec) instance.set_name = validated_data.get('set_name', instance.set_name) templates_manager = validated_data.get('templates_manager', []) if len(templates_manager) > 0: templates_manager = [ version_manager_api.get(id_) for id_ in templates_manager ] instance.templates_manager = templates_manager instance.description = validated_data.get('description', instance.description) return oai_provider_set_api.upsert(instance)
def update(self, instance, validated_data): instance.set_spec = validated_data.get("set_spec", instance.set_spec) instance.set_name = validated_data.get("set_name", instance.set_name) templates_manager = validated_data.get("templates_manager", []) if len(templates_manager) > 0: templates_manager = [ version_manager_api.get(id_, request=self.context["request"]) for id_ in templates_manager ] instance.templates_manager = templates_manager instance.description = validated_data.get("description", instance.description) return oai_provider_set_api.upsert(instance)
def create(self, validated_data): return oai_provider_set_api.upsert(OaiProviderSet(**validated_data))