wd_newrelic_rpm 3.3.4.1 → 3.5.5

data/test/multiverse/suites/agent_only/method_visibility_test.rb

@@ -0,0 +1,98 @@
+require 'multiverse/color'
+
+class MethodVisibilityTest < Test::Unit::TestCase
+  extend Multiverse::Color
+  
+  class InstrumentedClass
+    include NewRelic::Agent::MethodTracer
+    include NewRelic::Agent::Instrumentation::ControllerInstrumentation
+
+    def public_method!
+    end
+
+    def public_transaction!
+    end
+
+    private
+    def private_method!
+    end
+
+    def private_transaction!
+    end
+
+    protected
+    def protected_method!
+    end
+
+    def protected_transaction!
+    end
+
+    add_method_tracer :public_method!
+    add_method_tracer :private_method!
+    add_method_tracer :protected_method!
+
+    add_transaction_tracer :public_transaction!
+    add_transaction_tracer :private_transaction!
+    add_transaction_tracer :protected_transaction!
+  end
+
+
+  class ObjectWithInstrumentation
+    include NewRelic::Agent::MethodTracer
+    include NewRelic::Agent::Instrumentation::ControllerInstrumentation
+  end
+
+  class ObjectWithTracers < ObjectWithInstrumentation
+    private
+    def a_private_method
+    end
+    def a_private_transaction
+    end
+    protected
+    def a_protected_method
+    end
+    def a_protected_transaction
+    end
+
+    add_method_tracer :a_private_method
+    add_method_tracer :a_protected_method
+    add_transaction_tracer :a_private_transaction
+    add_transaction_tracer :a_protected_transaction
+  end
+
+
+  def setup
+    @instance = InstrumentedClass.new
+  end
+  
+  if NewRelic::VERSION::STRING >= '3.4.0.2'
+    %w| public private protected |.each do |visibility|
+      define_method "test_should_preserve_visibility_of_#{visibility}_traced_method" do
+        assert @instance.send("#{visibility}_methods").map{|s|s.to_sym}.include?(:"#{visibility}_method!"), "Method #{visibility}_method should be #{visibility}"
+      end
+      
+      define_method "test_should_preserve_visibility_of_#{visibility}_traced_transaction" do
+        assert @instance.send("#{visibility}_methods").map{|s|s.to_sym}.include?(:"#{visibility}_transaction!"), "Transcation #{visibility}_transaction should be #{visibility}"
+      end
+    end
+
+    def test_tracing_non_public_methods_doesnt_add_public_methods
+      assert_equal [], ObjectWithTracers.public_instance_methods - ObjectWithInstrumentation.public_instance_methods
+    end
+
+    # FIXME: Currently including MethodTracer and ControllerInstrumentation
+    # adds a bunch of public methods to the class.  It probably shouldn't do this.
+    #def test_instrumentation_doesnt_add_any_public_methods
+    #  assert_equal [], ObjectWithInstrumentation.public_instance_methods - Object.public_instance_methods
+    #end
+    #
+
+
+
+  else
+    def test_truth
+      assert true # jruby freaks out if there are no tests defined in the test class
+    end
+    puts yellow('SKIPPED! 3.4.0.1 and earlier fail preserve_visibility tests')
+  end
+end

data/test/multiverse/suites/agent_only/pipe_manager_test.rb

@@ -0,0 +1,33 @@
+# https://newrelic.atlassian.net/browse/RUBY-669
+
+class PipeManagerTest < Test::Unit::TestCase
+  def setup
+    @listener = NewRelic::Agent::PipeChannelManager.listener
+  end
+
+  def teardown
+    @listener.stop
+  end
+  
+  def test_old_pipes_are_cleaned_up_after_timeout
+    @listener.timeout = 1
+    NewRelic::Agent::PipeChannelManager.register_report_channel(:timeout_test)
+    sleep 2
+    @listener.start
+    sleep 0.5 # give the thread some time to start, and clean things up
+    assert_nil NewRelic::Agent::PipeChannelManager.channels[:timeout_test]
+  end
+
+  def test_pipes_are_regularly_checked_for_freshness
+    @listener.select_timeout = 1
+    @listener.timeout = 2
+    NewRelic::Agent::PipeChannelManager.register_report_channel(:select_test)
+    
+    sleep 1.5
+    @listener.start
+    assert NewRelic::Agent::PipeChannelManager.channels[:select_test]
+    
+    sleep 1.5
+    assert_nil NewRelic::Agent::PipeChannelManager.channels[:select_test]
+  end
+end

