mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
[Feature] support bos download retry (#5137)
* support bos download retry * update code * update code
This commit is contained in:
@@ -20,7 +20,6 @@ class TestResourceManagerV1(unittest.TestCase):
|
||||
max_num_seqs=max_num_seqs,
|
||||
num_gpu_blocks_override=102,
|
||||
max_num_batched_tokens=3200,
|
||||
enable_async_download_features=True,
|
||||
)
|
||||
args = asdict(engine_args)
|
||||
|
||||
@@ -130,9 +129,9 @@ class TestResourceManagerV1(unittest.TestCase):
|
||||
self.manager.bos_client = mock_client
|
||||
result = self.manager._download_features(self.request)
|
||||
self.assertIsNone(result)
|
||||
self.assertEqual(
|
||||
self.assertIn(
|
||||
"request test_request download features error",
|
||||
self.request.error_message,
|
||||
"request test_request download features error: link bucket-name/path/to/object1 download error: network error",
|
||||
)
|
||||
self.assertEqual(self.request.error_code, 530)
|
||||
|
||||
@@ -151,12 +150,27 @@ class TestResourceManagerV1(unittest.TestCase):
|
||||
self.manager.bos_client = mock_client
|
||||
result = self.manager._download_features(self.request)
|
||||
self.assertIsNone(result)
|
||||
self.assertEqual(
|
||||
self.assertIn(
|
||||
"request test_request download features error",
|
||||
self.request.error_message,
|
||||
"request test_request download features error: link bucket-name/path/to/object2 download error: timeout",
|
||||
)
|
||||
self.assertEqual(self.request.error_code, 530)
|
||||
|
||||
def test_download_features_retry(self):
|
||||
"""Test image feature download with error"""
|
||||
mock_client = MagicMock()
|
||||
mock_client.get_object_as_string.side_effect = Exception(
|
||||
"Your request rate is too high. We have put limits on your bucket."
|
||||
)
|
||||
|
||||
self.request.multimodal_inputs = {"image_feature_urls": ["bos://bucket-name/path/to/object1"]}
|
||||
|
||||
self.manager.bos_client = mock_client
|
||||
result = self.manager._download_features(self.request)
|
||||
self.assertIsNone(result)
|
||||
self.assertIn("Failed after 1 retries for bos://bucket-name/path/to/object1", self.request.error_message)
|
||||
self.assertEqual(self.request.error_code, 530)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user