intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 45 insertions, 0 deletions

diff --git a/library/cpp/messagebus/cc_semaphore_ut.cpp b/library/cpp/messagebus/cc_semaphore_ut.cpp
new file mode 100644
index 0000000000..206bb7c96a
--- /dev/null
+++ b/library/cpp/messagebus/cc_semaphore_ut.cpp
@@ -0,0 +1,45 @@
+#include <library/cpp/testing/unittest/registar.h>
+
+#include "cc_semaphore.h"
+
+#include <util/system/atomic.h>
+
+namespace {
+    struct TTestSemaphore: public TComplexConditionSemaphore<TTestSemaphore> {
+        TAtomic Current;
+
+        TTestSemaphore()
+            : Current(0)
+        {
+        }
+
+        bool TryWait() {
+            return AtomicGet(Current) > 0;
+        }
+
+        void Aquire() {
+            Wait();
+            AtomicDecrement(Current);
+        }
+
+        void Release() {
+            AtomicIncrement(Current);
+            Updated();
+        }
+    };
+}
+
+Y_UNIT_TEST_SUITE(TComplexConditionSemaphore) {
+    Y_UNIT_TEST(Simple) {
+        TTestSemaphore sema;
+        UNIT_ASSERT(!sema.TryWait());
+        sema.Release();
+        UNIT_ASSERT(sema.TryWait());
+        sema.Release();
+        UNIT_ASSERT(sema.TryWait());
+        sema.Aquire();
+        UNIT_ASSERT(sema.TryWait());
+        sema.Aquire();
+        UNIT_ASSERT(!sema.TryWait());
+    }
+}