RubyGems - newrelic_rpm - Versions diffs - 4.0.0.332 → 4.4.0.336 - Mend

newrelic_rpm 4.0.0.332 → 4.4.0.336

Files changed (917) hide show

data/test/new_relic/agent/commands/agent_command_router_test.rb DELETED Viewed

@@ -1,256 +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 File.expand_path(File.join(File.dirname(__FILE__),'..','..','data_container_tests'))
-require 'new_relic/agent/commands/agent_command_router'
-require 'new_relic/agent/commands/xray_session'
-class AgentCommandRouterTest < Minitest::Test
-  DEFAULT_ARGS = {
-    "profile_id" => 42
-  }
-  BAZZLE_ID = 123
-  BAZZLE = [BAZZLE_ID,{
-    "name" => "bazzle",
-    "arguments" => DEFAULT_ARGS
-  }]
-  BOOM_ID = 666
-  BOOM = [BOOM_ID,{
-    "name" => "boom",
-    "arguments" => DEFAULT_ARGS
-  }]
-  UNRECOGNIZED_ID = 42
-  UNRECOGNIZED = [UNRECOGNIZED_ID, {
-    "name" => "JIBBERISH",
-    "arguments" => {}
-  }]
-  attr_reader :service, :agent_commands, :calls
-  def setup
-    @service = stub
-    NewRelic::Agent.agent.stubs(:service).returns(@service)
-    @calls = []
-    @events = NewRelic::Agent::EventListener.new
-    @agent_commands = NewRelic::Agent::Commands::AgentCommandRouter.new(@events)
-    @agent_commands.handlers["bazzle"] = Proc.new { |args| handle_bazzle_command(args) }
-    @agent_commands.handlers["boom"]   = Proc.new { |args| handle_boom_command(args) }
-  end
-  def teardown
-    agent_commands.backtrace_service.worker_thread.join if agent_commands.backtrace_service.worker_thread
-  end
-  # Helpers for DataContainerTests
-  def create_container
-    @agent_commands
-  end
-  def max_data_items
-    1
-  end
-  def populate_container(container, n)
-    start_profile('duration' => 1.0)
-    advance_time(1.1)
-    agent_commands.backtrace_service.worker_thread.join if agent_commands.backtrace_service.worker_thread
-  end
-  # General command routing
-  def test_check_for_and_handle_agent_commands_dispatches_command
-    service.stubs(:get_agent_commands).returns([BAZZLE])
-    service.stubs(:agent_command_results)
-    agent_commands.check_for_and_handle_agent_commands
-    assert_equal [DEFAULT_ARGS], calls
-  end
-  def test_check_for_and_handle_agent_commands_generates_results
-    service.stubs(:get_agent_commands).returns([BAZZLE])
-    service.expects(:agent_command_results).with({ BAZZLE_ID.to_s => {} })
-    agent_commands.check_for_and_handle_agent_commands
-  end
-  def test_check_for_and_handle_agent_commands_dispatches_with_error
-    service.stubs(:get_agent_commands).returns([BOOM])
-    service.expects(:agent_command_results).with({ BOOM_ID.to_s => { "error" => "BOOOOOM" }})
-    agent_commands.check_for_and_handle_agent_commands
-  end
-  def test_check_for_and_handle_agent_commands_allows_multiple
-    service.stubs(:get_agent_commands).returns([BAZZLE, BOOM])
-    service.expects(:agent_command_results).with({ BAZZLE_ID.to_s => {},
-                                                   BOOM_ID.to_s => { "error" => "BOOOOOM" }})
-    agent_commands.check_for_and_handle_agent_commands
-  end
-  def test_check_for_and_handle_agent_commands_doesnt_call_results_if_no_commands
-    service.stubs(:get_agent_commands).returns([])
-    service.expects(:agent_command_results).never
-    agent_commands.check_for_and_handle_agent_commands
-  end
-  def test_unrecognized_commands
-    service.stubs(:get_agent_commands).returns([UNRECOGNIZED])
-    service.stubs(:agent_command_results)
-    expects_logging(:debug, regexp_matches(/unrecognized/i))
-    agent_commands.check_for_and_handle_agent_commands
-  end
-  # Start/stop X-Ray tests
-  def test_empty_agent_commands_stops_running_xray
-    start_xray_session(123)
-    service.stubs(:get_agent_commands).returns([])
-    agent_commands.check_for_and_handle_agent_commands
-    assert_false agent_commands.xray_session_collection.include?(123)
-  end
-  # Harvesting tests
-  if NewRelic::Agent::Threading::BacktraceService.is_supported?
-    include NewRelic::BasicDataContainerTests
-    def test_harvest_not_started
-      result = agent_commands.harvest!
-      assert_equal([], result)
-    end
-    def test_harvest_with_profile_in_progress
-      start_profile('duration' => 1.0)
-      result = agent_commands.harvest!
-      assert_equal([], result)
-    end
-    def test_harvest_with_profile_completed
-      start_profile('duration' => 1.0)
-      advance_time(1.1)
-      result = agent_commands.harvest!
-      refute_empty result
-    end
-    def test_can_stop_multiple_times_safely
-      start_profile('duration' => 1.0)
-      advance_time(1.1)
-      agent_commands.thread_profiler_session.stop(true)
-      result = agent_commands.harvest!
-      refute_empty result
-    end
-    def test_transmits_after_forced_stop
-      start_profile('duration' => 1.0)
-      agent_commands.thread_profiler_session.stop(true)
-      result = agent_commands.harvest!
-      refute_empty result
-    end
-    def test_harvest_following_before_shutdown_with_no_profile
-      @events.notify(:before_shutdown)
-      result = agent_commands.harvest!
-      assert_empty result
-    end
-    def test_harvest_following_before_shutdown_with_active_profile
-      start_profile('duration' => 1.0)
-      @events.notify(:before_shutdown)
-      result = agent_commands.harvest!
-      refute_empty result
-    end
-    def test_harvest_with_xray_sessions_in_progress
-      start_xray_session(123)
-      start_xray_session(456)
-      sample_on_profiles
-      result = agent_commands.harvest!
-      assert_equal 2, result.length
-    end
-    def test_harvest_with_xray_sessions_and_thread_profile_in_progress
-      start_xray_session(123)
-      start_xray_session(456)
-      start_profile('duration' => 1.0)
-      sample_on_profiles
-      result = agent_commands.harvest!
-      assert_equal 2, result.length
-    end
-    def test_harvest_with_xray_sessions_and_completed_thread_profile
-      start_xray_session(123)
-      start_xray_session(456)
-      start_profile('duration' => 1.0)
-      sample_on_profiles
-      advance_time(1.1)
-      result = agent_commands.harvest!
-      assert_equal 3, result.length
-    end
-  end
-  # Helpers
-  def handle_bazzle_command(command)
-    calls << command.arguments
-  end
-  def handle_boom_command(command)
-    raise NewRelic::Agent::Commands::AgentCommandRouter::AgentCommandError.new("BOOOOOM")
-  end
-  def start_profile(args={})
-    freeze_time
-    agent_commands.backtrace_service.worker_loop.stubs(:run)
-    agent_commands.thread_profiler_session.start(create_agent_command(args))
-  end
-  def start_xray_session(id)
-    args = { 'x_ray_id' => id, 'key_transaction_name' => "txn_#{id}" }
-    session = NewRelic::Agent::Commands::XraySession.new(args)
-    agent_commands.backtrace_service.worker_loop.stubs(:run)
-    agent_commands.xray_session_collection.add_session(session)
-  end
-  def sample_on_profiles
-    agent_commands.backtrace_service.profiles.each do |(_, profile)|
-      profile.aggregate([], :request, Thread.current)
-    end
-  end
-end

