|
15 | 15 | """Test cases for the firebase_admin.app_check module.""" |
16 | 16 | import base64 |
17 | 17 | import pytest |
| 18 | +import requests |
18 | 19 |
|
19 | 20 | from jwt import PyJWK, InvalidAudienceError, InvalidIssuerError |
20 | 21 | from jwt import ExpiredSignatureError, InvalidSignatureError |
@@ -280,36 +281,47 @@ def test_verify_token_with_consume_true_not_consumed(self, mocker): |
280 | 281 | mocker.patch("jwt.get_unverified_header", return_value=JWT_PAYLOAD_SAMPLE.get("headers")) |
281 | 282 | app = firebase_admin.get_app() |
282 | 283 | app_check_service = app_check._get_app_check_service(app) |
283 | | - mock_body = mocker.patch.object(app_check_service._http_client, "body", return_value={"alreadyConsumed": False}) |
| 284 | + mock_body = mocker.patch.object( |
| 285 | + app_check_service._http_client, "body", return_value={"alreadyConsumed": False} |
| 286 | + ) |
284 | 287 |
|
285 | 288 | payload = app_check.verify_token("encoded", app=app, consume=True) |
286 | 289 | expected = JWT_PAYLOAD_SAMPLE.copy() |
287 | 290 | expected["app_id"] = APP_ID |
288 | 291 | expected["already_consumed"] = False |
289 | 292 | assert payload == expected |
290 | 293 |
|
291 | | - expected_url = f"https://firebaseappcheck.googleapis.com/v1/projects/{PROJECT_ID}:verifyAppCheckToken" |
292 | | - mock_body.assert_called_once_with("post", expected_url, json={"app_check_token": "encoded"}) |
| 294 | + expected_url = ( |
| 295 | + f"https://firebaseappcheck.googleapis.com/v1/projects/{PROJECT_ID}:verifyAppCheckToken" |
| 296 | + ) |
| 297 | + mock_body.assert_called_once_with( |
| 298 | + "post", expected_url, json={"app_check_token": "encoded"} |
| 299 | + ) |
293 | 300 |
|
294 | 301 | def test_verify_token_with_consume_true_already_consumed(self, mocker): |
295 | 302 | mocker.patch("jwt.decode", return_value=JWT_PAYLOAD_SAMPLE) |
296 | 303 | mocker.patch("jwt.PyJWKClient.get_signing_key_from_jwt", return_value=PyJWK(signing_key)) |
297 | 304 | mocker.patch("jwt.get_unverified_header", return_value=JWT_PAYLOAD_SAMPLE.get("headers")) |
298 | 305 | app = firebase_admin.get_app() |
299 | 306 | app_check_service = app_check._get_app_check_service(app) |
300 | | - mock_body = mocker.patch.object(app_check_service._http_client, "body", return_value={"alreadyConsumed": True}) |
| 307 | + mock_body = mocker.patch.object( |
| 308 | + app_check_service._http_client, "body", return_value={"alreadyConsumed": True} |
| 309 | + ) |
301 | 310 |
|
302 | 311 | payload = app_check.verify_token("encoded", app=app, consume=True) |
303 | 312 | expected = JWT_PAYLOAD_SAMPLE.copy() |
304 | 313 | expected["app_id"] = APP_ID |
305 | 314 | expected["already_consumed"] = True |
306 | 315 | assert payload == expected |
307 | 316 |
|
308 | | - expected_url = f"https://firebaseappcheck.googleapis.com/v1/projects/{PROJECT_ID}:verifyAppCheckToken" |
309 | | - mock_body.assert_called_once_with("post", expected_url, json={"app_check_token": "encoded"}) |
| 317 | + expected_url = ( |
| 318 | + f"https://firebaseappcheck.googleapis.com/v1/projects/{PROJECT_ID}:verifyAppCheckToken" |
| 319 | + ) |
| 320 | + mock_body.assert_called_once_with( |
| 321 | + "post", expected_url, json={"app_check_token": "encoded"} |
| 322 | + ) |
310 | 323 |
|
311 | 324 | def test_verify_token_with_consume_true_backend_error(self, mocker): |
312 | | - import requests |
313 | 325 | mocker.patch("jwt.decode", return_value=JWT_PAYLOAD_SAMPLE) |
314 | 326 | mocker.patch("jwt.PyJWKClient.get_signing_key_from_jwt", return_value=PyJWK(signing_key)) |
315 | 327 | mocker.patch("jwt.get_unverified_header", return_value=JWT_PAYLOAD_SAMPLE.get("headers")) |
|
0 commit comments