instana 1.11.6 → 1.209.2

data/test/snapshot/google_cloud_run_instance_test.rb

@@ -0,0 +1,74 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+require 'test_helper'
+
+class GoogleCloudRunInstanceTest < Minitest::Test
+  def test_snapshot
+    ENV['K_SERVICE'] = 'test_service'
+    ENV['K_CONFIGURATION'] = 'test_config'
+    ENV['K_REVISION'] = 'test_revision'
+    ENV['PORT'] = 'test_port'
+
+    stub_request(:get, 'http://10.10.10.10//computeMetadata/v1/instance/id')
+      .to_return(status: 200, body: 'test_instance_id')
+    stub_request(:get, 'http://10.10.10.10//computeMetadata/v1/instance/zone')
+      .to_return(status: 200, body: 'region/number/test_region')
+    stub_request(:get, 'http://10.10.10.10//computeMetadata/v1/project/numeric-project-id')
+      .to_return(status: 200, body: 'numericProjectId')
+    stub_request(:get, 'http://10.10.10.10//computeMetadata/v1/project/project-id')
+      .to_return(status: 200, body: 'projectId')
+
+    subject = Instana::Snapshot::GoogleCloudRunInstance.new(metadata_uri: 'http://10.10.10.10/')
+    snapshot = subject.snapshot
+
+    assert_equal Instana::Snapshot::GoogleCloudRunInstance::ID, snapshot[:name]
+    assert_equal 'test_instance_id', snapshot[:entityId]
+
+    assert_equal "ruby", snapshot[:data][:runtime]
+    assert_equal "test_region", snapshot[:data][:region]
+    assert_equal "test_service", snapshot[:data][:service]
+    assert_equal "test_config", snapshot[:data][:configuration]
+    assert_equal "test_revision", snapshot[:data][:revision]
+    assert_equal "test_instance_id", snapshot[:data][:instanceId]
+    assert_equal "test_port", snapshot[:data][:port]
+    assert_equal "numericProjectId", snapshot[:data][:numericProjectId]
+    assert_equal "projectId", snapshot[:data][:projectId]
+  ensure
+    ENV['K_SERVICE'] = nil
+    ENV['K_CONFIGURATION'] = nil
+    ENV['K_REVISION'] = nil
+    ENV['PORT'] = nil
+  end
+
+  def test_snapshot_error
+    stub_request(:get, 'http://10.10.10.10//computeMetadata/v1/instance/id')
+      .to_return(status: 500)
+
+    subject = Instana::Snapshot::GoogleCloudRunInstance.new(metadata_uri: 'http://10.10.10.10/')
+
+    assert_raises do
+      subject.snapshot
+    end
+  end
+
+  def test_source
+    stub_request(:get, 'http://10.10.10.10//computeMetadata/v1/instance/id')
+      .to_return(status: 200, body: 'test_instance_id')
+    subject = Instana::Snapshot::GoogleCloudRunInstance.new(metadata_uri: 'http://10.10.10.10/')
+    source = subject.source
+
+    assert source[:hl]
+    assert_equal 'gcp', source[:cp]
+    assert_equal 'test_instance_id', source[:e]
+  end
+
+  def test_host_name
+    ENV['K_REVISION'] = 'test_revision'
+    subject = Instana::Snapshot::GoogleCloudRunInstance.new(metadata_uri: 'http://10.10.10.10/')
+
+    assert_equal 'gcp:cloud-run:revision:test_revision', subject.host_name
+  ensure
+    ENV['K_REVISION'] = nil
+  end
+end

data/test/snapshot/google_cloud_run_process_test.rb

@@ -0,0 +1,33 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+require 'test_helper'
+
+class GoogleCloudRunProcessTest < Minitest::Test
+  def test_snapshot
+    ENV['K_REVISION'] = 'test'
+    stub_request(:get, 'http://10.10.10.10//computeMetadata/v1/instance/id')
+      .to_return(status: 200, body: 'test_instance_id')
+
+    subject = Instana::Snapshot::GoogleCloudRunProcess.new(metadata_uri: 'http://10.10.10.10/')
+    snapshot = subject.snapshot
+
+    assert_equal Instana::Snapshot::GoogleCloudRunProcess::ID, snapshot[:name]
+    assert_equal 'test_instance_id', snapshot[:data][:container]
+    assert_equal 'gcpCloudRunInstance', snapshot[:data][:containerType]
+    assert_equal 'gcp:cloud-run:revision:test', snapshot[:data][:'com.instana.plugin.host.name']
+  ensure
+    ENV['K_REVISION'] = nil
+  end
+
+  def test_snapshot_error
+    stub_request(:get, 'http://10.10.10.10//computeMetadata/v1/instance/id')
+      .to_return(status: 500)
+
+    subject = Instana::Snapshot::GoogleCloudRunProcess.new(metadata_uri: 'http://10.10.10.10/')
+
+    assert_raises do
+      subject.snapshot
+    end
+  end
+end

