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

# Test the logic for loading the newrelic.yml config file.
#
# We will look at (in this priority):
# 1. ENV['NRCONFIG'] (if set)
# 2. ./config/newrelic.yml
# 3. ./newrelic.yml
# 4. ~/.newrelic/newrelic.yml
# 5. ~/newrelic.yml
#
# If this fails the agent will attempt to dial Lew Cirne's cell phone and ask
# that he verbally describe how it should be configured.

require 'multiverse_helpers'

class ConfigFileLoadingTest < Minitest::Test
  include MultiverseHelpers

  def setup
    # While FakeFS stubs out all the File system related libraries (i.e. File,
    # FileUtils, Dir, etc.)  It doesn't handle the Logger library that is used
    # to create the agent log file.  Therefore we check for the presence of the
    # log directory in the FakeFS, then error when we try to open the file in
    # the Real FS.  Set the log file to /dev/null to work around this issue.
    # It's incidental to the functionality being tested.
    #
    # '/dev/null' can be changed to 'stdout' if more debugging output is needed
    # in these tests.
    ENV['NEW_RELIC_LOG'] = '/dev/null'

    # Figure out where multiverse is in the real file system.
    @cwd = Dir.pwd

    # require the agent before we're in FakeFS so require doesn't hit the fake
    require 'newrelic_rpm'

    # Use a fake file system so we don't damage the real one.
    FakeFS.activate!
    FakeFS::FileSystem.clear

    FileUtils.mkdir_p(@cwd)
  end

  def teardown
    ENV['NEW_RELIC_LOG'] = nil
    FakeFS.deactivate!
  end

  def setup_config(path, manual_config_options = {})
    teardown_agent

    FileUtils.mkdir_p(File.dirname(path))
    Dir.chdir @cwd
    File.open(path, 'w') do |f|
      f.print <<-YAML
development:
  foo: "success!!"
test:
  foo: "success!!"
bazbangbarn:
  i_am: "bazbangbarn"
      YAML
    end
    setup_agent(manual_config_options)
  end

  def assert_config_read_from(path)
    setup_config(path)
    assert NewRelic::Agent.config[:foo] == "success!!", "Failed to read yaml config from #{path.inspect}\n\n#{NewRelic::Agent.config.inspect}"
  end

  def assert_config_not_read_from(path)
    setup_config(path)
    assert NewRelic::Agent.config[:foo] != "success!!", "Read yaml config from #{path.inspect}\n\n#{NewRelic::Agent.config.inspect}"
  end

  def test_config_loads_from_config_newrelic_yml
    assert_config_read_from(File.dirname(__FILE__) + "/config/newrelic.yml")
  end

  def test_config_loads_from_newrelic_yml
    assert_config_read_from(File.dirname(__FILE__) + "/newrelic.yml")
  end

  def test_config_loads_from_home_newrelic_yml
    assert_config_read_from(ENV['HOME'] + "/newrelic.yml")
  end

  def test_config_loads_from_home_dot_newrelic_newrelic_yml
    assert_config_read_from(ENV['HOME'] + "/.newrelic/newrelic.yml")
  end

  def test_config_loads_from_env_NRCONFIG
    ENV["NRCONFIG"] = "/tmp/foo/bar.yml"
    assert_config_read_from("/tmp/foo/bar.yml")
  ensure
    ENV["NRCONFIG"] = nil
  end

  def test_config_isnt_loaded_from_somewhere_crazy
    assert_config_not_read_from(File.dirname(__FILE__) + "/somewhere/crazy/newrelic.yml")
  end

  def test_config_will_load_settings_for_environment_passed_manual_start
    path = File.dirname(__FILE__) + "/config/newrelic.yml"

    # pass an env key to NewRelic::Agent.manual_start which should cause it to
    # load that section of newrelic.yml
    setup_config(path, {:env => 'bazbangbarn'} )
    assert_equal 'bazbangbarn', NewRelic::Agent.config[:i_am], "Agent.config did not load bazbangbarn config as requested"
  end
end