nerve_pharmeasy 0.7.0

data/spec/lib/nerve_spec.rb

@@ -0,0 +1,186 @@
+require 'spec_helper'
+require 'nerve/configuration_manager'
+require 'nerve/service_watcher'
+require 'nerve/reporter'
+require 'nerve/reporter/base'
+require 'nerve'
+
+def make_mock_service_watcher
+    mock_service_watcher = instance_double(Nerve::ServiceWatcher)
+    allow(mock_service_watcher).to receive(:start)
+    allow(mock_service_watcher).to receive(:stop)
+    allow(mock_service_watcher).to receive(:alive?).and_return(true)
+    allow(mock_service_watcher).to receive(:was_up).and_return(true)
+    mock_service_watcher
+end
+
+describe Nerve::Nerve do
+  let(:config_manager) { Nerve::ConfigurationManager.new() }
+  let(:mock_config_manager) { instance_double(Nerve::ConfigurationManager) }
+  let(:nerve_config) { "#{File.dirname(__FILE__)}/../../example/nerve.conf.json" }
+  let(:nerve_instance_id) { 'testid' }
+  let(:mock_service_watcher_one) { make_mock_service_watcher() }
+  let(:mock_service_watcher_two) { make_mock_service_watcher() }
+  let(:mock_reporter) { Nerve::Reporter::Base.new({}) }
+
+  describe 'check run' do
+    subject {
+      expect(config_manager).to receive(:parse_options_from_argv!).and_return({
+        :config => nerve_config,
+        :instance_id => nerve_instance_id,
+        :check_config => true
+      })
+      config_manager.parse_options!
+      Nerve::Nerve.new(config_manager)
+    }
+
+    it 'starts up and checks config' do
+      expect{subject.run}.not_to raise_error
+    end
+  end
+
+  describe 'full application run' do
+    before(:each) {
+      $EXIT = false
+
+      allow(Nerve::Reporter).to receive(:new_from_service) {
+        mock_reporter
+      }
+      allow(Nerve::ServiceWatcher).to receive(:new) { |config|
+        if config['name'] == 'service1'
+          mock_service_watcher_one
+        else
+          mock_service_watcher_two
+        end
+      }
+
+      allow(mock_config_manager).to receive(:reload!) { }
+      allow(mock_config_manager).to receive(:config) { {
+        'instance_id' => nerve_instance_id,
+        'services' => {
+          'service1' => {
+            'host' => 'localhost',
+            'port' => 1234
+          },
+          'service2' => {
+            'host' => 'localhost',
+            'port' => 1235
+          },
+        }
+      } }
+      allow(mock_config_manager).to receive(:options) { {
+        :config => 'noop',
+        :instance_id => nerve_instance_id,
+        :check_config => false
+      } }
+
+    }
+
+    it 'does a regular run and finishes' do
+      nerve = Nerve::Nerve.new(mock_config_manager)
+
+      expect(nerve).to receive(:heartbeat) {
+        $EXIT = true
+      }
+
+      expect{ nerve.run }.not_to raise_error
+    end
+
+    it 'relaunches dead watchers' do
+      nerve = Nerve::Nerve.new(mock_config_manager)
+
+      iterations = 2
+
+      # One service will fail an alive? call and need to be respawned
+      expect(nerve).to receive(:launch_watcher).twice.with('service1', anything).and_call_original
+      expect(nerve).to receive(:reap_watcher).twice.with('service1').and_call_original
+      expect(nerve).to receive(:launch_watcher).once.with('service2', anything).and_call_original
+      expect(nerve).to receive(:reap_watcher).once.with('service2').and_call_original
+
+      expect(nerve).to receive(:heartbeat).exactly(iterations + 1).times do
+        if iterations == 2
+          expect(mock_service_watcher_one).to receive(:alive?).and_return(false)
+          nerve.instance_variable_set(:@config_to_load, true)
+        elsif iterations == 1
+          expect(mock_service_watcher_one).to receive(:alive?).and_return(true)
+          nerve.instance_variable_set(:@config_to_load, true)
+        else
+          $EXIT = true
+        end
+        iterations -= 1
+      end
+
+      expect{ nerve.run }.not_to raise_error
+    end
+
+    it 'responds to changes in configuration' do
+      nerve = Nerve::Nerve.new(mock_config_manager)
+
+      iterations = 4
+      expect(nerve).to receive(:heartbeat).exactly(iterations + 1).times do
+        if iterations == 4
+          expect(nerve.instance_variable_get(:@watchers).keys).to contain_exactly('service1', 'service2')
+
+          # Remove service2 from the config
+          expect(mock_config_manager).to receive(:config).and_return({
+            'instance_id' => nerve_instance_id,
+            'services' => {
+              'service1' => {
+                'host' => 'localhost',
+                'port' => 1234
+              },
+            }
+          })
+          nerve.instance_variable_set(:@config_to_load, true)
+        elsif iterations == 3
+          expect(nerve.instance_variable_get(:@watchers).keys).to contain_exactly('service1')
+          expect(nerve.instance_variable_get(:@config_to_load)).to eq(false)
+
+          # Change the configuration of service1
+          expect(mock_config_manager).to receive(:config).and_return({
+            'instance_id' => nerve_instance_id,
+            'services' => {
+              'service1' => {
+                'host' => 'localhost',
+                'port' => 1236
+              },
+            }
+          })
+          nerve.instance_variable_set(:@config_to_load, true)
+        elsif iterations == 2
+          expect(nerve.instance_variable_get(:@watchers).keys).to contain_exactly('service1')
+          expect(nerve.instance_variable_get(:@watchers_desired).keys).to contain_exactly('service1')
+          expect(nerve.instance_variable_get(:@watchers_desired)['service1']['port']).to eq(1236)
+          expect(nerve.instance_variable_get(:@config_to_load)).to eq(false)
+
+          # Add another service
+          expect(mock_config_manager).to receive(:config) { {
+            'instance_id' => nerve_instance_id,
+            'services' => {
+              'service1' => {
+                'host' => 'localhost',
+                'port' => 1236
+              },
+              'service4' => {
+                'host' => 'localhost',
+                'port' => 1235
+              },
+            }
+          } }
+
+          nerve.instance_variable_set(:@config_to_load, true)
+        elsif iterations == 1
+          expect(nerve.instance_variable_get(:@watchers).keys).to contain_exactly('service1', 'service4')
+          nerve.instance_variable_set(:@config_to_load, true)
+        else
+          $EXIT = true
+        end
+        iterations -= 1
+      end
+
+      expect{ nerve.run }.not_to raise_error
+    end
+
+  end
+end
+

