wd_newrelic_rpm 3.3.4.1 → 3.5.5

data/test/multiverse/script/run_one

@@ -0,0 +1,3 @@
+#!/bin/sh
+
+test/multiverse/lib/multiverse/suite.rb $PWD/test/multiverse/suites/$1 "$2" true

data/test/multiverse/script/runner

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+require 'fileutils'
+require 'test/unit'
+
+module Multiverse
+  require File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib', 'multiverse', 'environment'))
+end
+
+Multiverse::Runner.run ARGV[0].to_s

data/test/multiverse/suites/active_record/Envfile

@@ -0,0 +1,13 @@
+gemfile <<-RB
+  if RUBY_VERSION == '1.8.6'
+    gem 'activerecord', '2.3.14'
+  elsif RUBY_PLATFORM == 'java'
+    gem 'jdbc-sqlite3'
+    gem 'activerecord-jdbcsqlite3-adapter'
+    gem 'activerecord'
+  else
+    gem 'activerecord'
+    gem 'sqlite3'
+  end
+  
+RB

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

@@ -0,0 +1,94 @@
+
+require 'rubygems'
+
+require 'active_record'
+require 'test/unit'
+require 'newrelic_rpm'
+require 'multiverse/color'
+
+
+class InstrumentActiveRecordMethods < Test::Unit::TestCase
+  extend Multiverse::Color
+
+  if RUBY_VERSION >= '1.8.7'
+    if RUBY_PLATFORM == 'java'
+      require 'jdbc/sqlite3'
+      @@adapter = 'jdbcsqlite3'
+    else
+      require 'sqlite3'
+      @@adapter = 'sqlite3'
+    end
+
+    class User < ActiveRecord::Base 
+      include NewRelic::Agent::MethodTracer 
+      has_many :aliases
+
+      add_method_tracer :save!
+      add_method_tracer :persisted?
+    end
+
+    class Alias < ActiveRecord::Base
+      include NewRelic::Agent::MethodTracer
+
+      add_method_tracer :save!
+      add_method_tracer :persisted?
+      add_method_tracer :destroyed?
+    end
+    
+    def setup
+      puts "adapter : #{@@adapter}"
+      @db_connection = ActiveRecord::Base.establish_connection( :adapter => @@adapter, :database => "testdb.sqlite3")
+      ActiveRecord::Migration.class_eval do
+        @connection = @db_connection
+        create_table :users do |t|
+              t.string  :name
+        end
+
+        create_table :aliases do |t|
+            t.integer :user_id
+            t.string :aka
+        end
+      end
+    end
+
+    def teardown
+      @db_connection = ActiveRecord::Base.establish_connection( :adapter => "sqlite3", :database => "testdb.sqlite3")
+      ActiveRecord::Migration.class_eval do
+        @connection = @db_connection
+        drop_table :users
+        drop_table :aliases
+      end
+    end
+
+    def test_basic_creation
+      a_user = User.new :name => "Bob"
+      assert a_user.new_record?
+      a_user.save!
+      assert User.connected?
+      assert a_user.persisted?
+      assert a_user.id == 1 
+    end
+
+    def test_alias_collection_query_method
+      a_user = User.new :name => "Bob"
+      a_user.save!
+      a_user = User.find(1)
+      assert User.connected?
+      assert a_user.id = 1  
+
+      an_alias = Alias.new :user_id => a_user.id, :aka => "the Blob"
+      assert an_alias.new_record?
+      an_alias.save!
+      assert an_alias.persisted?
+      an_alias.destroy
+      assert an_alias.destroyed?
+    end
+
+  else
+    def test_truth
+      assert true # jruby freaks out if there are no tests defined in the test class
+    end
+    puts yellow('SKIPPED! skipped until ruby 1.8.6 compatibilites ironed out')
+  end
+end
+

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,4 @@
+gemfile <<-RB
+  gem 'rack'
+  gem 'mocha', '< 0.13'
+RB

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

@@ -0,0 +1,99 @@
+# RUBY-981 Audit Log
+
+require 'newrelic_rpm'
+require 'fake_collector'
+require 'mocha'
+
+class AuditLogTest < Test::Unit::TestCase
+  # Initialization
+  def setup
+    $collector ||= NewRelic::FakeCollector.new
+    $collector.reset
+    $collector.run
+
+    @string_log = StringIO.new
+    NewRelic::Agent::AuditLogger.any_instance.stubs(:ensure_log_path).returns(@string_log)
+  end
+
+  def teardown
+    $collector.reset
+  end
+
+  def audit_log_contents
+    if @contents.nil?
+      @string_log.rewind
+      @contents = @string_log.read
+    end
+    @contents
+  end
+
+  def assert_audit_log_contains(needle)
+    # Original request bodies dumped to the log have symbol keys, but once
+    # they go through a dump/load, they're strings again, so we strip
+    # double-quotes and colons from the log, and the strings we searching for.
+    regex = /[:"]/
+    needle = needle.gsub(regex, '')
+    haystack = audit_log_contents.gsub(regex, '')
+    assert(haystack.include?(needle), "Expected log to contain '#{needle}'")
+  end
+
+  # Because we don't generate a strictly machine-readable representation of
+  # request bodies for the audit log, the transformation into strings is
+  # effectively one-way. This, combined with the fact that Hash traversal order
+  # is arbitrary in Ruby 1.8.x means that it's difficult to directly assert that
+  # some object graph made it into the audit log (due to different possible 
+  # orderings of the key/value pairs in Hashes that were embedded in the request
+  # body). So, this method traverses an object graph and only makes assertions
+  # about the terminal (non-Array-or-Hash) nodes therein.
+  def assert_audit_log_contains_object(o, format)
+    if format == :json
+      assert_audit_log_contains(JSON.dump(o))
+    else
+      case o
+      when Hash
+        o.each do |k,v|
+          assert_audit_log_contains_object(v, format)
+        end
+      when Array
+        o.each do |el|
+          assert_audit_log_contains_object(el, format)
+        end
+      else
+        assert_audit_log_contains(o.inspect)
+      end
+    end
+  end
+
+  def run_agent_with_options(options)
+    NewRelic::Agent.manual_start(options)
+    yield NewRelic::Agent.agent if block_given?
+    NewRelic::Agent.shutdown    
+  end
+
+  def test_logs_nothing_by_default
+    run_agent_with_options({})
+    assert_equal('', audit_log_contents)
+  end
+
+  def test_logs_nothing_when_disabled
+    run_agent_with_options({ :'audit_log.enabled' => false })
+    assert_equal('', audit_log_contents)
+  end
+
+  def test_logs_request_bodies_human_readably_ish
+    format = NewRelic::Agent::NewRelicService::JsonMarshaller.is_supported? ? :json : :pruby
+    run_agent_with_options({ :'audit_log.enabled' => true }) do |agent|
+      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.send(:harvest_and_send_slowest_sql)
+    end
+
+    $collector.agent_data.each do |req|
+      body = $collector.unpack_inner_blobs(req)
+      assert_audit_log_contains_object(body, format)
+    end
+  end
+end

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.default_encoder),
+                 $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
+