roundhouse-x 0.1.0

data/test/test_testing.rb

@@ -0,0 +1,78 @@
+require_relative 'helper'
+require 'roundhouse'
+require 'roundhouse/worker'
+require 'roundhouse/rails'
+
+Roundhouse.hook_rails!
+
+class TestTesting < Roundhouse::Test
+  describe 'roundhouse testing' do
+    describe 'require/load roundhouse/testing.rb' do
+      before do
+        require 'roundhouse/testing.rb'
+      end
+
+      after do
+        Roundhouse::Testing.disable!
+      end
+
+      it 'enables fake testing' do
+        Roundhouse::Testing.fake!
+        assert_equal true, Roundhouse::Testing.enabled?
+        assert_equal true, Roundhouse::Testing.fake?
+      end
+
+      it 'enables fake testing in a block' do
+        Roundhouse::Testing.disable!
+        assert_equal true, Roundhouse::Testing.disabled?
+
+        Roundhouse::Testing.fake! do
+          assert_equal true, Roundhouse::Testing.enabled?
+          assert_equal true, Roundhouse::Testing.fake?
+        end
+
+        assert_equal false, Roundhouse::Testing.enabled?
+        assert_equal false, Roundhouse::Testing.fake?
+      end
+
+      it 'disables testing in a block' do
+        Roundhouse::Testing.fake!
+
+        Roundhouse::Testing.disable! do
+          assert_equal true, Roundhouse::Testing.disabled?
+        end
+
+        assert_equal true, Roundhouse::Testing.enabled?
+      end
+    end
+
+    describe 'require/load roundhouse/testing/inline.rb' do
+      before do
+        require 'roundhouse/testing/inline.rb'
+      end
+
+      after do
+        Roundhouse::Testing.disable!
+      end
+
+      it 'enables inline testing' do
+        Roundhouse::Testing.inline!
+        assert_equal true, Roundhouse::Testing.enabled?
+        assert_equal true, Roundhouse::Testing.inline?
+      end
+
+      it 'enables inline testing in a block' do
+        Roundhouse::Testing.disable!
+        assert_equal true, Roundhouse::Testing.disabled?
+
+        Roundhouse::Testing.inline! do
+          assert_equal true, Roundhouse::Testing.enabled?
+          assert_equal true, Roundhouse::Testing.inline?
+        end
+
+        assert_equal false, Roundhouse::Testing.enabled?
+        assert_equal false, Roundhouse::Testing.inline?
+      end
+    end
+  end
+end

data/test/test_testing_fake.rb

