techarch-newrelic_rpm 2.10.2

data/recipes/newrelic.rb

@@ -0,0 +1,6 @@
+# The capistrano recipes in plugins are automatically
+# loaded from here.  From gems, they are available from
+# the lib directory.  We have to make them available from
+# both locations
+
+require File.join(File.dirname(__FILE__),'..','lib','new_relic','recipes')

data/techarch-newrelic_rpm.gemspec

@@ -0,0 +1,32 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{techarch-newrelic_rpm}
+  s.version = "2.10.2"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Bill Kayser" , "Philippe F. Monnet"]
+  s.email = ["bkayser@newrelic.com", "pfmonnet@gmail.com"]
+  s.date = %q{2009-11-04}
+  s.description = %q{New Relic Ruby Performance Monitoring Agent}
+  s.executables = ["mongrel_rpm", "newrelic_cmd"]
+  s.extra_rdoc_files = ["CHANGELOG", "LICENSE", "README.md", "bin/mongrel_rpm", "bin/newrelic_cmd", "lib/new_relic/agent.rb", "lib/new_relic/agent/agent.rb", "lib/new_relic/agent/chained_call.rb", "lib/new_relic/agent/collection_helper.rb", "lib/new_relic/agent/error_collector.rb", "lib/new_relic/agent/instrumentation/active_merchant.rb", "lib/new_relic/agent/instrumentation/active_record_instrumentation.rb", "lib/new_relic/agent/instrumentation/authlogic.rb", "lib/new_relic/agent/instrumentation/controller_instrumentation.rb", "lib/new_relic/agent/instrumentation/data_mapper.rb", "lib/new_relic/agent/instrumentation/dispatcher_instrumentation.rb", "lib/new_relic/agent/instrumentation/memcache.rb", "lib/new_relic/agent/instrumentation/merb/controller.rb", "lib/new_relic/agent/instrumentation/merb/dispatcher.rb", "lib/new_relic/agent/instrumentation/merb/errors.rb", "lib/new_relic/agent/instrumentation/net.rb", "lib/new_relic/agent/instrumentation/passenger_instrumentation.rb", "lib/new_relic/agent/instrumentation/rack.rb", "lib/new_relic/agent/instrumentation/rails/action_controller.rb", "lib/new_relic/agent/instrumentation/rails/action_web_service.rb", "lib/new_relic/agent/instrumentation/rails/dispatcher.rb", "lib/new_relic/agent/instrumentation/rails/errors.rb", "lib/new_relic/agent/instrumentation/sinatra.rb", "lib/new_relic/agent/method_tracer.rb", "lib/new_relic/agent/patch_const_missing.rb", "lib/new_relic/agent/sampler.rb", "lib/new_relic/agent/samplers/cpu_sampler.rb", "lib/new_relic/agent/samplers/memory_sampler.rb", "lib/new_relic/agent/samplers/mongrel_sampler.rb", "lib/new_relic/agent/shim_agent.rb", "lib/new_relic/agent/stats_engine.rb", "lib/new_relic/agent/stats_engine/metric_stats.rb", "lib/new_relic/agent/stats_engine/samplers.rb", "lib/new_relic/agent/stats_engine/transactions.rb", "lib/new_relic/agent/transaction_sampler.rb", "lib/new_relic/agent/worker_loop.rb", "lib/new_relic/commands/deployments.rb", "lib/new_relic/commands/new_relic_commands.rb", "lib/new_relic/control.rb", "lib/new_relic/control/external.rb", "lib/new_relic/control/merb.rb", "lib/new_relic/control/rails.rb", "lib/new_relic/control/ruby.rb", "lib/new_relic/control/sinatra.rb", "lib/new_relic/histogram.rb", "lib/new_relic/local_environment.rb", "lib/new_relic/merbtasks.rb", "lib/new_relic/metric_data.rb", "lib/new_relic/metric_parser.rb", "lib/new_relic/metric_parser/action_mailer.rb", "lib/new_relic/metric_parser/active_merchant.rb", "lib/new_relic/metric_parser/active_record.rb", "lib/new_relic/metric_parser/controller.rb", "lib/new_relic/metric_parser/controller_cpu.rb", "lib/new_relic/metric_parser/errors.rb", "lib/new_relic/metric_parser/external.rb", "lib/new_relic/metric_parser/mem_cache.rb", "lib/new_relic/metric_parser/view.rb", "lib/new_relic/metric_parser/web_frontend.rb", "lib/new_relic/metric_parser/web_service.rb", "lib/new_relic/metric_spec.rb", "lib/new_relic/metrics.rb", "lib/new_relic/noticed_error.rb", "lib/new_relic/rack/metric_app.rb", "lib/new_relic/rack/newrelic.ru", "lib/new_relic/rack/newrelic.yml", "lib/new_relic/rack_app.rb", "lib/new_relic/recipes.rb", "lib/new_relic/stats.rb", "lib/new_relic/transaction_analysis.rb", "lib/new_relic/transaction_sample.rb", "lib/new_relic/version.rb", "lib/new_relic_api.rb", "lib/newrelic_rpm.rb", "lib/tasks/all.rb", "lib/tasks/install.rake", "lib/tasks/tests.rake"]
+  s.files = ["CHANGELOG", "LICENSE", "README.md", "Rakefile", "bin/mongrel_rpm", "bin/newrelic_cmd", "cert/cacert.pem", "init.rb", "install.rb", "lib/new_relic/agent.rb", "lib/new_relic/agent/agent.rb", "lib/new_relic/agent/chained_call.rb", "lib/new_relic/agent/collection_helper.rb", "lib/new_relic/agent/error_collector.rb", "lib/new_relic/agent/instrumentation/active_merchant.rb", "lib/new_relic/agent/instrumentation/active_record_instrumentation.rb", "lib/new_relic/agent/instrumentation/authlogic.rb", "lib/new_relic/agent/instrumentation/controller_instrumentation.rb", "lib/new_relic/agent/instrumentation/data_mapper.rb", "lib/new_relic/agent/instrumentation/dispatcher_instrumentation.rb", "lib/new_relic/agent/instrumentation/memcache.rb", "lib/new_relic/agent/instrumentation/merb/controller.rb", "lib/new_relic/agent/instrumentation/merb/dispatcher.rb", "lib/new_relic/agent/instrumentation/merb/errors.rb", "lib/new_relic/agent/instrumentation/net.rb", "lib/new_relic/agent/instrumentation/passenger_instrumentation.rb", "lib/new_relic/agent/instrumentation/rack.rb", "lib/new_relic/agent/instrumentation/rails/action_controller.rb", "lib/new_relic/agent/instrumentation/rails/action_web_service.rb", "lib/new_relic/agent/instrumentation/rails/dispatcher.rb", "lib/new_relic/agent/instrumentation/rails/errors.rb", "lib/new_relic/agent/instrumentation/sinatra.rb", "lib/new_relic/agent/method_tracer.rb", "lib/new_relic/agent/patch_const_missing.rb", "lib/new_relic/agent/sampler.rb", "lib/new_relic/agent/samplers/cpu_sampler.rb", "lib/new_relic/agent/samplers/memory_sampler.rb", "lib/new_relic/agent/samplers/mongrel_sampler.rb", "lib/new_relic/agent/shim_agent.rb", "lib/new_relic/agent/stats_engine.rb", "lib/new_relic/agent/stats_engine/metric_stats.rb", "lib/new_relic/agent/stats_engine/samplers.rb", "lib/new_relic/agent/stats_engine/transactions.rb", "lib/new_relic/agent/transaction_sampler.rb", "lib/new_relic/agent/worker_loop.rb", "lib/new_relic/commands/deployments.rb", "lib/new_relic/commands/new_relic_commands.rb", "lib/new_relic/control.rb", "lib/new_relic/control/external.rb", "lib/new_relic/control/merb.rb", "lib/new_relic/control/rails.rb", "lib/new_relic/control/ruby.rb", "lib/new_relic/control/sinatra.rb", "lib/new_relic/histogram.rb", "lib/new_relic/local_environment.rb", "lib/new_relic/merbtasks.rb", "lib/new_relic/metric_data.rb", "lib/new_relic/metric_parser.rb", "lib/new_relic/metric_parser/action_mailer.rb", "lib/new_relic/metric_parser/active_merchant.rb", "lib/new_relic/metric_parser/active_record.rb", "lib/new_relic/metric_parser/controller.rb", "lib/new_relic/metric_parser/controller_cpu.rb", "lib/new_relic/metric_parser/errors.rb", "lib/new_relic/metric_parser/external.rb", "lib/new_relic/metric_parser/mem_cache.rb", "lib/new_relic/metric_parser/view.rb", "lib/new_relic/metric_parser/web_frontend.rb", "lib/new_relic/metric_parser/web_service.rb", "lib/new_relic/metric_spec.rb", "lib/new_relic/metrics.rb", "lib/new_relic/noticed_error.rb", "lib/new_relic/rack/metric_app.rb", "lib/new_relic/rack/newrelic.ru", "lib/new_relic/rack/newrelic.yml", "lib/new_relic/rack_app.rb", "lib/new_relic/recipes.rb", "lib/new_relic/stats.rb", "lib/new_relic/transaction_analysis.rb", "lib/new_relic/transaction_sample.rb", "lib/new_relic/version.rb", "lib/new_relic_api.rb", "lib/newrelic_rpm.rb", "lib/tasks/all.rb", "lib/tasks/install.rake", "lib/tasks/tests.rake", "newrelic.yml", "recipes/newrelic.rb", "test/active_record_fixtures.rb", "test/config/newrelic.yml", "test/config/test_control.rb", "test/new_relic/agent/active_record_instrumentation_test.rb", "test/new_relic/agent/agent_controller_test.rb", "test/new_relic/agent/agent_test.rb", "test/new_relic/agent/agent_test_controller.rb", "test/new_relic/agent/classloader_patch_test.rb", "test/new_relic/agent/collection_helper_test.rb", "test/new_relic/agent/dispatcher_instrumentation_test.rb", "test/new_relic/agent/error_collector_test.rb", "test/new_relic/agent/method_tracer_test.rb", "test/new_relic/agent/metric_data_test.rb", "test/new_relic/agent/mock_ar_connection.rb", "test/new_relic/agent/mock_scope_listener.rb", "test/new_relic/agent/net_instrumentation_test.rb", "test/new_relic/agent/stats_engine/metric_stats_test.rb", "test/new_relic/agent/stats_engine/samplers_test.rb", "test/new_relic/agent/stats_engine/stats_engine_test.rb", "test/new_relic/agent/task_instrumentation_test.rb", "test/new_relic/agent/testable_agent.rb", "test/new_relic/agent/transaction_sample_builder_test.rb", "test/new_relic/agent/transaction_sample_test.rb", "test/new_relic/agent/transaction_sampler_test.rb", "test/new_relic/agent/worker_loop_test.rb", "test/new_relic/control_test.rb", "test/new_relic/deployments_api_test.rb", "test/new_relic/environment_test.rb", "test/new_relic/metric_parser_test.rb", "test/new_relic/metric_spec_test.rb", "test/new_relic/shim_agent_test.rb", "test/new_relic/stats_test.rb", "test/new_relic/version_number_test.rb", "test/test_helper.rb", "test/ui/newrelic_controller_test.rb", "test/ui/newrelic_helper_test.rb", "ui/controllers/newrelic_controller.rb", "ui/helpers/google_pie_chart.rb", "ui/helpers/newrelic_helper.rb", "ui/views/layouts/newrelic_default.rhtml", "ui/views/newrelic/_explain_plans.rhtml", "ui/views/newrelic/_sample.rhtml", "ui/views/newrelic/_segment.rhtml", "ui/views/newrelic/_segment_limit_message.rhtml", "ui/views/newrelic/_segment_row.rhtml", "ui/views/newrelic/_show_sample_detail.rhtml", "ui/views/newrelic/_show_sample_sql.rhtml", "ui/views/newrelic/_show_sample_summary.rhtml", "ui/views/newrelic/_sql_row.rhtml", "ui/views/newrelic/_stack_trace.rhtml", "ui/views/newrelic/_table.rhtml", "ui/views/newrelic/explain_sql.rhtml", "ui/views/newrelic/images/arrow-close.png", "ui/views/newrelic/images/arrow-open.png", "ui/views/newrelic/images/blue_bar.gif", "ui/views/newrelic/images/file_icon.png", "ui/views/newrelic/images/gray_bar.gif", "ui/views/newrelic/images/new_relic_rpm_desktop.gif", "ui/views/newrelic/images/textmate.png", "ui/views/newrelic/index.rhtml", "ui/views/newrelic/javascript/prototype-scriptaculous.js", "ui/views/newrelic/javascript/transaction_sample.js", "ui/views/newrelic/sample_not_found.rhtml", "ui/views/newrelic/show_sample.rhtml", "ui/views/newrelic/show_source.rhtml", "ui/views/newrelic/stylesheets/style.css", "ui/views/newrelic/threads.rhtml", "Manifest", "techarch-newrelic_rpm.gemspec"]
+  s.homepage = %q{http://www.newrelic.com}
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "techarch-newrelic_rpm", "--main", "README.md"]
+  s.require_paths = ["lib"]
+  s.rubyforge_project = %q{newrelic}
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{New Relic Ruby Performance Monitoring Agent}
+  s.test_files = ["test/config/test_control.rb", "test/new_relic/agent/active_record_instrumentation_test.rb", "test/new_relic/agent/agent_controller_test.rb", "test/new_relic/agent/agent_test.rb", "test/new_relic/agent/classloader_patch_test.rb", "test/new_relic/agent/collection_helper_test.rb", "test/new_relic/agent/dispatcher_instrumentation_test.rb", "test/new_relic/agent/error_collector_test.rb", "test/new_relic/agent/method_tracer_test.rb", "test/new_relic/agent/metric_data_test.rb", "test/new_relic/agent/net_instrumentation_test.rb", "test/new_relic/agent/stats_engine/metric_stats_test.rb", "test/new_relic/agent/stats_engine/samplers_test.rb", "test/new_relic/agent/stats_engine/stats_engine_test.rb", "test/new_relic/agent/task_instrumentation_test.rb", "test/new_relic/agent/transaction_sample_builder_test.rb", "test/new_relic/agent/transaction_sample_test.rb", "test/new_relic/agent/transaction_sampler_test.rb", "test/new_relic/agent/worker_loop_test.rb", "test/new_relic/control_test.rb", "test/new_relic/deployments_api_test.rb", "test/new_relic/environment_test.rb", "test/new_relic/metric_parser_test.rb", "test/new_relic/metric_spec_test.rb", "test/new_relic/shim_agent_test.rb", "test/new_relic/stats_test.rb", "test/new_relic/version_number_test.rb", "test/test_helper.rb", "test/ui/newrelic_controller_test.rb", "test/ui/newrelic_helper_test.rb"]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+    else
+    end
+  else
+  end
+end

