plugin_testing.rb

# frozen_string_literal: true

require "test_helper"

module PluginTesting
  extend ActiveSupport::Concern
  extend ActiveSupport::Testing::Declarative

  included do
    self.use_transactional_tests = false

    setup do
      FileUtils.mkdir_p Rails.root.join("tmp", "pids")

      Dir.chdir("test/dummy") do
        cmd = %W[
          bundle exec puma
            -b tcp://127.0.0.1:9222
            -C config/puma_#{solid_queue_mode}.rb
            -s
            config.ru
        ]

        @pid = fork do
          exec(*cmd)
        end
      end

      wait_for_registered_processes(5, timeout: 5.second)
    end

    teardown do
      terminate_process(@pid, signal: :INT) if process_exists?(@pid)
      wait_for_registered_processes(0, timeout: 5.seconds)
    end
  end

  test "perform jobs inside puma's process" do
    StoreResultJob.perform_later(:puma_plugin)

    wait_for_jobs_to_finish_for(5.seconds)
    assert_equal 1, JobResult.where(queue_name: :background, status: "completed", value: :puma_plugin).count
  end

  test "stop the queue on puma's restart" do
    signal_process(@pid, :SIGUSR2)
    # Ensure the restart finishes before we try to continue with the test
    wait_for_registered_processes(0, timeout: 5.second)
    wait_for_registered_processes(5, timeout: 5.second)

    StoreResultJob.perform_later(:puma_plugin)
    wait_for_jobs_to_finish_for(2.seconds)
    assert_equal 1, JobResult.where(queue_name: :background, status: "completed", value: :puma_plugin).count
  end

  private
    def solid_queue_mode
      raise NotImplementedError
    end
end