data/spec/spec_helper.rb

@@ -0,0 +1,33 @@
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# Require this file using `require "spec_helper"` to ensure that it is only
+# loaded once.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+require "#{File.dirname(__FILE__)}/../lib/nerve"
+
+require 'factory_girl'
+
+FactoryGirl.find_definitions
+
+RSpec.configure do |config|
+  config.run_all_when_everything_filtered = true
+  config.filter_run :focus
+  config.include RbConfig
+  config.color = true
+
+  # verify every double we can think of
+  config.mock_with :rspec do |mocks|
+    mocks.verify_doubled_constant_names = true
+    mocks.verify_partial_doubles = true
+  end
+
+  # include factory-girl when running tests
+  config.include FactoryGirl::Syntax::Methods
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = 'random'
+end

metadata

@@ -0,0 +1,216 @@
+--- !ruby/object:Gem::Specification
+name: nerve_pharmeasy
+version: !ruby/object:Gem::Version
+  version: 0.7.0
+platform: ruby
+authors:
+- Martin Rhoads
+- Igor Serebryany
+- Pierre Carrier
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-08-23 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: zk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.9.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.9.2
+- !ruby/object:Gem::Dependency
+  name: bunny
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 1.1.0
+- !ruby/object:Gem::Dependency
+  name: etcd
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.3
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.1.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.1.0
+- !ruby/object:Gem::Dependency
+  name: factory_girl
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Nerve is a service registration daemon. It performs health checks on
+  your service and then publishes success or failure into one of several registries
+  (currently, zookeeper or etcd). Nerve is half or SmartStack, and is designed to
+  be operated along with Synapse to provide a full service discovery framework
+email:
+- martin.rhoads@airbnb.com
+- igor.serebryany@airbnb.com
+executables:
+- nerve
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".mailmap"
+- ".nerve.rc"
+- ".travis.yml"
+- CONTRIBUTING.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- Vagrantfile
+- bin/nerve
+- example/nerve.conf.json
+- example/nerve_services/etcd_service1.json
+- example/nerve_services/zookeeper_service1.json
+- lib/nerve.rb
+- lib/nerve/configuration_manager.rb
+- lib/nerve/log.rb
+- lib/nerve/reporter.rb
+- lib/nerve/reporter/base.rb
+- lib/nerve/reporter/etcd.rb
+- lib/nerve/reporter/zookeeper.rb
+- lib/nerve/ring_buffer.rb
+- lib/nerve/service_watcher.rb
+- lib/nerve/service_watcher/base.rb
+- lib/nerve/service_watcher/http.rb
+- lib/nerve/service_watcher/rabbitmq.rb
+- lib/nerve/service_watcher/tcp.rb
+- lib/nerve/utils.rb
+- lib/nerve/version.rb
+- nerve.conf.json
+- nerve.gemspec
+- spec/.gitkeep
+- spec/configuration_manager_spec.rb
+- spec/example_services_spec.rb
+- spec/factories/check.rb
+- spec/factories/service.rb
+- spec/lib/nerve/reporter_etcd_spec.rb
+- spec/lib/nerve/reporter_spec.rb
+- spec/lib/nerve/reporter_zookeeper_spec.rb
+- spec/lib/nerve/service_watcher_spec.rb
+- spec/lib/nerve_spec.rb
+- spec/spec_helper.rb
+homepage: https://github.com/airbnb/nerve
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: A service registration daemon
+test_files:
+- spec/.gitkeep
+- spec/configuration_manager_spec.rb
+- spec/example_services_spec.rb
+- spec/factories/check.rb
+- spec/factories/service.rb
+- spec/lib/nerve/reporter_etcd_spec.rb
+- spec/lib/nerve/reporter_spec.rb
+- spec/lib/nerve/reporter_zookeeper_spec.rb
+- spec/lib/nerve/service_watcher_spec.rb
+- spec/lib/nerve_spec.rb
+- spec/spec_helper.rb