continuity 0.0.1

data/test/test_cron_entry.rb

@@ -0,0 +1,191 @@
+require 'helper'
+require 'minitest/autorun'
+require 'continuity'
+
+include Continuity
+
+describe CronEntry do
+  describe "0 * * * * * (every minute)" do
+    before do
+      @ce_short = CronEntry.new("* * * * *")
+      @ce = CronEntry.new("0 * * * * *")
+    end
+
+    it "should run at 2010-12-20 00:00:00" do
+      @ce_short.at?(Time.parse("2010-12-20 00:00:00")).must_equal true
+      @ce.at?(Time.parse("2010-12-20 00:00:00")).must_equal true
+    end
+
+    it "should not run at 2010-12-20 00:00:01" do
+      @ce_short.at?(Time.parse("2010-12-20 00:00:01")).must_equal false
+      @ce.at?(Time.parse("2010-12-20 00:00:01")).must_equal false
+    end
+
+    it "should run 60 times in a one hour period" do
+      count = 0
+      start = Time.parse("2010-12-20 00:00:00").to_i
+      start.upto(start + 3599) do |t|
+        count += 1 if @ce_short.at?(Time.at(t))
+        count += 1 if @ce.at?(Time.at(t))
+      end
+      count.must_equal 120
+    end
+  end
+
+  describe "0 1 * * * * (every hour)" do
+    before do
+      @ce = CronEntry.new("0 1 * * * *")
+    end
+
+    it "should run at 2010-12-20 01:01:00" do
+      @ce.at?(Time.parse("2010-12-20 01:01:00")).must_equal true
+    end
+
+    it "should not run at 2010-12-20 00:01:01" do
+      @ce.at?(Time.parse("2010-12-20 00:01:01")).must_equal false
+    end
+
+    it "should run 24 times in a one day period" do
+      count = 0
+      start = Time.parse("2010-12-20 00:01:00").to_i
+      start.upto(start + 86399) do |t|
+        count += 1 if @ce.at?(Time.at(t))
+      end
+      count.must_equal 24 
+    end
+  end
+
+  describe "0 0 0 1 * * (first of the month)" do
+    before do
+      @ce = CronEntry.new("0 0 0 1 * *")
+    end
+
+    it "should run at 2010-12-01 00:00:00" do
+      @ce.at?(Time.parse("2010-12-01 00:00:00")).must_equal true
+    end
+
+    it "should run at 2010-08-01 00:00:00" do
+      @ce.at?(Time.parse("2010-08-01 00:00:00")).must_equal true
+    end
+
+    it "should not run at 2010-12-02 00:00:00" do
+      @ce.at?(Time.parse("2010-12-02 00:00:00")).must_equal false
+    end
+
+    it "should not run at 2010-12-01 01:00:00" do
+      @ce.at?(Time.parse("2010-12-01 01:00:00")).must_equal false
+    end
+
+    it "should run 12 times in a year" do
+      count = 0
+      start = Time.parse("2010-01-01 00:00:00").to_i
+      year_end = Time.parse("2010-12-31 23:59:59").to_i
+      start.step(year_end, 3600) do |t|
+        if @ce.at?(Time.at(t))
+          count += 1 
+        end
+      end
+      count.must_equal 12 
+    end
+  end
+
+  describe "0 0 0 1 1,4,7,10 * (quarterly months)" do
+    before do
+      @ce = CronEntry.new("0 0 0 1 1,4,7,10 *")
+    end
+
+    it "should run at 2010-04-01 00:00:00" do
+      @ce.at?(Time.parse("2010-04-01 00:00:00")).must_equal true
+    end
+
+    it "should not run at 2010-03-01 00:00:00" do
+      @ce.at?(Time.parse("2010-03-01 00:00:00")).must_equal false
+    end
+
+    it "should run 4 times in a year" do
+      count = 0
+      start = Time.parse("2010-01-01 00:00:00").to_i
+      year_end = Time.parse("2010-12-31 23:59:59").to_i
+      start.step(year_end, 3600) do |t|
+        if @ce.at?(Time.at(t))
+          count += 1 
+        end
+      end
+      count.must_equal 4
+    end
+  end
+
+  describe "0 0 7 * * 0 (sunday @ 7am)" do
+    before do
+      @ce = CronEntry.new("0 0 7 * * 0")
+    end
+
+    it "should run at 2010-12-19 07:00:00" do
+      @ce.at?(Time.parse("2010-12-19 07:00:00")).must_equal true
+    end
+
+    it "should run at 2010-12-26 07:00:00" do
+      @ce.at?(Time.parse("2010-12-26 07:00:00")).must_equal true
+    end
+
+    it "should not run at 2010-12-27 07:00:00" do
+      @ce.at?(Time.parse("2010-12-27 07:00:00")).must_equal false
+    end
+
+    it "should not run at 2010-12-26 06:00:00" do
+      @ce.at?(Time.parse("2010-12-26 06:00:00")).must_equal false
+    end
+  end
+
+  describe "ranges w/intervals" do
+    before do
+      # 10 * 20 * 3 = 600 times in a day
+      @ce = CronEntry.new("0-9/2,20-39/4 40-59/1 2,4,8 * * *")
+    end
+
+    it "should run at 2010-12-20 04:40:06" do
+      @ce.at?(Time.parse("2010-12-20 04:40:06")).must_equal true
+    end
+
+    it "should run at 2010-12-20 02:50:24" do
+      @ce.at?(Time.parse("2010-12-20 02:50:24")).must_equal true
+    end
+
+    it "should not run at 2010-12-20 03:50:22" do
+      @ce.at?(Time.parse("2010-12-20 03:50:22")).must_equal false
+    end
+
+    it "should run 600 times in a day" do
+      count = 0
+      start = Time.parse("2010-12-20 02:40:09").to_i
+      start.upto(start + 86399) do |t|
+        if @ce.at?(Time.at(t))
+          count += 1
+        end
+      end
+      count.must_equal 600
+    end
+  end
+
+  describe "invalid intervals" do
+    it "should raise CronFormatError on '1 2 3'" do
+      lambda { CronEntry.new("1 2 3") }.must_raise CronFormatError
+    end
+
+    it "should raise CronFormatError on '1 2 3 4 5 6 7'" do
+      lambda { CronEntry.new("1 2 3 4 5 6 7") }.must_raise CronFormatError
+    end
+
+    it "should raise CronFormatError on 'im not a cron task'" do
+      lambda { CronEntry.new("im not a cron task") }.must_raise CronFormatError
+    end
+
+    it "should raise CronFormatError on 'SrslyFake'" do
+      lambda { CronEntry.new("SrslyFake") }.must_raise CronFormatError
+    end
+
+    it "should raise CronFormatError on '0 0 0 0 0 0'" do
+      lambda { CronEntry.new("0 0 0 0 0 0") }.must_raise CronFormatError
+    end
+  end
+end

