test/scheduler_5_test.rb in rufus-scheduler-1.0.11 vs test/scheduler_5_test.rb in rufus-scheduler-1.0.12

@@ -5,22 +5,18 @@
 # John Mettraux at openwfe.org
 #
 # Sat Jan 26 20:05:57 JST 2008
 #
 
+$:.unshift(File.dirname(__FILE__) + '/../lib')
+
 require 'test/unit'
 require 'rufus/scheduler'
 
 
 class Scheduler5Test < Test::Unit::TestCase
 
-  #def setup
-  #end
-
-  #def teardown
-  #end
-
   #
   # Testing the :first_at parameter
   #
   def test_0
 
@@ -29,11 +25,11 @@
 
     $count = 0
 
     fa = Time.now + 3
 
-    s.schedule_every "1s", :first_at => fa do
+    s.schedule_every '1s', :first_at => fa do
       $count += 1
     end
 
     sleep 1
 
@@ -58,11 +54,11 @@
     s = Rufus::Scheduler.new
     s.start
 
     $count = 0
 
-    s.schedule_every "1s", :first_in => "3s" do
+    s.schedule_every '1s', :first_in => '3s' do
       $count += 1
     end
 
     sleep 1
 
@@ -73,9 +69,83 @@
     assert_equal 1, $count
 
     sleep 1
 
     assert_equal 2, $count
+
+    s.stop
+  end
+
+  #
+  # Testing the :timeout parameter
+  #
+  def test_2
+
+    s = Rufus::Scheduler.start_new
+
+    $count = 0
+    $error = nil
+
+    def s.log_exception (e)
+      $error = e
+    end
+
+    s.in '3s', :timeout => '2s' do
+      loop do
+        $count += 1
+        sleep 3
+      end
+    end
+
+    sleep 6
+
+    assert_kind_of Rufus::TimeOutError, $error
+    assert_equal 1, $count
+
+    s.stop
+  end
+
+  #
+  # Testing the :first_in + :timeout parameters
+  #
+  def test_3
+
+    s = Rufus::Scheduler.start_new
+
+    $count = 0
+    $error = nil
+    $jobs = nil
+    $status = :out
+
+    def s.log_exception (e)
+      $error = e
+      $status = :timedout
+    end
+
+    s.every '5s', :first_in => '2s', :timeout => '1s' do
+      $status = :in
+      Thread.pass # let the timeout job get scheduled
+      jobs = s.all_jobs
+      $jobs = [ jobs.size, jobs.collect { |j| j.tags }.flatten ]
+      $count += 1
+      sleep 4
+      $status = :out
+    end
+
+    sleep 4 # until after 1st timeout
+
+    assert_kind_of Rufus::TimeOutError, $error
+    assert_equal 1, $count
+    assert_equal [ 2, [ 'timeout' ] ], $jobs
+    assert_equal :timedout, $status
+
+    assert_equal 1, s.all_jobs.size
+
+    sleep 3.5 # right in the middle of the 2nd timeout
+
+    assert_equal :in, $status
+    assert_equal 2, s.all_jobs.size
+    assert_equal [ 'timeout' ], s.all_jobs.collect { |j| j.tags }.flatten
 
     s.stop
   end
 end