shivam 0.0.0-java

data/spec/hutch/config_spec.rb

@@ -0,0 +1,259 @@
+require 'hutch/config'
+require 'tempfile'
+
+describe Hutch::Config do
+  let(:new_value) { 'not-localhost' }
+
+  before do
+    Hutch::Config.instance_variable_set(:@config, nil)
+    Hutch::Config.initialize
+  end
+
+  after do
+    Hutch::Config.instance_variable_set(:@config, nil)
+  end
+
+  describe '.get' do
+    context 'for valid attributes' do
+      subject { Hutch::Config.get(:mq_host) }
+
+      context 'with no overridden value' do
+        it { is_expected.to eq('127.0.0.1') }
+      end
+
+      context 'with an overridden value' do
+        before do
+          Hutch::Config.set(:mq_host, new_value)
+        end
+
+        it { is_expected.to eq(new_value) }
+      end
+    end
+
+    context 'for invalid attributes' do
+      let(:invalid_get) { Hutch::Config.get(:invalid_attr) }
+
+      specify do
+        expect { invalid_get }.to raise_error Hutch::UnknownAttributeError
+      end
+    end
+  end
+
+  describe '.set' do
+    context 'for valid attributes' do
+      before  { Hutch::Config.set(:mq_host, new_value) }
+      subject { Hutch::Config.user_config[:mq_host] }
+
+      context 'sets value in user config hash' do
+        it { is_expected.to eq(new_value) }
+      end
+
+      context 'type casting' do
+        context 'number attributes' do
+          before  { Hutch::Config.set(:heartbeat, new_value) }
+          subject(:value) { Hutch::Config.user_config[:heartbeat] }
+
+          let(:new_value) { "0" }
+
+
+          specify 'casts values to integers' do
+            expect(value).to eq 0
+          end
+        end
+      end
+
+      context 'boolean attributes' do
+        context 'from non-empty string' do
+          before  { Hutch::Config.set(:autoload_rails, new_value) }
+          subject(:value) { Hutch::Config.user_config[:autoload_rails] }
+
+          let(:new_value) { "t" }
+
+
+          specify 'casts values to booleans' do
+            expect(value).to eq true
+          end
+        end
+
+        context 'from empty string' do
+          before  { Hutch::Config.set(:autoload_rails, new_value) }
+          subject(:value) { Hutch::Config.user_config[:autoload_rails] }
+
+          let(:new_value) { "" }
+
+
+          specify 'casts values to booleans' do
+            expect(value).to eq false
+          end
+        end
+
+        context 'from boolean' do
+          before  { Hutch::Config.set(:autoload_rails, new_value) }
+          subject(:value) { Hutch::Config.user_config[:autoload_rails] }
+
+          let(:new_value) { true }
+
+
+          specify 'casts values to booleans' do
+            expect(value).to eq true
+          end
+        end
+
+        context 'from nil' do
+          before  { Hutch::Config.set(:autoload_rails, new_value) }
+          subject(:value) { Hutch::Config.user_config[:autoload_rails] }
+
+          let(:new_value) { nil }
+
+
+          specify 'casts values to booleans' do
+            expect(value).to eq false
+          end
+        end
+      end
+
+      context 'string attributes' do
+        before  { Hutch::Config.set(:mq_exchange_type, new_value) }
+        subject(:value) { Hutch::Config.user_config[:mq_exchange_type] }
+
+        let(:new_value) { 1 }
+
+
+        specify 'does not perform any typecasting' do
+          expect(value).to eq new_value
+        end
+      end
+    end
+
+    context 'for invalid attributes' do
+      let(:invalid_set) { Hutch::Config.set(:invalid_attr, new_value) }
+
+      specify do
+        expect { invalid_set }.to raise_error Hutch::UnknownAttributeError
+      end
+    end
+  end
+
+  describe 'a magic getter' do
+    context 'for a valid attribute' do
+      it 'calls get' do
+        expect(Hutch::Config).to receive(:get).with(:mq_host)
+        Hutch::Config.mq_host
+      end
+    end
+
+    context 'for an invalid attribute' do
+      let(:invalid_getter) { Hutch::Config.invalid_attr }
+      specify { expect { invalid_getter }.to raise_error NoMethodError }
+    end
+
+    context 'for an ENV-overriden value attribute' do
+      around do |example|
+        ENV['HUTCH_MQ_HOST'] = 'example.com'
+        ENV['HUTCH_MQ_PORT'] = '10001'
+        ENV['HUTCH_MQ_TLS'] = 'true'
+        example.run
+        ENV.delete('HUTCH_MQ_HOST')
+        ENV.delete('HUTCH_MQ_PORT')
+        ENV.delete('HUTCH_MQ_TLS')
+      end
+
+      it 'returns the override' do
+        expect(Hutch::Config.mq_host).to eq 'example.com'
+      end
+
+      it 'returns the override for integers' do
+        expect(Hutch::Config.mq_port).to eq 10_001
+      end
+
+      it 'returns the override for booleans' do
+        expect(Hutch::Config.mq_tls).to eq true
+      end
+    end
+  end
+
+  describe 'a magic setter' do
+    context 'for a valid attribute' do
+      it 'calls set' do
+        expect(Hutch::Config).to receive(:set).with(:mq_host, new_value)
+        Hutch::Config.mq_host = new_value
+      end
+    end
+
+    context 'for an invalid attribute' do
+      let(:invalid_setter) { Hutch::Config.invalid_attr = new_value }
+      specify { expect { invalid_setter }.to raise_error NoMethodError }
+    end
+  end
+
+  describe '.load_from_file' do
+    let(:host) { 'broker.yourhost.com' }
+    let(:username) { 'calvin' }
+    let(:file) do
+      Tempfile.new('configs.yaml').tap do |t|
+        t.write(YAML.dump(config_data))
+        t.rewind
+      end
+    end
+
+    context 'when an attribute is invalid' do
+      let(:config_data) { { random_attribute: 'socks' } }
+      it 'raises an error' do
+        expect do
+          Hutch::Config.load_from_file(file)
+        end.to raise_error(NoMethodError)
+      end
+    end
+
+    context 'when attributes are valid' do
+      let(:config_data) { { mq_host: host, mq_username: username } }
+
+      it 'loads in the config data' do
+        Hutch::Config.load_from_file(file)
+        expect(Hutch::Config.mq_host).to eq host
+        expect(Hutch::Config.mq_username).to eq username
+      end
+    end
+
+    context 'when using ERB' do
+      let(:host) { 'localhost' }
+      let(:file) do
+        Tempfile.new('configs.yaml').tap do |t|
+          t.write(config_contents)
+          t.rewind
+        end
+      end
+      let(:config_contents) do
+        <<-YAML
+mq_host: 'localhost'
+mq_username: '<%= "calvin" %>'
+YAML
+      end
+      it 'loads in the config data' do
+        Hutch::Config.load_from_file(file)
+        expect(Hutch::Config.mq_host).to eq host
+        expect(Hutch::Config.mq_username).to eq username
+      end
+    end
+  end
+
+  context 'developer ergonomics' do
+    it 'will accept strings and symbols as config keys' do
+      expect(Hutch::Config.get(:mq_host)).to eq '127.0.0.1'
+      expect(Hutch::Config.get('mq_host')).to eq '127.0.0.1'
+    end
+
+    describe 'it will not overwrite existing config' do
+      it 'with defaults' do
+        expect(Hutch::Config.get(:mq_host)).to eq '127.0.0.1'
+        Hutch::Config.initialize
+
+        Hutch::Config.set(:mq_host, 'example2.com')
+
+        expect(Hutch::Config.get(:mq_host)).to eq 'example2.com'
+        Hutch::Config.initialize
+        expect(Hutch::Config.get(:mq_host)).to eq 'example2.com'
+      end
+    end
+  end
+end

