fix: preserve HiCache load-back allocator headroom

test/registered/unit/mem_cache/test_cp_shared_kv_layout.py

@@ -6,6 +6,7 @@ import torch
 
 from sglang.srt.mem_cache.cp_shared_kv_layout import CpSharedKVLayout
 from sglang.test.ci.ci_register import register_cpu_ci
+from sglang.test.test_utils import CustomTestCase
 
 register_cpu_ci(est_time=1, suite="stage-a-test-cpu")
 
@@ -52,7 +53,7 @@ class TestCpSharedKVLayout(unittest.TestCase):
         self.assertEqual(positions.tolist(), [11, 15])
 
 
-class TestCPSharedPagedAllocator(unittest.TestCase):
+class TestCPSharedPagedAllocator(CustomTestCase):
     def test_compute_owner_alloc_fallback_logs_every_event(self):
         from sglang.srt.mem_cache import common
 
@@ -95,6 +96,31 @@ class TestCPSharedPagedAllocator(unittest.TestCase):
         allocator.free(locs)
         self.assertEqual(allocator.available_size(), 64 * 8)
 
+    def test_allocator_state_str_reports_free_and_release_pages(self):
+        from sglang.srt.mem_cache.allocator import CPSharedPagedTokenToKVPoolAllocator
+
+        allocator = CPSharedPagedTokenToKVPoolAllocator(
+            logical_size=64 * 8,
+            physical_size=64 * 2,
+            page_size=64,
+            dtype=torch.bfloat16,
+            device="cpu",
+            kvcache=None,
+            need_sort=True,
+            cp_size=4,
+            cp_rank=0,
+        )
+        allocator.free_pages = torch.tensor([1, 5], dtype=torch.int64)
+        allocator.release_pages = torch.tensor([9], dtype=torch.int64)
+
+        state = allocator.allocator_state_str()
+
+        self.assertIn("allocator_available_size=192", state)
+        self.assertIn("allocator_free_size=128", state)
+        self.assertIn("free_pages=2", state)
+        self.assertIn("allocator_release_size=64", state)
+        self.assertIn("release_pages=1", state)
+
     def test_compute_owner_page_assignment_matches_in_seq_zigzag(self):
         from sglang.srt.mem_cache.cp_shared_kv_compute_owner import (
             build_in_seq_page_compute_owners,

test/registered/unit/mem_cache/test_radix_cache_unit.py

@@ -39,6 +39,7 @@ from sglang.srt.mem_cache.base_prefix_cache import (
     MatchPrefixParams,
 )
 from sglang.srt.mem_cache.radix_cache import RadixCache, RadixKey, TreeNode
+from sglang.test.test_utils import CustomTestCase
 
 # Test constants
 DEFAULT_PAGE_SIZE = 4
@@ -235,6 +236,16 @@ class TestTreeNode(unittest.TestCase):
         self.assertFalse(node2 < node1)
 
 
+class TestTreeNodeChildren(CustomTestCase):
+    def test_missing_child_access_does_not_create_orphan_node(self):
+        node = TreeNode()
+
+        with self.assertRaises(KeyError):
+            _ = node.children["missing"]
+
+        self.assertEqual(node.children, {})
+
+
 class TestRadixCache(unittest.TestCase):
     """Test cases for RadixCache class."""