copperegg-apm 1.0.0.pre1

data/spec/ethon_spec.rb

@@ -0,0 +1,73 @@
+require 'spec_helper'
+
+describe Ethon::Easy::Operations do
+  describe "#perform" do
+    it "should benchmark time for a get request" do
+      easy = Ethon::Easy.new(:url => "https://api.twitter.com/1.1/help/configuration.json")
+      easy.perform
+
+      expect(easy.response_code.to_s).to match(/\A\d{3}\Z/)
+      JSON.parse(easy.response_body)
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["id", "inst"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["inst"].keys.sort).to eq ["time", "url"]
+      expect(hash["inst"]["url"]).to eq "https://api.twitter.com/1.1/help/configuration.json {Ruby}"
+      expect(hash["inst"]["time"].to_s).to match(/\A\d+\Z/)
+    end
+
+    it "should not have authentication nor query params in the url" do
+      easy = Ethon::Easy.new
+      easy.http_request("http://mikeapikey:U@api.rails.dev/v2/revealmetrics/metric_groups.json?show_hidden=true", :get)
+      easy.perform
+
+      expect(easy.response_code.to_s).to match(/\A\d{3}\Z/)
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["id", "inst"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["inst"].keys.sort).to eq ["time", "url"]
+      expect(hash["inst"]["url"]).to eq "http://api.rails.dev/v2/revealmetrics/metric_groups.json {Ruby}"
+      expect(hash["inst"]["time"].to_s).to match(/\A\d+\Z/)
+    end
+
+    it "should benchmark time for a post request" do
+      easy = Ethon::Easy.new
+      easy.http_request("http://mikeapikey:U@api.rails.dev/v2/revealmetrics/tags.json", :post, {:params => {:tag => "asjasfe", :ids => "smtp"}})
+      easy.perform
+
+      expect(easy.response_code.to_s).to match(/\A\d{3}\Z/)
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["id", "inst"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["inst"].keys.sort).to eq ["time", "url"]
+      expect(hash["inst"]["url"]).to eq "http://api.rails.dev/v2/revealmetrics/tags.json {Ruby}"
+      expect(hash["inst"]["time"].to_s).to match(/\A\d+\Z/)
+    end
+
+    it "should benchmark time for a delete request" do
+      easy = Ethon::Easy.new
+      easy.http_request("http://mikeapikey:U@api.rails.dev/v2/revealmetrics/tags/smtp.json", :delete, {:params => {:ids => "smtp"}})
+      easy.perform
+
+      expect(easy.response_code.to_s).to match(/\A\d{3}\Z/)
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["id", "inst"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["inst"].keys.sort).to eq ["time", "url"]
+      expect(hash["inst"]["url"]).to eq "http://api.rails.dev/v2/revealmetrics/tags/smtp.json {Ruby}"
+      expect(hash["inst"]["time"].to_s).to match(/\A\d+\Z/)
+    end
+  end
+end

data/spec/helpers/mysql2_setup.rb

@@ -0,0 +1,51 @@
+begin
+  require 'mysql2'
+  require 'faker'
+rescue LoadError
+  require 'rubygems'
+  require 'mysql2'
+  require 'faker'
+end
+
+begin
+  client = Mysql2::Client.new :host => "localhost", :username => ENV["MYSQL_USER"]
+
+  create_db_sql = <<-SQL
+    CREATE DATABASE IF NOT EXISTS copperegg_apm_test
+    DEFAULT CHARACTER SET = utf8
+  SQL
+
+  client.query create_db_sql
+  client.query "GRANT ALL PRIVILEGES ON copperegg_apm_test.* TO '%'@'%'"
+  client.query "USE copperegg_apm_test"
+
+  create_table_sql = <<-SQL
+    CREATE TABLE IF NOT EXISTS users (
+      id MEDIUMINT NOT NULL AUTO_INCREMENT,
+      username VARCHAR(32),
+      email VARCHAR(100),
+      password VARCHAR(100),
+      details TEXT,
+      created_at TIMESTAMP,
+      updated_at TIMESTAMP,
+      PRIMARY KEY (id)
+    ) DEFAULT CHARSET = utf8
+  SQL
+
+  client.query create_table_sql
+
+  client.query "TRUNCATE users"
+
+  insert_sql = <<-SQL
+    INSERT INTO users (username, email, password, details, created_at, updated_at)
+    VALUES
+  SQL
+
+  10.times do |i|
+    insert_sql << "  ('#{Faker::Internet.user_name}', '#{Faker::Internet.email}', '#{Faker::Lorem.characters(16)}', '#{Faker::Lorem.paragraph(2)}', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)#{"," if i < 9}\n"
+  end
+
+  client.query insert_sql
+rescue Exception => e
+  puts "Could not create database copperegg_apm_test using user set in ENV['MYSQL_USER']. #{e.message}."
+end