data/spec/hutch/consumer_spec.rb

@@ -0,0 +1,208 @@
+require 'spec_helper'
+
+describe Hutch::Consumer do
+  around(:each) do |example|
+    isolate_constants do
+      example.run
+    end
+  end
+
+  let(:simple_consumer) do
+    unless defined? SimpleConsumer
+      class SimpleConsumer
+        include Hutch::Consumer
+        consume 'hutch.test1'
+      end
+    end
+    SimpleConsumer
+  end
+
+  let(:complex_consumer) do
+    unless defined? ComplexConsumer
+      class ComplexConsumer
+        include Hutch::Consumer
+        consume 'hutch.test1', 'hutch.test2'
+        arguments foo: :bar
+      end
+    end
+    ComplexConsumer
+  end
+
+  let(:consumer_using_quorum_queue) do
+    unless defined? ConsumerUsingQuorumQueue
+      class ConsumerUsingQuorumQueue
+        include Hutch::Consumer
+        consume 'hutch.test1'
+        arguments foo: :bar
+        
+        quorum_queue
+      end
+    end
+    ConsumerUsingQuorumQueue
+  end
+
+  let(:consumer_using_classic_queue) do
+    unless defined? ConsumerUsingLazyQueue
+      class ConsumerUsingLazyQueue
+        include Hutch::Consumer
+        consume 'hutch.test1'
+        arguments foo: :bar
+        lazy_queue
+        classic_queue
+      end
+    end
+    ConsumerUsingLazyQueue
+  end
+
+  describe 'module inclusion' do
+    it 'registers the class as a consumer' do
+      expect(Hutch).to receive(:register_consumer) do |klass|
+        expect(klass).to eq(simple_consumer)
+      end
+
+      simple_consumer
+    end
+  end
+
+
+  describe '.consume' do
+    it 'saves the routing key to the consumer' do
+      expect(simple_consumer.routing_keys).to include 'hutch.test1'
+    end
+
+    context 'with multiple routing keys' do
+      it 'registers the class once for each routing key' do
+        expect(complex_consumer.routing_keys).to include 'hutch.test1'
+        expect(complex_consumer.routing_keys).to include 'hutch.test2'
+      end
+    end
+
+    context 'when given the same routing key multiple times' do
+      subject { simple_consumer.routing_keys }
+      before { simple_consumer.consume 'hutch.test1' }
+
+      describe '#length' do
+        subject { super().length }
+        it { is_expected.to eq(1)}
+      end
+    end
+  end
+
+  describe '.queue_name' do
+    let(:queue_name) { 'foo' }
+
+    it 'overrides the queue name' do
+      simple_consumer.queue_name(queue_name)
+      expect(simple_consumer.get_queue_name).to eq(queue_name)
+    end
+  end
+
+  describe 'default queue mode' do
+    it 'does not specify any mode by default' do
+      expect(simple_consumer.queue_mode).to eq(nil)
+      expect(simple_consumer.queue_type).to eq(nil)
+    end
+  end
+
+  describe '.lazy_queue' do
+    context 'when queue mode has been set explicitly to lazy' do
+      it 'sets queue mode to lazy' do
+        expect(consumer_using_classic_queue.queue_mode).to eq('lazy')
+      end
+    end
+  end
+
+  describe '.classic_queue' do
+    context 'when queue type has been set explicitly to classic' do
+      it 'sets queue type to classic' do
+        expect(consumer_using_classic_queue.queue_type).to eq('classic')
+      end
+    end
+  end
+
+  describe '.quorum_queue' do
+    context 'when queue type has been set explicitly to quorum' do
+      it 'sets queue type to quorum' do
+        expect(consumer_using_quorum_queue.queue_type).to eq('quorum')
+      end
+
+      it 'accepts initial group size as an option' do
+        consumer = simple_consumer
+        expect { consumer.quorum_queue(initial_group_size: 3) }
+          .to change { consumer.initial_group_size }.to(3)
+      end
+    end
+  end
+
+  describe '.arguments' do
+    let(:args) { { foo: :bar} }
+
+    it 'overrides the arguments' do
+      simple_consumer.arguments(args)
+      expect(simple_consumer.get_arguments).to eq(args)
+    end
+
+  end
+
+  describe '.get_arguments' do
+    context 'when defined' do
+      it { expect(complex_consumer.get_arguments).to include(foo: :bar) }
+    end
+
+    context 'when queue is lazy' do
+      it 'has the x-queue-mode argument set to lazy' do
+        expect(consumer_using_classic_queue.get_arguments['x-queue-mode'])
+          .to eq('lazy')
+      end
+    end
+
+    context "when queue's type is quorum" do
+      let(:arguments) { consumer_using_quorum_queue.get_arguments }
+      it 'has the x-queue-type argument set to quorum' do
+        expect(arguments['x-queue-type']).to eq('quorum')
+        expect(arguments).to_not have_key('x-quorum-initial-group-size')
+      end
+
+      it 'has the x-quorum-initial-group-size argument set to quorum' do
+        consumer_using_quorum_queue.quorum_queue(initial_group_size: 5)
+        expect(arguments['x-queue-type']).to eq('quorum')
+        expect(arguments['x-quorum-initial-group-size']).to eq(5)
+      end
+    end
+  end
+
+  describe '.get_queue_name' do
+
+    context 'when queue name has been set explicitly' do
+      it 'returns the give queue name' do
+        class Foo
+          include Hutch::Consumer
+          queue_name "bar"
+        end
+
+        expect(Foo.get_queue_name).to eq("bar")
+      end
+    end
+
+    context 'when no queue name has been set' do
+      it 'replaces module separators with colons' do
+        module Foo
+          class Bar
+            include Hutch::Consumer
+          end
+        end
+
+        expect(Foo::Bar.get_queue_name).to eq('foo:bar')
+      end
+
+      it 'converts camelcase class names to snake case' do
+        class FooBarBAZ
+          include Hutch::Consumer
+        end
+
+        expect(FooBarBAZ.get_queue_name).to eq('foo_bar_baz')
+      end
+    end
+  end
+end
+