data/test/active_record_fixtures.rb

@@ -0,0 +1,55 @@
+# Some test models
+# These will get created and dropped during the active record tests
+# Be sure and call setup and teardown
+module ActiveRecordFixtures
+  extend self
+  def setup
+    ActiveRecordFixtures::Order.setup
+    ActiveRecordFixtures::Shipment.setup
+  end
+  def teardown
+    ActiveRecordFixtures::Shipment.teardown
+    ActiveRecordFixtures::Order.teardown
+  end
+  class Order < ActiveRecord::Base
+    self.table_name = 'newrelic_test_orders'
+    has_and_belongs_to_many :shipments, :class_name => 'ActiveRecordFixtures::Shipment'
+    def self.setup
+      connection.create_table self.table_name, :force => true do |t|
+        t.column :name, :string
+      end
+    end
+    def self.add_delay
+      # Introduce a 500 ms delay into db operations on Orders
+      def connection.log_info *args
+        sleep 0.5
+        super *args
+      end
+    end
+    def self.teardown
+      connection.drop_table table_name
+      def connection.log_info *args
+        super *args
+      end
+    end
+  end
+  
+  class Shipment < ActiveRecord::Base
+    self.table_name = 'newrelic_test_shipment'
+    has_and_belongs_to_many :orders, :class_name => 'ActiveRecordFixtures::Order'
+    def self.setup
+      connection.create_table self.table_name, :force => true do |t|
+        # no other columns
+      end
+      connection.create_table 'orders_shipments', :force => true, :id => false do |t|
+        t.column :order_id, :integer 
+        t.column :shipment_id, :integer 
+      end
+    end
+    
+    def self.teardown
+      connection.drop_table 'orders_shipments'
+      connection.drop_table self.table_name
+    end
+  end    
+end

