Merge pull request #1 from reflex-dev/masenf/pre-commits

fix pre-commit isues

Author: masenf

demos/azure/azure.py

@@ -20,7 +20,8 @@ def index():
                         rx.button("Logout", on_click=AzureAuthState.redirect_to_logout),
                     ),
                     rx.button(
-                        "Log In with Microsoft", on_click=AzureAuthState.redirect_to_login
+                        "Log In with Microsoft",
+                        on_click=AzureAuthState.redirect_to_login,
                     ),
                 ),
                 rx.spinner(),

demos/requirements.txt

@@ -1 +1 @@
-reflex-azure-auth>=0.1
+reflex-azure-auth>=0

demos/rxconfig.py

@@ -4,5 +4,5 @@
     app_name="azure",
     plugins=[
         rx.plugins.SitemapPlugin(),
-    ]
+    ],
 )

reflex_azure_auth/__init__.py

@@ -23,7 +23,7 @@
 from reflex_enterprise import App
 
 # Simple cached OIDC metadata + JWKS loader
-_OIDC_CACHE: dict[str, object] = {}
+_OIDC_CACHE: dict[str, dict] = {}
 
 
 async def _fetch_oidc_metadata(issuer: str) -> dict:
@@ -36,7 +36,6 @@ async def _fetch_oidc_metadata(issuer: str) -> dict:
         resp.raise_for_status()
         md = resp.json()
         _OIDC_CACHE[key] = md
-        print(md)
         return md
 
 
@@ -114,7 +113,7 @@ def _get_jwt_header_payload(token: str) -> tuple[dict, dict]:
             _b64_json_decode(parts[0]),
             _b64_json_decode(parts[1]),
         )
-    return {}
+    return {}, {}
 
 
 def _compute_at_hash(access_token: str, alg: str | None) -> str:
@@ -312,10 +311,9 @@ async def _validate_tokens(self, expiration_only: bool = False) -> bool:
                 at_hash_claim
                 and hasattr(self, "_expected_at_hash")
                 and self._expected_at_hash
-            ):
-                if at_hash_claim != self._expected_at_hash:
-                    print("at_hash mismatch")  # noqa: T201
-                    return False
+            ) and at_hash_claim != self._expected_at_hash:
+                print("at_hash mismatch")  # noqa: T201
+                return False
         except Exception:
             return False
 

tests/test_oidc.py

@@ -1,4 +1,3 @@
-import asyncio
 import datetime
 
 import pytest
@@ -16,7 +15,6 @@ def json(self):
 
 @pytest.mark.asyncio
 async def test_fetch_oidc_and_jwks(monkeypatch):
-    issuer = "https://login.microsoftonline.com/common"
     md = {"jwks_uri": "https://example.com/jwks"}
     jwks = {"keys": []}
 
@@ -29,22 +27,15 @@ async def fake_get(url, timeout=10):
 
     class FakeClient:
         async def __aenter__(self):
-            import asyncio
-            import datetime
-
             import pytest
 
-            import reflex_azure_auth as raa
-
-
             class DummyResp:
                 def __init__(self, data):
                     self._data = data
 
                 def json(self):
                     return self._data
 
-
             @pytest.mark.asyncio
             async def test_fetch_oidc_and_jwks(monkeypatch):
                 issuer = "https://login.microsoftonline.com/common"
@@ -76,7 +67,6 @@ async def get(self, url, timeout=10):
                 got_jwks = await raa._fetch_jwks(md["jwks_uri"])
                 assert got_jwks == jwks
 
-
             @pytest.mark.asyncio
             async def test_verify_jwt_claim_checks(monkeypatch):
                 # Monkeypatch network and jwt.decode to return controlled claims
@@ -92,11 +82,26 @@ async def fake_fetch_jwks(j):
 
                 def fake_decode(token, key_set=None):
                     # return a token that is valid and not expired
-                    return {"iss": issuer, "aud": "api://default", "exp": int((datetime.datetime.utcnow() + datetime.timedelta(minutes=5)).timestamp())}
-
-                monkeypatch.setattr(raa, "_fetch_oidc_metadata", fake_fetch_oidc_metadata)
+                    return {
+                        "iss": issuer,
+                        "aud": "api://default",
+                        "exp": int(
+                            (
+                                datetime.datetime.utcnow()
+                                + datetime.timedelta(minutes=5)
+                            ).timestamp()
+                        ),
+                    }
+
+                monkeypatch.setattr(
+                    raa, "_fetch_oidc_metadata", fake_fetch_oidc_metadata
+                )
                 monkeypatch.setattr(raa, "_fetch_jwks", fake_fetch_jwks)
-                monkeypatch.setattr(raa.jwt, "decode", lambda token, ks: fake_decode(token, ks))
+                monkeypatch.setattr(
+                    raa.jwt, "decode", lambda token, ks: fake_decode(token, ks)
+                )
 
-                claims = await raa.verify_jwt("dummy", audience="api://default", issuer=issuer)
+                claims = await raa.verify_jwt(
+                    "dummy", audience="api://default", issuer=issuer
+                )
                 assert claims["aud"] == "api://default"