Add scripted-runtime unit, core integration, and chunked-prefill tests (#27413)

2026-06-06 09:08:35 +08:00
parent 5a82db85f0
commit bf4f2ccc78
10 changed files with 1912 additions and 0 deletions
@@ -0,0 +1,79 @@
+from __future__ import annotations
+
+import json
+import os
+import sys
+import unittest
+
+from sglang.test.ci.ci_register import register_cpu_ci
+from sglang.test.scripted_runtime.utils import ensure_script_importable, resolve_fn
+from sglang.test.test_utils import CustomTestCase
+
+register_cpu_ci(est_time=2, suite="base-a-test-cpu")
+
+
+class TestResolveFn(CustomTestCase):
+
+    def test_resolves_top_level_function(self):
+        self.assertIs(resolve_fn("json:dumps"), json.dumps)
+
+    def test_resolves_nested_attribute_path(self):
+        self.assertIs(resolve_fn("os:path.join"), os.path.join)
+
+    def test_rejects_missing_colon(self):
+        with self.assertRaisesRegex(ValueError, "module.path:function_name"):
+            resolve_fn("json.dumps")
+
+    def test_rejects_empty_module(self):
+        with self.assertRaisesRegex(ValueError, "module.path:function_name"):
+            resolve_fn(":dumps")
+
+    def test_rejects_empty_function(self):
+        with self.assertRaisesRegex(ValueError, "module.path:function_name"):
+            resolve_fn("json:")
+
+    def test_rejects_non_callable_target(self):
+        with self.assertRaisesRegex(TypeError, "not callable"):
+            resolve_fn("math:pi")
+
+    def test_propagates_missing_module_error(self):
+        with self.assertRaises(ModuleNotFoundError):
+            resolve_fn("sglang_no_such_module_zzz:foo")
+
+    def test_propagates_missing_attribute_error(self):
+        with self.assertRaises(AttributeError):
+            resolve_fn("json:no_such_attribute")
+
+
+class TestEnsureScriptImportable(CustomTestCase):
+
+    _FAKE_ENTRY = "/tmp/__scripted_runtime_ut_fake_sys_path__"
+
+    def setUp(self):
+        self._orig_path = list(sys.path)
+
+    def tearDown(self):
+        sys.path[:] = self._orig_path
+
+    def test_inserts_new_entry_at_front(self):
+        self.assertNotIn(self._FAKE_ENTRY, sys.path)
+
+        ensure_script_importable(self._FAKE_ENTRY)
+
+        self.assertEqual(sys.path[0], self._FAKE_ENTRY)
+
+    def test_noop_when_entry_is_none(self):
+        ensure_script_importable(None)
+
+        self.assertEqual(sys.path, self._orig_path)
+
+    def test_noop_when_entry_already_present(self):
+        sys.path.insert(0, self._FAKE_ENTRY)
+
+        ensure_script_importable(self._FAKE_ENTRY)
+
+        self.assertEqual(sys.path.count(self._FAKE_ENTRY), 1)
+
+
+if __name__ == "__main__":
+    unittest.main()