mmtrix_rpm 1.0.0.1

data/test/multiverse/suites/rack/rack_env_mutation_test.rb

@@ -0,0 +1,53 @@
+# encoding: utf-8
+# This file is distributed under Mmtrix's license terms.
+# See https://github.com/mmtrix/rpm/blob/master/LICENSE for complete details.
+
+require 'mmtrix/rack/browser_monitoring'
+require 'mmtrix/rack/agent_hooks'
+require 'mmtrix/rack/error_collector'
+
+if Mmtrix::Agent::Instrumentation::RackHelpers.rack_version_supported?
+
+class RackEnvMutationTest < Minitest::Test
+  attr_accessor :inner_app
+
+  include MultiverseHelpers
+  include Rack::Test::Methods
+
+  setup_and_teardown_agent
+
+  class BadApp
+    def call(env)
+      Thread.new do
+        100.times do
+          env.each do |k, v|
+            # allow main thread to run while we're still in the middle of
+            # iterating
+            Thread.pass
+          end
+        end
+      end
+
+      # Give the thread we just spawned a chance to start up
+      Thread.pass
+
+      [200, {}, ['cool story']]
+    end
+  end
+
+  def app
+    inner_app = BadApp.new
+    Rack::Builder.app do
+      use Mmtrix::Rack::AgentHooks
+      run inner_app
+    end
+  end
+
+  def test_safe_from_iterations_over_rack_env_from_background_threads
+    100.times do
+      get '/'
+    end
+  end
+end
+
+end

data/test/multiverse/suites/rack/rack_parameter_filtering_test.rb

@@ -0,0 +1,49 @@
+# encoding: utf-8
+# This file is distributed under Mmtrix's license terms.
+# See https://github.com/mmtrix/rpm/blob/master/LICENSE for complete details.
+
+require 'filtering_test_app'
+
+if Mmtrix::Agent::Instrumentation::RackHelpers.rack_version_supported?
+
+  class RackParameterFilteringTest < Minitest::Test
+    include MultiverseHelpers
+
+    setup_and_teardown_agent
+
+    include Rack::Test::Methods
+
+    def app
+      Rack::Builder.app { run FilteringTestApp.new }
+    end
+
+    def test_file_upload_params_are_filtered
+      with_config(:capture_params => true) do
+        params = {
+          :title => "blah",
+          :file => Rack::Test::UploadedFile.new(__FILE__, 'text/plain')
+        }
+        post '/', params
+
+        expected = {
+          "request.parameters.title" => "blah",
+          "request.parameters.file" => "[FILE]"
+        }
+        assert_equal expected, last_transaction_trace_request_params
+      end
+    end
+
+    def test_apply_filters_returns_params_when_rails_is_not_present
+      with_config(:capture_params => true) do
+        params = {"name" => "name", "password" => "mypass"}
+        post '/', params
+
+        expected = {
+          "request.parameters.name" => "name",
+          "request.parameters.password" => "mypass"
+        }
+        assert_equal expected, last_transaction_trace_request_params
+      end
+    end
+  end
+end

data/test/multiverse/suites/rack/rack_unsupported_version_test.rb

@@ -0,0 +1,43 @@
+# encoding: utf-8
+# This file is distributed under Mmtrix's license terms.
+# See https://github.com/mmtrix/rpm/blob/master/LICENSE for complete details.
+
+if !Mmtrix::Agent::Instrumentation::RackHelpers.rack_version_supported?
+
+class RackUnsupportedVersionTest < Minitest::Test
+  include MultiverseHelpers
+
+  setup_and_teardown_agent
+
+  include Rack::Test::Methods
+
+  class SimpleMiddleware
+    def initialize(app)
+      @app = app
+    end
+
+    def call(env)
+      @app.call(env)
+    end
+  end
+
+  class ExampleApp
+    def call(env)
+      [200, {}, [self.class.name]]
+    end
+  end
+
+  def app
+    Rack::Builder.app do
+      use SimpleMiddleware
+      run ExampleApp.new
+    end
+  end
+
+  def test_no_instrumentation_when_not_supported
+    get '/'
+    assert_metrics_recorded_exclusive([])
+  end
+end
+
+end