data/test/multiverse/suites/agent_only/service_timeout_test.rb

@@ -0,0 +1,29 @@
+require 'socket'
+
+class ServiceTimeoutTest < Test::Unit::TestCase
+  def setup
+    hk = TCPServer.new('127.0.0.1',8081)
+
+    Thread.new {
+      client = hk.accept
+      what = client.gets
+      sleep 4
+      client.close
+      Thread.exit
+    }
+  end
+
+  def test_service_timeout
+    server = NewRelic::Control::Server.new('localhost', 8081,'127.0.0.1')  
+    NewRelic::Agent.config.apply_config(:timeout => 2)
+
+    service = NewRelic::Agent::NewRelicService.new('deadbeef', server)	
+
+    assert_raise Timeout::Error do
+      service.send('send_request',
+                   :uri => '/agent_listener/8/bd0e1d52adade840f7ca727d29a86249e89a6f1c/get_redirect_host',
+                   :encoding => 'UTF-8', :collector => server, :data => 'blah')
+    end
+  end
+
+end

data/test/multiverse/suites/agent_only/test_trace_method_with_punctuation.rb

@@ -0,0 +1,30 @@
+
+require 'test/unit'
+require 'newrelic_rpm'
+require 'new_relic/agent/method_tracer'
+
+
+class MethodInterrobangTest < Test::Unit::TestCase
+	include NewRelic::Agent::MethodTracer
+
+	def interrogate?
+		"say what?"
+	end
+
+	def mutate!
+		"oh yeah!"
+	end
+
+	add_method_tracer :interrogate?
+	add_method_tracer :mutate!
+
+	def test_alias_method_ending_in_question_mark
+		assert_respond_to self,:interrogate?
+		assert_equal "say what?", interrogate?
+	end
+
+	def test_aliase_method_ending_in_exclamation_makr
+		assert_respond_to self,:mutate!
+		assert_equal "oh yeah!",mutate!
+	end
+end

data/test/multiverse/suites/agent_only/test_trace_transaction_with_punctuation.rb

@@ -0,0 +1,32 @@
+
+require 'test/unit'
+require 'newrelic_rpm'
+require 'new_relic/agent/instrumentation/controller_instrumentation'
+
+
+class TransactionInterrobangTest < Test::Unit::TestCase
+  include NewRelic::Agent::Instrumentation::ControllerInstrumentation
+
+  def interrogate?
+    "say what?"
+  end
+
+  def mutate!
+    "oh yeah!"
+  end
+
+  add_transaction_tracer :interrogate?
+  add_transaction_tracer :mutate!
+
+  def test_alias_method_ending_in_question_mark
+    assert_respond_to self,:interrogate?
+    assert_respond_to self, :interrogate_with_newrelic_transaction_trace?
+    assert_equal "say what?", interrogate?
+  end
+
+  def test_aliase_method_ending_in_exclamation_makr
+    assert_respond_to self,:mutate!
+    assert_respond_to self, :mutate_with_newrelic_transaction_trace!
+    assert_equal "oh yeah!",mutate!
+  end
+end

data/test/multiverse/suites/agent_only/thread_profiling_test.rb