data/spec/helpers/mysql_setup.rb

@@ -0,0 +1,45 @@
+require 'mysql'
+require 'faker'
+
+begin
+  connection = Mysql.new("localhost", ENV["MYSQL_USER"])
+
+  create_db_sql = <<-SQL
+    CREATE DATABASE IF NOT EXISTS copperegg_apm_test
+    DEFAULT CHARACTER SET = utf8
+  SQL
+  
+  connection.query create_db_sql
+  connection.query "GRANT ALL PRIVILEGES ON copperegg_apm_test.* TO '%'@'%'"
+  connection.query "USE copperegg_apm_test"
+
+  create_table_sql = <<-SQL
+    CREATE TABLE IF NOT EXISTS users (
+      id MEDIUMINT NOT NULL AUTO_INCREMENT,
+      username VARCHAR(32),
+      email VARCHAR(100),
+      password VARCHAR(100),
+      details TEXT,
+      created_at TIMESTAMP,
+      updated_at TIMESTAMP,
+      PRIMARY KEY (id)
+    ) DEFAULT CHARSET = utf8
+  SQL
+
+  connection.query create_table_sql
+
+  connection.query "TRUNCATE users"
+
+  insert_sql = <<-SQL
+    INSERT INTO users (username, email, password, details, created_at, updated_at)
+    VALUES
+  SQL
+
+  10.times do |i|
+    insert_sql << "  ('#{Faker::Internet.user_name}', '#{Faker::Internet.email}', '#{Faker::Lorem.characters(16)}', '#{Faker::Lorem.paragraph(2)}', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)#{"," if i < 9}\n"
+  end
+
+  connection.query insert_sql
+rescue Mysql::Error => e
+  puts "Could not create database copperegg_apm_test using user set in ENV['MYSQL_USER']. #{e.message}."
+end

data/spec/helpers/pg_setup.rb

@@ -0,0 +1,43 @@
+require 'pg'
+require 'faker'
+
+begin
+  connection = PG.connect(:dbname => 'copperegg_apm_test', :user => ENV["PG_USER"])
+rescue PG::Error => e
+  if e.message =~ /database "copperegg_apm_test" does not exist/
+    if ENV["PG_USER"]
+      `createdb -U #{ENV["PG_USER"]} -w copperegg_apm_test`
+    else
+      `createdb -w copperegg_apm_test`
+    end
+    connection = PG.connect(:dbname => 'copperegg_apm_test', :user => ENV["PG_USER"])
+  else
+    raise e
+  end
+end
+
+create_table_sql = <<-SQL
+  CREATE TABLE IF NOT EXISTS users (
+    username VARCHAR,
+    email VARCHAR,
+    password VARCHAR,
+    details TEXT,
+    created_at TIMESTAMP,
+    updated_at TIMESTAMP
+  )
+SQL
+
+connection.exec create_table_sql
+
+connection.exec "TRUNCATE users"
+
+insert_sql = <<-SQL
+  INSERT INTO users (username, email, password, details, created_at, updated_at)
+  VALUES
+SQL
+
+10.times do |i|
+  insert_sql << "  ('#{Faker::Internet.user_name}', '#{Faker::Internet.email}', '#{Faker::Lorem.characters(16)}', '#{Faker::Lorem.paragraph(2)}', CURRENT_TIMESTAMP, CURRENT_TIMESTAMP)#{"," if i < 9}\n"
+end
+
+connection.exec insert_sql

data/spec/helpers/rails.rb

@@ -0,0 +1,18 @@
+require 'active_support/all'
+require 'action_controller'
+require 'action_dispatch'
+require 'rspec/rails'
+
+module Rails
+  class App
+    def env_config; {} end
+    def routes
+      return @routes if defined?(@routes)
+      @routes = ActionDispatch::Routing::RouteSet.new
+    end
+  end
+
+  def self.application
+    @app ||= App.new
+  end
+end

