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::WORK_STATE.set("a", {"b" => 1})

      @proctitle = $0
    end

    after do
      Sidekiq::Processor::WORK_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