Repo created

libs/base/base_tests/cancellable_tests.cpp

@@ -0,0 +1,83 @@
+#include "testing/testing.hpp"
+
+#include "base/cancellable.hpp"
+#include "base/math.hpp"
+#include "base/thread.hpp"
+
+#include <chrono>
+#include <cmath>
+#include <future>
+
+using namespace std;
+
+namespace base
+{
+UNIT_TEST(Cancellable_Smoke)
+{
+  Cancellable cancellable;
+
+  promise<void> syncPromise;
+  auto syncFuture = syncPromise.get_future();
+
+  double x = 0.123;
+
+  auto const fn = [&]
+  {
+    for (size_t it = 0;; it++)
+    {
+      if (it > 100 && cancellable.IsCancelled())
+        break;
+
+      x = cos(x);
+    }
+
+    syncPromise.set_value();
+  };
+
+  threads::SimpleThread thread(fn);
+  cancellable.Cancel();
+  syncFuture.wait();
+  thread.join();
+
+  TEST(cancellable.IsCancelled(), ());
+  TEST_EQUAL(cancellable.CancellationStatus(), Cancellable::Status::CancelCalled, ());
+  TEST(AlmostEqualAbs(x, 0.739, 1e-3), ());
+}
+
+UNIT_TEST(Cancellable_Deadline)
+{
+  Cancellable cancellable;
+  chrono::steady_clock::duration kTimeout = chrono::milliseconds(20);
+  cancellable.SetDeadline(chrono::steady_clock::now() + kTimeout);
+
+  promise<void> syncPromise;
+  auto syncFuture = syncPromise.get_future();
+
+  double x = 0.123;
+
+  auto const fn = [&]
+  {
+    while (true)
+    {
+      if (cancellable.IsCancelled())
+        break;
+
+      x = cos(x);
+    }
+
+    syncPromise.set_value();
+  };
+
+  threads::SimpleThread thread(fn);
+  syncFuture.wait();
+  thread.join();
+
+  TEST(cancellable.IsCancelled(), ());
+  TEST_EQUAL(cancellable.CancellationStatus(), Cancellable::Status::DeadlineExceeded, ());
+  TEST(AlmostEqualAbs(x, 0.739, 1e-3), ());
+
+  cancellable.Cancel();
+  TEST(cancellable.IsCancelled(), ());
+  TEST_EQUAL(cancellable.CancellationStatus(), Cancellable::Status::CancelCalled, ());
+}
+}  // namespace base