data/test/multiverse/suites/rack/url_map_test.rb

@@ -0,0 +1,117 @@
+# encoding: utf-8
+# This file is distributed under Mmtrix's license terms.
+# See https://github.com/mmtrix/rpm/blob/master/LICENSE for complete details.
+
+# These tests are for confirming that our direct support Rack::URLMap works
+# properly. Tests against the builder interface more commonly used (i.e. map)
+# can be found elsewhere in this suite.
+
+if Mmtrix::Agent::Instrumentation::RackHelpers.rack_version_supported?
+
+class UrlMapTest < Minitest::Test
+  include MultiverseHelpers
+
+  setup_and_teardown_agent
+
+  include Rack::Test::Methods
+
+  class SimpleMiddleware
+    def initialize(app)
+      @app = app
+    end
+
+    def call(env)
+      @app.call(env)
+    end
+  end
+
+  class MiddlewareOne   < SimpleMiddleware; end
+  class MiddlewareTwo   < SimpleMiddleware; end
+
+  class ExampleApp
+    def call(env)
+      [200, {}, [self.class.name]]
+    end
+  end
+
+  class PrefixAppOne < ExampleApp; end
+  class PrefixAppTwo < ExampleApp; end
+
+  def app
+    Rack::Builder.app do
+      use MiddlewareOne
+      use MiddlewareTwo
+
+      run Rack::URLMap.new(
+        '/prefix1' => PrefixAppOne.new,
+        '/prefix2' => PrefixAppTwo.new)
+    end
+  end
+
+  if Rack::VERSION[1] >= 4
+    def test_metrics_for_default_prefix
+      get '/'
+
+      assert_metrics_recorded_exclusive([
+        'Apdex',
+        'ApdexAll',
+        'HttpDispatcher',
+        'Middleware/all',
+        'Controller/Rack/UrlMapTest::ExampleApp/call',
+        'Apdex/Rack/UrlMapTest::ExampleApp/call',
+        'Middleware/Rack/UrlMapTest::MiddlewareOne/call',
+        'Middleware/Rack/UrlMapTest::MiddlewareTwo/call',
+        'Nested/Controller/Rack/Rack::URLMap/call',
+        'Nested/Controller/Rack/UrlMapTest::ExampleApp/call',
+        ['Middleware/Rack/UrlMapTest::MiddlewareOne/call', 'Controller/Rack/UrlMapTest::ExampleApp/call'],
+        ['Middleware/Rack/UrlMapTest::MiddlewareTwo/call', 'Controller/Rack/UrlMapTest::ExampleApp/call'],
+        ['Nested/Controller/Rack/UrlMapTest::ExampleApp/call', 'Controller/Rack/UrlMapTest::ExampleApp/call'],
+        ['Nested/Controller/Rack/Rack::URLMap/call', 'Controller/Rack/UrlMapTest::ExampleApp/call']
+      ])
+    end
+  end
+
+  def test_metrics_for_mapped_prefix
+    get '/prefix1'
+
+    assert_metrics_recorded_exclusive([
+      'Apdex',
+      'ApdexAll',
+      'HttpDispatcher',
+      'Middleware/all',
+      'Controller/Rack/UrlMapTest::PrefixAppOne/call',
+      'Apdex/Rack/UrlMapTest::PrefixAppOne/call',
+      'Middleware/Rack/UrlMapTest::MiddlewareOne/call',
+      'Middleware/Rack/UrlMapTest::MiddlewareTwo/call',
+      'Nested/Controller/Rack/Rack::URLMap/call',
+      'Nested/Controller/Rack/UrlMapTest::PrefixAppOne/call',
+      ['Middleware/Rack/UrlMapTest::MiddlewareOne/call', 'Controller/Rack/UrlMapTest::PrefixAppOne/call'],
+      ['Middleware/Rack/UrlMapTest::MiddlewareTwo/call', 'Controller/Rack/UrlMapTest::PrefixAppOne/call'],
+      ['Nested/Controller/Rack/Rack::URLMap/call', 'Controller/Rack/UrlMapTest::PrefixAppOne/call'],
+      ['Nested/Controller/Rack/UrlMapTest::PrefixAppOne/call', 'Controller/Rack/UrlMapTest::PrefixAppOne/call']
+    ])
+  end
+
+  def test_metrics_for_mapped_prefix_with_extra_middleware
+    get '/prefix2'
+
+    assert_metrics_recorded_exclusive([
+      'Apdex',
+      'ApdexAll',
+      'HttpDispatcher',
+      'Middleware/all',
+      'Controller/Rack/UrlMapTest::PrefixAppTwo/call',
+      'Apdex/Rack/UrlMapTest::PrefixAppTwo/call',
+      'Middleware/Rack/UrlMapTest::MiddlewareOne/call',
+      'Middleware/Rack/UrlMapTest::MiddlewareTwo/call',
+      'Nested/Controller/Rack/Rack::URLMap/call',
+      'Nested/Controller/Rack/UrlMapTest::PrefixAppTwo/call',
+      ['Middleware/Rack/UrlMapTest::MiddlewareOne/call', 'Controller/Rack/UrlMapTest::PrefixAppTwo/call'],
+      ['Middleware/Rack/UrlMapTest::MiddlewareTwo/call', 'Controller/Rack/UrlMapTest::PrefixAppTwo/call'],
+      ['Nested/Controller/Rack/Rack::URLMap/call', 'Controller/Rack/UrlMapTest::PrefixAppTwo/call'],
+      ['Nested/Controller/Rack/UrlMapTest::PrefixAppTwo/call', 'Controller/Rack/UrlMapTest::PrefixAppTwo/call']
+    ])
+  end
+end
+
+end