data/test/snapshot/lambda_function_test.rb

@@ -0,0 +1,37 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+require 'test_helper'
+
+class LambdaFunctionTest < Minitest::Test
+  def setup
+    @subject = Instana::Snapshot::LambdaFunction.new
+  end
+
+  def test_snapshot
+    Thread.current[:instana_function_arn] = 'test'
+
+    assert_equal Instana::Snapshot::LambdaFunction::ID, @subject.snapshot[:name]
+    assert_equal Thread.current[:instana_function_arn], @subject.snapshot[:entityId]
+  ensure
+    Thread.current[:instana_function_arn] = nil
+  end
+
+  def test_source
+    Thread.current[:instana_function_arn] = 'test'
+
+    assert @subject.source[:hl]
+    assert_equal 'aws', @subject.source[:cp]
+    assert_equal Thread.current[:instana_function_arn], @subject.source[:e]
+  ensure
+    Thread.current[:instana_function_arn] = nil
+  end
+
+  def test_host_name
+    Thread.current[:instana_function_arn] = 'test'
+
+    assert_equal Thread.current[:instana_function_arn], @subject.host_name
+  ensure
+    Thread.current[:instana_function_arn] = nil
+  end
+end

data/test/snapshot/ruby_process_test.rb

@@ -0,0 +1,14 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+require 'test_helper'
+
+class RubyProcessTest < Minitest::Test
+  def test_snapshot
+    subject = Instana::Snapshot::RubyProcess.new
+    snapshot = subject.snapshot
+
+    assert_equal Instana::Snapshot::RubyProcess::ID, snapshot[:name]
+    assert_equal Process.pid.to_s, snapshot[:entityId]
+  end
+end

data/test/support/apps/active_record/active_record.rb

@@ -0,0 +1,24 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+require 'rails'
+require 'active_record/railtie'
+
+migration_class = if ActiveRecord::Migration.respond_to?(:[])
+                    ActiveRecord::Migration[4.2]
+                  else
+                    ActiveRecord::Migration
+                  end
+
+class CreateBlocks < migration_class
+  def change
+    create_table :blocks do |t|
+      t.string :name
+      t.string :color
+      t.timestamps
+    end
+  end
+end
+
+class Block < ActiveRecord::Base
+end

data/test/{servers/grpc_50051.rb → support/apps/grpc/boot.rb}

@@ -1,4 +1,7 @@
-require File.expand_path(File.dirname(__FILE__) + '/../apps/grpc_server.rb')
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2017
+
+require_relative 'grpc_server'
 
 ::Instana.logger.info "Booting instrumented gRPC server on port 50051 for tests."
 

data/test/{apps → support/apps/grpc}/grpc_server.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2017
+
 require 'google/protobuf'
 
 Google::Protobuf::DescriptorPool.generated_pool.build do

data/test/{servers/rackapp_6511.rb → support/apps/http_endpoint/boot.rb}

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2016
+
 require 'rack/handler/puma'
 require 'rack/builder'
 require 'instana/rack'

data/test/{servers/rails_3205.rb → support/apps/rails/boot.rb}

@@ -1,22 +1,19 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2017
 
 ::Instana.logger.warn "Starting background Ruby on Rails #{Rails::VERSION::STRING} application on port 3205"
 
-if Rails::VERSION::STRING >= '5.0' && ::Instana::Test.mysql?
-  ::Instana.logger.fatal "Rails 5.x doesn't support the mysql adapter (discontinued).  Set DB_FLAVOR=msyql2 instead.  This will fail as is."
-end
-
-require "rails/all"
-require "action_controller/railtie" # require more if needed
+require "rails"
+require "active_record/railtie"
+require "active_model/railtie"
+require "action_controller/railtie"
+require "active_model/railtie"
 require 'rack/handler/puma'
 
 if Rails::VERSION::STRING >= '6.0'
