From c00b863ac6e9f6dfb295d4fc5291726a1112c080 Mon Sep 17 00:00:00 2001
From: Mariusz Felisiak <felisiak.mariusz@gmail.com>
Date: Mon, 27 Jan 2020 11:30:56 +0100
Subject: [PATCH] Made OnDeleteTests.test_protect() assert more specific
 ProtectedError.

---
 tests/delete/tests.py | 4 ++--
 1 file 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):