# 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.

require File.expand_path(File.join(File.dirname(__FILE__),'..','test_helper'))

require 'new_relic/environment_report'

class EnvironmentReportTest < Test::Unit::TestCase
  def setup
    @old_logic = ::NewRelic::EnvironmentReport.registered_reporters.dup
    @report = ::NewRelic::EnvironmentReport.new
  end

  def teardown
    ::NewRelic::EnvironmentReport.registered_reporters = @old_logic
  end

  def test_converts_to_array
    ::NewRelic::EnvironmentReport.report_on("something"){"awesome"}
    data = Array(::NewRelic::EnvironmentReport.new)
    expected = ["something", "awesome"]
    assert data.include?(expected), "expected to find #{expected} in #{data.inspect}"
  end

  def test_register_a_value_to_report_on
    ::NewRelic::EnvironmentReport.report_on("What time is it?") do
      "beer-o-clock"
    end
    assert_equal 'beer-o-clock', ::NewRelic::EnvironmentReport.new["What time is it?"]
  end

  def test_report_on_handles_errors_gracefully
    assert_nothing_raised do
      ::NewRelic::EnvironmentReport.report_on("What time is it?") do
        raise ArgumentError, "woah! something blew up"
      end
      assert_nil ::NewRelic::EnvironmentReport.new["What time is it?"]
    end
  end

  def test_it_does_not_set_keys_for_nil_values
    ::NewRelic::EnvironmentReport.report_on("What time is it?") do
      nil
    end
    assert ! NewRelic::EnvironmentReport.new.data.has_key?("What time is it?")
  end

  def test_can_set_an_environment_value_directly
    @report['My Value'] = "so awesome!!"
    assert_equal "so awesome!!", @report['My Value']
  end

  def test_it_knows_what_gems_are_in_the_environment
    assert(@report['Gems'].size > 5, "Expected at least 5 gems in #{@report['Gems'].inspect}")
    rake = @report['Gems'].detect{|s| s.include? 'rake'}
    assert_match(/^rake\([\d\.]+\)$/, rake)
  end

  def test_gathers_ruby_version
    assert_equal RUBY_VERSION, @report['Ruby version']
  end

  def test_gathers_system_info
    NewRelic::Agent::SystemInfo.stubs({
      :processor_count    => 8,
      :processor_arch     => 'x86_64',
      :os_version         => 'WiggleOS 1.1.1',
      :ruby_os_identifier => 'wiggleos'
    })
    report = ::NewRelic::EnvironmentReport.new
    assert_equal(8, report['Processors'])
    assert_equal('x86_64', report['Arch'])
    assert_equal('WiggleOS 1.1.1', report['OS version'])
    assert_equal('wiggleos', report['OS'])
  end

  def test_has_logic_for_keys
    [
      "Gems",
      "Plugin List",
      "Ruby version",
      "Ruby description",
      "Ruby platform",
      "Ruby patchlevel",
      'JRuby version',
      'Java VM version',
      'Processors',
      'Database adapter',
      'Framework',
      'Dispatcher',
      'Environment',
      'Arch',
      'OS version',
      'OS',
      'Rails Env',
      'Rails version',
      'Rails threadsafe',
    ].each do |key|
      assert NewRelic::EnvironmentReport.registered_reporters.has_key?(key), "Expected logic for #{key.inspect} in EnvironmentReport."
    end
  end
end