mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 17:11:21 +08:00
* abort api bug fix * bug fix * bug fix
This commit is contained in:
@@ -3700,6 +3700,8 @@ class TestCommonEngineAdditionalCoverage(unittest.TestCase):
|
||||
"""_wait_abort_complete exits when background thread cleans up."""
|
||||
eng = self._make_abort_engine()
|
||||
eng.resource_manager.waiting_abort_req_id_set = {"req-1_0"}
|
||||
# Add the request to requests dict so it won't be filtered out
|
||||
eng.resource_manager.requests = {"req-1_0": self._make_fake_request()}
|
||||
|
||||
call_count = [0]
|
||||
|
||||
@@ -3718,6 +3720,8 @@ class TestCommonEngineAdditionalCoverage(unittest.TestCase):
|
||||
"""Stall timeout triggers force cleanup for requests in to_be_aborted_req_id_set."""
|
||||
eng = self._make_abort_engine()
|
||||
eng.resource_manager.to_be_aborted_req_id_set = {"req-1_0"}
|
||||
# Add the request to requests dict so it won't be filtered out
|
||||
eng.resource_manager.requests = {"req-1_0": self._make_fake_request()}
|
||||
|
||||
def mock_recycle(req_id):
|
||||
eng.resource_manager.to_be_aborted_req_id_set.discard(req_id)
|
||||
|
||||
Reference in New Issue
Block a user