data/spec/hutch/error_handlers/airbrake_spec.rb

@@ -0,0 +1,49 @@
+require 'spec_helper'
+
+describe Hutch::ErrorHandlers::Airbrake do
+  let(:error_handler) { Hutch::ErrorHandlers::Airbrake.new }
+
+  describe '#handle' do
+    let(:error) do
+      begin
+        raise "Stuff went wrong"
+      rescue RuntimeError => err
+        err
+      end
+    end
+
+    it "logs the error to Airbrake" do
+      message_id = "1"
+      properties = OpenStruct.new(message_id: message_id)
+      payload = "{}"
+      consumer = double
+      ex = error
+      message = {
+        payload: payload,
+        consumer: consumer,
+        cgi_data: ENV.to_hash,
+      }
+      expect(::Airbrake).to receive(:notify).with(ex, message)
+      error_handler.handle(properties, payload, consumer, ex)
+    end
+  end
+
+  describe '#handle_setup_exception' do
+    let(:error) do
+      begin
+        raise "Stuff went wrong"
+      rescue RuntimeError => err
+        err
+      end
+    end
+
+    it "logs the error to Airbrake" do
+      ex = error
+      message = {
+        cgi_data: ENV.to_hash,
+      }
+      expect(::Airbrake).to receive(:notify).with(ex, message)
+      error_handler.handle_setup_exception(ex)
+    end
+  end
+end