data/spec/helpers/sqlite3_setup.rb

@@ -0,0 +1,30 @@
+require 'sqlite3'
+require 'faker'
+
+database = SQLite3::Database.new "copperegg_apm_test.db"
+
+database.execute "DROP TABLE IF EXISTS users"
+  
+create_table_sql = <<-SQL
+  CREATE TABLE IF NOT EXISTS users (
+    id MEDIUMINT NOT NULL,
+    username VARCHAR(32),
+    email VARCHAR(100),
+    password VARCHAR(100),
+    details TEXT,
+    PRIMARY KEY (id)
+  );
+SQL
+
+database.execute create_table_sql
+
+insert_sql = <<-SQL
+  INSERT INTO users (id, username, email, password, details)
+  VALUES
+SQL
+
+10.times do |i|
+  insert_sql << "  (#{i}, '#{Faker::Internet.user_name}', '#{Faker::Internet.email}', '#{Faker::Lorem.characters(16)}', '#{Faker::Lorem.paragraph(2)}')#{"," if i < 9}\n"
+end
+
+database.execute insert_sql

data/spec/kernel_spec.rb

@@ -0,0 +1,37 @@
+require 'spec_helper'
+
+describe Kernel do
+  describe "#raise" do
+    it "should instrument any exception" do
+      expect { raise "the roof" }.to raise_error(RuntimeError)
+      print "."
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["excp", "id"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["excp"].keys.sort).to eq ["error", "stacktrace", "ts"]
+      expect(hash["excp"]["error"]).to match(/RuntimeError\|/)
+      expect(hash["excp"]["stacktrace"]).to match(/\Athe roof\n/)
+      expect(hash["excp"]["ts"]).to be_an_instance_of(Fixnum)
+    end
+  end
+
+  describe "#fail" do
+    it "should instrument any exception" do      
+      expect { fail "epically" }.to raise_error(RuntimeError)
+      print "."
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["excp", "id"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["excp"].keys.sort).to eq ["error", "stacktrace", "ts"]
+      expect(hash["excp"]["error"]).to match(/RuntimeError\|/)
+      expect(hash["excp"]["stacktrace"]).to match(/\Aepically\n/)
+      expect(hash["excp"]["ts"]).to be_an_instance_of(Fixnum)
+    end
+  end
+end

data/spec/mysql2_spec.rb

@@ -0,0 +1,58 @@
+require File.dirname(__FILE__) + '/helpers/mysql2_setup'
+require 'spec_helper'
+
+describe Mysql2::Client do
+  describe "#query" do
+    before do
+      @client = Mysql2::Client.new :host => "localhost", :database => "copperegg_apm_test", :username => ENV["MYSQL_USER"]
+    end
+
+    it "should benchmark time" do
+      result = @client.query "select * from users"
+      expect(result).to be_an_instance_of(Mysql2::Result)
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["id", "inst"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["inst"].keys.sort).to eq ["sql", "time"]
+      expect(hash["inst"]["sql"]).to eq "select * from users {Ruby}"
+      expect(hash["inst"]["time"].to_s).to match(/\A\d+\Z/)
+    end
+
+    it "should benchmark update statements" do
+      @client.query "update users set details = 'blah' where id = 1"
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["id", "inst"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["inst"].keys.sort).to eq ["sql", "time"]
+      expect(hash["inst"]["sql"]).to eq "update users set details = ? where id = ? {Ruby}"
+      expect(hash["inst"]["time"].to_s).to match(/\A\d+\Z/)
+    end    
+
+    it "should not benchmark transaction statements" do
+      payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache)
+
+      @client.query "begin"
+
+      expect(CopperEgg::APM.send(:class_variable_get, :@@payload_cache)).to eq payload
+
+      @client.query "select * from users"
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+
+      expect(hash["inst"]["sql"]).to eq "select * from users {Ruby}"
+
+      @client.query "commit"
+
+      final_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+
+      expect(final_payload).to eq last_payload
+    end
+  end
+end

data/spec/mysql_spec.rb

