newrelic_rpm 4.0.0.332 → 6.13.1

data/test/new_relic/agent/obfuscator_test.rb

@@ -1,77 +0,0 @@
-# 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/obfuscator"
-
-class NewRelic::Agent::ObfuscatorTest < Minitest::Test
-
-  OBFUSCATION_KEY = (1..40).to_a.pack('c*')
-  RUM_KEY_LENGTH  = 13
-
-  def setup
-    @config = { :license_key => OBFUSCATION_KEY }
-    NewRelic::Agent.config.add_config_for_testing(@config)
-  end
-
-  def teardown
-    NewRelic::Agent.config.remove_config(@config)
-    @obfuscator = nil
-  end
-
-  def obfuscator(length=nil)
-    @obfuscator ||= NewRelic::Agent::Obfuscator.new(OBFUSCATION_KEY, length)
-  end
-
-  def test_obfuscate_basic
-    assert_encoded(RUM_KEY_LENGTH,
-                   'a happy piece of small text',
-                   'YCJrZXV2fih5Y25vaCFtZSR2a2ZkZSp/aXV1')
-  end
-
-  def test_obfuscate_long_string
-    assert_encoded(RUM_KEY_LENGTH,
-                   'a happy piece of small text' * 5,
-                   'YCJrZXV2fih5Y25vaCFtZSR2a2ZkZSp/aXV1YyNsZHZ3cSl6YmluZCJsYiV1amllZit4aHl2YiRtZ3d4cCp7ZWhiZyNrYyZ0ZWhmZyx5ZHp3ZSVuZnh5cyt8ZGRhZiRqYCd7ZGtnYC11Z3twZCZvaXl6cix9aGdgYSVpYSh6Z2pgYSF2Znxx')
-  end
-
-  def test_obfuscate_utf8
-    assert_encoded(RUM_KEY_LENGTH,
-                   "foooooééoooo - blah",
-                   "Z21sa2ppxKHKo2RjYm4iLiRnamZg")
-  end
-
-
-  def test_decoding_blank
-    obfuscator = NewRelic::Agent::Obfuscator.new('query')
-    assert_equal "", obfuscator.deobfuscate("")
-  end
-
-  def test_decoding_empty_key
-    obfuscator = NewRelic::Agent::Obfuscator.new("")
-    assert_equal "querty", obfuscator.encode('querty')
-  end
-
-  def test_encode_with_nil_uses_empty_key
-    obfuscator = NewRelic::Agent::Obfuscator.new(nil)
-    assert_equal "querty", obfuscator.encode('querty')
-  end
-
-  def test_encoding_functions_can_roundtrip_utf8_text
-    str = 'Анастасі́я Олексі́ївна Каме́нських'
-    obfuscator = NewRelic::Agent::Obfuscator.new('potap')
-    encoded = obfuscator.obfuscate(str)
-    decoded = obfuscator.deobfuscate(encoded)
-    decoded.force_encoding( 'utf-8' ) if decoded.respond_to?( :force_encoding )
-    assert_equal str, decoded
-  end
-
-  def assert_encoded(key_length, text, expected)
-    output = obfuscator(key_length).obfuscate(text)
-    assert_equal(expected, output)
-
-    unoutput = obfuscator.obfuscate(Base64.decode64(output))
-    assert_equal Base64.encode64(text).gsub("\n", ''), unoutput
-  end
-end

data/test/new_relic/agent/parameter_filtering_test.rb

@@ -1,39 +0,0 @@
-# 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/parameter_filtering'
-
-module NewRelic
-  module Agent
-    class ParameterFilteringTest < Minitest::Test
-
-      def test_apply_filters_returns_params_when_rails_is_not_present
-        undefine_constant(:"ActionDispatch::Http::ParameterFilter") do
-          params = {"password" => "mypass"}
-          result = ParameterFiltering.apply_filters({}, params)
-          assert_equal params, result
-        end
-      end
-
-      def test_apply_filters_replaces_file_uploads_with_placeholder
-        env = {"CONTENT_TYPE" => "multipart/form-data"}
-        params = {
-          :name => "name",
-          :file => {
-            :filename => "data.jpg",
-            :tempfile => "file_data"
-          }
-        }
-
-        expected = {:name => "name", :file => "[FILE]"}
-        result = ParameterFiltering.apply_filters(env, params)
-        assert_equal expected, result
-
-        # argument should not be mutated
-        assert_equal({ :filename => "data.jpg", :tempfile => "file_data" }, params[:file])
-      end
-    end
-  end
-end

data/test/new_relic/agent/payload_metric_mapping_test.rb

