roundhouse-x 0.1.0

data/test/test_api.rb

@@ -0,0 +1,521 @@
+require_relative 'helper'
+require 'roundhouse/api'
+require 'active_support/core_ext/date/conversions'
+require 'active_support/core_ext/date_time/conversions'
+
+class TestApi < Roundhouse::Test
+
+  describe "stats" do
+    before do
+      @before = DateTime::DATE_FORMATS[:default]
+      DateTime::DATE_FORMATS[:default] = "%d/%m/%Y %H:%M:%S"
+      Roundhouse.redis = REDIS
+      Roundhouse.redis {|c| c.flushdb }
+    end
+
+    after do
+      DateTime::DATE_FORMATS[:default] = @before
+    end
+
+    describe "processed" do
+      it "is initially zero" do
+        s = Roundhouse::Stats.new
+        assert_equal 0, s.processed
+      end
+
+      it "returns number of processed jobs" do
+        Roundhouse.redis { |conn| conn.set("stat:processed", 5) }
+        s = Roundhouse::Stats.new
+        assert_equal 5, s.processed
+      end
+    end
+
+    describe "failed" do
+      it "is initially zero" do
+        s = Roundhouse::Stats.new
+        assert_equal 0, s.failed
+      end
+
+      it "returns number of failed jobs" do
+        Roundhouse.redis { |conn| conn.set("stat:failed", 5) }
+        s = Roundhouse::Stats.new
+        assert_equal 5, s.failed
+      end
+    end
+
+    describe "reset" do
+      before do
+        Roundhouse.redis do |conn|
+          conn.set('stat:processed', 5)
+          conn.set('stat:failed', 10)
+        end
+      end
+
+      it 'will reset all stats by default' do
+        Roundhouse::Stats.new.reset
+        Roundhouse.redis do |conn|
+          assert_equal '0', conn.get('stat:processed')
+          assert_equal '0', conn.get('stat:failed')
+        end
+      end
+
+      it 'can reset individual stats' do
+        Roundhouse::Stats.new.reset('failed')
+        Roundhouse.redis do |conn|
+          assert_equal '5', conn.get('stat:processed')
+          assert_equal '0', conn.get('stat:failed')
+        end
+      end
+
+      it 'can accept anything that responds to #to_s' do
+        Roundhouse::Stats.new.reset(:failed)
+        Roundhouse.redis do |conn|
+          assert_equal '5', conn.get('stat:processed')
+          assert_equal '0', conn.get('stat:failed')
+        end
+      end
+
+      it 'ignores anything other than "failed" or "processed"' do
+        Roundhouse::Stats.new.reset((1..10).to_a, ['failed'])
+        Roundhouse.redis do |conn|
+          assert_equal '5', conn.get('stat:processed')
+          assert_equal '0', conn.get('stat:failed')
+        end
+      end
+    end
+
+    describe "queues" do
+      it "returns all queues" do
+        assert_equal Roundhouse::Stats.new.queues, Roundhouse::Stats::Queues.new.lengths
+      end
+
+      it "is initially empty" do
+        s = Roundhouse::Stats::Queues.new
+        assert_equal 0, s.lengths.size
+      end
+
+      it "returns a hash of queue and size in order" do
+        Roundhouse.redis do |conn|
+          conn.rpush 'queue:foo', '{}'
+          conn.sadd 'queues', 'foo'
+
+          3.times { conn.rpush 'queue:bar', '{}' }
+          conn.sadd 'queues', 'bar'
+        end
+
+        s = Roundhouse::Stats::Queues.new
+        assert_equal ({ "foo" => 1, "bar" => 3 }), s.lengths
+        assert_equal "bar", s.lengths.first.first
+      end
+    end
+
+    describe 'queue stats' do
+      describe 'initially empty' do
+        it "is initially empty" do
+          s = Roundhouse::Stats.new
+          assert_equal 0, s.in_rotation
+        end
+      end
+
+      describe 'with enqueued jobs' do
+        before do
+          Roundhouse.redis do |conn|
+            Roundhouse::Monitor.push_job conn, [{'queue_id' => 2500, 'class' => 'MyWorker', 'args' => [1]}]
+            4.times { Roundhouse::Monitor.push_job conn, [{'queue_id' => 500, 'class' => 'MyWorker', 'args' => [1]}] }
+            Roundhouse::Monitor.set_queue_is_empty(conn, 3000)
+            Roundhouse::Monitor.suspend(conn, 200)
+          end
+        end
+
+        it 'should give back correct stats' do
+          s = Roundhouse::Stats.new
+          assert_equal 2, s.in_rotation
+          assert_equal 5, s.enqueued
+          assert_equal (5/4), s.avg_queue_len
+          assert_equal 4, s.num_queues
+          assert_equal 1, s.num_empty_queues
+          assert_equal 1, s.num_suspended_queues
+        end
+
+      end
+
+    end
+
+    describe "over time" do
+      describe "processed" do
+        it 'retrieves hash of dates' do
+          Roundhouse.redis do |c|
+            c.incrby("stat:processed:2012-12-24", 4)
+            c.incrby("stat:processed:2012-12-25", 1)
+            c.incrby("stat:processed:2012-12-26", 6)
+            c.incrby("stat:processed:2012-12-27", 2)
+          end
+          Time.stub(:now, Time.parse("2012-12-26 1:00:00 -0500")) do
+            s = Roundhouse::Stats::History.new(2)
+            assert_equal ({ "2012-12-26" => 6, "2012-12-25" => 1 }), s.processed
+
+            s = Roundhouse::Stats::History.new(3)
+            assert_equal ({ "2012-12-26" => 6, "2012-12-25" => 1, "2012-12-24" => 4 }), s.processed
+
+            s = Roundhouse::Stats::History.new(2, Date.parse("2012-12-25"))
+            assert_equal ({ "2012-12-25" => 1, "2012-12-24" => 4 }), s.processed
+          end
+        end
+      end
+
+      describe "failed" do
+        it 'retrieves hash of dates' do
+          Roundhouse.redis do |c|
+            c.incrby("stat:failed:2012-12-24", 4)
+            c.incrby("stat:failed:2012-12-25", 1)
+            c.incrby("stat:failed:2012-12-26", 6)
+            c.incrby("stat:failed:2012-12-27", 2)
+          end
+          Time.stub(:now, Time.parse("2012-12-26 1:00:00 -0500")) do
+            s = Roundhouse::Stats::History.new(2)
+            assert_equal ({ "2012-12-26" => 6, "2012-12-25" => 1 }), s.failed
+
+            s = Roundhouse::Stats::History.new(3)
+            assert_equal ({ "2012-12-26" => 6, "2012-12-25" => 1, "2012-12-24" => 4 }), s.failed
+
+            s = Roundhouse::Stats::History.new(2, Date.parse("2012-12-25"))
+            assert_equal ({ "2012-12-25" => 1, "2012-12-24" => 4 }), s.failed
+          end
+        end
+      end
+    end
+  end
+
+  describe 'with an empty database' do
+    include Roundhouse::Util
+
+    before do
+      Roundhouse.redis = REDIS
+      Roundhouse.redis {|c| c.flushdb }
+    end
+
+    it 'shows queue as empty' do
+      q = Roundhouse::Queue.new(100)
+      assert_equal 0, q.size
+      assert_equal 0, q.latency
+    end
+
+    class ApiWorker
+      include Roundhouse::Worker
+    end
+
+    it 'can enumerate jobs' do
+      q = Roundhouse::Queue.new(100)
+      Time.stub(:now, Time.new(2012, 12, 26)) do
+        ApiWorker.perform_async(100, 1, 'mike')
+        assert_equal ['TestApi::ApiWorker'], q.map(&:klass)
+
+        job = q.first
+        assert_equal 24, job.jid.size
+        assert_equal [1, 'mike'], job.args
+        assert_equal Time.new(2012, 12, 26), job.enqueued_at
+      end
+
+      assert q.latency > 10_000_000
+
+      q = Roundhouse::Queue.new('200')
+      assert_equal 0, q.size
+    end
+
+    it 'has no enqueued_at time for jobs enqueued in the future' do
+      job_id = ApiWorker.perform_in(100, 1, 'foo')
+      job = Roundhouse::ScheduledSet.new.find_job(job_id)
+      assert_nil job.enqueued_at
+    end
+
+    it 'unwraps delayed jobs' do
+      skip 'no delay'
+      ApiWorker.delay.foo(1,2,3)
+      q = Roundhouse::Queue.new(100)
+      x = q.first
+      assert_equal "TestApi::ApiWorker.foo", x.display_class
+      assert_equal [1,2,3], x.display_args
+    end
+
+    it 'can delete jobs' do
+      q = Roundhouse::Queue.new(100)
+      ApiWorker.perform_async(100, 1, 'mike')
+      assert_equal 1, q.size
+
+      x = q.first
+      assert_equal "TestApi::ApiWorker", x.display_class
+      assert_equal [1,'mike'], x.display_args
+
+      assert_equal [true], q.map(&:delete)
+      assert_equal 0, q.size
+    end
+
+    it "can move scheduled job to queue" do
+      remain_id = ApiWorker.perform_in(100, 1, 'jason')
+      job_id = ApiWorker.perform_in(100, 1, 'jason')
+      job = Roundhouse::ScheduledSet.new.find_job(job_id)
+      q = Roundhouse::Queue.new(100)
+      job.add_to_queue
+      queued_job = q.find_job(job_id)
+      refute_nil queued_job
+      assert_equal queued_job.jid, job_id
+      assert_nil Roundhouse::ScheduledSet.new.find_job(job_id)
+      refute_nil Roundhouse::ScheduledSet.new.find_job(remain_id)
+    end
+
+    it "handles multiple scheduled jobs when moving to queue" do
+      jids = Roundhouse::Client.push_bulk('class' => ApiWorker,
+                                       'queue_id' => 100,
+                                       'args' => [[1, 'jason'], [2, 'jason']],
+                                       'at' => Time.now.to_f)
+      assert_equal 2, jids.size
+      (remain_id, job_id) = jids
+      job = Roundhouse::ScheduledSet.new.find_job(job_id)
+      q = Roundhouse::Queue.new(100)
+      job.add_to_queue
+      queued_job = q.find_job(job_id)
+      refute_nil queued_job
+      assert_equal queued_job.jid, job_id
+      assert_nil Roundhouse::ScheduledSet.new.find_job(job_id)
+      refute_nil Roundhouse::ScheduledSet.new.find_job(remain_id)
+    end
+
+    it 'can find job by id in sorted sets' do
+      job_id = ApiWorker.perform_in(100, 1, 'jason')
+      job = Roundhouse::ScheduledSet.new.find_job(job_id)
+      refute_nil job
+      assert_equal job_id, job.jid
+      assert_in_delta job.latency, 0.0, 0.1
+    end
+
+    it 'can remove jobs when iterating over a sorted set' do
+      # scheduled jobs must be greater than SortedSet#each underlying page size
+      51.times do
+        ApiWorker.perform_in(100, 100, 'aaron')
+      end
+      set = Roundhouse::ScheduledSet.new
+      set.map(&:delete)
+      assert_equal set.size, 0
+    end
+
+    it 'can remove jobs when iterating over a queue' do
+      # initial queue size must be greater than Queue#each underlying page size
+      51.times do
+        ApiWorker.perform_async(1, 'aaron')
+      end
+      q = Roundhouse::Queue.new(100)
+      q.map(&:delete)
+      assert_equal q.size, 0
+    end
+
+    it 'can find job by id in queues' do
+      q = Roundhouse::Queue.new(100)
+      job_id = ApiWorker.perform_async(100, 1, 'jason')
+      job = q.find_job(job_id)
+      refute_nil job
+      assert_equal job_id, job.jid
+    end
+
+    it 'can clear a queue' do
+      q = Roundhouse::Queue.new(100)
+      2.times { ApiWorker.perform_async(1, 'mike') }
+      q.clear
+
+      Roundhouse.redis do |conn|
+        refute conn.smembers('queues').include?('foo')
+        refute conn.exists('queue:foo')
+      end
+    end
+
+    it 'can fetch by score' do
+      same_time = Time.now.to_f
+      add_retry('bob1', same_time)
+      add_retry('bob2', same_time)
+      r = Roundhouse::RetrySet.new
+      assert_equal 2, r.fetch(same_time).size
+    end
+
+    it 'can fetch by score and jid' do
+      same_time = Time.now.to_f
+      add_retry('bob1', same_time)
+      add_retry('bob2', same_time)
+      r = Roundhouse::RetrySet.new
+      assert_equal 1, r.fetch(same_time, 'bob1').size
+    end
+
+    it 'shows empty retries' do
+      r = Roundhouse::RetrySet.new
+      assert_equal 0, r.size
+    end
+
+    it 'can enumerate retries' do
+      add_retry
+
+      r = Roundhouse::RetrySet.new
+      assert_equal 1, r.size
+      array = r.to_a
+      assert_equal 1, array.size
+
+      retri = array.first
+      assert_equal 'ApiWorker', retri.klass
+      assert_equal 100, retri.queue_id
+      assert_equal 'bob', retri.jid
+      assert_in_delta Time.now.to_f, retri.at.to_f, 0.02
+    end
+
+    it 'requires a jid to delete an entry' do
+      start_time = Time.now.to_f
+      add_retry('bob2', Time.now.to_f)
+      assert_raises(ArgumentError) do
+        Roundhouse::RetrySet.new.delete(start_time)
+      end
+    end
+
+    it 'can delete a single retry from score and jid' do
+      same_time = Time.now.to_f
+      add_retry('bob1', same_time)
+      add_retry('bob2', same_time)
+      r = Roundhouse::RetrySet.new
+      assert_equal 2, r.size
+      Roundhouse::RetrySet.new.delete(same_time, 'bob1')
+      assert_equal 1, r.size
+    end
+
+    it 'can retry a retry' do
+      add_retry
+      r = Roundhouse::RetrySet.new
+      assert_equal 1, r.size
+      r.first.retry
+      assert_equal 0, r.size
+      assert_equal 1, Roundhouse::Queue.new(100).size
+      job = Roundhouse::Queue.new(100).first
+      assert_equal 'bob', job.jid
+      assert_equal 1, job['retry_count']
+    end
+
+    it 'can clear retries' do
+      add_retry
+      add_retry('test')
+      r = Roundhouse::RetrySet.new
+      assert_equal 2, r.size
+      r.clear
+      assert_equal 0, r.size
+    end
+
+    it 'can enumerate processes' do
+      identity_string = "identity_string"
+      odata = {
+        'pid' => 123,
+        'hostname' => hostname,
+        'key' => identity_string,
+        'identity' => identity_string,
+        'started_at' => Time.now.to_f - 15,
+      }
+
+      time = Time.now.to_f
+      Roundhouse.redis do |conn|
+        conn.multi do
+          conn.sadd('processes', odata['key'])
+          conn.hmset(odata['key'], 'info', Roundhouse.dump_json(odata), 'busy', 10, 'beat', time)
+          conn.sadd('processes', 'fake:pid')
+        end
+      end
+
+      ps = Roundhouse::ProcessSet.new.to_a
+      assert_equal 1, ps.size
+      data = ps.first
+      assert_equal 10, data['busy']
+      assert_equal time, data['beat']
+      assert_equal 123, data['pid']
+      data.quiet!
+      data.stop!
+      signals_string = "#{odata['key']}-signals"
+      assert_equal "TERM", Roundhouse.redis{|c| c.lpop(signals_string) }
+      assert_equal "USR1", Roundhouse.redis{|c| c.lpop(signals_string) }
+    end
+
+    it 'can enumerate workers' do
+      w = Roundhouse::Workers.new
+      assert_equal 0, w.size
+      w.each do
+        assert false
+      end
+
+      key = "#{hostname}:#{$$}"
+      pdata = { 'pid' => $$, 'hostname' => hostname, 'started_at' => Time.now.to_i }
+      Roundhouse.redis do |conn|
+        conn.sadd('processes', key)
+        conn.hmset(key, 'info', Roundhouse.dump_json(pdata), 'busy', 0, 'beat', Time.now.to_f)
+      end
+
+      s = "#{key}:workers"
+      data = Roundhouse.dump_json({ 'payload' => {}, 'queue' => 'default', 'run_at' => Time.now.to_i })
+      Roundhouse.redis do |c|
+        c.hmset(s, '1234', data)
+      end
+
+      w.each do |p, x, y|
+        assert_equal key, p
+        assert_equal "1234", x
+        assert_equal 'default', y['queue']
+        assert_equal Time.now.year, Time.at(y['run_at']).year
+      end
+
+      s = "#{key}:workers"
+      data = Roundhouse.dump_json({ 'payload' => {}, 'queue' => 'default', 'run_at' => (Time.now.to_i - 2*60*60) })
+      Roundhouse.redis do |c|
+        c.multi do
+          c.hmset(s, '5678', data)
+          c.hmset("b#{s}", '5678', data)
+        end
+      end
+
+      assert_equal ['1234', '5678'], w.map { |_, tid, _| tid }
+    end
+
+    it 'can reschedule jobs' do
+      add_retry('foo1')
+      add_retry('foo2')
+
+      retries = Roundhouse::RetrySet.new
+      assert_equal 2, retries.size
+      refute(retries.map { |r| r.score > (Time.now.to_f + 9) }.any?)
+
+      retries.each do |retri|
+        retri.reschedule(Time.now.to_f + 10) if retri.jid == 'foo2'
+      end
+
+      assert_equal 2, retries.size
+      assert(retries.map { |r| r.score > (Time.now.to_f + 9) }.any?)
+    end
+
+    it 'prunes processes which have died' do
+      data = { 'pid' => rand(10_000), 'hostname' => "app#{rand(1_000)}", 'started_at' => Time.now.to_f }
+      key = "#{data['hostname']}:#{data['pid']}"
+      Roundhouse.redis do |conn|
+        conn.sadd('processes', key)
+        conn.hmset(key, 'info', Roundhouse.dump_json(data), 'busy', 0, 'beat', Time.now.to_f)
+      end
+
+      ps = Roundhouse::ProcessSet.new
+      assert_equal 1, ps.size
+      assert_equal 1, ps.to_a.size
+
+      Roundhouse.redis do |conn|
+        conn.sadd('processes', "bar:987")
+        conn.sadd('processes', "bar:986")
+      end
+
+      ps = Roundhouse::ProcessSet.new
+      assert_equal 1, ps.size
+      assert_equal 1, ps.to_a.size
+    end
+
+    def add_retry(jid = 'bob', at = Time.now.to_f)
+      payload = Roundhouse.dump_json('class' => 'ApiWorker', 'args' => [1, 'mike'], 'queue_id' => 100, 'jid' => jid, 'retry_count' => 2, 'failed_at' => Time.now.to_f)
+      Roundhouse.redis do |conn|
+        conn.zadd('retry', at.to_s, payload)
+      end
+    end
+  end
+end