django Postgres IntegrityError

Question

¿Puede alguien explicarme cómo probar correctamente los errores de Postgres DB, particularmente IntegrityError? Por ejemplo, he siguiente prueba:

class TestSlugs(TestCase): 
    # This slug must be unique 
    b = BookPublisher(slug=self.duplicate_slug) 
    self.assertRaises(IntegrityError, b.save) 

    #check if there's only one BookPublisher 
    self.assertEquals(BookPublisher.objects.count(), 1) 

Aquí llama la IntegrityError pero entonces todas las operaciones se producirá un error, porque así es como postgres obras, bien. Veo en los documentos que puedo usar transaction.rollback() pero donde: en la prueba o en el método save()?

Además, no me gusta la idea de escribir retrocesos a mano, ¿por qué no puede django solo tratar de guardar, y si falla? Dame IntegrityError y déjame continuar trabajando.

estoy usando Django 1.1

Answer 1

No altere el método save(), ya que desea que se propagarse a través de operaciones normales. Debería detectar la excepción en su clase de prueba y deshacerla (tenga en cuenta que, dado que está probando transacciones, debe crear una TransactionalTestCase en lugar de la TestCase normal).

Answer 2

no es 100% seguro de que esto es válido, pero se puede hacer:

def save(self): 
    transaction.commit() 
    try: 
     super(MyModel, self).save() 
    except IntegrityError: 
     transaction.rollback() 
    else: 
     transaction.commit()