data/test/multiverse/suites/rails/Envfile

@@ -0,0 +1,61 @@
+if RUBY_VERSION >= '1.9.3'
+  gemfile <<-RB
+    gem 'rails', '~>4.2.1'
+    gem 'haml', :require => false
+    gem 'minitest', '5.2.3'
+  RB
+
+  gemfile <<-RB
+    gem 'rails', '~>4.1.10'
+    # Multiverse has an incompatibility with Minitest 5.3.0, so lock here for
+    # now
+    gem 'minitest', '5.2.3'
+
+    # Load error with beta haml + rails, so hold require until test needs it.
+    gem 'haml', '~>4.1.0.beta1', :require => false
+  RB
+
+  gemfile <<-RB
+    gem 'rails', '~>4.0.13'
+    gem 'haml', '4.0.2'   # Getting load issues with haml 4.0.3
+  RB
+end
+
+gemfile <<-RB
+  gem 'rails', '~>3.2.20'
+  gem 'i18n', '~>0.6.11'
+  gem 'haml', '4.0.2'   # Getting load issues with haml 4.0.3
+  gem 'minitest_tu_shim', :require => false
+RB
+
+gemfile <<-RB
+  gem 'rails', '~>3.2.20'
+  gem 'i18n', '~>0.6.11'
+  gem 'sinatra', '~> 1.4.5'
+  gem 'haml', '4.0.2'   # Getting load issues with haml 4.0.3
+  gem 'minitest_tu_shim', :require => false
+RB
+
+gemfile <<-RB
+  gem 'rails', '~>3.1.12'
+  gem 'i18n', '~>0.6.11'
+  gem 'haml', '4.0.2'   # Getting load issues with haml 4.0.3
+  gem 'minitest_tu_shim', :require => false
+RB
+
+gemfile <<-RB
+  gem 'rails', '~>3.0.20'
+  gem 'haml', '4.0.2'   # Getting load issues with haml 4.0.3
+  gem 'minitest_tu_shim', :require => false
+RB
+
+if RUBY_VERSION <= '1.9.3' && (!defined?(RUBY_ENGINE) || RUBY_ENGINE != 'jruby')
+  gemfile <<-RB
+    gem 'mmtrix_rpm', :require => false, :path => File.expand_path('../../../../')
+    gem 'rails', '~> 2.3.18'
+    gem 'haml', '~> 3.1.0'   # last haml version that supports Rails 2
+    gem 'minitest_tu_shim', :require => false
+    gem 'sqlite3'
+    gem 'i18n', '0.6.11'
+  RB
+end