@@ -0,0 +1,80 @@
+# https://newrelic.atlassian.net/wiki/display/eng/Agent+Thread+Profiling
+# https://newrelic.atlassian.net/browse/RUBY-917
+
+if RUBY_VERSION >= '1.9'
+class ThreadProfilingTest < Test::Unit::TestCase
+  def setup
+    NewRelic::Agent.manual_start(:'thread_profiler.enabled' => true)
+
+    @agent = NewRelic::Agent.instance
+    @thread_profiler = @agent.thread_profiler
+
+    $collector ||= NewRelic::FakeCollector.new
+    $collector.reset
+    $collector.mock['connect'] = [200, {'return_value' => {"agent_run_id" => 666 }}]
+    $collector.mock['get_agent_commands'] = [200, {'return_value' => START_COMMAND}]
+    $collector.mock['agent_command_results'] = [200, {'return_value' => []}]
+    $collector.run
+  end
+
+  def teardown
+    $collector.reset
+  end
+
+  START_COMMAND = [[666,{
+      "name" => "start_profiler",
+      "arguments" => {
+        "profile_id" => -1,
+        "sample_period" => 0.01,
+        "duration" => 0.5,
+        "only_runnable_threads" => false,
+        "only_request_threads" => false,
+        "profile_agent_code" => false
+      }
+    }]]
+
+  STOP_COMMAND = [[666,{
+      "name" => "stop_profiler",
+      "arguments" => {
+        "profile_id" => -1,
+        "report_data" => true
+      }
+    }]]
+
+  # These are potentially fragile for being timing based
+  # START_COMMAND with 0.01 sampling and 0.5 duration expects to get
+  # roughly 50 polling cycles in. We check signficiantly less than that.
+
+  # STOP_COMMAND when immediately issued after a START_COMMAND is expected
+  # go only let a few cycles through, so we check less than 10
+
+  def test_thread_profiling
+    @agent.send(:check_for_agent_commands)
+    sleep(1)
+    NewRelic::Agent.shutdown
+
+    profile_data = $collector.calls_for('profile_data')[0]
+    assert_equal(666, profile_data[0])
+
+    poll_count = profile_data[1][0][3]
+    assert poll_count > 25, "Expected poll_count > 25, but was #{poll_count}"
+  end
+
+  def test_thread_profiling_can_stop
+    @agent.send(:check_for_agent_commands)
+
+    $collector.mock['get_agent_commands'] = [200, {'return_value' => STOP_COMMAND}]
+    @agent.send(:check_for_agent_commands)
+
+    sleep(0.1)
+    NewRelic::Agent.shutdown
+
+    profile_data = $collector.calls_for('profile_data')[0]
+    assert_equal(666, profile_data[0])
+
+    poll_count = profile_data[1][0][3]
+    assert poll_count < 10, "Expected poll_count < 10, but was #{poll_count}"
+  end
+end
+end
+

data/test/multiverse/suites/config_file_loading/Envfile

@@ -0,0 +1,7 @@
+# don't start the agent
+self.newrelic_gemfile_options=(':require => false')
+gemfile <<-RB
+  # stub file system so we can test that newrelic.yml can be loaded from
+  # various places.
+  gem 'fakefs', :require => 'fakefs/safe'
+RB

data/test/multiverse/suites/config_file_loading/config_file_loading_test.rb

@@ -0,0 +1,106 @@
+# 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.
+class ConfigFileLoadingTest < Test::Unit::TestCase
+
+  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
+    # Use a fake file system so we don't damage the real one.
+    FakeFS.activate!
+    FakeFS::FileSystem.clear
+
+    # require the agent after we're in FakeFS
+    require 'newrelic_rpm'
+
+    FileUtils.mkdir_p(@cwd)
+  end
+
+  def teardown
+    ENV['NEW_RELIC_LOG'] = nil
+    FakeFS.deactivate!
+  end
+
+  def setup_config(path, manual_config_options = {})
+    NewRelic::Agent.shutdown
+    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
+    NewRelic::Agent.reset_config
+    NewRelic::Agent.manual_start(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

data/test/multiverse/suites/datamapper/Envfile

@@ -0,0 +1,8 @@
+suite_condition("Datamapper not supported in 1.8.6") do
+  RUBY_VERSION != '1.8.6'
+end
+
+gemfile <<-RB
+  gem 'datamapper'
+RB
+

data/test/multiverse/suites/datamapper/config/newrelic.yml

@@ -0,0 +1,22 @@
+---
+development:
+  error_collector:
+    capture_source: true
+    enabled: true
+  apdex_t: 0.5
+  ssl: false
+  monitor_mode: true
+  license_key: bootstrap_newrelic_admin_license_key_000
+  developer_mode: false
+  app_name: test
+  host: 127.0.0.1
+  api_host: 127.0.0.1
+  port: 30303
+  transaction_tracer:
+    record_sql: obfuscated
+    enabled: true
+    stack_trace_threshold: 0.5
+    transaction_threshold: 1.0
+  capture_params: false
+  log_level: debug
+  log_file_path: agent.log