test: fix Task 1 CPU unit coverage

test/registered/unit/managers/test_prefill_adder.py

@@ -1,7 +1,12 @@
+import sys
 import unittest
 from types import SimpleNamespace
 from unittest.mock import MagicMock
 
+for _mod in ("sgl_kernel", "sgl_kernel.kvcacheio"):
+    if _mod not in sys.modules:
+        sys.modules[_mod] = MagicMock()
+
 from sglang.srt.managers.schedule_batch import Req
 from sglang.srt.managers.schedule_policy import AddReqResult, PrefillAdder
 from sglang.srt.mem_cache.base_prefix_cache import (
@@ -453,7 +458,6 @@ class TestPrefillAdder(CustomTestCase):
             running_batch,
             page_size=64,
             rem_input_tokens=4096,
-            rem_total_tokens=4096,
         )
         req = self.create_mock_req("req", priority=0, max_new_tokens=16)
         req.extend_input_len = 256
@@ -474,7 +478,7 @@ class TestPrefillAdder(CustomTestCase):
 
         self.assertNotEqual(result, AddReqResult.NO_TOKEN)
         params = self.mock_tree_cache.init_load_back.call_args.args[0]
-        self.assertEqual(params.mem_quota, 256)
+        self.assertEqual(params.mem_quota, 320)
 
 
 if __name__ == "__main__":