@@ -1,74 +0,0 @@
-# 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/transaction_metrics'
-require 'new_relic/agent/payload_metric_mapping'
-
-module NewRelic
-  module Agent
-    class PayloadMetricMappingTest < Minitest::Test
-      def setup
-        @metrics = TransactionMetrics.new
-      end
-
-      def test_maps_datastore_all
-        @metrics.record_unscoped 'Datastore/all' do |stats|
-         stats.total_call_time = 42
-         stats.call_count = 3
-        end
-
-        result = {}
-        PayloadMetricMapping.append_mapped_metrics @metrics, result
-
-        assert_equal 42, result['databaseDuration']
-        assert_equal 3, result['databaseCallCount']
-      end
-
-      def test_maps_gc_transaction_all
-        @metrics.record_unscoped 'GC/Transaction/all', 42
-
-        result = {}
-        PayloadMetricMapping.append_mapped_metrics @metrics, result
-
-        assert_equal 42, result['gcCumulative']
-      end
-
-      def test_maps_web_frontend_queue_time
-        @metrics.record_unscoped 'WebFrontend/QueueTime', 42
-
-        result = {}
-        PayloadMetricMapping.append_mapped_metrics @metrics, result
-
-        assert_equal 42, result['queueDuration']
-      end
-
-      def test_maps_external_all_web
-        @metrics.record_unscoped 'External/allWeb' do |stats|
-         stats.total_call_time = 42
-         stats.call_count = 3
-        end
-
-        result = {}
-        PayloadMetricMapping.append_mapped_metrics @metrics, result
-
-        assert_equal 42, result['externalDuration']
-        assert_equal 3, result['externalCallCount']
-      end
-
-      def test_maps_external_all_other
-        @metrics.record_unscoped 'External/allOther' do |stats|
-         stats.total_call_time = 42
-         stats.call_count = 3
-        end
-
-        result = {}
-        PayloadMetricMapping.append_mapped_metrics @metrics, result
-
-        assert_equal 42, result['externalDuration']
-        assert_equal 3, result['externalCallCount']
-      end
-    end
-  end
-end

data/test/new_relic/agent/pipe_channel_manager_test.rb

