hydraulic_brake 0.0.0

data/test/logger_test.rb

@@ -0,0 +1,79 @@
+require File.expand_path '../helper', __FILE__
+
+class LoggerTest < Test::Unit::TestCase
+  def stub_http(response, body = nil)
+    response.stubs(:body => body) if body
+    @http = stub(:post => response,
+                 :read_timeout= => nil,
+                 :open_timeout= => nil,
+                 :use_ssl= => nil)
+    Net::HTTP.stubs(:new).returns(@http)
+  end
+
+  def send_notice
+    HydraulicBrake.sender.send_to_airbrake({'foo' => "bar"})
+  end
+
+  def stub_verbose_log
+    HydraulicBrake.stubs(:write_verbose_log)
+  end
+
+  def configure
+    HydraulicBrake.configure { |config| }
+  end
+
+  should "report that notifier is ready when configured" do
+    stub_verbose_log
+    configure
+    assert_logged /Notifier (.*) ready/
+  end
+
+  should "not report that notifier is ready when internally configured" do
+    stub_verbose_log
+    HydraulicBrake.configure(true) { |config| }
+    assert_not_logged /.*/
+  end
+
+  should "print environment info a successful notification without a body" do
+    reset_config
+    stub_verbose_log
+    stub_http(Net::HTTPSuccess)
+    send_notice
+    assert_logged /Environment Info:/
+    assert_not_logged /Response from Airbrake:/
+  end
+
+  should "print environment info on a failed notification without a body" do
+    reset_config
+    stub_verbose_log
+    stub_http(Net::HTTPError)
+    send_notice
+    assert_logged /Environment Info:/
+    assert_not_logged /Response from Airbrake:/
+  end
+
+  should "print environment info and response on a success with a body" do
+    reset_config
+    stub_verbose_log
+    stub_http(Net::HTTPSuccess, 'test')
+    send_notice
+    assert_logged /Environment Info:/
+    assert_logged /Response from Airbrake:/
+  end
+
+  should "print environment info and response on a failure with a body" do
+    reset_config
+    stub_verbose_log
+    stub_http(Net::HTTPError, 'test')
+    send_notice
+    assert_logged /Environment Info:/
+    assert_logged /Response from Airbrake:/
+  end
+
+  should "print information about the notice when Airbrake server fails" do
+    stub_verbose_log
+    stub_http(Net::HTTPError, "test")
+    send_notice
+    assert_logged /Notice details:/
+  end
+end

data/test/notice_test.rb

