[TransferEngine] initiator_test script: make it works with P2PHANDSHAKE. (#907)

This commit is contained in:
doujiang24 2025-10-05 11:41:13 +08:00 committed by GitHub
parent 66fd297401
commit 2d80ed15b7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions

View File

@ -22,6 +22,10 @@ class TestVLLMAdaptorTransfer(unittest.TestCase):
if ret != 0:
raise RuntimeError(f"Initialization failed with code {ret}")
if cls.metadata_server == "P2PHANDSHAKE":
cls.initiator_server_name = cls.initiator_server_name.rpartition(':')[0] + ":" + str(cls.adaptor.get_rpc_port())
def test_random_write_circle_times(self):
"""Test circle times of random string write/read via buffer transfer."""
import random, string