From cc74499d511b719942f4e7a861b7a2f22d7bf1c6 Mon Sep 17 00:00:00 2001
From: Ke Bao <ISPObaoke@163.com>
Date: Mon, 9 Jun 2025 08:09:32 +0800
Subject: [PATCH] Fix draft extend ut stability with flush cache (#6979)

---
 test/srt/test_eagle_infer.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/test/srt/test_eagle_infer.py b/test/srt/test_eagle_infer.py
index 1caf447ec..f6dc2cf1f 100644
--- a/test/srt/test_eagle_infer.py
+++ b/test/srt/test_eagle_infer.py
@@ -612,6 +612,9 @@ class TestEAGLEDraftExtend(CustomTestCase):
         kill_process_tree(cls.process.pid)
 
     def test_one_batch_accept_length(self):
+        resp = requests.get(self.base_url + "/flush_cache")
+        self.assertEqual(resp.status_code, 200)
+
         prompts = [
             "Hello, my name is",
             "The president of the United States is",