Test bulk operations in models

tests/test_models.py

@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+from unittest import mock
+
 from django.core.exceptions import ValidationError
 from django.test import TestCase
 
@@ -24,6 +26,18 @@ def test_get_version(self):
             self.cookie_group.get_version(), self.cookie.created.isoformat()
         )
 
+    @mock.patch("cookie_consent.models.delete_cache")
+    def test_bulk_delete(self, mock_delete_cache):
+        CookieGroup.objects.filter(id=self.cookie_group.id).delete()
+
+        self.assertEqual(mock_delete_cache.call_count, 1)
+
+    @mock.patch("cookie_consent.models.delete_cache")
+    def test_bulk_update(self, mock_delete_cache):
+        CookieGroup.objects.filter(id=self.cookie_group.id).update(name="Optional2")
+
+        self.assertEqual(mock_delete_cache.call_count, 1)
+
 
 class CookieTest(TestCase):
     def setUp(self):
@@ -42,6 +56,18 @@ def setUp(self):
     def test_varname(self):
         self.assertEqual(self.cookie.varname, "optional=foo:.example.com")
 
+    @mock.patch("cookie_consent.models.delete_cache")
+    def test_bulk_delete(self, mock_delete_cache):
+        Cookie.objects.filter(id=self.cookie.id).delete()
+
+        self.assertEqual(mock_delete_cache.call_count, 1)
+
+    @mock.patch("cookie_consent.models.delete_cache")
+    def test_bulk_update(self, mock_delete_cache):
+        Cookie.objects.filter(id=self.cookie.id).update(name="foo2")
+
+        self.assertEqual(mock_delete_cache.call_count, 1)
+
 
 class ValidateCookieNameTest(TestCase):
     def test_valid(self):