require 'integration/shared_examples/rack_examples'

RSpec.describe "Rails integration specs" do
  include Warden::Test::Helpers

  let(:app) { Rails.application }

  include_examples 'rack examples'

  if ::Rails.version.start_with?('5.')
    it "inserts the Airbrake Rack middleware after DebugExceptions" do
      middlewares = Rails.configuration.middleware.middlewares.map(&:inspect)
      own_idx = middlewares.index('Airbrake::Rack::Middleware')

      expect(middlewares[own_idx - 1]).to eq('ActionDispatch::DebugExceptions')
    end
  else
    it "inserts the Airbrake Rack middleware after ConnectionManagement" do
      middlewares = Rails.configuration.middleware.middlewares.map(&:inspect)
      own_idx = middlewares.index('Airbrake::Rack::Middleware')

      expect(middlewares[own_idx - 1])
        .to eq('ActiveRecord::ConnectionAdapters::ConnectionManagement')
    end
  end

  shared_examples "context payload content" do |route|
    let(:user) do
      OpenStruct.new(
        id: 1,
        email: 'qa@example.com',
        username: 'qa-dept'
      )
    end

    before do
      login_as(user)
      get(route, foo: :bar)
      sleep 2
    end

    it "includes component information" do
      body = /"context":{.*"component":"dummy".*}/
      expect(a_request(:post, endpoint).with(body: body)).to have_been_made
    end

    it "includes action information" do
      body =
        case route
        when '/crash'
          /"context":{.*"action":"crash".*}/
        when '/notify_airbrake_helper'
          /"context":{.*"action":"notify_airbrake_helper".*}/
        when '/notify_airbrake_sync_helper'
          /"context":{.*"action":"notify_airbrake_sync_helper".*}/
        else
          raise 'Unknown route'
        end
      expect(a_request(:post, endpoint).with(body: body)).to have_been_made
    end

    it "includes version" do
      body = /"context":{.*"versions":{"rails":"\d\./
      expect(a_request(:post, endpoint).with(body: body)).to have_been_made
    end

    it "includes session" do
      body = /"context":{.*"session":{.*"session_id":"\w+".*}/
      expect(a_request(:post, endpoint).with(body: body)).to have_been_made
    end

    it "includes params" do
      action = route[1..-1]
      body = /"context":{.*"params":{.*"controller":"dummy","action":"#{action}".*}/
      expect(a_request(:post, endpoint).with(body: body)).to have_been_made
    end

    it "includes route" do
      body = /"context":{.*"route":"#{route}\(\.:format\)".*}/
      expect(a_request(:post, endpoint).with(body: body)).to have_been_made
    end
  end

  describe "context payload" do
    context "when exception reported through middleware" do
      include_examples('context payload content', '/crash')
    end

    context "when exception reported through the notify_airbrake helper" do
      include_examples('context payload content', '/notify_airbrake_helper')
    end

    context "when exception reported through the notify_airbrake_sync helper" do
      include_examples('context payload content', '/notify_airbrake_sync_helper')
    end
  end

  describe(
    "Active Record callbacks",
    skip: Gem::Version.new(Rails.version) < Gem::Version.new('5.1.0')
  ) do
    it "reports exceptions in after_commit callbacks" do
      expect(Airbrake).to receive(:notify).with(
        an_instance_of(AirbrakeTestError)
      ) do |exception|
        expect(exception.message).to eq('after_commit')
      end

      get '/active_record_after_commit'
    end

    it "reports exceptions in after_rollback callbacks" do
      expect(Airbrake).to receive(:notify).with(
        an_instance_of(AirbrakeTestError)
      ) do |exception|
        expect(exception.message).to eq('after_rollback')
      end

      get '/active_record_after_rollback'
    end
  end

  if Gem::Version.new(Rails.version) >= Gem::Version.new('4.2')
    describe "ActiveJob jobs" do
      it "reports exceptions occurring in ActiveJob workers" do
        expect(Airbrake).to receive(:notify)
          .with(an_instance_of(Airbrake::Notice)).at_least(:once)

        get '/active_job'
        sleep 2 # Wait for ActiveJob job exiting
      end

      context "when Airbrake is not configured" do
        before do
          # Make sure we don't call `build_notice` more than 1 time. Rack
          # integration will try to handle error 500 and we want to prevent
          # that: https://github.com/airbrake/airbrake/pull/583
          allow_any_instance_of(Airbrake::Rack::Middleware).to(
            receive(:notify_airbrake).and_return(nil)
          )
        end

        it "doesn't report errors" do
          expect(Airbrake).to receive(:notify).with(
            an_instance_of(Airbrake::Notice)
          ) { |notice|
            # TODO: this doesn't actually fail but prints a failure. Figure
            # out how to test properly.
            expect(notice[:errors].first[:message]).to eq('active_job error')
          }.at_least(:once)

          get '/active_job'
          sleep 2
        end
      end
    end
  end

  describe "Resque workers" do
    it "reports exceptions occurring in Resque workers" do
      expect(Airbrake).to receive(:notify_sync).with(
        anything,
        hash_including(
          'class' => 'BingoWorker',
          'args' => %w[bango bongo]
        )
      )
      with_resque { get '/resque' }
    end
  end

  describe "DelayedJob jobs" do
    it "reports exceptions occurring in DelayedJob jobs" do
      skip if Gem::Version.new(Rails.version) > Gem::Version.new('3.2.22.5')

      expect(Airbrake).to receive(:notify).with(
        anything,
        'job' => hash_including(
          'handler' => "--- !ruby/struct:BangoJob\nbingo: bingo\nbongo: bongo\n"
        )
      )

      get '/delayed_job'
    end

    it "reports exceptions occurring in DelayedJob jobs on Rails 4.2" do
      skip if Gem::Version.new(Rails.version) == Gem::Version.new('3.2.22.5')

      expect(Airbrake).to receive(:notify).with(
        anything,
        hash_including(
          'handler' => "--- !ruby/struct:BangoJob\nbingo: bingo\nbongo: bongo\n"
        )
      )

      get '/delayed_job'
    end
  end

  describe "user extraction" do
    context "when Warden is not available but 'current_user' is defined" do
      let(:user) do
        OpenStruct.new(
          id: 1,
          email: 'qa@example.com',
          username: 'qa-dept'
        )
      end

      before do
        allow_message_expectations_on_nil
        allow(Warden::Proxy).to receive(:new).and_return(nil)
        # Mock on_request to make the test pass. Started failing in warden 1.2.8
        # due to: https://github.com/wardencommunity/warden/pull/162
        allow(nil).to receive(:on_request).and_return(nil)
        allow_any_instance_of(DummyController).to receive(:current_user) { user }
      end

      it "sends user info" do
        get '/crash'
        sleep 2

        body = /"user":{"id":"1","username":"qa-dept","email":"qa@example.com"}/
        wait_for(a_request(:post, endpoint).with(body: body)).to have_been_made
      end
    end
  end

  describe "request performance hook" do
    before { allow(Airbrake).to receive(:notify).and_return(nil) }

    it "notifies request" do
      expect(Airbrake).to receive(:notify_request).with(
        hash_including(
          route: '/crash(.:format)',
          method: 'GET'
        )
      )
      get '/crash'
    end

    it "defaults to 500 when status code for exception returns 0" do
      allow(ActionDispatch::ExceptionWrapper)
        .to receive(:status_code_for_exception).and_return(0)

      expect(Airbrake).to receive(:notify_request).with(
        hash_including(
          route: '/crash(.:format)',
          method: 'HEAD',
          status_code: 500
        )
      )
      head '/crash'
    end
  end

  describe "query performance hook" do
    before { allow(Airbrake).to receive(:notify).and_return(nil) }

    it "sends queries to Airbrake" do
      expect(Airbrake).to receive(:notify_query).with(
        hash_including(
          route: '/crash(.:format)',
          method: 'GET',
          func: 'tap',
          file: 'lib/airbrake/rails/active_record_subscriber.rb',
          line: anything
        )
      ).at_least(:once)

      get '/crash'
    end

    context "when caller location cannot be found for a query" do
      before { allow(Kernel).to receive(:caller).and_return([]) }

      it "sends query to Airbrake without caller location" do
        expect(Airbrake).to receive(:notify_query).with(
          hash_including(
            route: '/crash(.:format)',
            method: 'GET',
            func: nil,
            file: nil,
            line: nil
          )
        ).at_least(:once)

        get '/crash'
      end
    end
  end

  describe "performance breakdown hook" do
    before { allow(Airbrake).to receive(:notify).and_return(nil) }

    it "sends performance breakdown info to Airbrake" do
      expect(Airbrake).to receive(:notify_performance_breakdown).with(
        hash_including(
          route: '/breakdown(.:format)',
          method: 'GET',
          response_type: :html,
          groups: hash_including(db: an_instance_of(Float))
        )
      ).at_least(:once)

      get '/breakdown'
    end
  end
end