Add a test for acquire/release

Shopify · Oct 16, 2024 · 7f911a6 · 7f911a6
1 parent 126fdf7
commit 7f911a6
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/test/resource_test.rb b/test/resource_test.rb
@@ -140,6 +140,22 @@ def test_acquire
     assert(acquired)
   end
 
+  def test_acquire_and_release_semaphore
+    resource = create_resource(:testing, tickets: 1)
+
+    assert_equal(1, resource.count)
+
+    begin
+      resource.acquire_semaphore
+
+      assert_equal(0, resource.count)
+    ensure
+      resource.release_semaphore
+
+      assert_equal(1, resource.count)
+    end
+  end
+
   def test_acquire_return_val
     resource = create_resource(:testing, tickets: 1)
     val = resource.acquire { 1234 }