core/semaphore: Add unit test for semaphore_units<>::split()

Signed-off-by: Duarte Nunes <[email protected]>
donovan68 · Oct 7, 2018 · fddd9d6 · fddd9d6
1 parent 310bd99
commit fddd9d6
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/semaphore_test.cc b/tests/semaphore_test.cc
@@ -230,3 +230,19 @@ SEASTAR_TEST_CASE(test_with_semaphore) {
         });
     });
 }
+
+SEASTAR_THREAD_TEST_CASE(test_semaphore_units_splitting) {
+    auto sm = semaphore(2);
+    auto units = get_units(sm, 2, 1min).get0();
+    {
+        BOOST_REQUIRE_EQUAL(sm.available_units(), 0);
+        auto split = units.split(1);
+        BOOST_REQUIRE_EQUAL(sm.available_units(), 0);
+    }
+    BOOST_REQUIRE_EQUAL(sm.available_units(), 1);
+    units.~semaphore_units();
+    units = get_units(sm, 2, 1min).get0();
+    BOOST_REQUIRE_EQUAL(sm.available_units(), 0);
+    BOOST_REQUIRE_THROW(units.split(10), std::invalid_argument);
+    BOOST_REQUIRE_EQUAL(sm.available_units(), 0);
+}