[Fix] Adapt ASU tests to BufferManager handles

Author: Fengli5355

ucm/transport/kv/asu/test/case/connection_manager_test.cc

@@ -64,9 +64,15 @@ class StubTransProvider : public TransProvider {
     {
         return {};
     }
-    Status RegisterMemory(ConnectionHandle, const std::vector<RegisterMemoryDesc>&,
-                          std::vector<MemHandle>&) override
+    Status RegisterMemory(ConnectionHandle, const std::vector<RegisterMemoryDesc>& descs,
+                          std::vector<MemHandle>& handles) override
     {
+        handles.clear();
+        handles.reserve(descs.size());
+        for (std::size_t index = 0; index < descs.size(); ++index) {
+            handles.push_back(reinterpret_cast<MemHandle>(static_cast<std::uintptr_t>(index) +
+                                                          static_cast<std::uintptr_t>(1)));
+        }
         return Status::OK();
     }
     std::vector<Status> UnregisterMemory(const std::vector<UnregisterMemoryDesc>&) override
@@ -78,7 +84,11 @@ class StubTransProvider : public TransProvider {
         return Status::OK();
     }
     std::vector<Status> FreeThread(const std::vector<ThreadHandle>&) override { return {}; }
-    Status GetMemTokenId(MemHandle, uint32_t&) override { return Status::OK(); }
+    Status GetMemTokenId(MemHandle, uint32_t& tokenId) override
+    {
+        tokenId = 1;
+        return Status::OK();
+    }
 };
 
 AsuEndpoint MakeEndpoint(const std::string& ip = "10.0.0.1")

ucm/transport/kv/asu/trans/test/asu_submit_flow_test.cpp

@@ -64,9 +64,15 @@ class StubTransProvider : public TransProvider {
         return std::vector<Status>(ioBatches.size(), Status::OK());
     }
 
-    Status RegisterMemory(ConnectionHandle, const std::vector<RegisterMemoryDesc>&,
-                          std::vector<MemHandle>&) override
+    Status RegisterMemory(ConnectionHandle, const std::vector<RegisterMemoryDesc>& descs,
+                          std::vector<MemHandle>& handles) override
     {
+        handles.clear();
+        handles.reserve(descs.size());
+        for (std::size_t index = 0; index < descs.size(); ++index) {
+            handles.push_back(reinterpret_cast<MemHandle>(static_cast<std::uintptr_t>(index) +
+                                                          static_cast<std::uintptr_t>(1)));
+        }
         return Status::OK();
     }
 
@@ -82,7 +88,11 @@ class StubTransProvider : public TransProvider {
 
     std::vector<Status> FreeThread(const std::vector<ThreadHandle>&) override { return {}; }
 
-    Status GetMemTokenId(MemHandle, uint32_t&) override { return Status::OK(); }
+    Status GetMemTokenId(MemHandle, uint32_t& tokenId) override
+    {
+        tokenId = 1;
+        return Status::OK();
+    }
 };
 
 class AsuSubmitFlowBufferTest : public ::testing::Test {
@@ -290,4 +300,4 @@ TEST(AsuSubmitFlowTest, SendSubBatchBuffersFailsAllSentSubBatchesWhenStatusCount
 }
 
 }  // namespace
-}  // namespace UC::ASU
+}  // namespace UC::ASU

ucm/transport/kv/asu/trans/test/transport_task_completion_test.cpp

@@ -23,8 +23,10 @@
  * */
 #include <acl/acl.h>
 #include <atomic>
+#include <cstddef>
 #include <cstdint>
 #include <gtest/gtest.h>
+#include <vector>
 #define private public
 #include "asu_transport_impl.h"
 #undef private
@@ -56,9 +58,15 @@ class StubTransProvider : public TransProvider {
     {
         return {};
     }
-    Status RegisterMemory(ConnectionHandle, const std::vector<RegisterMemoryDesc>&,
-                          std::vector<MemHandle>&) override
+    Status RegisterMemory(ConnectionHandle, const std::vector<RegisterMemoryDesc>& descs,
+                          std::vector<MemHandle>& handles) override
     {
+        handles.clear();
+        handles.reserve(descs.size());
+        for (std::size_t index = 0; index < descs.size(); ++index) {
+            handles.push_back(reinterpret_cast<MemHandle>(static_cast<std::uintptr_t>(index) +
+                                                          static_cast<std::uintptr_t>(1)));
+        }
         return Status::OK();
     }
     std::vector<Status> UnregisterMemory(const std::vector<UnregisterMemoryDesc>&) override
@@ -70,7 +78,11 @@ class StubTransProvider : public TransProvider {
         return Status::OK();
     }
     std::vector<Status> FreeThread(const std::vector<ThreadHandle>&) override { return {}; }
-    Status GetMemTokenId(MemHandle, uint32_t&) override { return Status::OK(); }
+    Status GetMemTokenId(MemHandle, uint32_t& tokenId) override
+    {
+        tokenId = 1;
+        return Status::OK();
+    }
 };
 
 class TransportTaskCompletionTest : public ::testing::Test {