data/test/multiverse/suites/rails/action_controller_live_rum_test.rb

@@ -0,0 +1,39 @@
+# encoding: utf-8
+# This file is distributed under Mmtrix's license terms.
+# See https://github.com/mmtrix/rpm/blob/master/LICENSE for complete details.
+
+require './app'
+
+if defined?(ActionController::Live)
+
+class UndeadController < ApplicationController
+  RESPONSE_BODY = "<html><head></head><body>Brains!</body></html>"
+
+  def brains
+    render :inline => RESPONSE_BODY
+  end
+end
+
+class LiveController < UndeadController
+  include ActionController::Live
+end
+
+class ActionControllerLiveRumTest < RailsMultiverseTest
+  include MultiverseHelpers
+
+  JS_LOADER = "JS LOADER IN DA HOUSE"
+
+  setup_and_teardown_agent(:js_agent_loader => JS_LOADER, :beacon => "beacon", :browser_key => "key")
+
+  def test_rum_instrumentation_when_not_streaming
+    get '/undead/brains'
+    assert_includes(response.body, JS_LOADER)
+  end
+
+  def test_excludes_rum_instrumentation_when_streaming_with_action_controller_live
+    get '/live/brains'
+    assert_equal(LiveController::RESPONSE_BODY, response.body)
+  end
+end
+
+end

data/test/multiverse/suites/rails/activejob_test.rb

