Update test to use user_token and update setup_env_user_token to use token when possible (kubeflow#1663)

Signed-off-by: Debarati Basu-Nag <dbasunag@redhat.com>

Author: dbasunag

clients/python/tests/conftest.py

@@ -178,9 +178,10 @@ def client_attrs() -> dict[str, any]:
 
 
 @pytest.fixture(scope="module")
-def setup_env_user_token():
+def setup_env_user_token(user_token: str) -> str:
+    token_bytes = (user_token or "Token").encode("utf-8")
     with tempfile.NamedTemporaryFile(delete=False) as token_file:
-        token_file.write(b"Token")
+        token_file.write(token_bytes)
     old_token_path = os.getenv("KF_PIPELINES_SA_TOKEN_PATH")
     os.environ["KF_PIPELINES_SA_TOKEN_PATH"] = token_file.name
 

clients/python/tests/fuzz_api/conftest.py

@@ -29,11 +29,11 @@ def generated_schema(request: pytest.FixtureRequest, pytestconfig: pytest.Config
     return schema
 
 @pytest.fixture
-def auth_headers(setup_env_user_token):
+def auth_headers(user_token: str) -> dict[str, str]:
     """Provides authorization headers for API requests."""
     return {
         "Content-Type": "application/json",
-        "Authorization": f"Bearer {setup_env_user_token}"
+        "Authorization": f"Bearer {user_token}",
     }
 
 @pytest.fixture

clients/python/tests/test_client.py

@@ -862,7 +862,7 @@ def test_nested_recursive_store_in_s3(
 
 @pytest.mark.e2e
 def test_custom_async_runner_with_ray(
-    client_attrs: dict[str, any], client: ModelRegistry, monkeypatch
+    client_attrs: dict[str, any], client: ModelRegistry, user_token: str, monkeypatch
 ):
     """Test Ray integration with uvloop event loop policy"""
     import asyncio
@@ -901,6 +901,7 @@ def test_with_ray():
                     author=client_attrs["author"],
                     is_secure=client_attrs["ssl"],
                     async_runner=atr.run,
+                    user_token=user_token,
                 )
                 client.register_model(
                     name="test_model",