Made OnDeleteTests.test_protect() assert more specific ProtectedError.

1 files changed, 2 insertions, 2 deletions

diff --git a/tests/delete/tests.py b/tests/delete/tests.py
index 5046afba17..f78e61704a 100644
--- 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 @@ class OnDeleteTests(TestCase):
             "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):