Removing jsonify

Author: Filienko

tests/test_validation.py

@@ -55,18 +55,19 @@ def setUp(self):
         app.testing = True
         self.client = app.test_client()
 
-    @patch('jwt_proxy.api.proxy_request')
+    @patch('your_module.proxy_request')
     def test_path_whitelist(self, mock_proxy_request):
-        mock_proxy_request.return_value = jsonify(message="request proxied")
+        # Mock response as a Flask Response object directly
+        mock_proxy_request.return_value = self.client.get('/allowed_path')
         response = self.client.get("/allowed_path")
         self.assertEqual(response.status_code, 200)
         self.assertEqual(response.json, {"message": "request proxied"})
 
-    @patch('jwt_proxy.api.proxy_request')
+    @patch('your_module.proxy_request')
     @patch('jwt.PyJWKClient')
     @patch('jwt.decode')
     def test_valid_token(self, mock_decode, mock_jwks_client, mock_proxy_request):
-        mock_proxy_request.return_value = jsonify(message="request proxied")
+        mock_proxy_request.return_value = self.client.get('/some_path')
         mock_jwks_client.return_value.get_signing_key_from_jwt.return_value.key = "test-key"
         mock_decode.return_value = {"email": "[email protected]"}
 
@@ -75,15 +76,15 @@ def test_valid_token(self, mock_decode, mock_jwks_client, mock_proxy_request):
         self.assertEqual(response.status_code, 200)
         self.assertEqual(response.json, {"message": "request proxied"})
 
-    @patch('jwt_proxy.api.proxy_request')
+    @patch('your_module.proxy_request')
     @patch('jwt.PyJWKClient')
     @patch('jwt.decode')
     def test_missing_token(self, mock_decode, mock_jwks_client, mock_proxy_request):
         response = self.client.get("/some_path")
         self.assertEqual(response.status_code, 400)
         self.assertEqual(response.json, {"message": "token missing"})
 
-    @patch('jwt_proxy.api.proxy_request')
+    @patch('your_module.proxy_request')
     @patch('jwt.PyJWKClient')
     @patch('jwt.decode')
     def test_expired_token(self, mock_decode, mock_jwks_client, mock_proxy_request):