@@ -0,0 +1,265 @@ +/** + * Copyright (c) 2025 Huawei Technologies Co., Ltd. + * This program is free software, you can redistribute it and/or modify it under the terms and conditions of + * CANN Open Software License Agreement Version 2.0 (the "License"). + * Please refer to the License for details. You may not use this file except in compliance with the License. + * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, + * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE. + * See LICENSE in the root of the software repository for the full text of the License. + */ + +#include "gtest/gtest.h" +#include +#include +#define private public +#include "virtual_topo.h" +#include "dev_ub_connection.h" +#include "task.h" +#include "ub_transport_lite_impl.h" +#include "ub_mem_transport.h" +#include "mem_transport_lite.h" +#include "ub_local_notify.h" +#include "local_ub_rma_buffer.h" +#include "socket.h" +#include "rtsq_a5.h" +#include "ascend_hal.h" +#include "drv_api_exception.h" +#include "ub_conn_lite_mgr.h" +#include "ins_to_sqe_rule.h" +#include "stream_lite.h" +#include "mem_transport_callback.h" +#undef private + +using namespace Hccl; + +class UbTransportLiteImplWaitWithTimeoutTest : public testing::Test { +protected: + static void SetUpTestCase() + { + std::cout << "UbTransportLiteImplWaitWithTimeoutTest tests set up." << std::endl; + } + + static void TearDownTestCase() + { + std::cout << "UbTransportLiteImplWaitWithTimeoutTest tests tear down." << std::endl; + } + + virtual void SetUp() + { + MOCKER(HrtGetDeviceType).stubs().will(returnValue((DevType)DevType::DEV_TYPE_910A2)); + MOCKER_CPP(&RtsqBase::QuerySqBaseAddr).stubs().with(any()).will(returnValue(reinterpret_cast(&mockSq))); + MOCKER_CPP(&RtsqBase::QuerySqStatusByType).stubs().with(any()).will(returnValue(static_cast(0))); + MOCKER_CPP(&RtsqBase::ConfigSqStatusByType).stubs(); + std::cout << "A Test case in UbTransportLiteImplWaitWithTimeoutTest SetUp" << std::endl; + } + + virtual void TearDown() + { + std::cout << "A Test case in UbTransportLiteImplWaitWithTimeoutTest TearDown" << std::endl; + GlobalMockObject::verify(); + } + + std::vector GetNotifyUniqueId(u32 notifyId, u32 devPhyId) + { + BinaryStream binaryStream; + binaryStream << notifyId; + binaryStream << devPhyId; + std::vector result; + binaryStream.Dump(result); + return result; + } + + std::vector GetRmtBufferUniqueId(u64 addr, u64 size, u32 tokenId, u32 tokenValue) + { + BinaryStream binaryStream; + binaryStream << addr; + binaryStream << size; + binaryStream << tokenId; + binaryStream << tokenValue; + std::vector result; + binaryStream.Dump(result); + return result; + } + + std::vector GetConnUniqueId() + { + u32 dieId = 0; + u32 funcId = 0; + u32 jettyId = 0; + u32 jfcPollMode = 0; + bool dwqeCacheLocked = false; + u64 dbAddr = 0x100; + u64 sqVa = 0x100; + u32 sqDepth = 100; + u32 tpn = 100; + u32 cqId = 1; + u32 connVector = 1; + u32 protocolType = 1; + u32 roceVersion = 2; + u32 mtu = 1; + BinaryStream binaryStream; + binaryStream << dieId; + binaryStream << funcId; + binaryStream << jettyId; + binaryStream << jfcPollMode; + binaryStream << dwqeCacheLocked; + binaryStream << dbAddr; + binaryStream << sqVa; + binaryStream << sqDepth; + binaryStream << tpn; + binaryStream << cqId; + binaryStream << connVector; + binaryStream << protocolType; + binaryStream << roceVersion; + binaryStream << mtu; + std::vector result; + binaryStream.Dump(result); + return result; + } + + std::vector BuildUbTransportLiteUniqueId() + { + auto locNotify0 = GetNotifyUniqueId(1, 1); + auto locNotify1 = GetNotifyUniqueId(2, 2); + auto rmtNotify0 = GetRmtBufferUniqueId(1, 1, 1, 1); + auto rmtNotify1 = GetRmtBufferUniqueId(2, 2, 2, 2); + + u32 notifyNum = 2; + + auto rmtBuffer0 = GetRmtBufferUniqueId(300, 200, 3, 3); + auto rmtBuffer1 = GetRmtBufferUniqueId(300, 200, 4, 4); + u32 buffeNum = 2; + + auto conn0 = GetConnUniqueId(); + u32 connNum = 1; + + BinaryStream binaryStream; + + u32 type = (u32)TransportType::UB; + binaryStream << type; + binaryStream << notifyNum; + binaryStream << buffeNum; + binaryStream << connNum; + + std::vector data0; + data0.insert(data0.end(), locNotify0.begin(), locNotify0.end()); + data0.insert(data0.end(), locNotify1.begin(), locNotify1.end()); + binaryStream << data0; + + std::vector data1; + data1.insert(data1.end(), rmtNotify0.begin(), rmtNotify0.end()); + data1.insert(data1.end(), rmtNotify1.begin(), rmtNotify1.end()); + binaryStream << data1; + + std::vector data2; + data2.insert(data2.end(), rmtBuffer0.begin(), rmtBuffer0.end()); + data2.insert(data2.end(), rmtBuffer1.begin(), rmtBuffer1.end()); + binaryStream << data2; + + std::vector data4; + data4.insert(data4.end(), conn0.begin(), conn0.end()); + binaryStream << data4; + + std::vector liteData; + binaryStream.Dump(liteData); + return liteData; + } + + u8 mockSq[64]; +}; + +TEST_F(UbTransportLiteImplWaitWithTimeoutTest, UbTransportLiteImpl_WaitWithTimeout_Normal_Success) +{ + std::vector liteData = BuildUbTransportLiteUniqueId(); + + RmaConnLite rmaConnLite; + RmaConnLite *connLite = &rmaConnLite; + MOCKER_CPP(&UbConnLiteMgr::Get).stubs().will(returnValue(connLite)); + MOCKER_CPP(&MirrorTaskManager::AddTaskInfo).stubs().with(any()); + LinkData linkData(BasePortType(PortDeploymentType::DEV_NET, ConnectProtoType::UB), 0, 1, 0, 1); + MirrorTaskManager mirrorTaskMgr(0, &GlobalMirrorTasks::Instance(), true); + auto transportCallback = MemTransportCallback(linkData, mirrorTaskMgr); + MemTransportLite transportLite(liteData, transportCallback); + auto &ubTransportLite = *(dynamic_cast(transportLite.impl.get())); + + u32 fakeStreamId = 1; + u32 fakeSqId = 1; + u32 fakedevPhyId = 1; + BinaryStream liteBinaryStream; + liteBinaryStream << fakeStreamId; + liteBinaryStream << fakeSqId; + liteBinaryStream << fakedevPhyId; + std::vector uniqueId{}; + liteBinaryStream.Dump(uniqueId); + + StreamLite stream(uniqueId); + RtsqA5 rtsq(fakedevPhyId, fakeStreamId, fakeSqId); + stream.rtsq = std::make_unique(rtsq); + + u32 timeout = 1800; + EXPECT_NO_THROW(ubTransportLite.WaitWithTimeout(0, stream, timeout)); +} + +TEST_F(UbTransportLiteImplWaitWithTimeoutTest, UbTransportLiteImpl_WaitWithTimeout_ZeroTimeout_Success) +{ + std::vector liteData = BuildUbTransportLiteUniqueId(); + + RmaConnLite rmaConnLite; + RmaConnLite *connLite = &rmaConnLite; + MOCKER_CPP(&UbConnLiteMgr::Get).stubs().will(returnValue(connLite)); + MOCKER_CPP(&MirrorTaskManager::AddTaskInfo).stubs().with(any()); + LinkData linkData(BasePortType(PortDeploymentType::DEV_NET, ConnectProtoType::UB), 0, 1, 0, 1); + MirrorTaskManager mirrorTaskMgr(0, &GlobalMirrorTasks::Instance(), true); + auto transportCallback = MemTransportCallback(linkData, mirrorTaskMgr); + MemTransportLite transportLite(liteData, transportCallback); + auto &ubTransportLite = *(dynamic_cast(transportLite.impl.get())); + + u32 fakeStreamId = 1; + u32 fakeSqId = 1; + u32 fakedevPhyId = 1; + BinaryStream liteBinaryStream; + liteBinaryStream << fakeStreamId; + liteBinaryStream << fakeSqId; + liteBinaryStream << fakedevPhyId; + std::vector uniqueId{}; + liteBinaryStream.Dump(uniqueId); + + StreamLite stream(uniqueId); + RtsqA5 rtsq(fakedevPhyId, fakeStreamId, fakeSqId); + stream.rtsq = std::make_unique(rtsq); + + u32 timeout = 0; + EXPECT_NO_THROW(ubTransportLite.WaitWithTimeout(0, stream, timeout)); +} + +TEST_F(UbTransportLiteImplWaitWithTimeoutTest, UbTransportLiteImpl_WaitWithTimeout_MaxTimeout_Success) +{ + std::vector liteData = BuildUbTransportLiteUniqueId(); + + RmaConnLite rmaConnLite; + RmaConnLite *connLite = &rmaConnLite; + MOCKER_CPP(&UbConnLiteMgr::Get).stubs().will(returnValue(connLite)); + MOCKER_CPP(&MirrorTaskManager::AddTaskInfo).stubs().with(any()); + LinkData linkData(BasePortType(PortDeploymentType::DEV_NET, ConnectProtoType::UB), 0, 1, 0, 1); + MirrorTaskManager mirrorTaskMgr(0, &GlobalMirrorTasks::Instance(), true); + auto transportCallback = MemTransportCallback(linkData, mirrorTaskMgr); + MemTransportLite transportLite(liteData, transportCallback); + auto &ubTransportLite = *(dynamic_cast(transportLite.impl.get())); + + u32 fakeStreamId = 1; + u32 fakeSqId = 1; + u32 fakedevPhyId = 1; + BinaryStream liteBinaryStream; + liteBinaryStream << fakeStreamId; + liteBinaryStream << fakeSqId; + liteBinaryStream << fakedevPhyId; + std::vector uniqueId{}; + liteBinaryStream.Dump(uniqueId); + + StreamLite stream(uniqueId); + RtsqA5 rtsq(fakedevPhyId, fakeStreamId, fakeSqId); + stream.rtsq = std::make_unique(rtsq); + + u32 timeout = 0xFFFFFFFF; + EXPECT_NO_THROW(ubTransportLite.WaitWithTimeout(0, stream, timeout)); +}

hcomm

hcomm