@@ -0,0 +1,324 @@
+require File.expand_path '../helper', __FILE__
+
+class NoticeTest < Test::Unit::TestCase
+
+  include DefinesConstants
+
+  def configure
+    HydraulicBrake::Configuration.new.tap do |config|
+      config.api_key = 'abc123def456'
+    end
+  end
+
+  def build_notice(args = {})
+    configuration = args.delete(:configuration) || configure
+    HydraulicBrake::Notice.new(configuration.merge(args))
+  end
+
+  def stub_request(attrs = {})
+    stub('request', { :parameters  => { 'one' => 'two' },
+                      :protocol    => 'http',
+                      :host        => 'some.host',
+                      :request_uri => '/some/uri',
+                      :session     => { :to_hash => { 'a' => 'b' } },
+                      :env         => { 'three' => 'four' } }.update(attrs))
+  end
+
+  def assert_accepts_exception_attribute(attribute, args = {}, &block)
+    exception = build_exception
+    block ||= lambda { exception.send(attribute) }
+    value = block.call(exception)
+
+    notice_from_exception = build_notice(args.merge(:exception => exception))
+
+    assert_equal notice_from_exception.send(attribute),
+                 value,
+                 "#{attribute} was not correctly set from an exception"
+
+    notice_from_hash = build_notice(args.merge(attribute => value))
+    assert_equal notice_from_hash.send(attribute),
+                 value,
+                 "#{attribute} was not correctly set from a hash"
+  end
+
+  def assert_valid_notice_document(document)
+    xsd_path = File.join(File.dirname(__FILE__), "airbrake_2_3.xsd")
+    schema = Nokogiri::XML::Schema.new(IO.read(xsd_path))
+    errors = schema.validate(document)
+    assert errors.empty?, errors.collect{|e| e.message }.join
+  end
+
+  def build_backtrace_array
+    ["app/models/user.rb:13:in `magic'",
+      "app/controllers/users_controller.rb:8:in `index'"]
+  end
+
+  def hostname
+    `hostname`.chomp
+  end
+
+  def user
+    Struct.new(:email,:id,:name).
+      new("darth@vader.com",1,"Anakin Skywalker")
+  end
+
+  should "set the api key" do
+    api_key = 'key'
+    notice = build_notice(:api_key => api_key)
+    assert_equal api_key, notice.api_key
+  end
+
+  should "accept a project root" do
+    project_root = '/path/to/project'
+    notice = build_notice(:project_root => project_root)
+    assert_equal project_root, notice.project_root
+  end
+
+  should "accept a component" do
+    assert_equal 'users_controller', build_notice(:component => 'users_controller').controller
+  end
+
+  should "alias the component as controller" do
+    assert_equal 'users_controller', build_notice(:controller => 'users_controller').component
+    assert_equal 'users_controller', build_notice(:component => 'users_controller').controller
+  end
+
+  should "accept a action" do
+    assert_equal 'index', build_notice(:action => 'index').action
+  end
+
+  should "accept a url" do
+    url = 'http://some.host/uri'
+    notice = build_notice(:url => url)
+    assert_equal url, notice.url
+  end
+
+  should "set the host name" do
+    notice = build_notice
+    assert_equal hostname, notice.hostname
+  end
+
+  should "accept a backtrace from an exception or hash" do
+    array = ["user.rb:34:in `crazy'"]
+    exception = build_exception
+    exception.set_backtrace array
+    backtrace = HydraulicBrake::Backtrace.parse(array)
+    notice_from_exception = build_notice(:exception => exception)
+
+
+    assert_equal backtrace,
+                 notice_from_exception.backtrace,
+                 "backtrace was not correctly set from an exception"
+
+    notice_from_hash = build_notice(:backtrace => array)
+    assert_equal backtrace,
+                 notice_from_hash.backtrace,
+                 "backtrace was not correctly set from a hash"
+  end
+
+  should "accept user" do
+    assert_equal user.id, build_notice(:user => user).user.id
+    assert_equal user.email, build_notice(:user => user).user.email
+    assert_equal user.name, build_notice(:user => user).user.name
+  end
+
+  should "pass its backtrace filters for parsing" do
+    backtrace_array = ['my/file/backtrace:3']
+    exception = build_exception
+    exception.set_backtrace(backtrace_array)
+    HydraulicBrake::Backtrace.expects(:parse).with(backtrace_array, {:filters => 'foo'})
+
+    notice = HydraulicBrake::Notice.new({:exception => exception, :backtrace_filters => 'foo'})
+  end
+
+  should "set the error class from an exception or hash" do
+    assert_accepts_exception_attribute :error_class do |exception|
+      exception.class.name
+    end
+  end
+
+  should "set the error message from an exception or hash" do
+    assert_accepts_exception_attribute :error_message do |exception|
+      "#{exception.class.name}: #{exception.message}"
+    end
+  end
+
+  should "accept parameters from a request or hash" do
+    parameters = { 'one' => 'two' }
+    notice_from_hash = build_notice(:parameters => parameters)
+    assert_equal notice_from_hash.parameters, parameters
+  end
+
+  should "accept session data from a session_data hash" do
+    data = { 'one' => 'two' }
+    notice = build_notice(:session_data => data)
+    assert_equal data, notice.session_data
+  end
+
+  should "accept an environment name" do
+    assert_equal 'development', build_notice(:environment_name => 'development').environment_name
+  end
+
+  should "accept CGI data from a hash" do
+    data = { 'string' => 'value' }
+    notice = build_notice(:cgi_data => data)
+    assert_equal data, notice.cgi_data, "should take CGI data from a hash"
+  end
+
+  should "accept notifier information" do
+    params = { :notifier_name    => 'a name for a notifier',
+               :notifier_version => '1.0.5',
+               :notifier_url     => 'http://notifiers.r.us/download' }
+    notice = build_notice(params)
+    assert_equal params[:notifier_name], notice.notifier_name
+    assert_equal params[:notifier_version], notice.notifier_version
+    assert_equal params[:notifier_url], notice.notifier_url
+  end
+
+  should "set sensible defaults without an exception" do
+    backtrace = HydraulicBrake::Backtrace.parse(build_backtrace_array)
+    notice = build_notice(:backtrace => build_backtrace_array)
+
+    assert_equal 'Notification', notice.error_message
+    assert_array_starts_with backtrace.lines, notice.backtrace.lines
+    assert_equal({}, notice.parameters)
+    assert_equal({}, notice.session_data)
+  end
+
+  should "use the caller as the backtrace for an exception without a backtrace" do
+    filters = HydraulicBrake::Configuration.new.backtrace_filters
+    backtrace = HydraulicBrake::Backtrace.parse(caller, :filters => filters)
+    notice = build_notice(:exception => StandardError.new('error'), :backtrace => nil)
+
+    assert_array_starts_with backtrace.lines, notice.backtrace.lines
+  end
+
+  context "a Notice turned into XML" do
+    setup do
+      HydraulicBrake.configure do |config|
+        config.api_key = "1234567890"
+      end
+
+      @exception = build_exception
+
+      @notice = build_notice({
+        :notifier_name    => 'a name',
+        :notifier_version => '1.2.3',
+        :notifier_url     => 'http://some.url/path',
+        :exception        => @exception,
+        :controller       => "controller",
+        :action           => "action",
+        :url              => "http://url.com",
+        :parameters       => { "paramskey"     => "paramsvalue",
+                               "nestparentkey" => { "nestkey" => "nestvalue" } },
+        :session_data     => { "sessionkey" => "sessionvalue" },
+        :cgi_data         => { "cgikey" => "cgivalue" },
+        :project_root     => "RAILS_ROOT",
+        :environment_name => "RAILS_ENV"
+      })
+
+      @xml = @notice.to_xml
+
+      @document = Nokogiri::XML::Document.parse(@xml)
+    end
+
+    should "validate against the XML schema" do
+      assert_valid_notice_document @document
+    end
+
+    should "serialize a Notice to XML when sent #to_xml" do
+      assert_valid_node(@document, "//api-key", @notice.api_key)
+
+      assert_valid_node(@document, "//notifier/name",    @notice.notifier_name)
+      assert_valid_node(@document, "//notifier/version", @notice.notifier_version)
+      assert_valid_node(@document, "//notifier/url",     @notice.notifier_url)
+
+      assert_valid_node(@document, "//error/class",   @notice.error_class)
+      assert_valid_node(@document, "//error/message", @notice.error_message)
+
+      assert_valid_node(@document, "//error/backtrace/line/@number", @notice.backtrace.lines.first.number)
+      assert_valid_node(@document, "//error/backtrace/line/@file", @notice.backtrace.lines.first.file)
+      assert_valid_node(@document, "//error/backtrace/line/@method", @notice.backtrace.lines.first.method)
+
+      assert_valid_node(@document, "//request/url",        @notice.url)
+      assert_valid_node(@document, "//request/component", @notice.controller)
+      assert_valid_node(@document, "//request/action",     @notice.action)
+
+      assert_valid_node(@document, "//request/params/var/@key",     "paramskey")
+      assert_valid_node(@document, "//request/params/var",          "paramsvalue")
+      assert_valid_node(@document, "//request/params/var/@key",     "nestparentkey")
+      assert_valid_node(@document, "//request/params/var/var/@key", "nestkey")
+      assert_valid_node(@document, "//request/params/var/var",      "nestvalue")
+      assert_valid_node(@document, "//request/session/var/@key",    "sessionkey")
+      assert_valid_node(@document, "//request/session/var",         "sessionvalue")
+      assert_valid_node(@document, "//request/cgi-data/var/@key",   "cgikey")
+      assert_valid_node(@document, "//request/cgi-data/var",        "cgivalue")
+
+      assert_valid_node(@document, "//server-environment/project-root",     "RAILS_ROOT")
+      assert_valid_node(@document, "//server-environment/environment-name", "RAILS_ENV")
+      assert_valid_node(@document, "//server-environment/hostname", hostname)
+    end
+  end
+
+  should "not send empty request data" do
+    notice = build_notice
+    assert_nil notice.url
+    assert_nil notice.controller
+    assert_nil notice.action
+
+    xml = notice.to_xml
+    document = Nokogiri::XML.parse(xml)
+    assert_nil document.at('//request/url')
+    assert_nil document.at('//request/component')
+    assert_nil document.at('//request/action')
+
+    assert_valid_notice_document document
+  end
+
+  %w(url controller action).each do |var|
+    should "send a request if #{var} is present" do
+      notice = build_notice(var.to_sym => 'value')
+      xml = notice.to_xml
+      document = Nokogiri::XML.parse(xml)
+      assert_not_nil document.at('//request')
+    end
+  end
+
+  %w(parameters cgi_data session_data).each do |var|
+    should "send a request if #{var} is present" do
+      notice = build_notice(var.to_sym => { 'key' => 'value' })
+      xml = notice.to_xml
+      document = Nokogiri::XML.parse(xml)
+      assert_not_nil document.at('//request')
+    end
+  end
+
+  should "act like a hash" do
+    notice = build_notice(:error_message => 'some message')
+    assert_equal notice.error_message, notice[:error_message]
+  end
+
+  should "return params on notice[:request][:params]" do
+    params = { 'one' => 'two' }
+    notice = build_notice(:parameters => params)
+    assert_equal params, notice[:request][:params]
+  end
+
+  should "ensure #to_hash is called on objects that support it" do
+    assert_nothing_raised do
+      build_notice(:session => { :object => stub(:to_hash => {}) })
+    end
+  end
+
+  should "ensure #to_ary is called on objects that support it" do
+    assert_nothing_raised do
+      build_notice(:session => { :object => stub(:to_ary => {}) })
+    end
+  end
+
+  should "prefer passed error_message to exception message" do
+    exception = build_exception
+    notice = build_notice(:exception => exception,:error_message => "Random ponies")
+    assert_equal "BacktracedException: Random ponies", notice.error_message
+  end
+end

