kulesa-sidekiq 1.2.2

data/myapp/config/locales/en.yml

@@ -0,0 +1,5 @@
+# Sample localization file for English. Add more files in this directory for other locales.
+# See https://github.com/svenfuchs/rails-i18n/tree/master/rails%2Flocale for starting points.
+
+en:
+  hello: "Hello world"

data/myapp/config/routes.rb

@@ -0,0 +1,10 @@
+require 'sidekiq/web'
+
+Myapp::Application.routes.draw do
+  mount Sidekiq::Web => '/sidekiq'
+  get "work" => "work#index"
+  get "work/email" => "work#email"
+  get "work/post" => "work#delayed_post"
+  get "work/long" => "work#long"
+  get "work/crash" => "work#crash"
+end

data/myapp/db/migrate/20120123214055_create_posts.rb

@@ -0,0 +1,10 @@
+class CreatePosts < ActiveRecord::Migration
+  def change
+    create_table :posts do |t|
+      t.string :title
+      t.string :body
+
+      t.timestamps
+    end
+  end
+end

data/myapp/db/seeds.rb

@@ -0,0 +1,7 @@
+# This file should contain all the record creation needed to seed the database with its default values.
+# The data can then be loaded with the rake db:seed (or created alongside the db with db:setup).
+#
+# Examples:
+#
+#   cities = City.create([{ name: 'Chicago' }, { name: 'Copenhagen' }])
+#   Mayor.create(name: 'Emanuel', city: cities.first)

data/myapp/script/rails

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+# This command will automatically be run when you run "rails" with Rails 3 gems installed from the root of your application.
+
+APP_PATH = File.expand_path('../../config/application',  __FILE__)
+require File.expand_path('../../config/boot',  __FILE__)
+require 'rails/commands'

data/sidekiq.gemspec

@@ -0,0 +1,27 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path('../lib/sidekiq/version', __FILE__)
+
+Gem::Specification.new do |gem|
+  gem.authors       = ["Mike Perham"]
+  gem.email         = ["mperham@gmail.com"]
+  gem.description   = gem.summary = "Simple, efficient message processing for Ruby"
+  gem.homepage      = "http://mperham.github.com/sidekiq"
+
+  gem.executables   = ['sidekiq', 'sidekiqctl']
+  gem.files         = `git ls-files`.split("\n")
+  gem.test_files    = `git ls-files -- test/*`.split("\n")
+  gem.name          = "kulesa-sidekiq"
+  gem.require_paths = ["lib"]
+  gem.version       = Sidekiq::VERSION
+  gem.add_dependency                  'redis'
+  gem.add_dependency                  'redis-namespace'
+  gem.add_dependency                  'connection_pool', '~> 0.9.0'
+  gem.add_dependency                  'kulesa-celluloid'
+  gem.add_dependency                  'multi_json', '~> 1'
+  gem.add_development_dependency      'minitest'
+  gem.add_development_dependency      'sinatra'
+  gem.add_development_dependency      'slim'
+  gem.add_development_dependency      'rake'
+  gem.add_development_dependency      'actionmailer', '~> 3'
+  gem.add_development_dependency      'activerecord', '~> 3'
+end

data/test/config.yml

@@ -0,0 +1,9 @@
+---
+:verbose:      false
+:environment:  xzibit
+:require:      ./test/fake_env.rb
+:pidfile:      /tmp/sidekiq-config-test.pid
+:concurrency:  50
+:queues:
+  - [often, 2]
+  - [seldom, 1]

data/test/helper.rb

@@ -0,0 +1,16 @@
+ENV['RACK_ENV'] = ENV['RAILS_ENV'] = 'test'
+if ENV.has_key?("SIMPLECOV")
+  require 'simplecov'
+  SimpleCov.start
+end
+
+require 'minitest/unit'
+require 'minitest/pride'
+require 'minitest/autorun'
+
+require 'sidekiq'
+require 'sidekiq/util'
+Sidekiq.logger.level = Logger::ERROR
+
+require 'sidekiq/redis_connection'
+REDIS = Sidekiq::RedisConnection.create(:url => "redis://localhost/15", :namespace => 'testy')

