module ActiveSupport::Testing::TimeHelpers

def after_teardown
  travel_back
  super
end

def freeze_time(&block)
  travel_to Time.now, &block
end

def simple_stubs
  @simple_stubs ||= SimpleStubs.new
end

def travel(duration, &block)
  travel_to Time.now + duration, &block
end

def travel_back
  stubbed_time = Time.current if block_given? && simple_stubs.stubbed?
  simple_stubs.unstub_all!
  yield if block_given?
ensure
  travel_to stubbed_time if stubbed_time
end

def travel_to(date_or_time)
  if block_given? && in_block
    travel_to_nested_block_call = <<~MSG
Calling `travel_to` with a block, when we have previously already made a call to `travel_to`, can lead to confusing time stubbing.
Instead of:
   travel_to 2.days.from_now do
     # 2 days from today
     travel_to 3.days.from_now do
       # 5 days from today
     end
   end
preferred way to achieve above is:
   travel 2.days do
     # 2 days from today
   end
   travel 5.days do
     # 5 days from today
   end
    MSG
    raise travel_to_nested_block_call
  end
  if date_or_time.is_a?(Date) && !date_or_time.is_a?(DateTime)
    now = date_or_time.midnight.to_time
  elsif date_or_time.is_a?(String)
    now = Time.zone.parse(date_or_time)
  else
    now = date_or_time.to_time.change(usec: 0)
  end
  stubbed_time = Time.now if simple_stubs.stubbing(Time, :now)
  simple_stubs.stub_object(Time, :now) { at(now.to_i) }
  simple_stubs.stub_object(Date, :today) { jd(now.to_date.jd) }
  simple_stubs.stub_object(DateTime, :now) { jd(now.to_date.jd, now.hour, now.min, now.sec, Rational(now.utc_offset, 86400)) }
  if block_given?
    begin
      self.in_block = true
      yield
    ensure
      if stubbed_time
        travel_to stubbed_time
      else
        travel_back
      end
      self.in_block = false
    end
  end
end