newrelic_rpm 3.5.2.17 → 3.5.3.24

data/test/multiverse/suites/active_record/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

data/test/multiverse/suites/active_record/encoding_test.rb

@@ -0,0 +1,26 @@
+# https://support.newrelic.com/tickets/2101
+# https://github.com/newrelic/rpm/pull/42
+# https://github.com/newrelic/rpm/pull/45
+require 'new_relic/agent/instrumentation/active_record'
+
+class DatabaseAdapter
+  # we patch in here
+  def log(*args)
+  end
+  include ::NewRelic::Agent::Instrumentation::ActiveRecord
+end
+
+class EncodingTest < Test::Unit::TestCase
+
+  def test_should_not_bomb_out_if_a_query_is_in_an_invalid_encoding
+    # Contains invalid UTF8 Byte
+    query = "select ICS95095010000000000083320000BS01030000004100+\xFF00000000000000000"
+    if RUBY_VERSION >= '1.9'
+      # Force the query to an invalid encoding
+      query.force_encoding 'UTF-8'
+      assert_equal false, query.valid_encoding?
+    end
+    db = DatabaseAdapter.new
+    db.send(:log, query)
+  end
+end

data/test/multiverse/suites/agent_only/Envfile

@@ -0,0 +1,3 @@
+gemfile <<-RB
+  gem 'rack'
+RB

data/test/multiverse/suites/agent_only/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

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

@@ -0,0 +1,53 @@
+# https://newrelic.atlassian.net/browse/RUBY-765
+require 'fake_collector'
+
+class HttpResponseCodeTest < Test::Unit::TestCase
+  def setup
+    $collector ||= NewRelic::FakeCollector.new
+    $collector.reset
+    $collector.run
+    NewRelic::Agent.manual_start(:send_data_on_exit => false)
+    @agent = NewRelic::Agent.instance
+  end
+
+  def teardown
+    $collector.reset
+    NewRelic::Agent.shutdown
+  end
+
+  def test_request_entity_too_large
+    $collector.mock['metric_data'] = [413, {'exception' => {'error_type' => 'RuntimeError', 'message' => 'too much'}}]
+
+    @agent.stats_engine.get_stats_no_scope('Custom/too_big') \
+      .record_data_point(1)
+    assert_equal 1, @agent.stats_engine \
+      .get_stats_no_scope('Custom/too_big').call_count
+
+    @agent.send(:harvest_and_send_timeslice_data)
+
+    # make sure the data gets thrown away without crashing
+    assert_equal 0, @agent.stats_engine \
+      .get_stats_no_scope('Custom/too_big').call_count
+
+    # make sure we actually talked to the collector
+    assert_equal(1, $collector.agent_data.select{|x| x.action == 'metric_data'}.size)
+  end
+
+  def test_unsupported_media_type
+    $collector.mock['metric_data'] = [415, {'exception' => {'error_type' => 'RuntimeError', 'message' => 'looks bad'}}]
+
+    @agent.stats_engine.get_stats_no_scope('Custom/too_big') \
+      .record_data_point(1)
+    assert_equal 1, @agent.stats_engine \
+      .get_stats_no_scope('Custom/too_big').call_count
+
+    @agent.send(:harvest_and_send_timeslice_data)
+
+    # make sure the data gets thrown away without crashing
+    assert_equal 0, @agent.stats_engine \
+      .get_stats_no_scope('Custom/too_big').call_count
+
+    # make sure we actually talked to the collector
+    assert_equal(1, $collector.agent_data.select{|x| x.action == 'metric_data'}.size)
+  end
+end

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