data/test/test_cli.rb

@@ -0,0 +1,168 @@
+require 'helper'
+require 'sidekiq/cli'
+require 'tempfile'
+
+cli = Sidekiq::CLI.instance
+def cli.die(code)
+  @code = code
+end
+
+def cli.valid?
+  !@code
+end
+
+class TestCli < MiniTest::Unit::TestCase
+  describe 'with cli' do
+
+    before do
+      @cli = Sidekiq::CLI.instance
+    end
+
+    it 'blows up with an invalid require' do
+      assert_raises ArgumentError do
+        @cli.parse(['sidekiq', '-r', 'foobar'])
+      end
+    end
+
+    it 'requires the specified Ruby code' do
+      @cli.parse(['sidekiq', '-r', './test/fake_env.rb'])
+      assert($LOADED_FEATURES.any? { |x| x =~ /fake_env/ })
+      assert @cli.valid?
+    end
+
+    it 'changes concurrency' do
+      @cli.parse(['sidekiq', '-c', '60', '-r', './test/fake_env.rb'])
+      assert_equal 60, Sidekiq.options[:concurrency]
+    end
+
+    it 'changes queues' do
+      @cli.parse(['sidekiq', '-q', 'foo', '-r', './test/fake_env.rb'])
+      assert_equal ['foo'], Sidekiq.options[:queues]
+    end
+
+    it 'changes timeout' do
+      @cli.parse(['sidekiq', '-t', '30', '-r', './test/fake_env.rb'])
+      assert_equal 30, Sidekiq.options[:timeout]
+    end
+
+    it 'handles multiple queues with weights' do
+      @cli.parse(['sidekiq', '-q', 'foo,3', '-q', 'bar', '-r', './test/fake_env.rb'])
+      assert_equal %w(bar foo foo foo), Sidekiq.options[:queues].sort
+    end
+
+    it 'sets verbose' do
+      old = Sidekiq.logger.level
+      @cli.parse(['sidekiq', '-v', '-r', './test/fake_env.rb'])
+      assert_equal Logger::DEBUG, Sidekiq.logger.level
+      # If we leave the logger at DEBUG it'll add a lot of noise to the test output
+      Sidekiq.logger.level = old
+    end
+
+    describe 'with pidfile' do
+      before do
+        @tmp_file = Tempfile.new('sidekiq-test')
+        @tmp_path = @tmp_file.path
+        @tmp_file.close!
+
+        @cli.parse(['sidekiq', '-P', @tmp_path, '-r', './test/fake_env.rb'])
+      end
+
+      after do
+        File.unlink @tmp_path if File.exist? @tmp_path
+      end
+
+      it 'sets pidfile path' do
+        assert_equal @tmp_path, Sidekiq.options[:pidfile]
+      end
+
+      it 'writes pidfile' do
+        assert_equal File.read(@tmp_path).strip.to_i, Process.pid
+      end
+    end
+
+    describe 'with config file' do
+      before do
+        @cli.parse(['sidekiq', '-C', './test/config.yml'])
+      end
+
+      it 'takes a path' do
+        assert_equal './test/config.yml', Sidekiq.options[:config_file]
+      end
+
+      it 'sets verbose' do
+        refute Sidekiq.options[:verbose]
+      end
+
+      it 'sets require file' do
+        assert_equal './test/fake_env.rb', Sidekiq.options[:require]
+      end
+
+      it 'sets environment' do
+        assert_equal 'xzibit', Sidekiq.options[:environment]
+      end
+
+      it 'sets concurrency' do
+        assert_equal 50, Sidekiq.options[:concurrency]
+      end
+
+      it 'sets pid file' do
+        assert_equal '/tmp/sidekiq-config-test.pid', Sidekiq.options[:pidfile]
+      end
+
+      it 'sets queues' do
+        assert_equal 2, Sidekiq.options[:queues].count { |q| q == 'often' }
+        assert_equal 1, Sidekiq.options[:queues].count { |q| q == 'seldom' }
+      end
+    end
+
+    describe 'with config file and flags' do
+      before do
+        # We need an actual file here.
+        @tmp_lib_path = '/tmp/require-me.rb'
+        File.open(@tmp_lib_path, 'w') do |f|
+          f.puts "# do work"
+        end
+
+        @tmp_file = Tempfile.new('sidekiqr')
+        @tmp_path = @tmp_file.path
+        @tmp_file.close!
+
+        @cli.parse(['sidekiq',
+                    '-C', './test/config.yml',
+                    '-e', 'snoop',
+                    '-c', '100',
+                    '-r', @tmp_lib_path,
+                    '-P', @tmp_path,
+                    '-q', 'often,7',
+                    '-q', 'seldom,3'])
+      end
+
+      after do
+        File.unlink @tmp_lib_path if File.exist? @tmp_lib_path
+        File.unlink @tmp_path if File.exist? @tmp_path
+      end
+
+      it 'uses concurrency flag' do
+        assert_equal 100, Sidekiq.options[:concurrency]
+      end
+
+      it 'uses require file flag' do
+        assert_equal @tmp_lib_path, Sidekiq.options[:require]
+      end
+
+      it 'uses environment flag' do
+        assert_equal 'snoop', Sidekiq.options[:environment]
+      end
+
+      it 'uses pidfile flag' do
+        assert_equal @tmp_path, Sidekiq.options[:pidfile]
+      end
+
+      it 'sets queues' do
+        assert_equal 7, Sidekiq.options[:queues].count { |q| q == 'often' }
+        assert_equal 3, Sidekiq.options[:queues].count { |q| q == 'seldom' }
+      end
+    end
+  end
+
+end