data/test/config/newrelic.yml

@@ -0,0 +1,46 @@
+# This is the config file loaded when running unit tests locally.
+common: &default_settings
+  transaction_tracer:
+    enabled: true
+    record_sql: raw
+
+test:
+  <<: *default_settings
+  host: localhost
+  port: 3000
+  log_level: info
+  capture_params: true
+  enabled: false
+  developer: true
+  monitor_daemons: false
+  disable_samplers: true
+  api_host: 127.0.0.1
+  api_port: 443
+  message: '<%= generated_for_user %>'
+  license_key: '<%= license_key %>'
+  agent_enabled: auto
+  apdex_t: 1.1
+
+  # Some fixtures for newrelic.yml parsing tests
+  erb_value: <%= 'hey'*3 %>
+  tval: true
+  tstring: 'true'
+  fval: false
+  yval: yes
+  sval: sure
+    
+  # checks for loading classes on the background thread
+  check_bg_loading: true
+
+  transaction_tracer:
+    record_sql: raw
+    transaction_threshold: Apdex_f # case insensitive
+    explain_threshold: 0.1
+    explain_enabled: true
+    stack_trace_threshold: 0.1
+    
+  error_collector:
+    enabled: true
+    capture_source: true
+    
+  

data/test/config/test_control.rb

