@@ -55,14 +55,14 @@ def setUp(self):
55
55
app .testing = True
56
56
self .client = app .test_client ()
57
57
58
- @patch ('your_module .proxy_request' )
58
+ @patch ('jwt_proxy.api .proxy_request' )
59
59
def test_path_whitelist (self , mock_proxy_request ):
60
60
mock_proxy_request .return_value = jsonify (message = "request proxied" )
61
61
response = self .client .get ("/allowed_path" )
62
62
self .assertEqual (response .status_code , 200 )
63
63
self .assertEqual (response .json , {"message" : "request proxied" })
64
64
65
- @patch ('your_module .proxy_request' )
65
+ @patch ('jwt_proxy.api .proxy_request' )
66
66
@patch ('jwt.PyJWKClient' )
67
67
@patch ('jwt.decode' )
68
68
def test_valid_token (self , mock_decode , mock_jwks_client , mock_proxy_request ):
@@ -75,15 +75,15 @@ def test_valid_token(self, mock_decode, mock_jwks_client, mock_proxy_request):
75
75
self .assertEqual (response .status_code , 200 )
76
76
self .assertEqual (response .json , {"message" : "request proxied" })
77
77
78
- @patch ('your_module .proxy_request' )
78
+ @patch ('jwt_proxy.api .proxy_request' )
79
79
@patch ('jwt.PyJWKClient' )
80
80
@patch ('jwt.decode' )
81
81
def test_missing_token (self , mock_decode , mock_jwks_client , mock_proxy_request ):
82
82
response = self .client .get ("/some_path" )
83
83
self .assertEqual (response .status_code , 400 )
84
84
self .assertEqual (response .json , {"message" : "token missing" })
85
85
86
- @patch ('your_module .proxy_request' )
86
+ @patch ('jwt_proxy.api .proxy_request' )
87
87
@patch ('jwt.PyJWKClient' )
88
88
@patch ('jwt.decode' )
89
89
def test_expired_token (self , mock_decode , mock_jwks_client , mock_proxy_request ):
0 commit comments