From 1b2ae97f5f9fbae2f07ad8de895bc12b05eb8d1f Mon Sep 17 00:00:00 2001 From: Hlamalani Date: Wed, 4 Sep 2024 15:20:31 +0200 Subject: [PATCH] add assert for bearer token --- aaq/tests/test_utils.py | 4 ++-- aaq/tests/test_views.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/aaq/tests/test_utils.py b/aaq/tests/test_utils.py index 64a38e46..7aabdfe1 100644 --- a/aaq/tests/test_utils.py +++ b/aaq/tests/test_utils.py @@ -51,7 +51,7 @@ def test_search_function(self): self.assertIn("query_id", response) self.assertEqual(response["query_id"], 1) self.assertEqual(json.loads(search_request.request.body), payload) - self.assertIn("Bearer", search_request.request.headers['Authorization']) + self.assertIn("Bearer", search_request.request.headers["Authorization"]) assert response == { "message": "*0* - Example content title\n" "*1* - Another example content title", @@ -95,7 +95,7 @@ def test_urgency_check(self): self.assertIn("is_urgent", response) self.assertIn("matched_rules", response) self.assertEqual(json.loads(request.request.body), message_text) - self.assertIn("Bearer", request.request.headers['Authorization']) + self.assertIn("Bearer", request.request.headers["Authorization"]) assert response == { "details": { diff --git a/aaq/tests/test_views.py b/aaq/tests/test_views.py index a9ea3873..23281a6c 100644 --- a/aaq/tests/test_views.py +++ b/aaq/tests/test_views.py @@ -293,7 +293,7 @@ def test_response_feedback_view(self): request = responses.calls[0] assert response.status_code == 200 - self.assertIn("Bearer", request.request.headers['Authorization']) + self.assertIn("Bearer", request.request.headers["Authorization"]) def test_response_feedback_invalid_view(self): """Test that we can submit response feedback""" @@ -494,7 +494,7 @@ def test_content_feedback_view(self): request = responses.calls[0] assert response.status_code == 200 - self.assertIn("Bearer", request.request.headers['Authorization']) + self.assertIn("Bearer", request.request.headers["Authorization"]) def test_content_feedback_invalid_view(self): """Test that we can submit content feedback"""