Rust server: align launcher and request validation behavior (#37327)

2026-09-01 23:42:36 -07:00
parent 83bd2c473f
commit f8f04bafa8
13 changed files with 317 additions and 41 deletions
@@ -0,0 +1,99 @@
+"""Tests for declarative slash-command test groups."""
+
+import importlib.util
+import json
+import os
+import sys
+import tempfile
+import unittest
+from pathlib import Path
+from types import ModuleType
+from unittest.mock import patch
+
+from sglang.test.ci.ci_register import register_cpu_ci
+from sglang.test.test_utils import CustomTestCase
+
+register_cpu_ci(est_time=1, suite="base-a-test-cpu")
+
+_REPO_ROOT = Path(__file__).resolve().parents[4]
+_HANDLER_PATH = _REPO_ROOT / "scripts/ci/utils/slash_command_handler.py"
+
+
+def _load_handler():
+    github = ModuleType("github")
+    github.Auth = object()
+    github.Github = object()
+    spec = importlib.util.spec_from_file_location(
+        "slash_command_handler", _HANDLER_PATH
+    )
+    module = importlib.util.module_from_spec(spec)
+    with patch.dict(sys.modules, {"github": github}):
+        spec.loader.exec_module(module)
+    return module
+
+
+class TestConfiguredTestGroups(CustomTestCase):
+    def test_additional_group_requires_only_manifest_data(self):
+        handler = _load_handler()
+        previous_cwd = os.getcwd()
+        try:
+            os.chdir(_REPO_ROOT)
+            with tempfile.TemporaryDirectory() as temp_dir:
+                manifest = Path(temp_dir) / "groups.json"
+                manifest.write_text(
+                    json.dumps(
+                        {
+                            "mixed": [
+                                "registered/rust/test_run_rust_tests.py",
+                                "registered/core/test_srt_endpoint.py",
+                            ]
+                        }
+                    )
+                )
+                with patch.object(handler, "TEST_GROUPS_FILE_PATH", str(manifest)):
+                    specs, error = handler.resolve_test_group_specs("mixed")
+
+            self.assertIsNone(error)
+            self.assertEqual(
+                specs,
+                [
+                    "registered/rust/test_run_rust_tests.py",
+                    "registered/core/test_srt_endpoint.py",
+                ],
+            )
+        finally:
+            os.chdir(previous_cwd)
+
+    def test_rust_server_group(self):
+        handler = _load_handler()
+        previous_cwd = os.getcwd()
+        try:
+            os.chdir(_REPO_ROOT)
+            specs, error = handler.resolve_test_group_specs("rust-server")
+            self.assertIsNone(error)
+            self.assertEqual(
+                specs,
+                [
+                    "registered/rust/test_run_rust_tests.py",
+                    "registered/core/test_srt_endpoint.py",
+                    "registered/vlm/test_rust_native_mm_e2e.py",
+                    "registered/vlm/test_rust_native_mm_mmmu.py",
+                ],
+            )
+
+            resolved = [
+                item
+                for test_spec in specs
+                for item in handler._resolve_test_spec(test_spec)
+            ]
+            self.assertTrue(all(item["error"] is None for item in resolved), resolved)
+            self.assertEqual(
+                [item["mode"] for item in resolved],
+                ["cpu", "cuda", "cuda", "cuda"],
+            )
+        finally:
+            os.chdir(previous_cwd)
+
+
+if __name__ == "__main__":
+    unittest.main()