ghazel-newrelic_rpm 3.4.0.2 → 3.5.4

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,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