# encoding: utf-8
# This file is distributed under New Relic's license terms.
# See https://github.com/newrelic/rpm/blob/master/LICENSE for complete details.

if defined?(::Rails)

require File.expand_path(File.join(File.dirname(__FILE__),'..', '..',
                                   'test_helper'))
require 'rack/test'
require 'new_relic/rack/developer_mode'

ENV['RACK_ENV'] = 'test'

class DeveloperModeTest < Minitest::Test
  include Rack::Test::Methods
  include TransactionSampleTestHelper

  def app
    mock_app = lambda { |env| [500, {}, "Don't touch me!"] }
    NewRelic::Rack::DeveloperMode.new(mock_app)
  end

  def setup
    @test_config = { :developer_mode => true, :disable_harvest_thread => true }
    NewRelic::Agent.config.add_config_for_testing(@test_config)
    run_sample_trace('/here')
    run_sample_trace('/there')
    run_sample_trace('/somewhere')
    @sampler = NewRelic::Agent.instance.transaction_sampler
  end

  def teardown
    NewRelic::Agent.config.remove_config(@test_config)
  end

  def test_index_displays_all_samples
    get '/newrelic'

    assert last_response.ok?
    assert last_response.body.include?('/here')
    assert last_response.body.include?('/there')
    assert last_response.body.include?('/somewhere')
  end

  def test_show_sample_summary_displays_sample_details
    get "/newrelic/show_sample_summary?id=#{@sampler.dev_mode_sample_buffer.samples[0].sample_id}"

    assert last_response.ok?
    assert last_response.body.include?('/here')
    assert last_response.body.include?('SandwichesController')
    assert last_response.body.include?('index')
  end

  def test_explain_sql_displays_query_plan
    sample = @sampler.dev_mode_sample_buffer.samples[0]
    sql_segment = app.send(:sql_segments, sample)[0]
    explain_results = [["Id", "Select Type", "Table", "Type", "Possible Keys", "Key", "Key Length",
                        "Ref", "Rows", "Extra"],
                       [["1", "SIMPLE", "sandwiches", "range", "PRIMARY", "PRIMARY", "4", "", "1",
                         "Using index"]]]
    NewRelic::Agent::Transaction::TraceNode.any_instance.expects(:explain_sql).returns(explain_results)

    get "/newrelic/explain_sql?id=#{sample.sample_id}&segment=#{sql_segment.object_id}"

    assert last_response.ok?
    assert last_response.body.include?('PRIMARY')
    assert last_response.body.include?('Key Length')
    assert last_response.body.include?('Using index')
  end

  def test_doesnt_record_transaction
    NewRelic::Agent.instance.transaction_sampler.dev_mode_sample_buffer.reset!

    get "/newrelic"

    assert_empty NewRelic::Agent.instance.transaction_sampler.dev_mode_sample_buffer.samples
  end

  def test_request_attributes
    attributes = NewRelic::Agent::Transaction::Attributes.new(NewRelic::Agent.instance.attribute_filter)
    attributes.add_agent_attribute("request.parameters.foo", "bar", NewRelic::Agent::AttributeFilter::DST_ALL)
    attributes.add_agent_attribute(:"request.headers.referer", "/somewhere", NewRelic::Agent::AttributeFilter::DST_ALL)
    sample = stub(:attributes => attributes)

    expected = {
      "request.parameters.foo" => "bar"
    }

    result = app.send(:request_attributes_for, sample)
    assert_equal expected, result
  end
end

else
  puts "Skipping tests in #{__FILE__} because Rails is unavailable"
end