@@ -0,0 +1,240 @@
+require_relative 'helper'
+require 'roundhouse'
+require 'roundhouse/worker'
+require 'roundhouse/rails'
+
+require 'active_support/core_ext/numeric/time'
+
+Roundhouse.hook_rails!
+
+class TestTesting < Roundhouse::Test
+  describe 'roundhouse testing' do
+    class PerformError < RuntimeError; end
+
+    class DirectWorker
+      include Roundhouse::Worker
+      def perform(a, b)
+        a + b
+      end
+    end
+
+    class EnqueuedWorker
+      include Roundhouse::Worker
+      def perform(a, b)
+        a + b
+      end
+    end
+
+    class StoredWorker
+      include Roundhouse::Worker
+      def perform(error)
+        raise PerformError if error
+      end
+    end
+
+    before do
+      require 'roundhouse/testing.rb'
+      Roundhouse::Testing.fake!
+      EnqueuedWorker.jobs.clear
+      DirectWorker.jobs.clear
+    end
+
+    after do
+      Roundhouse::Testing.disable!
+    end
+
+    it 'stubs the async call' do
+      assert_equal 0, DirectWorker.jobs.size
+      assert DirectWorker.perform_async(100, 1, 2)
+      assert_equal 1, DirectWorker.jobs.size
+      assert DirectWorker.perform_in(100, 10, 1, 2)
+      assert_equal 2, DirectWorker.jobs.size
+      assert DirectWorker.perform_at(100, 10, 1, 2)
+      assert_equal 3, DirectWorker.jobs.size
+      assert_in_delta 10.seconds.from_now.to_f, DirectWorker.jobs.last['at'], 0.01
+    end
+
+    it 'stubs the enqueue call' do
+      assert_equal 0, EnqueuedWorker.jobs.size
+      assert Roundhouse::Client.enqueue_to(100, EnqueuedWorker, 1, 2)
+      assert_equal 1, EnqueuedWorker.jobs.size
+    end
+
+    it 'stubs the enqueue_to call' do
+      assert_equal 0, EnqueuedWorker.jobs.size
+      assert Roundhouse::Client.enqueue_to(1000, EnqueuedWorker, 1, 2)
+      assert_equal 1, EnqueuedWorker.jobs.size
+    end
+
+    it 'executes all stored jobs' do
+      assert StoredWorker.perform_async(100, false)
+      assert StoredWorker.perform_async(100, true)
+
+      assert_equal 2, StoredWorker.jobs.size
+      assert_raises PerformError do
+        StoredWorker.drain
+      end
+      assert_equal 0, StoredWorker.jobs.size
+
+    end
+
+    class SpecificJidWorker
+      include Roundhouse::Worker
+      class_attribute :count
+      self.count = 0
+      def perform(worker_jid)
+        return unless worker_jid == self.jid
+        self.class.count += 1
+      end
+    end
+
+    it 'execute only jobs with assigned JID' do
+      4.times do |i|
+        jid = SpecificJidWorker.perform_async(100, nil)
+        if i % 2 == 0
+          SpecificJidWorker.jobs[-1]["args"] = ["wrong_jid"]
+        else
+          SpecificJidWorker.jobs[-1]["args"] = [jid]
+        end
+      end
+
+      SpecificJidWorker.perform_one
+      assert_equal 0, SpecificJidWorker.count
+
+      SpecificJidWorker.perform_one
+      assert_equal 1, SpecificJidWorker.count
+
+      SpecificJidWorker.drain
+      assert_equal 2, SpecificJidWorker.count
+    end
+
+    it 'round trip serializes the job arguments' do
+      assert StoredWorker.perform_async(100, :mike)
+      job = StoredWorker.jobs.first
+      assert_equal "mike", job['args'].first
+      StoredWorker.clear
+    end
+
+    it 'perform_one runs only one job' do
+      DirectWorker.perform_async(100, 1, 2)
+      DirectWorker.perform_async(100, 3, 4)
+      assert_equal 2, DirectWorker.jobs.size
+
+      DirectWorker.perform_one
+      assert_equal 1, DirectWorker.jobs.size
+
+      DirectWorker.clear
+    end
+
+    it 'perform_one raise error upon empty queue' do
+      DirectWorker.clear
+      assert_raises Roundhouse::EmptyQueueError do
+        DirectWorker.perform_one
+      end
+    end
+
+    class FirstWorker
+      include Roundhouse::Worker
+      class_attribute :count
+      self.count = 0
+      def perform
+        self.class.count += 1
+      end
+    end
+
+    class SecondWorker
+      include Roundhouse::Worker
+      class_attribute :count
+      self.count = 0
+      def perform
+        self.class.count += 1
+      end
+    end
+
+    class ThirdWorker
+      include Roundhouse::Worker
+      class_attribute :count
+      def perform
+        FirstWorker.perform_async(100)
+        SecondWorker.perform_async(100)
+      end
+    end
+
+    it 'clears jobs across all workers' do
+      Roundhouse::Worker.jobs.clear
+      FirstWorker.count = 0
+      SecondWorker.count = 0
+
+      assert_equal 0, FirstWorker.jobs.size
+      assert_equal 0, SecondWorker.jobs.size
+
+      FirstWorker.perform_async(100)
+      SecondWorker.perform_async(100)
+
+      assert_equal 1, FirstWorker.jobs.size
+      assert_equal 1, SecondWorker.jobs.size
+
+      Roundhouse::Worker.clear_all
+
+      assert_equal 0, FirstWorker.jobs.size
+      assert_equal 0, SecondWorker.jobs.size
+
+      assert_equal 0, FirstWorker.count
+      assert_equal 0, SecondWorker.count
+    end
+
+    it 'drains jobs across all workers' do
+      Roundhouse::Worker.jobs.clear
+      FirstWorker.count = 0
+      SecondWorker.count = 0
+
+      assert_equal 0, FirstWorker.jobs.size
+      assert_equal 0, SecondWorker.jobs.size
+
+      assert_equal 0, FirstWorker.count
+      assert_equal 0, SecondWorker.count
+
+      FirstWorker.perform_async(100)
+      SecondWorker.perform_async(100)
+
+      assert_equal 1, FirstWorker.jobs.size
+      assert_equal 1, SecondWorker.jobs.size
+
+      Roundhouse::Worker.drain_all
+
+      assert_equal 0, FirstWorker.jobs.size
+      assert_equal 0, SecondWorker.jobs.size
+
+      assert_equal 1, FirstWorker.count
+      assert_equal 1, SecondWorker.count
+    end
+
+    it 'drains jobs across all workers even when workers create new jobs' do
+      Roundhouse::Worker.jobs.clear
+      FirstWorker.count = 0
+      SecondWorker.count = 0
+
+      assert_equal 0, ThirdWorker.jobs.size
+
+      assert_equal 0, FirstWorker.count
+      assert_equal 0, SecondWorker.count
+
+      ThirdWorker.perform_async(100)
+
+      assert_equal 1, ThirdWorker.jobs.size
+
+      Roundhouse::Worker.drain_all
+
+      assert_equal 0, ThirdWorker.jobs.size
+
+      assert_equal 1, FirstWorker.count
+      assert_equal 1, SecondWorker.count
+    end
+
+    it 'can execute a job' do
+      worker = Minitest::Mock.new
+      worker.expect(:perform, nil, [1, 2, 3])
+      DirectWorker.execute_job(worker, [1, 2, 3])
+    end
+  end
+end

