ENV['RACK_ENV'] ||= 'test'
$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))

require 'datadog-notifications'
require 'rspec'
require 'rack/test'
require 'grape'
require 'active_record'
require 'sqlite3'

### Active-record test preparation

ActiveRecord::Base.configurations["test"] = { 'adapter' => 'sqlite3', 'database' => ':memory:' }
ActiveRecord::Base.establish_connection(:test)
ActiveRecord::Base.connection.create_table :posts do |t|
  t.string :title
end
class Post < ActiveRecord::Base
end

### Mocks

module Mock
  class Reporter < Datadog::Notifications::Reporter
    def timing(stat, _ms, opts={})
      super(stat, 333, opts)
    end

    def flush_buffer; end
    @should_batch = true
  end

  class Instrumentable
    def initialize(opts={})
      @opts = opts
    end

    def perform
      ActiveSupport::Notifications.instrument("mock.start", @opts)
      ActiveSupport::Notifications.instrument("mock.perform", @opts) do |payload|
        payload[:status] = 200
      end
    end
  end
end

### Configuration

RSpec.configure do |c|
  helpers = Module.new do
    def buffered
      Datadog::Notifications.instance.send(:reporter).buffer
    end
  end

  c.include helpers
  c.before do
    buffered.clear
  end
end

Datadog::Notifications.configure do |c|
  c.hostname = "test.host"
  c.reporter = Mock::Reporter
  c.tags     = ["custom:tag"]

  c.use Datadog::Notifications::Plugins::ActiveRecord
  c.use Datadog::Notifications::Plugins::Grape,
    tags: ["more:tags"],
    metric_name: "api.request",
    exception_handler: lambda {|e|
      if e.message.include?('unauthorized')
        401
      else
        Datadog::Notifications::Plugins::Grape.exception_status(e)
      end
    }
end