# 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/agent/autostart'

class AutostartTest < Test::Unit::TestCase

  def test_typically_the_agent_should_autostart
    assert ::NewRelic::Agent::Autostart.agent_should_start?
  end

  if defined?(::Rails)
    def test_agent_wont_autostart_if_RAILS_CONSOLE_constant_is_defined
      assert !defined?(::Rails::Console), "precondition: Rails::Console shouldn't be defined"
      Rails.const_set(:Console, Class.new)
      assert ! ::NewRelic::Agent::Autostart.agent_should_start?, "Agent shouldn't autostart in Rails Console session"
    ensure
      Rails.send(:remove_const, :Console)
    end
  else
    puts "Skipping tests in #{__FILE__} because Rails is unavailable"
  end

  def test_agent_will_autostart_if_global_CONSOLE_constant_is_defined
    Object.const_set(:Console, Class.new)
    assert ::NewRelic::Agent::Autostart.agent_should_start?, "Agent shouldn't find ::Console"
  ensure
    Object.send(:remove_const, :Console)
  end

  def test_agent_wont_start_if_dollar_0_is_irb
    @orig_dollar_0, $0 = $0, '/foo/bar/irb'
    assert ! ::NewRelic::Agent::Autostart.agent_should_start?, "Agent shouldn't autostart when process is invoked by irb"
  ensure
    $0 = @orig_dollar_0
  end

  RAILS_DEFAULT_RAKE_TASKS = %w| about assets:clean assets:clobber
  assets:environment assets:precompile db:create db:drop db:fixtures:load
  db:migrate db:migrate:status db:rollback db:schema:cache:clear
  db:schema:cache:dump db:schema:dump db:schema:load db:seed db:setup
  db:structure:dump db:version doc:app log:clear middleware notes notes:custom
  rails:template rails:update routes secret spec spec:controllers spec:helpers
  spec:models spec:rcov stats test test:all test:all:db test:recent test:single
  test:uncommitted time:zones:all tmp:clear tmp:create |.each do |task|

    define_method("test_agent_wont_autostart_if_top_level_rake_task_is_#{task}") do
      Rake.stubs(:application => stub(:top_level_tasks => [task]))
      assert ! ::NewRelic::Agent::Autostart.agent_should_start?, "Agent shouldn't start during #{task.inspect} rake task"
    end
  end


  MyConst = true
  def test_blacklisted_constants_can_be_configured
    with_config('autostart.blacklisted_constants' => "IRB,::AutostartTest::MyConst") do
      assert ! ::NewRelic::Agent::Autostart.agent_should_start?, "Agent shouldn't autostart when environment contains blacklisted constant"
    end
  end

  def test_blacklisted_executable_can_be_configured
    @orig_dollar_0, $0 = $0, '/foo/bar/baz'
    with_config('autostart.blacklisted_executables' => 'boo,baz') do
      assert ! ::NewRelic::Agent::Autostart.agent_should_start?, "Agent shouldn't autostart when process is invoked by blacklisted executable"
    end
  ensure
    $0 = @orig_dollar_0
  end

  def test_blacklisted_rake_tasks_can_be_configured
    with_config('autostart.blacklisted_rake_tasks' => 'foo,bar,baz:bang') do
      Rake.stubs(:application => stub(:top_level_tasks => ['biz', 'baz:bang']))
      assert ! ::NewRelic::Agent::Autostart.agent_should_start?, "Agent shouldn't during blacklisted rake task"
    end
  end

  module ::Outer
    class Included
    end
  end

  class ::Excluded
  end

  module ::ContainsAnObject
    class ContainedObject
    end
  end

  def test_should_look_within_module
    assert_equal ::Outer::Included, NewRelic::Agent::Autostart.constant_is_defined?("Outer::Included")
  end

  def test_shouldnt_look_outside_module_for_class
    assert_equal false, NewRelic::Agent::Autostart.constant_is_defined?("Outer::Excluded")
  end

  def test_shouldnt_look_outside_module_for_module
    assert_equal false, NewRelic::Agent::Autostart.constant_is_defined?("Outer::Outer")
  end

  def test_should_allow_object_in_module_names
    assert_equal ::ContainsAnObject::ContainedObject,
                  NewRelic::Agent::Autostart.constant_is_defined?("ContainsAnObject::ContainedObject")
  end

end