data/test/notifier_test.rb

@@ -0,0 +1,220 @@
+require File.expand_path '../helper', __FILE__
+
+class NotifierTest < Test::Unit::TestCase
+
+  class OriginalException < Exception
+  end
+
+  class ContinuedException < Exception
+  end
+
+  include DefinesConstants
+
+  def setup
+    super
+    reset_config
+  end
+
+  def assert_sent(notice, notice_args)
+    assert_received(HydraulicBrake::Notice, :new) {|expect| expect.with(has_entries(notice_args)) }
+    assert_received(HydraulicBrake.sender, :send_to_airbrake) {|expect| expect.with(notice) }
+  end
+
+  def set_public_env
+    HydraulicBrake.configure { |config| config.environment_name = 'production' }
+  end
+
+  def set_development_env
+    HydraulicBrake.configure { |config| config.environment_name = 'development' }
+  end
+
+  should "yield and save a configuration when configuring" do
+    yielded_configuration = nil
+    HydraulicBrake.configure do |config|
+      yielded_configuration = config
+    end
+
+    assert_kind_of HydraulicBrake::Configuration, yielded_configuration
+    assert_equal yielded_configuration, HydraulicBrake.configuration
+  end
+
+  should "not remove existing config options when configuring twice" do
+    first_config = nil
+    HydraulicBrake.configure do |config|
+      first_config = config
+    end
+    HydraulicBrake.configure do |config|
+      assert_equal first_config, config
+    end
+  end
+
+  should "configure the sender" do
+    sender = stub_sender
+    HydraulicBrake::Sender.stubs(:new => sender)
+    configuration = nil
+
+    HydraulicBrake.configure { |yielded_config| configuration = yielded_config }
+
+    assert_received(HydraulicBrake::Sender, :new) { |expect| expect.with(configuration) }
+    assert_equal sender, HydraulicBrake.sender
+  end
+
+  should "create and send a notice for an exception" do
+    set_public_env
+    exception = build_exception
+    stub_sender!
+    notice = stub_notice!
+
+    HydraulicBrake.notify(exception)
+
+    assert_sent notice, :exception => exception
+  end
+
+  should "create and send a notice for a hash" do
+    set_public_env
+    notice = stub_notice!
+    notice_args = { :error_message => 'uh oh' }
+    stub_sender!
+
+    HydraulicBrake.notify(notice_args)
+
+    assert_sent(notice, notice_args)
+  end
+
+  should "not pass the hash as an exception when sending a notice for it" do
+    set_public_env
+    notice = stub_notice!
+    notice_args = { :error_message => 'uh oh' }
+    stub_sender!
+
+    HydraulicBrake.notify(notice_args)
+
+    assert_received(HydraulicBrake::Notice, :new) {|expect| expect.with(Not(has_key(:exception))) }
+  end
+
+  should "create and send a notice for an exception that responds to to_hash" do
+    set_public_env
+    exception = build_exception
+    notice = stub_notice!
+    notice_args = { :error_message => 'uh oh' }
+    exception.stubs(:to_hash).returns(notice_args)
+    stub_sender!
+
+    HydraulicBrake.notify(exception)
+
+    assert_sent(notice, notice_args.merge(:exception => exception))
+  end
+
+  should "create and sent a notice for an exception and hash" do
+    set_public_env
+    exception = build_exception
+    notice = stub_notice!
+    notice_args = { :error_message => 'uh oh' }
+    stub_sender!
+
+    HydraulicBrake.notify(exception, notice_args)
+
+    assert_sent(notice, notice_args.merge(:exception => exception))
+  end
+
+  should "not create a notice in a development environment" do
+    set_development_env
+    sender = stub_sender!
+
+    HydraulicBrake.notify(build_exception)
+
+    assert_received(sender, :send_to_airbrake) {|expect| expect.never }
+  end
+
+  should "pass config to created notices" do
+    exception = build_exception
+    config_opts = { 'one' => 'two', 'three' => 'four' }
+    stub_notice!
+    stub_sender!
+    HydraulicBrake.configuration = stub('config', :merge => config_opts, :public? => true)
+
+    HydraulicBrake.notify(exception)
+
+    assert_received(HydraulicBrake::Notice, :new) do |expect|
+      expect.with(has_entries(config_opts))
+    end
+  end
+
+  context "building notice JSON for an exception" do
+    setup do
+      @params    = { :controller => "users", :action => "create" }
+      @exception = build_exception
+      @hash      = HydraulicBrake.build_lookup_hash_for(@exception, @params)
+    end
+
+    should "set action" do
+      assert_equal @params[:action], @hash[:action]
+    end
+
+    should "set controller" do
+      assert_equal @params[:controller], @hash[:component]
+    end
+
+    should "set line number" do
+      assert @hash[:line_number] =~ /\d+/
+    end
+
+    should "set file" do
+      assert_match(/test\/helper\.rb$/, @hash[:file])
+    end
+
+    should "set env to production" do
+      assert_equal 'production', @hash[:environment_name]
+    end
+
+    should "set error class" do
+      assert_equal @exception.class.to_s, @hash[:error_class]
+    end
+
+    should "not set file or line number with no backtrace" do
+      @exception.stubs(:backtrace).returns([])
+
+      @hash = HydraulicBrake.build_lookup_hash_for(@exception)
+
+      assert_nil @hash[:line_number]
+      assert_nil @hash[:file]
+    end
+
+    should "not set action or controller when not provided" do
+      @hash = HydraulicBrake.build_lookup_hash_for(@exception)
+
+      assert_nil @hash[:action]
+      assert_nil @hash[:controller]
+    end
+
+    context "when an exception that provides #original_exception is raised" do
+      setup do
+        @exception.stubs(:original_exception).returns(begin
+          raise NotifierTest::OriginalException.new
+        rescue Exception => e
+          e
+        end)
+      end
+
+      should "unwrap exceptions that provide #original_exception" do
+        @hash = HydraulicBrake.build_lookup_hash_for(@exception)
+        assert_equal "NotifierTest::OriginalException", @hash[:error_class]
+      end
+    end
+
+    context "when an exception that provides #continued_exception is raised" do
+      setup do
+        @exception.stubs(:continued_exception).returns(begin
+          raise NotifierTest::ContinuedException.new
+        rescue Exception => e
+          e
+        end)
+      end
+
+      should "unwrap exceptions that provide #continued_exception" do
+        @hash = HydraulicBrake.build_lookup_hash_for(@exception)
+        assert_equal "NotifierTest::ContinuedException", @hash[:error_class]
+      end
+    end
+  end
+end

data/test/recursion_test.rb

@@ -0,0 +1,10 @@
+require File.expand_path '../helper', __FILE__
+
+class RecursionTest < Test::Unit::TestCase
+  should "not allow infinite recursion" do
+    hash = {:a => :a}
+    hash[:hash] = hash
+    notice = HydraulicBrake::Notice.new(:parameters => hash)
+    assert_equal "[possible infinite recursion halted]", notice.parameters[:hash]
+  end
+end