data/test/test_testing_inline.rb

@@ -0,0 +1,65 @@
+require_relative 'helper'
+require 'roundhouse'
+require 'roundhouse/worker'
+require 'roundhouse/rails'
+
+Roundhouse.hook_rails!
+
+class TestInline < Roundhouse::Test
+  describe 'roundhouse inline testing' do
+    class InlineError < RuntimeError; end
+    class ParameterIsNotString < RuntimeError; end
+
+    class InlineWorker
+      include Roundhouse::Worker
+      def perform(pass)
+        raise ArgumentError, "no jid" unless jid
+        raise InlineError unless pass
+      end
+    end
+
+    class InlineWorkerWithTimeParam
+      include Roundhouse::Worker
+      def perform(time)
+        raise ParameterIsNotString unless time.is_a?(String) || time.is_a?(Numeric)
+      end
+    end
+
+    before do
+      require 'roundhouse/testing/inline.rb'
+      Roundhouse::Testing.inline!
+    end
+
+    after do
+      Roundhouse::Testing.disable!
+    end
+
+    it 'stubs the async call when in testing mode' do
+      assert InlineWorker.perform_async(100, true)
+
+      assert_raises InlineError do
+        InlineWorker.perform_async(100, false)
+      end
+    end
+
+    it 'stubs the enqueue call when in testing mode' do
+      assert Roundhouse::Client.enqueue_to(100, InlineWorker, true)
+
+      assert_raises InlineError do
+        Roundhouse::Client.enqueue_to(100, InlineWorker, false)
+      end
+    end
+
+    it 'stubs the push_bulk call when in testing mode' do
+      assert Roundhouse::Client.push_bulk({'queue_id' => 100, 'class' => InlineWorker, 'args' => [[true], [true]]})
+
+      assert_raises InlineError do
+        Roundhouse::Client.push_bulk({'queue_id' => 100, 'class' => InlineWorker, 'args' => [[true], [false]]})
+      end
+    end
+
+    it 'should relay parameters through json' do
+      assert Roundhouse::Client.enqueue_to(100, InlineWorkerWithTimeParam, Time.now)
+    end
+  end
+end

data/test/test_util.rb

@@ -0,0 +1,18 @@
+require_relative 'helper'
+require 'roundhouse'
+require 'roundhouse/web_helpers'
+
+class TestUtil < Roundhouse::Test
+
+  class Helpers
+    include Roundhouse::Util
+  end
+
+  def test_hertz_donut
+    obj = Helpers.new
+    output = capture_logging(Logger::DEBUG) do
+      assert_equal false, obj.want_a_hertz_donut?
+    end
+    assert_includes output, "hz: 10"
+  end
+end