[ci/bazel][2] bazelize all other ray_release tests (ray-project#35032)

Bazelize all other ray-release tests Signed-off-by: Cuong Nguyen <[email protected]> Signed-off-by: e428265 <[email protected]>
lmco · Aug 31, 2023 · 8599a77 · 8599a77
1 parent 5e1d280
commit 8599a77
Show file tree

Hide file tree

Showing 3 changed files with 57 additions and 0 deletions.
diff --git a/release/BUILD b/release/BUILD
@@ -435,6 +435,21 @@ py_test(
  ],
 )
 
+py_test(
+ name = "test_anyscale_job_manager",
+ size = "small",
+ srcs = ["ray_release/tests/test_anyscale_job_manager.py"],
+ exec_compatible_with = [":hermetic_python"],
+ tags = [
+ "release_unit",
+ "team:ci",
+ ],
+ deps = [
+ ":ray_release",
+ bk_require("pytest"),
+ ],
+)
+
 py_test(
  name = "test_anyscale_job_wrapper",
  size = "small",
@@ -565,6 +580,21 @@ py_test(
  ],
 )
 
+py_test(
+ name = "test_result",
+ size = "small",
+ srcs = ["ray_release/tests/test_result.py"],
+ exec_compatible_with = [":hermetic_python"],
+ tags = [
+ "release_unit",
+ "team:ci",
+ ],
+ deps = [
+ ":ray_release", 
+ bk_require("pytest"),
+ ],
+)
+
 py_test(
  name = "test_run_script",
  size = "small",
@@ -585,6 +615,21 @@ py_test(
  ],
 )
 
+py_test(
+ name = "test_template",
+ size = "small",
+ srcs = ["ray_release/tests/test_template.py"],
+ exec_compatible_with = [":hermetic_python"],
+ tags = [
+ "release_unit",
+ "team:ci",
+ ],
+ deps = [
+ ":ray_release",
+ bk_require("pytest"),
+ ],
+)
+
 py_test(
  name = "test_wheels",
  size = "small",

diff --git a/release/ray_release/tests/test_anyscale_job_manager.py b/release/ray_release/tests/test_anyscale_job_manager.py
@@ -1,3 +1,5 @@
+import pytest
+import sys
 import tempfile
 import os
 
@@ -19,3 +21,7 @@ def test_get_ray_error_logs():
  ) = AnyscaleJobManager._find_job_driver_and_ray_error_logs(tmpdir)
  assert ray_error_log == "".join(ERROR_LOG_PATTERNS + ["haha"])
  assert job_driver_log == "w00t"
+
+
+if __name__ == "__main__":
+ sys.exit(pytest.main(["-v", __file__]))
diff --git a/release/ray_release/tests/test_result.py b/release/ray_release/tests/test_result.py
@@ -1,3 +1,5 @@
+import pytest
+import sys
 import os
 from unittest import mock
 from ray_release.result import handle_exception, ExitCode, ResultStatus
@@ -34,3 +36,7 @@ def test_handle_exception():
  ResultStatus.INFRA_ERROR,
  None,
  )
+
+
+if __name__ == "__main__":
+ sys.exit(pytest.main(["-v", __file__]))