70 lines
1.9 KiB
C++
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);
|
||
|
|
}
|
||
|
|
}
|