[CI]【Hackathon 9th Sprint No.41】NO.41 功能模块 fastdeploy/splitwise/splitwise_connector.py 单测补充 by essos-bot · Pull Request #5039 · PaddlePaddle/FastDeploy (original) (raw)

image

=================================== FAILURES ===================================
____________________ TestSplitwiseConnector.test_init_basic ____________________

self = <test_splitwise_connector.TestSplitwiseConnector testMethod=test_init_basic>

def test_init_basic(self):
    """Test basic initialization."""
    # Mock the environment variable directly in the module
    with patch("fastdeploy.splitwise.splitwise_connector.envs.FD_ENABLE_CACHE_TASK", "0"):
        connector = self.create_connector()

        self.assertEqual(connector.cfg, self.mock_cfg)
        self.assertEqual(connector.engine_worker_queue, self.mock_worker_queue)
        self.assertEqual(connector.resource_manager, self.mock_resource_manager)
        self.assertEqual(connector.idx, 0)
        self.assertEqual(connector.connect_innode_instances, {})
     self.assertEqual(connector.temp_cache_info, {})

E AttributeError: 'SplitwiseConnector' object has no attribute 'temp_cache_info'

tests/v1/test_splitwise_connector.py:199: AttributeError
___________ TestSplitwiseConnector.test_send_cache_infos_decode_ipc ____________

self = <test_splitwise_connector.TestSplitwiseConnector testMethod=test_send_cache_infos_decode_ipc>

def test_send_cache_infos_decode_ipc(self):
    """Test sending cache info for decode tasks with IPC protocol."""
    self.mock_cfg.cache_config.pd_comm_port = [5678]
    connector = self.create_connector()

    mock_task = Mock()
    mock_task.disaggregate_info = {
        "role": "decode",
        "transfer_protocol": "ipc",
        "cache_info": {"ipc": {"port": 12345}},
        "block_tables": [1, 2, 3],
    }
    mock_task.request_id = "test123"

    # Mock EngineWorkerQueue to avoid real connection
    with patch("fastdeploy.splitwise.splitwise_connector.EngineWorkerQueue") as mock_queue_class:
        mock_queue = Mock()
        mock_queue_class.return_value = mock_queue
     result = connector.send_cache_infos([mock_task], 1)

E AttributeError: 'SplitwiseConnector' object has no attribute 'send_cache_infos'

tests/v1/test_splitwise_connector.py:495: AttributeError
___________ TestSplitwiseConnector.test_send_cache_infos_decode_rdma ___________

self = <test_splitwise_connector.TestSplitwiseConnector testMethod=test_send_cache_infos_decode_rdma>

def test_send_cache_infos_decode_rdma(self):
    """Test sending cache info for decode tasks with RDMA protocol."""
    self.mock_cfg.cache_config.pd_comm_port = [5678]
    connector = self.create_connector()

    mock_task = Mock()
    mock_task.disaggregate_info = {
        "role": "decode",
        "transfer_protocol": "rdma",
        "cache_info": {"rdma": {"ip": "192.168.1.100", "port": 8080}},
        "block_tables": [1, 2, 3],
    }
    mock_task.request_id = "test123"

    with patch.object(connector, "_send_message") as mock_send:
     result = connector.send_cache_infos([mock_task], 1)

E AttributeError: 'SplitwiseConnector' object has no attribute 'send_cache_infos'

tests/v1/test_splitwise_connector.py:514: AttributeError
_____________ TestSplitwiseConnector.test_send_cache_infos_prefill _____________

self = <test_splitwise_connector.TestSplitwiseConnector testMethod=test_send_cache_infos_prefill>

def test_send_cache_infos_prefill(self):
    """Test sending cache info for prefill tasks."""
    self.mock_cfg.cache_config.pd_comm_port = [5678]
    connector = self.create_connector()

    mock_task = Mock()
    mock_task.disaggregate_info = {"role": "prefill", "cache_info": {"ipc": {"current_id": 1}}}
    mock_task.block_tables = [1, 2, 3]
    mock_task.request_id = "test123"

    with patch.object(connector.engine_worker_queue, "put_cache_info") as mock_put_cache:
     result = connector.send_cache_infos([mock_task], -1)

E AttributeError: 'SplitwiseConnector' object has no attribute 'send_cache_infos'

tests/v1/test_splitwise_connector.py:530: AttributeError
=========================== short test summary info ============================
FAILED tests/v1/test_splitwise_connector.py::TestSplitwiseConnector::test_init_basic - AttributeError: 'SplitwiseConnector' object has no attribute 'temp_cache_info'
FAILED tests/v1/test_splitwise_connector.py::TestSplitwiseConnector::test_send_cache_infos_decode_ipc - AttributeError: 'SplitwiseConnector' object has no attribute 'send_cache_infos'
FAILED tests/v1/test_splitwise_connector.py::TestSplitwiseConnector::test_send_cache_infos_decode_rdma - AttributeError: 'SplitwiseConnector' object has no attribute 'send_cache_infos'
FAILED tests/v1/test_splitwise_connector.py::TestSplitwiseConnector::test_send_cache_infos_prefill - AttributeError: 'SplitwiseConnector' object has no attribute 'send_cache_infos'
========================= 4 failed, 36 passed in 0.34s =========================

本地环境运行正常,但是 ci 失败,查看原因