DeepSeek/3FS
1
0
Fork 0
mirror of https://github.com/deepseek-ai/3FS synced 2025-06-26 18:16:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Initial commit

Browse Source
This commit is contained in:
dev
2025-02-27 21:53:53 +08:00
commit 815e55e4c0
1291 changed files with 185445 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

105
tests/common/net/ib/TestIBNotInitialized.cc Normal file
View File

@@ -0,0 +1,105 @@
#include <chrono>
#include <folly/experimental/coro/BlockingWait.h>
#include <folly/experimental/coro/Collect.h>
#include <folly/experimental/coro/GtestHelpers.h>
#include <folly/experimental/coro/Invoke.h>
#include <folly/logging/xlog.h>
#include <gflags/gflags.h>
#include <gtest/gtest.h>
#include <memory>
#include <tuple>
#include <utility>
#include <vector>
#include "common/net/Client.h"
#include "common/net/Listener.h"
#include "common/net/Server.h"
#include "common/net/WriteItem.h"
#include "common/net/ib/IBDevice.h"
#include "common/net/sync/Client.h"
#include "common/serde/ClientContext.h"
#include "common/utils/Address.h"
#include "common/utils/Coroutine.h"
#include "common/utils/Result.h"
#include "gtest/gtest.h"
#include "tests/GtestHelpers.h"
#include "tests/common/net/Echo.h"
namespace hf3fs::net {
using namespace test;
class TestIBNotInitialized : public ::testing::Test {
protected:
void SetUp() override { IBManager::stop(); }
};
TEST_F(TestIBNotInitialized, Basic) {
ASSERT_FALSE(IBManager::initialized());
ASSERT_TRUE(IBDevice::all().empty());
}
class EchoServiceImpl : public serde::ServiceWrapper<EchoServiceImpl, Echo> {
public:
CoTryTask<EchoRsp> echo(serde::CallContext &ctx, const EchoReq &req) {
EchoRsp rsp;
rsp.val = req.val;
co_return rsp;
}
CoTryTask<HelloRsp> hello(serde::CallContext &ctx, const HelloReq &req) {
co_return makeError(StatusCode::kNotImplemented);
}
CoTryTask<HelloRsp> fail(serde::CallContext &ctx, const HelloReq &) {
co_return makeError(StatusCode::kNotImplemented);
}
};
TEST_F(TestIBNotInitialized, TCP) {
Server::Config serverConfig;
serverConfig.groups(0).set_network_type(Address::TCP);
Server server{serverConfig};
ASSERT_TRUE(server.setup());
server.addSerdeService(std::make_unique<EchoServiceImpl>());
ASSERT_TRUE(server.start());
Client::Config clientConfig;
Client client{clientConfig};
client.start();
auto ctx = client.serdeCtx(server.groups().front()->addressList().front());
folly::coro::blockingWait(folly::coro::co_invoke([&]() -> CoTask<void> {
Echo<> client;
EchoReq req;
req.val = "foo";
auto result = co_await client.echo(ctx, req);
CO_ASSERT_OK(result);
CO_ASSERT_EQ(result->val, "foo");
}));
}
TEST_F(TestIBNotInitialized, RDMAServer) {
Server::Config serverConfig;
serverConfig.groups(0).set_network_type(Address::RDMA);
Server server{serverConfig};
ASSERT_OK(server.setup());
ASSERT_ERROR(server.start(), RPCCode::kIBDeviceNotInitialized);
}
TEST_F(TestIBNotInitialized, RDMAClient) {
Client::Config clientConfig;
Client client{clientConfig};
client.start();
auto ctx = client.serdeCtx(Address::fromString("rdma://127.0.0.1:8000"));
folly::coro::blockingWait(folly::coro::co_invoke([&]() -> CoTask<void> {
Echo<> client;
EchoReq req;
req.val = "foo";
auto result = co_await client.echo(ctx, req);
CO_ASSERT_TRUE(result.hasError());
CO_ASSERT_EQ(result.error().code(), RPCCode::kSendFailed);
}));
}
} // namespace hf3fs::net