data/test/test_client.rb

@@ -0,0 +1,119 @@
+require 'helper'
+require 'sidekiq/client'
+require 'sidekiq/worker'
+
+class TestClient < MiniTest::Unit::TestCase
+  describe 'with real redis' do
+    before do
+      Sidekiq.redis = REDIS
+      Sidekiq.redis {|c| c.flushdb }
+    end
+
+    class QueueWorker
+      include Sidekiq::Worker
+      sidekiq_options :queue => 'customqueue'
+    end
+
+    it 'does not push duplicate messages when configured for unique only' do
+      Sidekiq.client_middleware do |chain|
+        chain.add Sidekiq::Middleware::Client::UniqueJobs
+      end
+      QueueWorker.sidekiq_options :unique => true
+      10.times { Sidekiq::Client.push('class' => QueueWorker, 'args' => [1, 2]) }
+      assert_equal 1, Sidekiq.redis {|c| c.llen("queue:customqueue") }
+    end
+
+    it 'does push duplicate messages when not configured for unique only' do
+      Sidekiq.client_middleware do |chain|
+        chain.add Sidekiq::Middleware::Client::UniqueJobs
+      end
+      QueueWorker.sidekiq_options :unique => false
+      10.times { Sidekiq::Client.push('class' => QueueWorker, 'args' => [1, 2]) }
+      assert_equal 10, Sidekiq.redis {|c| c.llen("queue:customqueue") }
+    end
+  end
+
+  describe 'with mock redis' do
+    before do
+      @redis = MiniTest::Mock.new
+      def @redis.multi; [yield] * 2 if block_given?; end
+      def @redis.set(*); true; end
+      def @redis.sadd(*); true; end
+      def @redis.srem(*); true; end
+      def @redis.get(*); nil; end
+      def @redis.del(*); nil; end
+      def @redis.incrby(*); nil; end
+      def @redis.setex(*); true; end
+      def @redis.expire(*); true; end
+      def @redis.watch(*); true; end
+      def @redis.with_connection; yield self; end
+      def @redis.with; yield self; end
+      def @redis.exec; true; end
+      Sidekiq.instance_variable_set(:@redis, @redis)
+    end
+
+    it 'raises ArgumentError with invalid params' do
+      assert_raises ArgumentError do
+        Sidekiq::Client.push('foo', 1)
+      end
+
+      assert_raises ArgumentError do
+        Sidekiq::Client.push('foo', :class => 'Foo', :noargs => [1, 2])
+      end
+    end
+
+    it 'pushes messages to redis' do
+      @redis.expect :rpush, 1, ['queue:foo', String]
+      pushed = Sidekiq::Client.push('queue' => 'foo', 'class' => MyWorker, 'args' => [1, 2])
+      assert pushed
+      @redis.verify
+    end
+
+    class MyWorker
+      include Sidekiq::Worker
+    end
+
+    it 'handles perform_async' do
+      @redis.expect :rpush, 1, ['queue:default', String]
+      pushed = MyWorker.perform_async(1, 2)
+      assert pushed
+      @redis.verify
+    end
+
+    it 'handles perform_async on failure' do
+      @redis.expect :rpush, nil, ['queue:default', String]
+      pushed = MyWorker.perform_async(1, 2)
+      refute pushed
+      @redis.verify
+    end
+
+    it 'enqueues messages to redis' do
+      @redis.expect :rpush, 1, ['queue:default', String]
+      pushed = Sidekiq::Client.enqueue(MyWorker, 1, 2)
+      assert pushed
+      @redis.verify
+    end
+
+    class QueuedWorker
+      include Sidekiq::Worker
+      sidekiq_options :queue => :flimflam, :timeout => 1
+    end
+
+    it 'enqueues to the named queue' do
+      @redis.expect :rpush, 1, ['queue:flimflam', String]
+      pushed = QueuedWorker.perform_async(1, 2)
+      assert pushed
+      @redis.verify
+    end
+
+    it 'retrieves queues' do
+      @redis.expect :smembers, ['bob'], ['queues']
+      assert_equal ['bob'], Sidekiq::Client.registered_queues
+    end
+
+    it 'retrieves workers' do
+      @redis.expect :smembers, ['bob'], ['workers']
+      assert_equal ['bob'], Sidekiq::Client.registered_workers
+    end
+  end
+end