@@ -0,0 +1,66 @@
+require File.dirname(__FILE__) + '/helpers/mysql_setup'
+require 'spec_helper'
+
+describe Mysql do
+  describe "#query" do
+    before do
+      @client = Mysql.new("localhost", ENV["MYSQL_USER"], "", "copperegg_apm_test")
+    end
+
+    it "should benchmark time" do
+      result = @client.query("select * from users")
+      expect(result).to be_an_instance_of(Mysql::Result)
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["id", "inst"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["inst"].keys.sort).to eq ["sql", "time"]
+      expect(hash["inst"]["sql"]).to eq "select * from users {Ruby}"
+      expect(hash["inst"]["time"].to_s).to match(/\A\d+\Z/)
+    end
+
+    it "should not benchmark transaction statements" do
+      payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache)
+
+      @client.query "begin"
+
+      expect(CopperEgg::APM.send(:class_variable_get, :@@payload_cache)).to eq payload
+
+      @client.query "select * from users"
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+
+      expect(hash["inst"]["sql"]).to eq "select * from users {Ruby}"
+
+      @client.query "commit"
+
+      final_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+
+      expect(final_payload).to eq last_payload
+    end
+  end
+
+  describe "#query (with block)" do
+    before do
+      @client = Mysql.new("localhost", ENV["MYSQL_USER"] || "root", "", "copperegg_apm_test")
+    end
+
+    it "should benchmark time" do
+      @client.query("select * from users") do |result|
+        expect(result).to be_an_instance_of(Mysql::Result)
+      end
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["id", "inst"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["inst"].keys.sort).to eq ["sql", "time"]
+      expect(hash["inst"]["sql"]).to eq "select * from users {Ruby}"
+      expect(hash["inst"]["time"].to_s).to match(/\A\d+\Z/)
+    end
+  end
+end

data/spec/net_http_spec.rb

@@ -0,0 +1,52 @@
+require 'spec_helper'
+
+describe Net::HTTP do
+  describe "#request" do
+    it "should benchmark time for a get request" do
+      uri = URI("http://api.rails.dev/v2/revealmetrics/metric_groups.json")
+      request = Net::HTTP::Get.new(uri.request_uri)
+      request.basic_auth :mikeapikey, :U
+      response = Net::HTTP.start(uri.host, uri.port) {|http| http.request(request)}
+  
+      expect(response.code).to match(/\A\d{3}\Z/)
+      expect(response.body).to be_an_instance_of(String)
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["id", "inst"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["inst"].keys.sort).to eq ["time", "url"]
+      expect(hash["inst"]["url"]).to eq "http://api.rails.dev/v2/revealmetrics/metric_groups.json {Ruby}"
+      expect(hash["inst"]["time"].to_s).to match(/\A\d+\Z/)
+    end
+
+    it "should not have authentication nor query params in the url" do
+      Net::HTTP.get(URI("http://mikeapikey:U@api.rails.dev/v2/revealmetrics/metric_groups.json?show_hidden=true"))
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["id", "inst"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["inst"].keys.sort).to eq ["time", "url"]
+      expect(hash["inst"]["url"]).to eq "http://api.rails.dev/v2/revealmetrics/metric_groups.json {Ruby}"
+      expect(hash["inst"]["time"].to_s).to match(/\A\d+\Z/)
+    end
+
+    it "should benchmark time for a post request" do
+      response = Net::HTTP.post_form URI("http://mikeapikey:U@api.rails.dev/v2/revealmetrics/tags.json"), :tag => "asjasfe", :ids => "smtp"
+
+      expect(response.code).to match(/\A\d{3}\Z/)
+
+      last_payload = CopperEgg::APM.send(:class_variable_get, :@@payload_cache).split("\x00").select {|i| i.size > 2}.map {|i| i.sub(/^[^\{]+/,'')}.last
+      hash = JSON.parse last_payload
+      
+      expect(hash.keys.sort).to eq ["id", "inst"]
+      expect(hash["id"]).to match(/\A[0-1a-z]{16}\z/i)
+      expect(hash["inst"].keys.sort).to eq ["time", "url"]
+      expect(hash["inst"]["url"]).to eq "http://api.rails.dev/v2/revealmetrics/tags.json {Ruby}"
+      expect(hash["inst"]["time"].to_s).to match(/\A\d+\Z/)
+    end
+  end
+end