[Core] move authentication_test_utils into ray._private to fix macos tests (ray-project#58528)

the auth token test setup in `conftest.py` is breaking macos test. there
are two test scripts (`test_microbenchmarks.py` and `test_basic.py`)
that run after the wheel is installed but without editable mode. for
these test to pass,` conftest.py` cannot import anything under
`ray.tests`.

this pr moves `authentication_test_utils` into `ray._private` to fix
this issue

Signed-off-by: sampan <sampan@anyscale.com>
Co-authored-by: sampan <sampan@anyscale.com>
Signed-off-by: YK <1811651+ykdojo@users.noreply.github.com>

Author: 2 people

python/ray/tests/authentication_test_utils.py renamed to python/ray/_private/authentication_test_utils.py

@@ -24,6 +24,13 @@
 import ray._private.ray_constants as ray_constants
 from ray._common.network_utils import build_address, find_free_port
 from ray._common.test_utils import wait_for_condition
+from ray._private.authentication_test_utils import (
+    authentication_env_guard,
+    clear_auth_token_sources,
+    reset_auth_token_state,
+    set_auth_mode,
+    set_env_auth_token,
+)
 from ray._private.conftest_utils import set_override_dashboard_url  # noqa: F401
 from ray._private.runtime_env import virtualenv_utils
 from ray._private.test_utils import (
@@ -42,13 +49,6 @@
     teardown_tls,
 )
 from ray.cluster_utils import AutoscalingCluster, Cluster, cluster_not_supported
-from ray.tests.authentication_test_utils import (
-    authentication_env_guard,
-    clear_auth_token_sources,
-    reset_auth_token_state,
-    set_auth_mode,
-    set_env_auth_token,
-)
 
 import psutil
 

python/ray/tests/conftest.py

@@ -12,12 +12,12 @@
     format_authentication_http_error,
     get_auth_headers_if_auth_enabled,
 )
-from ray.core.generated import runtime_env_agent_pb2
-from ray.tests.authentication_test_utils import (
+from ray._private.authentication_test_utils import (
     reset_auth_token_state,
     set_auth_mode,
     set_env_auth_token,
 )
+from ray.core.generated import runtime_env_agent_pb2
 
 
 def _agent_url(agent_address: str, path: str) -> str:

python/ray/tests/test_runtime_env_agent_auth.py

@@ -4,14 +4,14 @@
     HTTP_REQUEST_INVALID_TOKEN_ERROR_MESSAGE,
     HTTP_REQUEST_MISSING_TOKEN_ERROR_MESSAGE,
 )
-from ray.dashboard.modules.dashboard_sdk import SubmissionClient
-from ray.dashboard.modules.job.sdk import JobSubmissionClient
-from ray.tests.authentication_test_utils import (
+from ray._private.authentication_test_utils import (
     clear_auth_token_sources,
     reset_auth_token_state,
     set_auth_mode,
     set_env_auth_token,
 )
+from ray.dashboard.modules.dashboard_sdk import SubmissionClient
+from ray.dashboard.modules.job.sdk import JobSubmissionClient
 from ray.util.state import StateApiClient
 
 

python/ray/tests/test_submission_client_auth.py

@@ -19,7 +19,7 @@
     _RAYLET_AVAILABLE = False
     AuthenticationTokenLoader = None
 
-from ray.tests.authentication_test_utils import (
+from ray._private.authentication_test_utils import (
     clear_auth_token_sources,
     reset_auth_token_state,
     set_auth_mode,