data/test/test_periodic_entry.rb

@@ -0,0 +1,109 @@
+require 'helper'
+require 'minitest/autorun'
+require 'continuity'
+
+include Continuity
+
+describe PeriodicEntry do
+  describe "10s" do
+    before do
+      @pe = PeriodicEntry.new("10s")
+    end
+
+    it "should trigger at Time.at(10)" do
+      assert @pe.at?(Time.at(10))
+    end
+
+    it "should trigger 6 times in a minute" do
+      times = 0
+
+      time = Time.parse("2010-12-21 00:01:00").to_i
+      time.upto(time + 59) do |n|
+        times += 1 if @pe.at?(Time.at(n))
+      end
+
+      times.must_equal 6
+    end
+  end
+
+  describe "1m" do
+    before do
+      @pe = PeriodicEntry.new("1m")
+    end
+
+    it "should trigger on the minute" do
+      time = Time.parse("2010-12-21 00:01:00").to_i
+
+      assert @pe.at?(time)
+    end
+
+    it "should trigger 60 times in a hour" do
+      times = 0
+
+      time = Time.parse("2010-12-21 00:00:00").to_i
+      time.upto(time + 3599) do |n|
+        times += 1 if @pe.at?(Time.at(n))
+      end
+
+      times.must_equal 60
+    end
+  end
+
+  describe "2h" do
+    before do
+      @pe = PeriodicEntry.new("2h")
+    end
+
+    it "should trigger at 2am" do
+      time = Time.parse("2010-12-21 02:00:00").to_i
+
+      assert @pe.at?(time)
+    end
+
+    it "should trigger 12 times in a 24hour period" do
+      times = 0
+
+      time = Time.parse("2010-12-21 00:00:00").to_i
+      time.upto(time + 86399) do |n|
+        times += 1 if @pe.at?(Time.at(n))
+      end
+
+      times.must_equal 12
+    end
+  end
+
+  describe "2d" do
+    before do
+      @pe = PeriodicEntry.new("2d")
+    end
+
+    it "should trigger 5 times in a 10day period" do
+      times = 0
+
+      time = Time.parse("2010-12-21 00:00:00").to_i
+      time.upto(time + (86400 * 10) - 1) do |n|
+        times += 1 if @pe.at?(Time.at(n))
+      end
+
+      times.must_equal 5 
+    end
+  end
+
+  describe "1w" do
+    before do
+      @pe = PeriodicEntry.new("1w")
+    end
+
+    it "should trigger 4 times in 28 days" do
+      times = 0
+
+      time = Time.parse("2010-12-01 00:00:00").to_i
+      time_end = Time.parse("2010-12-27 23:59:59").to_i
+      time.upto(time_end) do |n|
+        times += 1 if @pe.at?(Time.at(n))
+      end
+
+      times.must_equal 4 
+    end
+  end
+end

data/test/test_race_issues.rb

