def test_oai_xsl_template_upsert_raises_error_if_save_failed(self, mock_save):
# Arrange
mock_save.side_effect = Exception()
# Act # Assert
with self.assertRaises(Exception):
oai_xsl_template_api.upsert(self.mock_oai_xsl_template)
def test_oai_xsl_template_upsert_returns_object(self, mock_save):
# Arrange
mock_save.return_value = self.mock_oai_xsl_template
# Act
result = oai_xsl_template_api.upsert(self.mock_oai_xsl_template)
# Assert
self.assertIsInstance(result, OaiXslTemplate)
def update(self, instance, validated_data):
instance.xslt = oai_xslt_api.get_by_id(validated_data['xslt'])
return oai_xsl_template_api.upsert(instance)
def create(self, validated_data):
return oai_xsl_template_api.upsert(OaiXslTemplate(**validated_data))
def _save(self, form):
# Save treatment.
try:
oai_xsl_template_api.upsert(self.object)
except Exception as e:
form.add_error(None, str(e))
