tdlight/tdutils/test/WaitFreeVector.cpp
2022-08-02 14:27:18 +03:00

70 lines
1.9 KiB
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2022
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#include "td/utils/common.h"
#include "td/utils/Random.h"
#include "td/utils/tests.h"
#include "td/utils/WaitFreeVector.h"
TEST(WaitFreeVector, stress_test) {
td::Random::Xorshift128plus rnd(123);
td::vector<td::uint64> reference;
td::WaitFreeVector<td::uint64> vector;
td::vector<td::RandomSteps::Step> steps;
auto add_step = [&](td::uint32 weight, auto f) {
steps.emplace_back(td::RandomSteps::Step{std::move(f), weight});
};
auto gen_key = [&] {
return static_cast<size_t>(rnd() % reference.size());
};
add_step(2000, [&] {
ASSERT_EQ(reference.size(), vector.size());
ASSERT_EQ(reference.empty(), vector.empty());
if (reference.empty()) {
return;
}
auto key = gen_key();
ASSERT_EQ(reference[key], vector[key]);
auto value = rnd();
reference[key] = value;
vector[key] = value;
ASSERT_EQ(reference[key], vector[key]);
});
add_step(2000, [&] {
ASSERT_EQ(reference.size(), vector.size());
ASSERT_EQ(reference.empty(), vector.empty());
auto value = rnd();
reference.emplace_back(value);
if (rnd() & 1) {
vector.emplace_back(value);
} else if (rnd() & 1) {
vector.push_back(value);
} else {
vector.push_back(std::move(value));
}
ASSERT_EQ(reference.back(), vector.back());
});
add_step(500, [&] {
ASSERT_EQ(reference.size(), vector.size());
ASSERT_EQ(reference.empty(), vector.empty());
if (reference.empty()) {
return;
}
reference.pop_back();
vector.pop_back();
});
td::RandomSteps runner(std::move(steps));
for (size_t i = 0; i < 1000000; i++) {
runner.step(rnd);
}
}