data/spec/hutch/error_handlers/bugsnag_spec.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+require "spec_helper"
+
+describe Hutch::ErrorHandlers::Bugsnag do
+  let(:error_handler) { described_class.new }
+
+  before do
+    Bugsnag.configure do |bugsnag|
+      bugsnag.api_key = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"
+    end
+  end
+
+  describe "#handle" do
+    let(:error) do
+      begin
+        raise "Stuff went wrong"
+      rescue RuntimeError => err
+        err
+      end
+    end
+
+    it "logs the error to Bugsnag" do
+      message_id = "1"
+      properties = OpenStruct.new(message_id: message_id)
+      payload = "{}"
+      consumer = double
+      ex = error
+      message = {
+        payload: payload,
+        consumer: consumer
+      }
+
+      expect(::Bugsnag).to receive(:notify).with(ex).and_call_original
+      expect_any_instance_of(::Bugsnag::Report).to receive(:add_tab).with(:hutch, message)
+      error_handler.handle(properties, payload, consumer, ex)
+    end
+  end
+
+  describe "#handle_setup_exception" do
+    let(:error) do
+      begin
+        raise "Stuff went wrong"
+      rescue RuntimeError => err
+        err
+      end
+    end
+
+    it "logs the error to Bugsnag" do
+      ex = error
+      expect(::Bugsnag).to receive(:notify).with(ex)
+      error_handler.handle_setup_exception(ex)
+    end
+  end
+end