@@ -0,0 +1,109 @@
+# https://newrelic.atlassian.net/wiki/display/eng/The+Terror+and+Glory+of+Transaction+Traces
+# https://newrelic.atlassian.net/browse/RUBY-914
+
+class MarshalingTest < Test::Unit::TestCase
+  def setup
+    NewRelic::Agent.manual_start(:'transaction_tracer.transaction_threshold' => 0.0)
+    @agent = NewRelic::Agent.instance
+    @sampler = @agent.transaction_sampler
+
+    $collector ||= NewRelic::FakeCollector.new
+    $collector.reset
+    $collector.mock['connect'] = [200, {'return_value' => { 'agent_run_id' => 666 }}]
+    $collector.run
+  end
+
+  def teardown
+    Thread.current[:transaction_sample_builder] = nil
+    $collector.reset
+  end
+
+  def test_transaction_trace_marshaling
+    # create fake transaction trace
+    time = Time.now.to_f
+    @sampler.notice_first_scope_push time
+    @sampler.notice_transaction '/path', nil, {}
+    @sampler.notice_push_scope "a"
+    @sampler.notice_push_scope "ab"
+    sleep 1
+    @sampler.notice_pop_scope "ab"
+    @sampler.notice_pop_scope "a"
+    @sampler.notice_scope_empty
+
+    expected_sample = @sampler.instance_variable_get(:@slowest_sample)
+
+    @agent.service.connect
+    @agent.send(:harvest_and_send_slowest_sample)
+
+    if NewRelic::Agent::NewRelicService::JsonMarshaller.is_supported?
+      marshaller = NewRelic::Agent::NewRelicService::JsonMarshaller.new
+    else
+      marshaller = NewRelic::Agent::NewRelicService::PrubyMarshaller.new
+    end
+
+    assert_equal(666,
+                 $collector.agent_data.select{|x| x.action == 'transaction_sample_data'}[0].body[0])
+    assert_equal(expected_sample.to_collector_array(marshaller),
+                 $collector.agent_data.select{|x| x.action == 'transaction_sample_data'}[0].body[1][0])
+  end
+
+  def test_metric_data_marshalling
+    stats = NewRelic::Agent.instance.stats_engine.get_stats_no_scope('Custom/test/method')
+    stats.record_data_point(1.0)
+    stats.record_data_point(2.0, 1.0)
+    expected = [ [ {'name' => 'Custom/test/method', 'scope' => ''},
+                           [2, 3.0, 2.0, 1.0, 2.0, 5.0] ] ]
+
+    @agent.service.connect
+    @agent.send(:harvest_and_send_timeslice_data)
+
+    assert_equal(666,
+       $collector.agent_data.select{|x| x.action == 'metric_data'}[0].body[0])
+
+    metric_data = $collector.agent_data \
+      .select{|x| x.action == 'metric_data'}[0].body[3]
+    assert metric_data
+
+    custom_metric = metric_data \
+      .select{|m| m[0]['name'] == 'Custom/test/method' }
+    assert_equal(expected, custom_metric)
+  end
+
+  def test_error_data_marshalling
+    @agent.error_collector.notice_error(Exception.new('test error'))
+    @agent.service.connect
+    @agent.send(:harvest_and_send_errors)
+
+    assert_equal(666,
+       $collector.agent_data.select{|x| x.action == 'error_data'}[0].body[0])
+
+    error_data = $collector.agent_data \
+      .select{|x| x.action == 'error_data'}[0].body[1][0]
+    assert_equal('test error', error_data[2])
+  end
+
+  def test_sql_trace_data_marshalling
+    @agent.sql_sampler.notice_first_scope_push(nil)
+    @agent.sql_sampler.notice_sql("select * from test",
+                                  "Database/test/select",
+                                  nil, 1.5)
+    @agent.sql_sampler.notice_scope_empty
+
+    @agent.service.connect
+    @agent.send(:harvest_and_send_slowest_sql)
+
+    sql_data = $collector.agent_data \
+      .select{|x| x.action == 'sql_trace_data'}[0].body[0]
+    assert_equal('select * from test', sql_data[0][3])
+  end
+
+  def test_connect_marshalling
+    @agent.service.connect('pid' => 1, 'agent_version' => '9000',
+                           'app_name' => 'test')
+
+    connect_data = $collector.agent_data \
+      .select{|x| x.action == 'connect'}[0].body[0]
+    assert_equal '9000', connect_data['agent_version']
+  end
+end
+

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