data/test/test_extensions.rb

@@ -0,0 +1,69 @@
+require 'helper'
+require 'sidekiq'
+require 'active_record'
+require 'action_mailer'
+require 'sidekiq/extensions/action_mailer'
+require 'sidekiq/extensions/active_record'
+require 'sidekiq/rails'
+
+Sidekiq.hook_rails!
+
+class TestExtensions < MiniTest::Unit::TestCase
+  describe 'sidekiq extensions' do
+    before do
+      Sidekiq.redis = REDIS
+      Sidekiq.redis {|c| c.flushdb }
+    end
+
+    class MyModel < ActiveRecord::Base
+      def self.long_class_method
+        raise "Should not be called!"
+      end
+    end
+
+    it 'allows delayed execution of ActiveRecord class methods' do
+      assert_equal [], Sidekiq::Client.registered_queues
+      assert_equal 0, Sidekiq.redis {|c| c.llen('queue:default') }
+      MyModel.delay.long_class_method
+      assert_equal ['default'], Sidekiq::Client.registered_queues
+      assert_equal 1, Sidekiq.redis {|c| c.llen('queue:default') }
+    end
+
+    class UserMailer < ActionMailer::Base
+      def greetings(a, b)
+        raise "Should not be called!"
+      end
+    end
+
+    it 'allows delayed delivery of ActionMailer mails' do
+      assert_equal [], Sidekiq::Client.registered_queues
+      assert_equal 0, Sidekiq.redis {|c| c.llen('queue:default') }
+      UserMailer.delay.greetings(1, 2)
+      assert_equal ['default'], Sidekiq::Client.registered_queues
+      assert_equal 1, Sidekiq.redis {|c| c.llen('queue:default') }
+    end
+  end
+
+  describe 'sidekiq rails extensions configuration' do
+    before do
+      @options = Sidekiq.options
+    end
+
+    after do
+      Sidekiq.options = @options
+    end
+
+    it 'should set enable_rails_extensions option to true by default' do
+      assert Sidekiq.options[:enable_rails_extensions]
+    end
+
+    it 'should extend ActiveRecord and ActiveMailer if enable_rails_extensions is true' do
+      assert Sidekiq.hook_rails!
+    end
+
+    it 'should not extend ActiveRecord and ActiveMailer if enable_rails_extensions is false' do
+      Sidekiq.options = { :enable_rails_extensions => false }
+      refute Sidekiq.hook_rails!
+    end
+  end
+end