mperham--sidekiq/test/test_systemd.rb

require_relative 'helper'
require "sidekiq/sd_notify"
require 'sidekiq/systemd'

class TestSystemd < Minitest::Test
  def setup
    ::Dir::Tmpname.create("sidekiq_socket") do |sockaddr|
      @sockaddr = sockaddr
      @socket = Socket.new(:UNIX, :DGRAM, 0)
      socket_ai = Addrinfo.unix(sockaddr)
      @socket.bind(socket_ai)
      ENV["NOTIFY_SOCKET"] = sockaddr
    end
  end

  def teardown
    @socket.close if @socket
    File.unlink(@sockaddr) if @sockaddr
    @socket = nil
    @sockaddr = nil
  end

  def socket_message
    @socket.recvfrom(10)[0]
  end

  def test_notify
    count = Sidekiq::SdNotify.ready
    assert_equal(socket_message, "READY=1")
    assert_equal(ENV["NOTIFY_SOCKET"], @sockaddr)
    assert_equal(count, 7)

    count = Sidekiq::SdNotify.stopping
    assert_equal(socket_message, "STOPPING=1")
    assert_equal(ENV["NOTIFY_SOCKET"], @sockaddr)
    assert_equal(count, 10)

    refute Sidekiq::SdNotify.watchdog?
  end
end