mperham--sidekiq/test/test_launcher.rb

# frozen_string_literal: true

require_relative 'helper'
require 'sidekiq/launcher'

describe Sidekiq::Launcher do
  subject { Sidekiq::Launcher.new(options) }
  before do
    Sidekiq.redis {|c| c.flushdb }
  end

  def new_manager(opts)
    Sidekiq::Manager.new(opts)
  end

  describe 'memory collection' do
    it 'works in any test environment' do
      kb = Sidekiq::Launcher::MEMORY_GRABBER.call($$)
      refute_nil kb
      assert kb > 0
    end
  end

  describe 'heartbeat' do
    before do
      @mgr = new_manager(options)
      @launcher = Sidekiq::Launcher.new(options)
      @launcher.manager = @mgr
      @id = @launcher.identity

      Sidekiq::Processor::WORKER_STATE.set('a', {'b' => 1})

      @proctitle = $0
    end

    after do
      Sidekiq::Processor::WORKER_STATE.clear
      $0 = @proctitle
    end

    describe '#heartbeat' do
      describe 'run' do
        it 'sets sidekiq version, tag and the number of busy workers to proctitle' do
          subject.heartbeat

          assert_equal "sidekiq #{Sidekiq::VERSION} myapp [1 of 3 busy]", $0
        end

        it 'stores process info in redis' do
          subject.heartbeat

          workers, rtt = Sidekiq.redis { |c| c.hmget(subject.identity, 'busy', 'rtt_us') }

          assert_equal "1", workers
          refute_nil rtt
          assert_in_delta 1000, rtt.to_i, 1000

          expires = Sidekiq.redis { |c| c.pttl(subject.identity) }

          assert_in_delta 60000, expires, 500
        end

        describe 'events' do
          before do
            @cnt = 0

            Sidekiq.on(:heartbeat) do
              @cnt += 1
            end
          end

          it 'fires start heartbeat event only once' do
            assert_equal 0, @cnt
            subject.heartbeat
            assert_equal 1, @cnt
            subject.heartbeat
            assert_equal 1, @cnt
          end
        end
      end

      describe 'quiet' do
        before do
          subject.quiet
        end

        it 'sets stopping proctitle' do
          subject.heartbeat

          assert_equal "sidekiq #{Sidekiq::VERSION} myapp [1 of 3 busy] stopping", $0
        end

        it 'stores process info in redis' do
          subject.heartbeat

          info = Sidekiq.redis { |c| c.hmget(subject.identity, 'busy') }

          assert_equal ["1"], info

          expires = Sidekiq.redis { |c| c.pttl(subject.identity) }

          assert_in_delta 60000, expires, 50
        end
      end

    it 'fires new heartbeat events' do
      i = 0
      Sidekiq.on(:heartbeat) do
        i += 1
      end
      assert_equal 0, i
      @launcher.heartbeat
      assert_equal 1, i
      @launcher.heartbeat
      assert_equal 1, i
    end

    describe 'when manager is active' do
      before do
        Sidekiq::Launcher::PROCTITLES << proc { "xyz" }
        @launcher.heartbeat
        Sidekiq::Launcher::PROCTITLES.pop
      end

      it 'sets useful info to proctitle' do
        assert_equal "sidekiq #{Sidekiq::VERSION} myapp [1 of 3 busy] xyz", $0
      end

      it 'stores process info in redis' do
        info = Sidekiq.redis { |c| c.hmget(@id, 'busy') }
        assert_equal ["1"], info
        expires = Sidekiq.redis { |c| c.pttl(@id) }
        assert_in_delta 60000, expires, 500
      end
    end
  end

    describe 'when manager is stopped' do
      before do
        @launcher.quiet
        @launcher.heartbeat
      end

      #after do
        #puts system('redis-cli -n 15 keys  "*" | while read LINE ; do TTL=`redis-cli -n 15 ttl "$LINE"`; if [ "$TTL" -eq -1 ]; then echo "$LINE"; fi; done;')
      #end

      it 'indicates stopping status in proctitle' do
        assert_equal "sidekiq #{Sidekiq::VERSION} myapp [1 of 3 busy] stopping", $0
      end

      it 'stores process info in redis' do
        info = Sidekiq.redis { |c| c.hmget(@id, 'busy') }
        assert_equal ["1"], info
        expires = Sidekiq.redis { |c| c.pttl(@id) }
        assert_in_delta 60000, expires, 50
      end
    end
  end

  def options
    { :concurrency => 3, :queues => ['default'], :tag => 'myapp' }
  end

end