require File.dirname(__FILE__) + "/../spec_helper"

RSpec.describe "ActiveRecord SQL metrics", type: :request do
  let(:tags_middleware) do
    lambda do |tags|
      tags.merge(tags_middleware: :tags_middleware)
    end
  end
  before do
    allow_any_instance_of(ActionDispatch::Request).to receive(:request_id).and_return(:request_id)
    allow_any_instance_of(InfluxDB::Rails::Configuration).to receive(:application_name).and_return(:app_name)
    allow_any_instance_of(InfluxDB::Rails::Configuration).to receive(:tags_middleware).and_return(tags_middleware)
  end

  it "writes metric" do
    skip("https://github.com/rails/rails/issues/30586") unless payload_names_fixed_in_rails?

    get "/metrics"

    expect_metric(
      tags:   a_hash_including(
        location:        "MetricsController#index",
        hook:            "sql",
        name:            "Metric Create",
        class_name:      "Metric",
        operation:       "INSERT",
        additional_tag:  :value,
        server:          Socket.gethostname,
        app_name:        :app_name,
        tags_middleware: :tags_middleware
      ),
      values: a_hash_including(
        additional_value: :value,
        request_id:       :request_id,
        value:            be_between(1, 30),
        sql:              "INSERT INTO \"metrics\" (\"name\", \"created_at\", \"updated_at\") VALUES (xxx)"
      )
    )
  end

  it "writes metric" do
    skip("https://github.com/rails/rails/issues/30586") if payload_names_fixed_in_rails?

    get "/metrics"

    expect_metric(
      tags:   a_hash_including(
        location:        "MetricsController#index",
        hook:            "sql",
        name:            "SQL",
        class_name:      "SQL",
        operation:       "INSERT",
        additional_tag:  :value,
        server:          Socket.gethostname,
        app_name:        :app_name,
        tags_middleware: :tags_middleware
      ),
      values: a_hash_including(
        additional_value: :value,
        request_id:       :request_id,
        value:            be_between(1, 30),
        sql:              "INSERT INTO \"metrics\" (\"name\", \"created_at\", \"updated_at\") VALUES (xxx)"
      )
    )
  end

  it "includes correct timestamps" do
    travel_to Time.zone.local(2018, 1, 1, 9, 0, 0)

    get "/metrics"

    expect_metric(
      tags:      a_hash_including(
        location: "MetricsController#index",
        hook:     "sql"
      ),
      timestamp: 1_514_797_200
    )
  end

  it "does not write metric when hook is ignored" do
    allow_any_instance_of(InfluxDB::Rails::Configuration).to receive(:ignored_hooks).and_return(["sql.active_record"])

    get "/metrics"

    expect_no_metric(
      tags: a_hash_including(
        location: "MetricsController#index",
        hook:     "sql"
      )
    )
  end

  def payload_names_fixed_in_rails?
    Rails::VERSION::MAJOR > 5 ||
      rails_after_5_1?
  end

  def rails_after_5_1?
    Rails::VERSION::MAJOR == 5 &&
      Rails::VERSION::MINOR > 1
  end
end