data/spec/hutch/error_handlers/honeybadger_spec.rb

@@ -0,0 +1,58 @@
+require 'spec_helper'
+
+describe Hutch::ErrorHandlers::Honeybadger do
+  let(:error_handler) { Hutch::ErrorHandlers::Honeybadger.new }
+
+  describe '#handle' do
+    let(:error) do
+      begin
+        raise "Stuff went wrong"
+      rescue RuntimeError => err
+        err
+      end
+    end
+
+    it "logs the error to Honeybadger" do
+      message_id = "1"
+      properties = OpenStruct.new(message_id: message_id)
+      payload = "{}"
+      consumer = double
+      ex = error
+      message = {
+        :error_class => ex.class.name,
+          :error_message => "#{ ex.class.name }: #{ ex.message }",
+          :backtrace => ex.backtrace,
+          :context => {
+            :message_id => message_id,
+            :consumer => consumer
+          },
+          :parameters => {
+            :payload => payload
+          }
+      }
+      expect(error_handler).to receive(:notify_honeybadger).with(message)
+      error_handler.handle(properties, payload, consumer, ex)
+    end
+  end
+
+  describe '#handle_setup_exception' do
+    let(:error) do
+      begin
+        raise "Stuff went wrong during setup"
+      rescue RuntimeError => err
+        err
+      end
+    end
+
+    it "logs the error to Honeybadger" do
+      ex = error
+      message = {
+        :error_class => ex.class.name,
+          :error_message => "#{ ex.class.name }: #{ ex.message }",
+          :backtrace => ex.backtrace,
+      }
+      expect(error_handler).to receive(:notify_honeybadger).with(message)
+      error_handler.handle_setup_exception(ex)
+    end
+  end
+end

data/spec/hutch/error_handlers/logger_spec.rb

@@ -0,0 +1,28 @@
+require 'spec_helper'
+
+describe Hutch::ErrorHandlers::Logger do
+  let(:error_handler) { Hutch::ErrorHandlers::Logger.new }
+
+  describe '#handle' do
+    let(:properties) { OpenStruct.new(message_id: "1") }
+    let(:payload) { "{}" }
+    let(:error) { double(message: "Stuff went wrong", class: "RuntimeError",
+                       backtrace: ["line 1", "line 2"]) }
+
+    it "logs three separate lines" do
+      expect(Hutch::Logging.logger).to receive(:error).exactly(3).times
+      error_handler.handle(properties, payload, double, error)
+    end
+  end
+
+  describe '#handle_setup_exception' do
+    let(:error) { double(message: "Stuff went wrong during setup",
+                       class: "RuntimeError",
+                       backtrace: ["line 1", "line 2"]) }
+
+    it "logs two separate lines" do
+      expect(Hutch::Logging.logger).to receive(:error).exactly(2).times
+      error_handler.handle_setup_exception(error)
+    end
+  end
+end

data/spec/hutch/error_handlers/rollbar_spec.rb

@@ -0,0 +1,45 @@
+require 'spec_helper'
+
+describe Hutch::ErrorHandlers::Rollbar do
+  let(:error_handler) { Hutch::ErrorHandlers::Rollbar.new }
+
+  describe '#handle' do
+    let(:error) do
+      begin
+        raise "Stuff went wrong"
+      rescue RuntimeError => err
+        err
+      end
+    end
+
+    it "logs the error to Rollbar" do
+      message_id = "1"
+      properties = OpenStruct.new(message_id: message_id)
+      payload = "{}"
+      consumer = double
+      ex = error
+      message = {
+        payload: payload,
+        consumer: consumer
+      }
+      expect(::Rollbar).to receive(:error).with(ex, message)
+      error_handler.handle(properties, payload, consumer, ex)
+    end
+  end
+
+  describe '#handle_setup_exception' do
+    let(:error) do
+      begin
+        raise "Stuff went wrong"
+      rescue RuntimeError => err
+        err
+      end
+    end
+
+    it "logs the error to Rollbar" do
+      ex = error
+      expect(::Rollbar).to receive(:error).with(ex)
+      error_handler.handle_setup_exception(ex)
+    end
+  end
+end