@@ -0,0 +1,152 @@
+# encoding: utf-8
+# This file is distributed under Mmtrix's license terms.
+# See https://github.com/mmtrix/rpm/blob/master/LICENSE for complete details.
+
+require File.expand_path(File.join(__FILE__, '..', 'app'))
+
+require 'logger'
+require 'stringio'
+
+# ActiveJob is in Rails 4.2+, so make sure we're on an allowed version before
+# we try to load. Previously just tried to require it, but that had load issues
+# on Rubinius.
+if Rails::VERSION::STRING >= "4.2.0"
+
+require 'active_job'
+
+ActiveJob::Base.queue_adapter = :inline
+
+class MyJob < ActiveJob::Base
+  def perform
+    # Nothing needed!
+  end
+end
+
+class MyJobWithAlternateQueue < ActiveJob::Base
+  queue_as :my_jobs
+
+  def perform
+  end
+end
+
+class MyJobWithParams < ActiveJob::Base
+  def self.last_params
+    @@last_params
+  end
+
+  def perform(first, last)
+    @@last_params = [first, last]
+  end
+end
+
+class MyFailure < ActiveJob::Base
+
+  def perform
+    raise ArgumentError.new("No it isn't!")
+  end
+end
+
+class ActiveJobTest < Minitest::Test
+
+  include MultiverseHelpers
+
+  setup_and_teardown_agent do
+    @log = StringIO.new
+    ActiveJob::Base.logger = ::Logger.new(@log)
+  end
+
+  def after_teardown
+    unless passed?
+      puts "\nEmitting log from failure: #{self.name}"
+      @log.rewind
+      puts @log.read
+    end
+  end
+
+  ENQUEUE_PREFIX = "MessageBroker/ActiveJob::Inline/Queue/Produce/Named"
+  PERFORM_PREFIX = "MessageBroker/ActiveJob::Inline/Queue/Consume/Named"
+
+  PERFORM_TRANSACTION_NAME   = 'OtherTransaction/ActiveJob::Inline/MyJob/execute'
+  PERFORM_TRANSACTION_ROLLUP = 'OtherTransaction/ActiveJob::Inline/all'
+
+  def test_record_enqueue_metrics
+    in_web_transaction do
+      MyJob.perform_later
+    end
+
+    assert_metrics_recorded("#{ENQUEUE_PREFIX}/default")
+  end
+
+  def test_record_enqueue_metrics_with_alternate_queue
+    in_web_transaction do
+      MyJobWithAlternateQueue.perform_later
+    end
+
+    assert_metrics_recorded("#{ENQUEUE_PREFIX}/my_jobs")
+  end
+
+  def test_record_perform_metrics_in_web
+    in_web_transaction do
+      MyJob.perform_later
+    end
+
+    assert_metrics_recorded("#{PERFORM_PREFIX}/default")
+  end
+
+  def test_record_perform_metrics_with_alternate_queue_in_web
+    in_web_transaction do
+      MyJobWithAlternateQueue.perform_later
+    end
+
+    assert_metrics_recorded("#{PERFORM_PREFIX}/my_jobs")
+  end
+
+  def test_doesnt_record_perform_metrics_from_background
+    in_background_transaction do
+      MyJob.perform_later
+    end
+
+    assert_metrics_not_recorded("#{PERFORM_PREFIX}/default")
+  end
+
+  def test_starts_transaction_if_there_isnt_one
+    MyJob.perform_later
+    assert_metrics_recorded([PERFORM_TRANSACTION_ROLLUP,
+                             PERFORM_TRANSACTION_NAME])
+  end
+
+  def test_nests_other_transaction_if_already_running
+    in_background_transaction do
+      MyJob.perform_later
+    end
+
+    assert_metrics_recorded([PERFORM_TRANSACTION_ROLLUP,
+                             PERFORM_TRANSACTION_NAME])
+  end
+
+  # If running tasks inline, either in a dev environment or from
+  # misconfiguration we shouldn't accidentally rename our web transaction
+  def test_doesnt_nest_transactions_if_in_web
+    in_web_transaction do
+      MyJob.perform_later
+    end
+
+    assert_metrics_not_recorded([PERFORM_TRANSACTION_ROLLUP,
+                                 PERFORM_TRANSACTION_NAME])
+  end
+
+  def test_doesnt_interfere_with_params_on_job
+    MyJobWithParams.perform_later("1", "2")
+    assert_equal(["1", "2"], MyJobWithParams.last_params)
+  end
+
+  def test_captures_errors
+    # Because we're processing inline, we get the error raised here
+    assert_raises ArgumentError do
+      MyFailure.perform_later
+    end
+    assert_metrics_recorded(["Errors/all"])
+  end
+end
+
+end

data/test/multiverse/suites/rails/app/views/foos/_foo.html.haml

@@ -0,0 +1 @@
+= foo.inspect

data/test/multiverse/suites/rails/app/views/views/_a_partial.html.erb

@@ -0,0 +1,2 @@
+<%- advance_time(@delay) if @delay %>
+This is a partial

data/test/multiverse/suites/rails/app/views/views/_mid_partial.html.erb

@@ -0,0 +1 @@
+<%= render 'a_partial' %>

data/test/multiverse/suites/rails/app/views/views/_top_partial.html.erb

@@ -0,0 +1,3 @@
+<% (1..10).each do |i| %>
+  <%= render 'mid_partial', :i => i %>
+<% end %>

data/test/multiverse/suites/rails/app/views/views/deep_partial.html.erb

@@ -0,0 +1,3 @@
+<%- 10.times do |i| %>
+  <%= render 'top_partial' %>
+<%- end %>

data/test/multiverse/suites/rails/app/views/views/haml_view.html.haml

@@ -0,0 +1,6 @@
+!!!
+%html
+  %head
+    %title Hello
+  %body
+    hello from haml

data/test/multiverse/suites/rails/app/views/views/index.html.erb