@@ -1,316 +0,0 @@
-# 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 'timeout'
-require File.expand_path(File.join(File.dirname(__FILE__), '..', '..', 'test_helper'))
-require 'new_relic/agent/pipe_channel_manager'
-require 'new_relic/agent/transaction_event_primitive'
-
-class NewRelic::Agent::PipeChannelManagerTest < Minitest::Test
-  include TransactionSampleTestHelper
-
-  def setup
-    @test_config = { :monitor_mode => true }
-    NewRelic::Agent.agent.drop_buffered_data
-    NewRelic::Agent.config.add_config_for_testing(@test_config)
-
-    listener = NewRelic::Agent::PipeChannelManager::Listener.new
-    NewRelic::Agent::PipeChannelManager.instance_variable_set(:@listener, listener)
-
-    NewRelic::Agent.manual_start
-    NewRelic::Agent::TransactionState.tl_clear_for_testing
-  end
-
-  def teardown
-    NewRelic::Agent::PipeChannelManager.listener.close_all_pipes
-    NewRelic::Agent::PipeChannelManager.listener.stop
-    NewRelic::Agent.shutdown
-    NewRelic::Agent.config.remove_config(@test_config)
-  end
-
-  def test_registering_a_pipe
-    NewRelic::Agent::PipeChannelManager.listener.wake.in.expects(:<<).with('.')
-    NewRelic::Agent::PipeChannelManager.register_report_channel(1)
-    pipe = NewRelic::Agent::PipeChannelManager.channels[1]
-
-    assert pipe.out.kind_of?(IO)
-    assert pipe.in.kind_of?(IO)
-
-    NewRelic::Agent::PipeChannelManager.listener.close_all_pipes
-  end
-
-  if NewRelic::LanguageSupport.can_fork?
-    def test_listener_merges_timeslice_metrics
-      metric = 'Custom/test/method'
-
-      NewRelic::Agent.record_metric(metric, 1.0)
-
-      start_listener_with_pipe(666)
-
-      run_child(666) do
-        NewRelic::Agent.after_fork
-        new_engine = NewRelic::Agent::StatsEngine.new
-        new_engine.tl_record_unscoped_metrics(metric, 2.0)
-        service = NewRelic::Agent::PipeService.new(666)
-        service.metric_data(new_engine.harvest!)
-      end
-
-      assert_metrics_recorded(metric => { :total_call_time => 3.0 })
-    end
-
-    def test_listener_merges_transaction_traces
-      sampler = NewRelic::Agent.agent.transaction_sampler
-      sample = run_sample_trace
-      assert_equal(1, sampler.count)
-
-      start_listener_with_pipe(667)
-      run_child(667) do
-        NewRelic::Agent.after_fork
-        with_config(:'transaction_tracer.transaction_threshold' => 0.0) do
-          sample = run_sample_trace
-          service = NewRelic::Agent::PipeService.new(667)
-          service.transaction_sample_data(sampler.harvest!)
-        end
-      end
-
-      assert_equal(2, sampler.count)
-    end
-
-    def test_listener_merges_error_traces
-      sampler = NewRelic::Agent.agent.error_collector
-      sampler.notice_error(Exception.new("message"), :uri => '/myurl/',
-                           :metric => 'path', :referer => 'test_referer',
-                           :request_params => {:x => 'y'})
-      NewRelic::Agent.agent.merge_data_for_endpoint(:error_data, sampler.error_trace_aggregator.harvest!)
-
-      errors = NewRelic::Agent.agent.error_collector.error_trace_aggregator.instance_variable_get :@errors
-      assert_equal(1, errors.size)
-
-      start_listener_with_pipe(668)
-
-      run_child(668) do
-        NewRelic::Agent.after_fork
-        new_sampler = NewRelic::Agent::ErrorCollector.new
-        new_sampler.notice_error(Exception.new("new message"), :uri => '/myurl/',
-                                 :metric => 'path', :referer => 'test_referer',
-                                 :request_params => {:x => 'y'})
-        service = NewRelic::Agent::PipeService.new(668)
-        service.error_data(new_sampler.error_trace_aggregator.harvest!)
-      end
-      errors = NewRelic::Agent.agent.error_collector.error_trace_aggregator.instance_variable_get :@errors
-      assert_equal(2, errors.size)
-    end
-
-    def test_listener_merges_analytics_events
-      transaction_event_aggregator = NewRelic::Agent.agent.transaction_event_aggregator
-      reset_lifetime_counts! transaction_event_aggregator
-
-      start_listener_with_pipe(699)
-      NewRelic::Agent.agent.stubs(:connected?).returns(true)
-      run_child(699) do
-        NewRelic::Agent.after_fork(:report_to_channel => 699)
-        transaction_event_aggregator.append NewRelic::Agent::TransactionEventPrimitive.create({
-          :start_timestamp => Time.now,
-          :name => 'whatever',
-          :duration => 10,
-          :type => :controller
-        })
-        NewRelic::Agent.agent.send(:transmit_event_data)
-      end
-
-      _, samples = transaction_event_aggregator.harvest!
-
-      assert_equal(1, samples.size)
-      assert_lifetime_counts(transaction_event_aggregator, 1)
-    end
-
-    def test_listener_merges_error_events
-      error_event_aggregator = NewRelic::Agent.agent.error_collector.error_event_aggregator
-      reset_lifetime_counts! error_event_aggregator
-
-      sampler = NewRelic::Agent.agent.error_collector
-      sampler.notice_error(Exception.new("message"), :uri => '/myurl/',
-                           :metric => 'path', :referer => 'test_referer',
-                           :request_params => {:x => 'y'})
-
-      start_listener_with_pipe(668)
-
-      run_child(668) do
-        NewRelic::Agent.after_fork
-        new_sampler = NewRelic::Agent::ErrorCollector.new
-        new_sampler.notice_error(Exception.new("new message"), :uri => '/myurl/',
-                                 :metric => 'path', :referer => 'test_referer',
-                                 :request_params => {:x => 'y'})
-        service = NewRelic::Agent::PipeService.new(668)
-        service.error_event_data(new_sampler.error_event_aggregator.harvest!)
-      end
-      _, errors = error_event_aggregator.harvest!
-      assert_equal(2, errors.size)
-      assert_lifetime_counts(error_event_aggregator, 2)
-    end
-
-    def test_listener_merges_sql_traces
-      sampler = NewRelic::Agent.agent.sql_sampler
-      create_sql_sample(sampler)
-
-      start_listener_with_pipe(667)
-      run_child(667) do
-        NewRelic::Agent.after_fork
-        create_sql_sample(sampler)
-        service = NewRelic::Agent::PipeService.new(667)
-        service.sql_trace_data(sampler.harvest!)
-      end
-
-      assert_equal(2, sampler.harvest!.size)
-    end
-
-    def test_close_pipe_on_child_explicit_close
-      listener = start_listener_with_pipe(669)
-      pid = Process.fork do
-        NewRelic::Agent::PipeService.new(669)
-      end
-      Process.wait(pid)
-      listener.stop_listener_thread
-      assert_pipe_finished(669)
-    end
-
-    def test_close_pipe_on_child_exit
-      listener = start_listener_with_pipe(669)
-      pid = Process.fork do
-        NewRelic::Agent::PipeService.new(669)
-        exit!
-      end
-      Process.wait(pid)
-      listener.stop_listener_thread
-      assert_pipe_finished(669)
-    end
-
-    def test_manager_does_not_crash_when_given_bad_data
-      listener = start_listener_with_pipe(670)
-      pid = Process.fork do
-        listener.pipes[670].in << 'some unloadable garbage'
-      end
-      Process.wait(pid)
-      listener.stop
-    end
-
-    def test_manager_does_not_crash_when_given_unmarshallable_junk
-      listener = start_listener_with_pipe(671)
-      expects_logging(:error, any_parameters)
-
-      pid = Process.fork do
-        listener.pipes[671].write("\x00")
-      end
-      Process.wait(pid)
-      listener.stop
-    end
-
-    def pipe_finished?(id)
-      (!NewRelic::Agent::PipeChannelManager.channels[id] ||
-        NewRelic::Agent::PipeChannelManager.channels[id].closed?)
-    end
-
-    def assert_pipe_finished(id)
-      assert(pipe_finished?(id),
-        "Expected pipe with ID #{id} to be nil or closed")
-    end
-
-    def create_sql_sample(sampler)
-      state = NewRelic::Agent::TransactionState.tl_get
-      sampler.on_start_transaction(state, Time.now)
-      sampler.notice_sql("SELECT * FROM table", "ActiveRecord/Widgets/find", nil, 100, state)
-      sampler.on_finishing_transaction(state, 'noodles', Time.now)
-    end
-  end
-
-  def start_listener_with_pipe(pipe_id)
-    listener = NewRelic::Agent::PipeChannelManager.listener
-    listener.start
-    listener.register_pipe(pipe_id)
-    listener
-  end
-
-  def test_pipe_read_length_failure
-    write_pipe = stub(:set_encoding => nil, :closed? => false, :close => nil)
-
-    # If we only read three bytes, it isn't valid.
-    # We can't tell whether any four bytes or more are a "good" length or not.
-    read_pipe = stub(:read => "jrc")
-    IO.stubs(:pipe).returns([read_pipe, write_pipe])
-
-    # Includes the failed bytes
-    expects_logging(:error, includes("[6a 72 63]"))
-
-    pipe = NewRelic::Agent::PipeChannelManager::Pipe.new
-    assert_nil pipe.read
-  end
-
-  def test_pipe_read_length_nil_fails
-    write_pipe = stub(:set_encoding => nil, :closed? => false, :close => nil)
-
-    # No length at all returned on pipe, also a failure.
-    read_pipe = stub(:read => nil)
-    IO.stubs(:pipe).returns([read_pipe, write_pipe])
-
-    pipe = NewRelic::Agent::PipeChannelManager::Pipe.new
-    assert_nil pipe.read
-  end
-
-  def test_listener_pipes_race_condition
-    begin
-      listener = NewRelic::Agent::PipeChannelManager.listener
-      listener.instance_variable_set(:@select_timeout, 0.00001)
-      listener.start
-
-      listener.register_pipe(0)
-      pipe_out = listener.pipes[0].out
-
-      mutex = Mutex.new
-      mutex.lock
-      tried_to_close = false
-      pipe_out.singleton_class.send(:define_method, :closed?, Proc.new do
-        tried_to_close = true
-        mutex.synchronize do
-        end
-      end)
-
-      # Make sure we're actually paused in closed? call
-      until tried_to_close
-      end
-
-      t = Thread.new do
-        listener.register_pipe(1)
-      end
-
-      mutex.unlock
-      t.join
-    ensure
-      listener.close_all_pipes
-      listener.stop
-    end
-  end
-
-  def run_child(channel_id)
-    pid = Process.fork do
-      yield
-    end
-
-    Process.wait(pid)
-    until pipe_finished?(channel_id)
-      sleep 0.01
-    end
-  end
-
-  def assert_lifetime_counts container, value
-    buffer = container.instance_variable_get :@buffer
-    assert_equal value, buffer.captured_lifetime
-    assert_equal value, buffer.seen_lifetime
-  end
-
-  def reset_lifetime_counts! container
-    buffer = container.instance_variable_get :@buffer
-    buffer.instance_variable_set :@captured_lifetime, 0
-    buffer.instance_variable_set :@seen_lifetime, 0
-  end
-end