-  require File.expand_path(File.dirname(__FILE__) + '/../models/block6')
-  system("mkdir -p app/assets/config && echo '{}' > app/assets/config/manifest.js")
-  # Rails 6 Issue: https://github.com/rails/rails/issues/37183
-  class ApplicationController < ActionController::Base
-  end
+  require_relative 'models/block6'
 else
-  require File.expand_path(File.dirname(__FILE__) + '/../models/block')
+  require_relative 'models/block'
 end
 
 ActiveRecord::Base.establish_connection(ENV['DATABASE_URL'])
@@ -62,7 +59,7 @@ class RailsTestApp < Rails::Application
   # uncomment below to display errors
   # config.consider_all_requests_local = true
 
-  config.paths['app/views'].unshift(File.expand_path(File.dirname(__FILE__) + '/../views'))
+  config.paths['app/views'].unshift(File.join(__dir__, 'views'))
 
   config.active_support.deprecation = :stderr
 

data/test/{models → support/apps/rails/models}/block.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2017
+
 class Block < ActiveRecord::Base
   def do_work(*args)
     block = Block.first
@@ -7,7 +10,7 @@ class Block < ActiveRecord::Base
   end
 end
 
-class CreateBlocks < ActiveRecord::Migration
+class CreateBlocks < ActiveRecord::Migration[4.2]
   def change
     create_table :blocks do |t|
       t.string :name

data/test/{models → support/apps/rails/models}/block6.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2017
+
 class Block < ActiveRecord::Base
   def do_work(*args)
     block = Block.first

data/test/support/apps/resque/boot.rb

@@ -0,0 +1,5 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+require_relative 'jobs/resque_error_job'
+require_relative 'jobs/resque_fast_job'

data/test/{jobs → support/apps/resque/jobs}/resque_error_job.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2018
+
 require "redis"
 require "net/http"
 
@@ -7,8 +10,6 @@ class ErrorJob
   def self.perform
     if ENV.key?('REDIS_URL')
       redis = Redis.new(:url => ENV['REDIS_URL'])
-    elsif ENV.key?('REDIS_URL')
-      redis = Redis.new(:url => ENV['REDIS_URL'])
     else
       redis = Redis.new(:url => 'localhost:6379')
     end
@@ -17,6 +18,5 @@ class ErrorJob
     redis.set('ts', dt)
 
     raise Exception.new("Silly Rabbit, Trix are for kids.")
-    redis.set(:nb_id, 2)
   end
 end

data/test/{jobs → support/apps/resque/jobs}/resque_fast_job.rb

@@ -1,14 +1,17 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2018
+
 require "redis"
 require "net/http"
 
 class FastJob
   @queue = :critical
 
-  def self.perform
+  def self.perform(*args)
+    raise 'Invalid Args' unless args.empty?
+
     if ENV.key?('REDIS_URL')
       redis = Redis.new(:url => ENV['REDIS_URL'])
-    elsif ENV.key?('REDIS_URL')
-      redis = Redis.new(:url => ENV['REDIS_URL'])
     else
       redis = Redis.new(:url => 'redis://localhost:6379')
     end

data/test/support/apps/sidekiq/boot.rb

@@ -0,0 +1,25 @@
+# Hook into sidekiq to control the current mode
+
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+$sidekiq_mode = :client
+class << Sidekiq
+  def server?
+    $sidekiq_mode == :server
+  end
+end
+
+# Configure redis for sidekiq client
+Sidekiq.configure_client do |config|
+  config.redis = { url: ENV['REDIS_URL'] }
+end
+
+# Configure redis for sidekiq worker
+$sidekiq_mode = :server
+::Sidekiq.configure_server do |config|
+  config.redis = { url: ENV['REDIS_URL'] }
+end
+$sidekiq_mode = :client
+
+require_relative 'worker'

data/test/{jobs → support/apps/sidekiq/jobs}/sidekiq_job_1.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2017
+
 class SidekiqJobOne
   include Sidekiq::Worker
 

data/test/{jobs → support/apps/sidekiq/jobs}/sidekiq_job_2.rb

