🎨 [#339] Replaced ._headers with .headers

maykinmedia · Jun 13, 2023 · 7eb2277 · 7eb2277
1 parent 7f6d67f
commit 7eb2277
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 15 deletions.
diff --git a/src/objects/tests/v1/test_auth_fields.py b/src/objects/tests/v1/test_auth_fields.py
@@ -53,7 +53,7 @@ def test_retrieve_without_query(self):
             },
         )
         self.assertEqual(
-            set(response._headers["x-unauthorized-fields"][1].split(",")),
+            set(response.headers["x-unauthorized-fields"][1].split(",")),
             {
                 "uuid",
                 "record__data__name",
@@ -91,7 +91,7 @@ def test_retrieve_with_query_fields(self):
                 "record": {"data": {"name": record.data["name"]}},
             },
         )
-        self.assertNotIn("x-unauthorized-fields", response._headers)
+        self.assertNotIn("x-unauthorized-fields", response.headers)
 
     def test_retrieve_incorrect_auth_fields(self):
         PermissionFactory.create(
@@ -139,7 +139,7 @@ def test_retrieve_query_fields_not_allowed(self):
             },
         )
         self.assertEqual(
-            response._headers["x-unauthorized-fields"][1], "record__data__desc"
+            response.headers["x-unauthorized-fields"][1], "record__data__desc"
         )
 
     def test_retrieve_no_allowed_fields(self):
@@ -160,7 +160,7 @@ def test_retrieve_no_allowed_fields(self):
 
         self.assertEqual(response.status_code, status.HTTP_200_OK)
         self.assertEqual(response.json(), {})
-        self.assertIn("x-unauthorized-fields", response._headers)
+        self.assertIn("x-unauthorized-fields", response.headers)
 
 
 class ListAuthFieldsTests(TokenAuthMixin, APITestCase):
@@ -236,7 +236,7 @@ def test_list_without_query_different_object_types(self):
             ],
         )
         self.assertEqual(
-            response._headers["x-unauthorized-fields"][1],
+            response.headers["x-unauthorized-fields"][1],
             f"{self.other_object_type.url}(1)=type; {self.object_type.url}(1)=uuid",
         )
 
@@ -283,7 +283,7 @@ def test_list_with_query_fields(self):
                 },
             ],
         )
-        self.assertNotIn("x-unauthorized-fields", response._headers)
+        self.assertNotIn("x-unauthorized-fields", response.headers)
 
     def test_list_incorrect_auth_fields(self):
         PermissionFactory.create(
@@ -376,7 +376,7 @@ def test_list_no_allowed_fields(self):
 
         self.assertEqual(response.status_code, status.HTTP_200_OK)
         self.assertEqual(response.json(), [{}, {}])
-        self.assertIn("x-unauthorized-fields", response._headers)
+        self.assertIn("x-unauthorized-fields", response.headers)
 
 
 class SearchAuthFieldsTests(TokenAuthMixin, APITestCase):

diff --git a/src/objects/tests/v2/test_auth_fields.py b/src/objects/tests/v2/test_auth_fields.py
@@ -53,7 +53,7 @@ def test_retrieve_without_query(self):
             },
         )
         self.assertEqual(
-            set(response._headers["x-unauthorized-fields"][1].split(",")),
+            set(response.headers["x-unauthorized-fields"][1].split(",")),
             {
                 "uuid",
                 "record__data__name",
@@ -91,7 +91,7 @@ def test_retrieve_with_query_fields(self):
                 "record": {"data": {"name": record.data["name"]}},
             },
         )
-        self.assertNotIn("x-unauthorized-fields", response._headers)
+        self.assertNotIn("x-unauthorized-fields", response.headers)
 
     def test_retrieve_incorrect_auth_fields(self):
         PermissionFactory.create(
@@ -139,7 +139,7 @@ def test_retrieve_query_fields_not_allowed(self):
             },
         )
         self.assertEqual(
-            response._headers["x-unauthorized-fields"][1], "record__data__desc"
+            response.headers["x-unauthorized-fields"][1], "record__data__desc"
         )
 
     def test_retrieve_no_allowed_fields(self):
@@ -160,7 +160,7 @@ def test_retrieve_no_allowed_fields(self):
 
         self.assertEqual(response.status_code, status.HTTP_200_OK)
         self.assertEqual(response.json(), {})
-        self.assertIn("x-unauthorized-fields", response._headers)
+        self.assertIn("x-unauthorized-fields", response.headers)
 
 
 class ListAuthFieldsTests(TokenAuthMixin, APITestCase):
@@ -236,7 +236,7 @@ def test_list_without_query_different_object_types(self):
             ],
         )
         self.assertEqual(
-            response._headers["x-unauthorized-fields"][1],
+            response.headers["x-unauthorized-fields"][1],
             f"{self.other_object_type.url}(1)=type; {self.object_type.url}(1)=uuid",
         )
 
@@ -283,7 +283,7 @@ def test_list_with_query_fields(self):
                 },
             ],
         )
-        self.assertNotIn("x-unauthorized-fields", response._headers)
+        self.assertNotIn("x-unauthorized-fields", response.headers)
 
     def test_list_incorrect_auth_fields(self):
         PermissionFactory.create(
@@ -378,7 +378,7 @@ def test_list_no_allowed_fields(self):
 
         self.assertEqual(response.status_code, status.HTTP_200_OK)
         self.assertEqual(response.json()["results"], [{}, {}])
-        self.assertIn("x-unauthorized-fields", response._headers)
+        self.assertIn("x-unauthorized-fields", response.headers)
 
 
 class SearchAuthFieldsTests(TokenAuthMixin, APITestCase):

diff --git a/src/objects/tests/v2/test_notifications_kanaal.py b/src/objects/tests/v2/test_notifications_kanaal.py
@@ -6,7 +6,7 @@
 from django.test import override_settings
 
 from rest_framework.test import APITestCase
-from vng_api_common.notifications.kanalen import Kanaal
+from notifications_api_common.kanalen import Kanaal
 
 from objects.core.models import Object
 

diff --git a/src/objects/tests/v2/test_notifications_send.py b/src/objects/tests/v2/test_notifications_send.py
@@ -1,6 +1,7 @@
 from unittest.mock import patch
 
 from django.test import override_settings
+from django.test import TestCase
 
 import requests_mock
 from django_capture_on_commit_callbacks import capture_on_commit_callbacks