AIP-84 | Fix `test_python_client` after Adding Permission to API #47388

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged

Mar 5, 2025

clients/python/test_python_client.py

-Original file line number
+Diff line change
@@ Expand Up / @@ -42,6 +42,9 @@ @@
     from airflow_client.client.api import config_api, dag_api, dag_run_api
     from airflow_client.client.model.dag_run import DAGRun
+    from airflow.auth.managers.simple.simple_auth_manager import SimpleAuthManager
+    from airflow.auth.managers.simple.user import SimpleAuthManagerUser
     # The client must use the authentication and authorization parameters
     # in accordance with the API server security policy.
     # Examples for each auth method are provided below, use the example that
@@ Expand All / @@ -57,8 +60,17 @@ @@
     # privileges in Airflow
     # Configure HTTP basic authorization: Basic
+    auth_manager = SimpleAuthManager()
     configuration = airflow_client.client.Configuration(
-        host="http://localhost:8080/public", username="admin", password="admin"
+        host="http://localhost:8080/public",
+        api_key={
+            "Authorization": "Bearer "
+            + auth_manager._get_token_signer().generate_signed_token(
+                {
+                    **auth_manager.serialize_user(SimpleAuthManagerUser(username="test", role="admin")),
+                }
+            )
+        },
     )
     # Make sure in the [core] section, the  `load_examples` config is set to True in your airflow.cfg
@@ Expand Down @@

Provide feedback