|
| 1 | +import respx |
| 2 | +import httpx |
| 3 | + |
1 | 4 | from pytest import fixture, mark
|
2 | 5 | from fastapi import FastAPI
|
3 | 6 | from supertokens_python.framework.fastapi import get_middleware
|
|
21 | 24 |
|
22 | 25 | pytestmark = mark.asyncio
|
23 | 26 |
|
| 27 | +respx_mock = respx.MockRouter |
| 28 | + |
24 | 29 |
|
25 | 30 | @fixture(scope="function")
|
26 | 31 | async def fastapi_client():
|
@@ -64,3 +69,56 @@ async def test_thirdpary_parsing_works(fastapi_client: TestClient):
|
64 | 69 | res.content
|
65 | 70 | == b'<html><head><script>window.location.replace("http://supertokens.io/auth/callback/apple?state=afc596274293e1587315c&code=c7685e261f98e4b3b94e34b3a69ff9cf4.0.rvxt.eE8rO__6hGoqaX1B7ODPmA");</script></head></html>'
|
66 | 71 | )
|
| 72 | + |
| 73 | + |
| 74 | +async def test_apple_provider_can_fetch_keys(): |
| 75 | + from supertokens_python.recipe.thirdparty.providers.apple import Apple |
| 76 | + from cryptography.hazmat.primitives.asymmetric.rsa import RSAPublicKey |
| 77 | + |
| 78 | + def api_side_effect(_: httpx.Request): |
| 79 | + return httpx.Response( |
| 80 | + 200, |
| 81 | + json={ |
| 82 | + "keys": [ |
| 83 | + { |
| 84 | + "kty": "RSA", |
| 85 | + "kid": "W6WcOKB", |
| 86 | + "use": "sig", |
| 87 | + "alg": "RS256", |
| 88 | + "n": "2Zc5d0-zkZ5AKmtYTvxHc3vRc41YfbklflxG9SWsg5qXUxvfgpktGAcxXLFAd9Uglzow9ezvmTGce5d3DhAYKwHAEPT9hbaMDj7DfmEwuNO8UahfnBkBXsCoUaL3QITF5_DAPsZroTqs7tkQQZ7qPkQXCSu2aosgOJmaoKQgwcOdjD0D49ne2B_dkxBcNCcJT9pTSWJ8NfGycjWAQsvC8CGstH8oKwhC5raDcc2IGXMOQC7Qr75d6J5Q24CePHj_JD7zjbwYy9KNH8wyr829eO_G4OEUW50FAN6HKtvjhJIguMl_1BLZ93z2KJyxExiNTZBUBQbbgCNBfzTv7JrxMw", |
| 89 | + "e": "AQAB", |
| 90 | + }, |
| 91 | + { |
| 92 | + "kty": "RSA", |
| 93 | + "kid": "fh6Bs8C", |
| 94 | + "use": "sig", |
| 95 | + "alg": "RS256", |
| 96 | + "n": "u704gotMSZc6CSSVNCZ1d0S9dZKwO2BVzfdTKYz8wSNm7R_KIufOQf3ru7Pph1FjW6gQ8zgvhnv4IebkGWsZJlodduTC7c0sRb5PZpEyM6PtO8FPHowaracJJsK1f6_rSLstLdWbSDXeSq7vBvDu3Q31RaoV_0YlEzQwPsbCvD45oVy5Vo5oBePUm4cqi6T3cZ-10gr9QJCVwvx7KiQsttp0kUkHM94PlxbG_HAWlEZjvAlxfEDc-_xZQwC6fVjfazs3j1b2DZWsGmBRdx1snO75nM7hpyRRQB4jVejW9TuZDtPtsNadXTr9I5NjxPdIYMORj9XKEh44Z73yfv0gtw", |
| 97 | + "e": "AQAB", |
| 98 | + }, |
| 99 | + { |
| 100 | + "kty": "RSA", |
| 101 | + "kid": "YuyXoY", |
| 102 | + "use": "sig", |
| 103 | + "alg": "RS256", |
| 104 | + "n": "1JiU4l3YCeT4o0gVmxGTEK1IXR-Ghdg5Bzka12tzmtdCxU00ChH66aV-4HRBjF1t95IsaeHeDFRgmF0lJbTDTqa6_VZo2hc0zTiUAsGLacN6slePvDcR1IMucQGtPP5tGhIbU-HKabsKOFdD4VQ5PCXifjpN9R-1qOR571BxCAl4u1kUUIePAAJcBcqGRFSI_I1j_jbN3gflK_8ZNmgnPrXA0kZXzj1I7ZHgekGbZoxmDrzYm2zmja1MsE5A_JX7itBYnlR41LOtvLRCNtw7K3EFlbfB6hkPL-Swk5XNGbWZdTROmaTNzJhV-lWT0gGm6V1qWAK2qOZoIDa_3Ud0Gw", |
| 105 | + "e": "AQAB", |
| 106 | + }, |
| 107 | + ] |
| 108 | + }, |
| 109 | + ) |
| 110 | + |
| 111 | + with respx_mock(assert_all_mocked=False) as mocker: |
| 112 | + mocked_route = mocker.get("https://appleid.apple.com/auth/keys").mock( |
| 113 | + side_effect=api_side_effect |
| 114 | + ) |
| 115 | + |
| 116 | + apple = Apple( |
| 117 | + "client-id", "client-key-id", "client-private-key", "client-team-id" |
| 118 | + ) |
| 119 | + # pylint: disable=protected-access |
| 120 | + keys = await apple._fetch_apple_public_keys() # type: ignore |
| 121 | + |
| 122 | + assert mocked_route.call_count == 1 |
| 123 | + assert len(keys) == 3 |
| 124 | + assert isinstance(keys[0], RSAPublicKey) |
0 commit comments