mperham--sidekiq/test/test_cli.rb

require 'helper'
require 'sidekiq/cli'
require 'tempfile'

class TestCli < MiniTest::Unit::TestCase
  describe 'with cli' do
    before do
      @cli = new_cli
    end

    it 'blows up with an invalid require' do
      assert_raises ArgumentError do
        @cli.parse(['sidekiq', '-r', 'foobar'])
      end
    end

    it 'blows up with invalid Ruby' 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, @cli.options[:processor_count]
    end

    it 'changes queues' do
      @cli.parse(['sidekiq', '-q', 'foo', '-r', './test/fake_env.rb'])
      assert_equal ['foo'], @cli.options[:queues]
    end

    it 'handles weights' do
      @cli.parse(['sidekiq', '-q', 'foo,3', '-q', 'bar', '-r', './test/fake_env.rb'])
      assert_equal %w(bar foo foo foo), @cli.options[:queues].sort
    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, @cli.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', @cli.options[:config_file]
      end

      it 'sets verbose' do
        refute @cli.options[:verbose]
      end
      
      it 'sets namespace' do
        assert_equal "test_namespace", @cli.options[:namespace]
      end

      it 'sets require file' do
        assert_equal './test/fake_env.rb', @cli.options[:require]
      end

      it 'sets environment' do
        assert_equal 'xzibit', @cli.options[:environment]
      end

      it 'sets concurrency' do
        assert_equal 50, @cli.options[:concurrency]
      end

      it 'sets pid file' do
        assert_equal '/tmp/sidekiq-config-test.pid', @cli.options[:pidfile]
      end

      it 'sets queues' do
        assert_equal 2, @cli.options[:queues].select{ |q| q == 'often' }.length
        assert_equal 1, @cli.options[:queues].select{ |q| q == 'seldom' }.length
      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',
                    '-n', 'sweet_story_bro',
                    '-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 processor count flag' do
        assert_equal 100, @cli.options[:processor_count]
      end
      
      it 'uses namespace flag' do
        assert_equal "sweet_story_bro", @cli.options[:namespace]
      end

      it 'uses require file flag' do
        assert_equal @tmp_lib_path, @cli.options[:require]
      end

      it 'uses environment flag' do
        assert_equal 'snoop', @cli.options[:environment]
      end

      it 'uses concurrency flag' do
        assert_equal 100, @cli.options[:processor_count]
      end

      it 'uses pidfile flag' do
        assert_equal @tmp_path, @cli.options[:pidfile]
      end

      it 'sets queues' do
        assert_equal 7, @cli.options[:queues].select{ |q| q == 'often' }.length
        assert_equal 3, @cli.options[:queues].select{ |q| q == 'seldom' }.length
      end
    end

    def new_cli
      cli = Sidekiq::CLI.new
      def cli.die(code)
        @code = code
      end

      def cli.valid?
        !@code
      end
      cli
    end
  end
end