mirror of
https://github.com/deepseek-ai/3FS
synced 2025-06-26 18:16:45 +00:00
82 lines
2.5 KiB
C++
82 lines
2.5 KiB
C++
#include <algorithm>
|
|
#include <fmt/core.h>
|
|
#include <folly/Random.h>
|
|
#include <gtest/gtest.h>
|
|
#include <vector>
|
|
|
|
#include "common/utils/FaultInjection.h"
|
|
#include "common/utils/StatusCode.h"
|
|
#include "meta/store/MetaStore.h"
|
|
#include "tests/meta/MetaTestBase.h"
|
|
|
|
namespace hf3fs::meta::server {
|
|
|
|
template <typename KV>
|
|
class TestList : public MetaTestBase<KV> {};
|
|
|
|
using KVTypes = ::testing::Types<mem::MemKV, fdb::DB>;
|
|
TYPED_TEST_SUITE(TestList, KVTypes);
|
|
|
|
TYPED_TEST(TestList, FaultInjection) {
|
|
folly::coro::blockingWait([&]() -> CoTask<void> {
|
|
auto cluster = this->createMockCluster();
|
|
auto &meta = cluster.meta().getOperator();
|
|
|
|
std::vector<std::string> paths;
|
|
size_t numFiles = folly::Random::rand32(0, 4096);
|
|
for (size_t i = 0; i < numFiles; i++) {
|
|
std::string path = std::to_string(i);
|
|
auto result = co_await meta.create({SUPER_USER, path, {}, O_RDONLY, p644});
|
|
CO_ASSERT_OK(result);
|
|
paths.push_back(path);
|
|
}
|
|
std::sort(paths.begin(), paths.end());
|
|
|
|
for (size_t i = 0; i < 10; i++) {
|
|
FAULT_INJECTION_SET(10, 3); // 10%, 3 faults
|
|
std::vector<DirEntry> entries;
|
|
bool hasMore = true;
|
|
const bool needStatus = folly::Random::oneIn(2);
|
|
String prev;
|
|
|
|
// do load
|
|
while (hasMore) {
|
|
auto limit = folly::Random::rand32(0, 512);
|
|
auto req = i % 2 == 0 ? ListReq(SUPER_USER, Path("/"), prev, (int)limit, needStatus)
|
|
: ListReq(SUPER_USER, InodeId::root(), prev, (int)limit, needStatus);
|
|
auto list = co_await meta.list(req);
|
|
CO_ASSERT_OK(list);
|
|
if (needStatus) {
|
|
CO_ASSERT_TRUE(list->inodes.size() == list->entries.size());
|
|
}
|
|
for (size_t i = 0; i < list->entries.size(); i++) {
|
|
if (needStatus) {
|
|
auto entry = list->entries.at(i);
|
|
auto inode = list->inodes.at(i);
|
|
entries.push_back(entry);
|
|
CO_ASSERT_EQ(entry.id, inode.id);
|
|
} else {
|
|
entries.push_back(list->entries.at(i));
|
|
}
|
|
}
|
|
|
|
hasMore = list->more;
|
|
if (list->entries.size() == 0) {
|
|
CO_ASSERT_FALSE(hasMore);
|
|
}
|
|
if (hasMore) {
|
|
prev = list->entries.at(list->entries.size() - 1).name;
|
|
}
|
|
}
|
|
|
|
// check
|
|
CO_ASSERT_EQ(entries.size(), numFiles);
|
|
for (size_t i = 0; i < numFiles; i++) {
|
|
auto entry = entries[i];
|
|
CO_ASSERT_EQ(entry.name, paths[i]);
|
|
}
|
|
}
|
|
}());
|
|
}
|
|
} // namespace hf3fs::meta::server
|