@@ -1,3 +1,6 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2017
+
 class SidekiqJobTwo
   include Sidekiq::Worker
 

data/test/{servers → support/apps}/sidekiq/worker.rb

@@ -1,10 +1,13 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2017
+
 require 'sidekiq/launcher'
 require 'sidekiq/cli'
 require 'sidekiq/api'
 require 'sidekiq/processor'
 
-require Dir.pwd + '/test/jobs/sidekiq_job_1.rb'
-require Dir.pwd + '/test/jobs/sidekiq_job_2.rb'
+require_relative 'jobs/sidekiq_job_1'
+require_relative 'jobs/sidekiq_job_2'
 
 ::Instana.logger.info "Booting instrumented sidekiq worker for tests."
 ::Sidekiq.logger.level = ::Logger::FATAL

data/test/support/helpers.rb

@@ -0,0 +1,97 @@
+# frozen_string_literal: true
+
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+module Instana
+  module TestHelpers
+    # Used to reset the gem to boot state.  It clears out any queued and/or staged
+    # traces and resets the tracer to no active trace.
+    #
+    def clear_all!
+      ::Instana.processor.clear!
+      ::Instana.tracer.clear!
+      nil
+    end
+
+    def disable_redis_instrumentation
+      ::Instana.config[:redis][:enabled] = false
+    end
+
+    def enable_redis_instrumentation
+      ::Instana.config[:redis][:enabled] = true
+    end
+
+    def validate_sdk_span(json_span, sdk_hash = {}, errored = false, _ec = 1)
+      assert_equal :sdk, json_span[:n]
+      assert json_span.key?(:k)
+      assert json_span.key?(:d)
+      assert json_span.key?(:ts)
+
+      sdk_hash.each do |k, v|
+        assert_equal v, json_span[:data][:sdk][k]
+      end
+
+      return unless errored
+
+      assert_equal true, json_span[:error]
+      assert_equal 1, json_span[:ec]
+    end
+
+    def find_spans_by_name(spans, name)
+      spans.select do |span|
+        (span[:n] == :sdk && span[:data][:sdk][:name] == name) || span[:n] == name
+      end.tap do |result|
+        raise StandardError, "No SDK spans (#{name}) could be found" if result.empty?
+      end
+    end
+
+    def find_first_span_by_name(spans, name)
+      spans.each do |span|
+        case span[:n]
+        when :sdk
+          return span if span[:data][:sdk][:name] == name
+        when name
+          return span
+        end
+      end
+      raise StandardError, "Span (#{name}) not found"
+    end
+
+    def find_span_by_id(spans, id)
+      spans.each do |span|
+        return span if span[:s] == id
+      end
+      raise StandardError, "Span with id (#{id}) not found"
+    end
+
+    # Finds the first span in +spans+ for which +block+ returns true
+    #
+    #     ar_span = find_first_span_by_qualifier(ar_spans) do |span|
+    #       span[:data][:activerecord][:sql] == sql
+    #     end
+    #
+    # This helper will raise an exception if no span evaluates to true against he provided block.
+    #
+    # +spans+: +Array+ of spans to search
+    # +block+: The Ruby block to evaluate against each span
+    def find_first_span_by_qualifier(spans, &block)
+      spans.each do |span|
+        return span if block.call(span)
+      end
+      raise StandardError, 'Span with qualifier not found'
+    end
+
+    def has_postgres_database?
+      URI(ENV.fetch('DATABASE_URL', '')).scheme == 'postgres'
+    end
+
+    def has_mysql2_database?
+      URI(ENV.fetch('DATABASE_URL', '')).scheme == 'mysql2'
+    end
+
+    def has_mysql_database?
+      URI(ENV.fetch('DATABASE_URL', '')).scheme == 'mysql'
+    end
+  end
+end

data/test/support/mock_timer.rb

@@ -0,0 +1,20 @@
+# (c) Copyright IBM Corp. 2021
+# (c) Copyright Instana Inc. 2021
+
+class MockTimer
+  attr_reader :opts, :block, :running
+
+  def initialize(*args, &blk)
+    @opts = args.first
+    @block = blk
+    @running = false
+  end
+
+  def shutdown
+    @running = false
+  end
+
+  def execute
+    @running = true
+  end
+end