@@ -0,0 +1,62 @@
+require 'helper'
+require 'minitest/autorun'
+require 'continuity'
+
+include Continuity
+
+def build_scheduler
+  redis = Redis.new(:port => 16379)
+  scheduler = Continuity::Scheduler.new_using_redis(redis)
+  scheduler.cron("*/10 * * * * *") do
+  end
+
+  scheduler.every("1m") do
+  end
+
+  scheduler
+end
+
+WORKER_COUNT = 100
+MUTEX        = Mutex.new
+TEST_LENGTH  = 60
+
+describe "simulation" do
+  before do
+    redis = Redis.new(:port => 16379)
+    redis.flushall
+  end
+
+  it "should schedule a continuous range" do
+    @workers = []
+    last_scheduled = nil
+
+    WORKER_COUNT.times do
+
+      @workers << Thread.new do
+        s = build_scheduler
+        
+        loop do
+          sleep rand(5)
+          range = s.maybe_schedule
+
+          if range
+            MUTEX.synchronize do
+              if last_scheduled.nil?
+                last_scheduled = range.last
+              else
+                assert_equal (last_scheduled + 1), range.first
+                last_scheduled = range.last
+              end
+            end
+          end
+        end
+
+      end
+
+    end
+
+    sleep TEST_LENGTH
+    @workers.each { |w| w.terminate }
+    assert last_scheduled > Time.now.to_i - 20 
+  end
+end

data/test/test_redis_backend.rb

@@ -0,0 +1,146 @@
+require 'helper'
+require 'minitest/autorun'
+require 'continuity'
+
+describe Continuity::RedisBackend do
+  before do
+    @rb = Continuity::RedisBackend.new(redis_clean, 10, 30)
+  end
+
+  describe "bootstrapping" do
+    it "should set current time as the last scheduled at time" do
+      now = Time.now.to_i
+      @rb.lock_for_scheduling(now) {}
+      last_scheduled_at = @rb.lock_for_scheduling(now+1) {}
+      last_scheduled_at.must_equal now
+    end
+
+    it "should yield" do
+      yielded = false
+
+      now = Time.now.to_i
+      @rb.lock_for_scheduling(now) { yielded = true }
+
+      yielded.must_equal true
+    end
+  end
+
+  describe "trying to lock before scheduling period is over" do
+    before do
+      now = Time.now.to_i
+      @rb.lock_for_scheduling(now) {}
+    end
+
+    it "should not yield" do
+      now = Time.now.to_i
+      @rb.lock_for_scheduling(now) {}
+
+      yielded = false
+
+      @rb.lock_for_scheduling(now+5) { yielded = true }
+
+      yielded.must_equal false
+    end
+  end
+
+  describe "trying to lock while another client holds to lock" do
+    it "should not yield" do
+      first_yields = false
+      second_yields = false
+      now = Time.now.to_i
+      @rb.lock_for_scheduling(now) {}
+
+      @rb.lock_for_scheduling(now+30) do
+        first_yields = true
+        20.times do
+          @rb.lock_for_scheduling(now+35) do
+            second_yields = true
+          end
+        end
+      end
+
+      first_yields.must_equal true
+      second_yields.must_equal false
+    end
+  end
+
+  describe "throwing exception in the block given to lock" do
+    it "should release lock" do
+      now = Time.now.to_i
+      @rb.lock_for_scheduling(now) {}
+
+      begin
+        @rb.lock_for_scheduling(now+30) do
+          raise StandardError
+        end
+      rescue
+      end
+      
+      acquired_lock = false
+      @rb.lock_for_scheduling(now+30) do
+        acquired_lock = true
+      end
+
+      acquired_lock.must_equal true
+    end
+  end
+
+  describe "expired lock" do
+    it "should be acquireable" do
+      now = Time.now.to_i
+      @rb.lock_for_scheduling(now) {}
+      
+      acquired_lock_early = false
+      acquired_lock = false
+
+      @rb.lock_for_scheduling(now+30) do
+        @rb.lock_for_scheduling(now+22) do
+          acquired_lock_early = true
+        end
+        @rb.lock_for_scheduling(now+62) do
+          acquired_lock = true
+        end
+      end
+
+      acquired_lock.must_equal true
+      acquired_lock_early.must_equal false
+    end
+  end
+
+  describe "locking after waiting period is up" do
+    it "should yield" do
+      now = Time.now.to_i
+      @rb.lock_for_scheduling(now) {}
+
+      yielded = false
+      @rb.lock_for_scheduling(now+15) { yielded = true }
+      yielded.must_equal true
+    end
+
+    it "should yield last scheduled time" do
+      now = Time.now.to_i
+      @rb.lock_for_scheduling(now) {}
+
+      yielded = false
+      @rb.lock_for_scheduling(now+11) { |t| 
+        t.must_equal now
+        yielded = true 
+      }
+      yielded.must_equal true
+    end
+
+    it "should continue to yield as time goes on" do
+      yielded_count = 0
+      now = Time.now.to_i
+      @rb.lock_for_scheduling(now) {}
+
+      10.times do |n|
+        @rb.lock_for_scheduling(now+((n+1)*30)) do
+          yielded_count += 1
+        end
+      end
+
+      yielded_count.must_equal 10
+    end
+  end
+end