data/test/new_relic/agent/commands/agent_command_test.rb DELETED Viewed

@@ -1,37 +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/commands/agent_command'
-module NewRelic::Agent::Commands
-  class AgentCommandTest < Minitest::Test
-    ID = 123
-    NAME = 'nuke_it_from_orbit'
-    ARGUMENTS = { "profile_id" => 42 }
-    NUKE_IT_FROM_ORBIT = [ID,{
-      "name" => NAME,
-      "arguments" => ARGUMENTS
-    }]
-    def test_destructures_name_from_collector_command
-      command = AgentCommand.new(NUKE_IT_FROM_ORBIT)
-      expected = NAME
-      assert_equal expected, command.name
-    end
-    def test_destructures_id_from_collector_command
-      command = AgentCommand.new(NUKE_IT_FROM_ORBIT)
-      expected = ID
-      assert_equal expected, command.id
-    end
-    def test_destructures_arguments_from_collector_command
-      command = AgentCommand.new(NUKE_IT_FROM_ORBIT)
-      expected = ARGUMENTS
-      assert_equal expected, command.arguments
-    end
-  end
-end

data/test/new_relic/agent/commands/thread_profiler_session_test.rb DELETED Viewed

@@ -1,216 +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 'base64'
-require 'thread'
-require 'timeout'
-require 'zlib'
-require 'new_relic/agent/threading/backtrace_service'
-require 'new_relic/agent/threading/threaded_test_case'
-require 'new_relic/agent/commands/thread_profiler_session'
-module ThreadProfilerSessionTestHelpers
-  START = {
-    "profile_id" => 42,
-    "sample_period" => 0.02,
-    "duration" => 0.025,
-    "only_runnable_threads" => false,
-    "only_request_threads" => false,
-    "profile_agent_code" => false,
-  }
-  STOP = {
-    "profile_id" => 42,
-    "report_data" => true,
-  }
-  STOP_AND_DISCARD = {
-    "profile_id" => 42,
-    "report_data" => false,
-  }
-  def start_command
-    create_agent_command(START)
-  end
-  def stop_command
-    create_agent_command(STOP)
-  end
-  def stop_and_discard_command
-    create_agent_command(STOP_AND_DISCARD)
-  end
-end
-if !NewRelic::Agent::Threading::BacktraceService.is_supported?
-  class ThreadProfilerUnsupportedTest < Minitest::Test
-    include ThreadProfilerSessionTestHelpers
-    def setup
-      backtrace_service = NewRelic::Agent::Threading::BacktraceService.new
-      @profiler = NewRelic::Agent::Commands::ThreadProfilerSession.new(backtrace_service)
-    end
-    def test_thread_profiling_isnt_supported
-      assert_equal false, NewRelic::Agent::Threading::BacktraceService.is_supported?
-    end
-    def test_stop_is_safe_when_not_supported
-      @profiler.start(start_command)
-      @profiler.stop(true)
-    end
-    def test_wont_start_and_reports_error
-      assert_raises NewRelic::Agent::Commands::AgentCommandRouter::AgentCommandError do
-        @profiler.handle_start_command(start_command)
-      end
-      assert_equal false, @profiler.running?
-    end
-  end
-else
-  require 'json'
-  class ThreadProfilerSessionTest < Minitest::Test
-    include ThreadedTestCase
-    include ThreadProfilerSessionTestHelpers
-    def setup
-      setup_fake_threads
-      backtrace_service = NewRelic::Agent::Threading::BacktraceService.new
-      backtrace_service.worker_loop.stubs(:run).returns(nil)
-      @profiler = NewRelic::Agent::Commands::ThreadProfilerSession.new(backtrace_service)
-    end
-    def teardown
-      NewRelic::Agent.instance.stats_engine.clear_stats
-      teardown_fake_threads
-    end
-    def test_is_supported
-      assert NewRelic::Agent::Threading::BacktraceService.is_supported?
-    end
-    def test_is_not_running
-      assert_false @profiler.running?
-    end
-    def test_is_running
-      @profiler.start(start_command)
-      assert @profiler.running?
-    end
-    def test_is_not_ready_to_harvest_if_no_profile_started
-      assert_false @profiler.ready_to_harvest?
-    end
-    def test_is_ready_to_harvest_if_duration_has_elapsed
-      freeze_time
-      @profiler.start(start_command)
-      assert_false @profiler.ready_to_harvest?
-      advance_time(0.026)
-      assert @profiler.ready_to_harvest?
-    end
-    def test_can_stop_a_running_profile
-      @profiler.start(start_command)
-      assert @profiler.running?
-      @profiler.stop(true)
-      assert_false @profiler.running?
-      refute_nil @profiler.harvest
-    end
-    def test_can_stop_a_running_profile_and_discard
-      @profiler.start(start_command)
-      assert @profiler.running?
-      @profiler.stop(false)
-      assert_nil @profiler.harvest
-    end
-    def test_wont_crash_if_stopping_when_not_started
-      @profiler.stop(true)
-      assert_equal false, @profiler.running?
-    end
-    def test_handle_start_command_starts_running
-      @profiler.handle_start_command(start_command)
-      assert_equal true, @profiler.running?
-    end
-    def test_config_can_disable_running
-      with_config(:'thread_profiler.enabled' => false) do
-        assert_raises NewRelic::Agent::Commands::AgentCommandRouter::AgentCommandError do
-          @profiler.handle_start_command(start_command)
-        end
-        assert_false @profiler.running?
-      end
-    end
-    def test_handle_stop_command
-      @profiler.start(start_command)
-      assert @profiler.running?
-      @profiler.handle_stop_command(stop_command)
-      assert_false @profiler.running?
-    end
-    def test_handle_stop_command_and_discard
-      @profiler.start(start_command)
-      assert @profiler.running?
-      @profiler.handle_stop_command(stop_and_discard_command)
-      assert_nil @profiler.harvest
-    end
-    def test_handle_start_command_wont_start_second_profile
-      @profiler.handle_start_command(start_command)
-      original_profile = @profiler.instance_variable_get(:@profile)
-      begin
-        @profiler.handle_start_command(start_command)
-      rescue NewRelic::Agent::Commands::AgentCommandRouter::AgentCommandError
-      end
-      assert_equal original_profile, @profiler.harvest
-    end
-    def test_start_command_sent_twice_raises_error
-      @profiler.handle_start_command(start_command)
-      assert_raises NewRelic::Agent::Commands::AgentCommandRouter::AgentCommandError do
-        @profiler.handle_start_command(start_command)
-      end
-    end
-    def test_harvested_profile_doesnt_still_report_as_ready_to_harvest
-      freeze_time
-      @profiler.handle_start_command(start_command)
-      advance_time(1.0)
-      @profiler.stop(true)
-      assert @profiler.ready_to_harvest?
-      @profiler.harvest
-      assert_false @profiler.ready_to_harvest?
-    end
-    def test_command_attributes_passed_along
-      @profiler.handle_start_command(start_command)
-      @profiler.handle_stop_command(stop_command)
-      profile = @profiler.harvest
-      assert_equal 42,  profile.profile_id
-      assert_equal 0.02, profile.requested_period
-    end
-  end
-end