@@ -0,0 +1,4 @@
+<%- advance_time(@delay) if @delay %>
+<%- 3.times do |i| %>
+<%= render "a_partial" %>
+<%- end%>

data/test/multiverse/suites/rails/app.rb

@@ -0,0 +1,44 @@
+# encoding: utf-8
+# This file is distributed under Mmtrix's license terms.
+# See https://github.com/mmtrix/rpm/blob/master/LICENSE for complete details.
+
+require File.join(File.dirname(__FILE__), 'middlewares')
+
+suite_dir = File.expand_path(File.dirname(__FILE__))
+
+if defined?(Rails::VERSION::MAJOR) && Rails::VERSION::MAJOR > 2
+  require File.join(suite_dir, 'rails3_app', 'app_rails3_plus')
+  class RailsMultiverseTest < ActionDispatch::IntegrationTest; end
+elsif !defined?(RAILS_ROOT)
+  RAILS_ROOT = File.join(suite_dir, 'rails2_app')
+  require File.join(RAILS_ROOT, 'config', 'environment')
+  class RailsMultiverseTest < ActionController::IntegrationTest; end
+end
+
+# a basic active model compliant model we can render
+class Foo
+  extend ActiveModel::Naming if defined?(ActiveModel::Naming)
+
+  def to_model
+    self
+  end
+
+  def to_partial_path
+    'foos/foo'
+  end
+
+  def valid?()      true end
+  def new_record?() true end
+  def destroyed?()  true end
+
+  def raise_error
+    raise 'this is an uncaught model error'
+  end
+
+  def errors
+    obj = Object.new
+    def obj.[](key)         [] end
+    def obj.full_messages() [] end
+    obj
+  end
+end

data/test/multiverse/suites/rails/bad_instrumentation_test.rb

@@ -0,0 +1,29 @@
+# encoding: utf-8
+# This file is distributed under Mmtrix's license terms.
+# See https://github.com/mmtrix/rpm/blob/master/LICENSE for complete details.
+
+require './app'
+
+class BadInstrumentationController < ApplicationController
+  # This action is intended to simulate a chunk of instrumentation that pushes
+  # a traced method frame, but then never pops it. Such a situation will break
+  # instrumentation of that request, but should not actually cause the request
+  # to fail.
+  # https://mmtrix.atlassian.net/browse/RUBY-1158
+  def failwhale
+    state = Mmtrix::Agent::TransactionState.tl_get
+    stack = state.traced_method_stack
+    stack.push_frame(state, 'failwhale')
+    render :text => 'everything went great'
+  end
+end
+
+class BadInstrumentationTest < RailsMultiverseTest
+  include MultiverseHelpers
+  setup_and_teardown_agent
+
+  def test_unbalanced_tt_stack_should_not_cause_request_to_fail
+    rsp = get '/bad_instrumentation/failwhale'
+    assert_equal(200, rsp.to_i)
+  end
+end

data/test/multiverse/suites/rails/config/mmtrix.yml

@@ -0,0 +1,30 @@
+common: &default_settings
+  license_key: 'bd0e1d52adade840f7ca727d29a86249e89a6f1c'
+  host: 127.0.0.1
+  port: <%= $collector && $collector.port %>
+  app_name: Rails multiverse test app
+  enabled: true
+  ssl: false
+  apdex_t: 1.0
+  capture_params: true
+  transaction_tracer:
+    enabled: true
+    transaction_threshold: apdex_f
+    record_sql: obfuscated
+    stack_trace_threshold: 0.500
+  error_collector:
+    enabled: true
+    ignore_errors: Mmtrix::TestHelpers::Exceptions::IgnoredError
+
+development:
+  <<: *default_settings
+  developer: true
+
+test:
+  <<: *default_settings
+
+production:
+  <<: *default_settings
+
+staging:
+  <<: *default_settings

data/test/multiverse/suites/rails/dummy.txt

@@ -0,0 +1 @@
+hello!