Test bucket scheduler with timers.

chadrem · Jul 18, 2015 · 84428e9 · 84428e9
1 parent eeabc74
commit 84428e9
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 3 deletions.
diff --git a/test/periodic_timer_test.rb b/test/periodic_timer_test.rb
@@ -1,11 +1,24 @@
 require 'test_helper'
 
 class PeriodicTimerTest < Minitest::Test
-  def test_basic_usage
+  def test_basic_usage_with_default_scheduler
+    perform_basic_test(Workers.scheduler)
+  end
+
+  def test_basic_usage_with_bucket_scheduler
+    scheduler = Workers::BucketScheduler.new
+    perform_basic_test(scheduler)
+  ensure
+    scheduler.dispose
+  end
+
+  private
+
+  def perform_basic_test(scheduler)
     counterA = 0
     counterB = 0
 
-    timer = Workers::PeriodicTimer.new(0.01) do
+    timer = Workers::PeriodicTimer.new(0.01, :scheduler => scheduler) do
       counterA += 1
     end
 

diff --git a/test/timer_test.rb b/test/timer_test.rb
@@ -1,7 +1,20 @@
 require 'test_helper'
 
 class TimerTest < Minitest::Test
-  def test_basic_usage
+  def test_basic_usage_with_default_scheduler
+    perform_basic_test(Workers.scheduler)
+  end
+
+  def test_basic_usage_with_bucket_scheduler
+    scheduler = Workers::BucketScheduler.new
+    perform_basic_test(scheduler)
+  ensure
+    scheduler.dispose
+  end
+
+  private
+
+  def perform_basic_test(scheduler)
     counterA = 0
     counterB = 0