chore: import upstream snapshot with attribution

2026-07-13 13:17:40 +08:00
commit f1825c8ceb
10096 changed files with 2364182 additions and 0 deletions
@@ -0,0 +1,115 @@
+load("@rules_python//python:defs.bzl", "py_library")
+load("//bazel:python.bzl", "py_test_module_list", "py_test_module_list_with_env_variants", "py_test_run_all_subdirectory")
+
+py_library(
+    name = "conftest",
+    srcs = ["conftest.py"],
+)
+
+py_test_run_all_subdirectory(
+    size = "small",
+    include = glob(["test_*.py"]),
+    exclude = [
+        "test_haproxy_binary.py",
+        "test_round_robin_router.py",
+    ],
+    extra_srcs = [],
+    tags = ["team:serve"],
+    deps = [
+        ":conftest",
+        "//python/ray/serve:serve_lib",
+        "//python/ray/serve/tests:common",
+    ],
+)
+
+py_test_module_list(
+    size = "small",
+    files = ["test_round_robin_router.py"],
+    tags = ["team:serve"],
+    deps = [
+        ":conftest",
+        "//python/ray/serve:serve_lib",
+        "//python/ray/serve/tests:common",
+    ],
+)
+
+py_test_module_list(
+    size = "small",
+    files = ["test_haproxy_binary.py"],
+    tags = [
+        "no_windows",
+        "team:serve",
+    ],
+    deps = [
+        ":conftest",
+        "//python/ray/serve:serve_lib",
+        "//python/ray/serve/tests:common",
+    ],
+)
+
+py_test_module_list(
+    size = "small",
+    timeout = "short",
+    env = {
+        "RAY_SERVE_USE_PACK_SCHEDULING_STRATEGY": "1",
+        "RAY_SERVE_FAIL_ON_RANK_ERROR": "1",
+    },
+    files = [
+        "test_deployment_scheduler.py",
+        "test_deployment_state.py",
+    ],
+    name_suffix = "_with_pack_scheduling",
+    tags = [
+        "no_windows",
+        "team:serve",
+    ],
+    deps = [
+        ":conftest",
+        "//python/ray/serve:serve_lib",
+        "//python/ray/serve/tests:common",
+    ],
+)
+
+py_test_module_list_with_env_variants(
+    size = "small",
+    env_variants = {
+        "metr_disab": {
+            "env": {
+                "RAY_SERVE_COLLECT_AUTOSCALING_METRICS_ON_HANDLE": "0",
+                "RAY_SERVE_FAIL_ON_RANK_ERROR": "1",
+            },
+            "name_suffix": "_metr_disab",
+        },
+        "metr_agg_at_controller": {
+            "env": {
+                "RAY_SERVE_AGGREGATE_METRICS_AT_CONTROLLER": "1",
+                "RAY_SERVE_COLLECT_AUTOSCALING_METRICS_ON_HANDLE": "1",
+                "RAY_SERVE_FAIL_ON_RANK_ERROR": "1",
+            },
+            "name_suffix": "_metr_agg_at_controller",
+        },
+        "metr_agg_at_controller_and_replicas": {
+            "env": {
+                "RAY_SERVE_AGGREGATE_METRICS_AT_CONTROLLER": "1",
+                "RAY_SERVE_COLLECT_AUTOSCALING_METRICS_ON_HANDLE": "0",
+                "RAY_SERVE_FAIL_ON_RANK_ERROR": "1",
+            },
+            "name_suffix": "_metr_agg_at_controller_and_replicas",
+        },
+    },
+    files = [
+        "test_autoscaling_policy.py",
+        "test_controller.py",
+        "test_deployment_state.py",
+        "test_router.py",
+    ],
+    tags = [
+        "no_windows",
+        "team:serve",
+    ],
+    deps = [
+        ":conftest",
+        "//python/ray/serve:serve_lib",
+        "//python/ray/serve/tests:common",
+    ],
+)