@@ -0,0 +1,39 @@
+require 'new_relic/control/rails'
+require 'new_relic/agent/agent_test_controller'
+
+class NewRelic::Control::Test < NewRelic::Control::Rails #:nodoc:
+  def env
+    'test'
+  end
+  def app
+    :rails
+  end
+  def config_file
+    File.join(File.dirname(__FILE__), "newrelic.yml")
+  end
+  def initialize local_env
+    super local_env
+    setup_log 
+  end
+  # when running tests, don't write out stderr
+  def log!(msg, level=:info)
+    log.send level, msg if log
+  end
+
+  # Add the default route in case it's missing.  Need it for testing.
+  def install_devmode_route
+    super
+    ActionController::Routing::RouteSet.class_eval do
+      return if defined? draw_without_test_route
+      def draw_with_test_route
+        draw_without_test_route do | map |
+          map.connect ':controller/:action/:id'
+          yield map        
+        end
+      end
+      alias_method_chain :draw, :test_route
+    end
+    # Force the routes to be reloaded
+    ActionController::Routing::Routes.reload!
+  end
+end

data/test/new_relic/agent/active_record_instrumentation_test.rb

@@ -0,0 +1,264 @@
+require File.expand_path(File.join(File.dirname(__FILE__),'..','..','test_helper')) 
+require 'active_record_fixtures'
+
+class ActiveRecordInstrumentationTest < Test::Unit::TestCase
+  
+  def setup
+    super
+    NewRelic::Agent.manual_start
+    ActiveRecordFixtures.setup
+    NewRelic::Agent.instance.transaction_sampler.harvest
+#    ActiveRecordFixtures::Order.execute "show tables"   # this warms up the connection - which can issue sql statements
+    NewRelic::Agent.instance.stats_engine.clear_stats
+  rescue
+    puts e
+    puts e.backtrace.join("\n")
+  end
+  
+  def teardown
+    super
+    ActiveRecordFixtures.teardown
+  end
+  def test_agent_setup
+    assert NewRelic::Agent.instance.class == NewRelic::Agent::Agent
+  end
+  def test_finder
+    ActiveRecordFixtures::Order.create :id => 0, :name => 'jeff'
+    ActiveRecordFixtures::Order.find(:all)
+    s = NewRelic::Agent.get_stats("ActiveRecord/ActiveRecordFixtures::Order/find")
+    assert_equal 1, s.call_count
+    ActiveRecordFixtures::Order.find_all_by_name "jeff"
+    s = NewRelic::Agent.get_stats("ActiveRecord/ActiveRecordFixtures::Order/find")
+    assert_equal 2, s.call_count
+  end
+  
+  # multiple duplicate find calls should only cause metric trigger on the first
+  # call.  the others are ignored.
+  def test_query_cache
+    ActiveRecordFixtures::Order.cache do
+      m = ActiveRecordFixtures::Order.create :id => 0, :name => 'jeff'
+      ActiveRecordFixtures::Order.find(:all)
+      s = NewRelic::Agent.get_stats("ActiveRecord/ActiveRecordFixtures::Order/find")
+      assert_equal 1, s.call_count
+      
+      10.times { ActiveRecordFixtures::Order.find m.id }
+    end
+    s = NewRelic::Agent.get_stats("ActiveRecord/ActiveRecordFixtures::Order/find")
+    assert_equal 2, s.call_count    
+  end
+  
+  def test_metric_names
+    m = ActiveRecordFixtures::Order.create :id => 0, :name => 'jeff'
+    m = ActiveRecordFixtures::Order.find(m.id)
+    m.id = 999
+    m.save!
+    
+    metrics = NewRelic::Agent.instance.stats_engine.metrics
+    #   This doesn't work on hudson because the sampler metrics creep in.    
+    #   metrics = NewRelic::Agent.instance.stats_engine.metrics.select { |mname| mname =~ /ActiveRecord\/ActiveRecordFixtures::Order\// }.sort
+    expected = %W[
+      ActiveRecord/all
+      ActiveRecord/find
+      ActiveRecord/ActiveRecordFixtures::Order/find
+      ]
+    expected += %W[Database/SQL/insert] if defined?(JRuby)      
+    expected += %W[ActiveRecord/create Database/SQL/other ActiveRecord/ActiveRecordFixtures::Order/create] unless defined?(JRuby)      
+    expected += %W[ActiveRecord/save ActiveRecord/ActiveRecordFixtures::Order/save] if NewRelic::Control.instance.rails_version < '2.1.0'   
+    compare_metrics expected, metrics
+    assert_equal 1, NewRelic::Agent.get_stats("ActiveRecord/ActiveRecordFixtures::Order/find").call_count
+    assert_equal (defined?(JRuby) ? 0 : 1), NewRelic::Agent.get_stats("ActiveRecord/ActiveRecordFixtures::Order/create").call_count
+  end
+  def test_join_metrics
+    m = ActiveRecordFixtures::Order.create :name => 'jeff'
+    m = ActiveRecordFixtures::Order.find(m.id)
+    s = m.shipments.create
+    m.shipments.to_a
+    m.destroy
+    
+    metrics = NewRelic::Agent.instance.stats_engine.metrics
+    #   This doesn't work on hudson because the sampler metrics creep in.    
+    #   metrics = NewRelic::Agent.instance.stats_engine.metrics.select { |mname| mname =~ /ActiveRecord\/ActiveRecordFixtures::Order\// }.sort
+    expected_metrics = %W[
+    ActiveRecord/all
+    ActiveRecord/destroy
+    ActiveRecord/ActiveRecordFixtures::Order/destroy
+    Database/SQL/insert
+    Database/SQL/delete
+    ActiveRecord/find
+    ActiveRecord/ActiveRecordFixtures::Order/find
+    ActiveRecord/ActiveRecordFixtures::Shipment/find
+    ]
+    
+    expected_metrics += %W[Database/SQL/other Database/SQL/show ActiveRecord/create
+                           ActiveRecord/ActiveRecordFixtures::Shipment/create
+                           ActiveRecord/ActiveRecordFixtures::Order/create
+                           ] unless defined? JRuby
+
+    compare_metrics expected_metrics, metrics
+    # This number may be different with different db adapters, not sure
+    # assert_equal 17, NewRelic::Agent.get_stats("ActiveRecord/all").call_count
+    assert_equal NewRelic::Agent.get_stats("ActiveRecord/all").total_exclusive_time, NewRelic::Agent.get_stats("ActiveRecord/all").total_call_time unless defined?(RUBY_DESCRIPTION) && RUBY_DESCRIPTION =~ /Enterprise Edition/
+    assert_equal 1, NewRelic::Agent.get_stats("ActiveRecord/ActiveRecordFixtures::Order/find").call_count
+    assert_equal 1, NewRelic::Agent.get_stats("ActiveRecord/ActiveRecordFixtures::Shipment/find").call_count
+    assert_equal 1, NewRelic::Agent.get_stats("Database/SQL/insert").call_count unless defined? JRuby
+    assert_equal 3, NewRelic::Agent.get_stats("Database/SQL/insert").call_count if defined? JRuby
+    assert_equal 1, NewRelic::Agent.get_stats("Database/SQL/delete").call_count
+  end
+  def test_direct_sql
+    list = ActiveRecordFixtures::Order.connection.select_rows "select * from #{ActiveRecordFixtures::Order.table_name}"
+    metrics = NewRelic::Agent.instance.stats_engine.metrics
+    compare_metrics %W[
+    ActiveRecord/all
+    Database/SQL/select
+    ], metrics
+    assert_equal 1, NewRelic::Agent.get_stats("Database/SQL/select").call_count
+  end
+  
+  def test_other_sql
+    list = ActiveRecordFixtures::Order.connection.execute "begin"
+    metrics = NewRelic::Agent.instance.stats_engine.metrics
+    compare_metrics %W[
+    ActiveRecord/all
+    Database/SQL/other
+    ], metrics
+    assert_equal 1, NewRelic::Agent.get_stats("Database/SQL/other").call_count
+  end
+  
+  def test_show_sql
+    list = ActiveRecordFixtures::Order.connection.execute "show tables"
+    metrics = NewRelic::Agent.instance.stats_engine.metrics
+    compare_metrics %W[
+    ActiveRecord/all
+    Database/SQL/show
+    ], metrics
+    assert_equal 1, NewRelic::Agent.get_stats("Database/SQL/show").call_count 
+  end
+
+  def test_blocked_instrumentation
+    ActiveRecordFixtures::Order.add_delay
+    NewRelic::Agent.disable_all_tracing do
+      ActiveRecordFixtures::Order.find(:all)
+    end
+    assert_nil NewRelic::Agent.instance.transaction_sampler.last_sample
+    metrics = NewRelic::Agent.instance.stats_engine.metrics
+    compare_metrics [], metrics
+  end
+  def test_run_explains
+    ActiveRecordFixtures::Order.add_delay
+    ActiveRecordFixtures::Order.find(:all)
+    
+    sample = NewRelic::Agent.instance.transaction_sampler.last_sample
+    
+    segment = sample.root_segment.called_segments.first.called_segments.first
+    assert_match /^SELECT \* FROM ["`]?#{ActiveRecordFixtures::Order.table_name}["`]?$/i, segment.params[:sql].strip
+    NewRelic::TransactionSample::Segment.any_instance.expects(:explain_sql).returns([])
+    sample = sample.prepare_to_send(:obfuscate_sql => true, :explain_enabled => true, :explain_sql => 0.0)
+    segment = sample.root_segment.called_segments.first.called_segments.first
+  end
+  def test_prepare_to_send
+    ActiveRecordFixtures::Order.add_delay
+    ActiveRecordFixtures::Order.find(:all)
+    
+    sample = NewRelic::Agent.instance.transaction_sampler.last_sample
+    # 
+    sql_segment = sample.root_segment.called_segments.first.called_segments.first
+    assert_match /^SELECT /, sql_segment.params[:sql]
+    assert sql_segment.duration > 0.0, "Segment duration must be greater than zero."
+    sample = sample.prepare_to_send(:record_sql => :raw, :explain_enabled => true, :explain_sql => 0.0)
+    sql_segment = sample.root_segment.called_segments.first.called_segments.first
+    assert_match /^SELECT /, sql_segment.params[:sql]
+    explanations = sql_segment.params[:explanation]
+    if isMysql? || isPostgres?
+      assert_not_nil explanations, "No explains in segment: #{sql_segment}"
+      assert_equal 1, explanations.size,"No explains in segment: #{sql_segment}" 
+      assert_equal 1, explanations.first.size
+    end
+  end
+  def test_transaction
+    ActiveRecordFixtures::Order.add_delay
+    ActiveRecordFixtures::Order.find(:all)
+    
+    sample = NewRelic::Agent.instance.transaction_sampler.last_sample
+    
+    sample = sample.prepare_to_send(:obfuscate_sql => true, :explain_enabled => true, :explain_sql => 0.0)
+    segment = sample.root_segment.called_segments.first.called_segments.first
+    assert_nil segment.params[:sql], "SQL should have been removed."
+    explanations = segment.params[:explanation]
+    if isMysql? || isPostgres?
+      assert_not_nil explanations, "No explains in segment: #{segment}"
+      assert_equal 1, explanations.size,"No explains in segment: #{segment}" 
+      assert_equal 1, explanations.first.size
+    end    
+    if isPostgres?
+      assert_equal Array, explanations.class
+      assert_equal Array, explanations[0].class
+      assert_equal Array, explanations[0][0].class
+      assert_match /Seq Scan on test_data/, explanations[0][0].join(";") 
+    elsif isMysql?
+      assert_equal "1;SIMPLE;#{ActiveRecordFixtures::Order.table_name};ALL;;;;;1;", explanations.first.first.join(";")
+    end
+    
+    s = NewRelic::Agent.get_stats("ActiveRecord/ActiveRecordFixtures::Order/find")
+    assert_equal 1, s.call_count
+  end
+  # These are only valid for rails 2.1 and later
+  if NewRelic::Control.instance.rails_version >= NewRelic::VersionNumber.new("2.1.0")
+    ActiveRecordFixtures::Order.class_eval do
+      named_scope :jeffs, :conditions => { :name => 'Jeff' }
+    end
+    def test_named_scope
+      ActiveRecordFixtures::Order.create :name => 'Jeff'
+      s = NewRelic::Agent.get_stats("ActiveRecord/ActiveRecordFixtures::Order/find")
+      before_count = s.call_count
+      x = ActiveRecordFixtures::Order.jeffs.find(:all)
+      assert_equal 1, x.size
+      se = NewRelic::Agent.instance.stats_engine
+      assert_equal before_count+1, s.call_count
+    end
+  end
+  
+  # This is to make sure the all metric is recorded for exceptional cases
+  def test_error_handling
+    # have the AR select throw an error
+    ActiveRecordFixtures::Order.connection.stubs(:log_info).with do | sql, x, y |
+      raise "Error" if sql =~ /select/
+      true
+    end
+    ActiveRecordFixtures::Order.connection.select_rows "select * from #{ActiveRecordFixtures::Order.table_name}" rescue nil
+    metrics = NewRelic::Agent.instance.stats_engine.metrics
+    compare_metrics %W[
+    ActiveRecord/all
+    Database/SQL/select
+    ], metrics
+    assert_equal 1, NewRelic::Agent.get_stats("Database/SQL/select").call_count
+    assert_equal 1, NewRelic::Agent.get_stats("ActiveRecord/all").call_count
+  end
+  
+  def test_rescue_handling
+    begin
+      ActiveRecordFixtures::Order.transaction do
+        raise ActiveRecord::ActiveRecordError.new('preserve-me!') 
+      end
+    rescue ActiveRecord::ActiveRecordError => e
+      assert_equal 'preserve-me!', e.message
+    rescue
+      fail "Rescue2: Got something COMPLETELY unexpected: $!:#{$!.inspect}"
+    end
+
+  end
+  
+  private
+  
+  def compare_metrics expected_list, actual_list
+    actual = Set.new actual_list
+    actual.delete('GC/cumulative') # in case we are in REE
+    expected = Set.new expected_list
+    assert_equal expected.to_a.sort, actual.to_a.sort, "extra: #{(actual - expected).to_a.join(", ")}; missing: #{(expected - actual).to_a.join(", ")}"
+  end
+  def isPostgres?
+    ActiveRecordFixtures::Order.configurations[RAILS_ENV]['adapter'] =~ /postgres/
+  end
+  def isMysql?
+    ActiveRecordFixtures::Order.connection.class.name =~ /mysql/i 
+  end
+end