Made OnDeleteTests.test_protect() assert more specific ProtectedError.

django · Jan 27, 2020 · ca9c2c1 · ca9c2c1
1 parent 99640fa
commit ca9c2c1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/delete/tests.py b/tests/delete/tests.py
@@ -1,6 +1,6 @@
 from math import ceil
 
-from django.db import IntegrityError, connection, models
+from django.db import connection, models
 from django.db.models.deletion import (
     Collector, ProtectedError, RestrictedError,
 )
@@ -76,7 +76,7 @@ def test_protect(self):
             "Cannot delete some instances of model 'R' because they are "
             "referenced through protected foreign keys: 'A.protect'."
         )
-        with self.assertRaisesMessage(IntegrityError, msg):
+        with self.assertRaisesMessage(ProtectedError, msg):
             a.protect.delete()
 
     def test_protect_multiple(self):