class RSpec::Rails::Matchers::ActiveJob::HaveEnqueuedJob

def at(date)
  @at = date
  self
end

def at_least(count)
  set_expected_number(:at_least, count)
  self
end

def at_most(count)
  set_expected_number(:at_most, count)
  self
end

def base_message
  "#{message_expectation_modifier} #{@expected_number} jobs,".tap do |msg|
    msg << " with #{@args}," if @args.any?
    msg << " on queue #{@queue}," if @queue
    msg << " at #{@at}," if @at
    msg << " but enqueued #{@matching_jobs_count}"
  end
end

def exactly(count)
  set_expected_number(:exactly, count)
  self
end

def failure_message
  "expected to enqueue #{base_message}"
end

def failure_message_when_negated
  "expected not to enqueue #{base_message}"
end

def initialize(job)
  @job = job
  @args = []
  @queue = nil
  @at = nil
  set_expected_number(:exactly, 1)
end

def matches?(proc)
  raise ArgumentError, "have_enqueued_jobs only supports block expectations" unless Proc === proc
  original_enqueued_jobs_count = queue_adapter.enqueued_jobs.count
  proc.call
  in_block_jobs = queue_adapter.enqueued_jobs.drop(original_enqueued_jobs_count)
  @matching_jobs_count = in_block_jobs.count do |job|
    serialized_attributes.all? { |key, value| value == job[key] }
  end
  case @expectation_type
  when :exactly then @expected_number == @matching_jobs_count
  when :at_most then @expected_number >= @matching_jobs_count
  when :at_least then @expected_number <= @matching_jobs_count
  end
end

def message_expectation_modifier
  case @expectation_type
  when :exactly then "exactly"
  when :at_most then "at most"
  when :at_least then "at least"
  end
end

def on_queue(queue)
  @queue = queue
  self
end

def queue_adapter
  ::ActiveJob::Base.queue_adapter
end

def serialized_attributes
  {}.tap do |attributes|
    attributes[:args]  = ::ActiveJob::Arguments.serialize(@args) if @args.any?
    attributes[:at]    = @at.to_f if @at
    attributes[:queue] = @queue if @queue
    attributes[:job]   = @job if @job
  end
end

def set_expected_number(relativity, count)
  @expectation_type = relativity
  @expected_number = case count
                     when :once then 1
                     when :twice then 2
                     when :thrice then 3
                     else Integer(count)
                     end
end

def supports_block_expectations?
  true
end

def times
  self
end

def with(*args)
  @args = args
  self
end