1 files changed, 69 insertions, 0 deletions
diff --git a/protocols/Telegram/tdlib/td/tdutils/test/WaitFreeVector.cpp b/protocols/Telegram/tdlib/td/tdutils/test/WaitFreeVector.cpp
new file mode 100644
index 0000000000..0f0cc58796
--- /dev/null
+++ b/protocols/Telegram/tdlib/td/tdutils/test/WaitFreeVector.cpp
@@ -0,0 +1,69 @@
+//
+// 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);
+  }
+}