chore: use condition variable to shutdown websocket pools (#5721)

2024-11-20 22:29:47 +01:00
parent 19f449866e
commit 1c827f6288
9 changed files with 237 additions and 30 deletions
@@ -0,0 +1,88 @@
+#include "util/OnceFlag.hpp"
+
+#include "Test.hpp"
+
+#include <thread>
+
+using namespace chatterino;
+
+// this test shouldn't time out (no assert necessary)
+TEST(OnceFlag, basic)
+{
+    OnceFlag startedFlag;
+    OnceFlag startedAckFlag;
+    OnceFlag stoppedFlag;
+
+    std::thread t([&] {
+        startedFlag.set();
+        startedAckFlag.wait();
+        std::this_thread::sleep_for(std::chrono::milliseconds{50});
+        stoppedFlag.set();
+    });
+
+    startedFlag.wait();
+    startedAckFlag.set();
+    stoppedFlag.wait();
+
+    t.join();
+}
+
+TEST(OnceFlag, waitFor)
+{
+    OnceFlag startedFlag;
+    OnceFlag startedAckFlag;
+    OnceFlag stoppedFlag;
+
+    std::thread t([&] {
+        startedFlag.set();
+        startedAckFlag.wait();
+
+        std::this_thread::sleep_for(std::chrono::milliseconds{100});
+        stoppedFlag.set();
+    });
+
+    startedFlag.wait();
+    startedAckFlag.set();
+
+    auto start = std::chrono::system_clock::now();
+    ASSERT_TRUE(stoppedFlag.waitFor(std::chrono::milliseconds{200}));
+    auto stop = std::chrono::system_clock::now();
+
+    ASSERT_LT(stop - start, std::chrono::milliseconds{150});
+
+    start = std::chrono::system_clock::now();
+    ASSERT_TRUE(stoppedFlag.waitFor(std::chrono::milliseconds{1000}));
+    stop = std::chrono::system_clock::now();
+
+    ASSERT_LT(stop - start, std::chrono::milliseconds{10});
+
+    start = std::chrono::system_clock::now();
+    stoppedFlag.wait();
+    stop = std::chrono::system_clock::now();
+
+    ASSERT_LT(stop - start, std::chrono::milliseconds{10});
+
+    t.join();
+}
+
+TEST(OnceFlag, waitForTimeout)
+{
+    OnceFlag startedFlag;
+    OnceFlag startedAckFlag;
+    OnceFlag stoppedFlag;
+
+    std::thread t([&] {
+        startedFlag.set();
+        startedAckFlag.wait();
+        std::this_thread::sleep_for(std::chrono::milliseconds{100});
+        stoppedFlag.set();
+    });
+
+    startedFlag.wait();
+    startedAckFlag.set();
+
+    ASSERT_FALSE(stoppedFlag.waitFor(std::chrono::milliseconds{25}));
+    stoppedFlag.wait();
+
+    t.join();
+}