lite_spec_helper.rb

# frozen_string_literal: true
# rubocop:todo all

$LOAD_PATH.unshift(File.dirname(__FILE__))
$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), "shared", "lib"))

# Load byebug before mongoid, to place breakpoints in the lib methods.
# But SpecConfig needs the driver code - require the driver here.
require "mongo"

# MRI 2.5 and JRuby 9.2 change visibility of Object#pp when 'pp' is required,
# which happens when RSpec reports anything. This creates an issue for tests
# that verify method forwarding. Work around by proactively loading 'pp'.
# https://github.com/jruby/jruby/issues/5599
require 'pp'

autoload :Timecop, 'timecop'

require 'support/spec_config'
require 'mrss/lite_constraints'

if Gem::Version.new(Mongo::VERSION) < Gem::Version.new('2.18.0.alpha')
  require "mrss/session_registry_legacy"
else
  require "mrss/session_registry"
end

Mrss.patch_mongo_for_session_registry

unless SpecConfig.instance.ci?
  begin
    require 'byebug'
  rescue LoadError
    # jruby - try pry
    begin
      require 'pry'
    # jruby likes to raise random error classes, in this case
    # NameError in addition to LoadError
    rescue Exception
    end
  end
end

require 'mongoid'

if SpecConfig.instance.mri? && !SpecConfig.instance.windows?
  require 'timeout_interrupt'
else
  require 'timeout'
  TimeoutInterrupt = Timeout
end

STANDARD_TIMEOUTS = {
  app: 500, # App tests under JRuby take a REALLY long time (over 5 minutes per test).
  default: 30,
}.freeze

def timeout_type
  if ENV['EXAMPLE_TIMEOUT'].to_i > 0
    :custom
  elsif SpecConfig.instance.app_tests?
    :app
  else
    :default
  end
end

def example_timeout_seconds
  STANDARD_TIMEOUTS.fetch(
    timeout_type,
    (ENV['EXAMPLE_TIMEOUT'] || STANDARD_TIMEOUTS[:default]).to_i
  )
end

RSpec.configure do |config|
  config.expect_with(:rspec) do |c|
    c.syntax = [:should, :expect]
  end

  if SpecConfig.instance.ci?
    config.add_formatter(RSpec::Core::Formatters::JsonFormatter, File.join(File.dirname(__FILE__), '../tmp/rspec.json'))
  end

  if SpecConfig.instance.ci? && !%w(1 true yes).include?(ENV['INTERACTIVE']&.downcase)
    config.around(:each) do |example|
      TimeoutInterrupt.timeout(example_timeout_seconds) do
        example.run
      end
    end
  end

  def local_env(env = nil, &block)
    around do |example|
      env ||= block.call
      saved_env = ENV.to_h
      ENV.update(env)

      example.run
    ensure
      ENV.replace(saved_env) if saved_env
    end
  end

  config.extend(Mrss::LiteConstraints)
end

# require all shared examples
Dir['./spec/support/shared/*.rb'].sort.each { |file| require file }