[Bugfix][CI] Fix stale Mooncake lookup expectation broken by a merge race (#49802)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com>
Harry Mellor committed
Jul 25, 2026 at 11:10 UTC
0b1a8bb1f6386dd0b60105f63cf67337cb775a56
1 file changed
+4
-2
tests/v1/kv_connector/unit/test_mooncake_store_worker.py
+4
-2
@@ -1905,13 +1905,15 @@ def test_lookup_rejects_boundary_missing_one_mamba_shard():
1905
)
1906
_refresh_group_tp_replication_factors(worker)
1907
1908
+ # 33 tokens for two 16-token blocks: the hit stops below the request end,
1909
+ # so the full-hit re-derivation stays out of the shard accounting.
1910
worker.store.batch_is_exist.side_effect = lambda keys: [1] * len(keys)
1909
- assert worker.lookup(32, [b"h0", b"h1"]) == 32
1911
+ assert worker.lookup(33, [b"h0", b"h1"]) == 32
1912
1913
worker.store.batch_is_exist.side_effect = lambda keys: [
1914
0 if "tp_rank:1" in k and "group:1" in k else 1 for k in keys
1915
]
1914
- assert worker.lookup(32, [b"h0", b"h1"]) == 0
1916
+ assert worker.lookup(33, [b"h0", b"h1"]) == 0
1917
1918
1919
def test_lookup_requires_all_dcp_rank_namespaces():