elasticsearch-drain 0.0.1

data/test/elasticsearch/drain/test_nodes.rb

@@ -0,0 +1,42 @@
+require 'test_helper'
+require 'pp'
+
+class TestNodes < Minitest::Test
+  def setup
+    VCR.insert_cassette 'nodes', record: :new_episodes
+    @drain = ::Elasticsearch::Drain.new(
+      'localhost:9250',
+      'esuilogs-razor-d0prod-r01-v000',
+      'us-west-2'
+    )
+    @nodes = ::Elasticsearch::Drain::Nodes.new(@drain.client, @drain.asg)
+  end
+
+  def teardown
+    VCR.eject_cassette
+  end
+
+  def test_has_nodes
+    assert_respond_to @nodes, :client
+  end
+
+  def test_nodes_is_array
+    assert_respond_to @nodes.nodes, :each
+  end
+
+  def test_has_info
+    assert_respond_to @nodes, :info
+  end
+
+  def test_info_has_value
+    assert_respond_to @nodes.info, :each
+  end
+
+  def test_has_stats
+    assert_respond_to @nodes, :stats
+  end
+
+  def test_stats_has_value
+    assert_respond_to @nodes.stats, :each
+  end
+end

data/test/elasticsearch/test_drain.rb

@@ -0,0 +1,44 @@
+require 'test_helper'
+
+class TestDrain < Minitest::Test
+  def setup
+    VCR.insert_cassette 'drain', record: :new_episodes
+    @drain = ::Elasticsearch::Drain.new(
+      'localhost:9250',
+      'esuilogs-razor-d0prod-r01-v000',
+      'us-west-2'
+    )
+  end
+
+  def teardown
+    VCR.eject_cassette
+  end
+
+  def test_nodes_method
+    assert_respond_to @drain, :nodes
+  end
+
+  def test_has_hosts_attribute
+    assert_respond_to @drain, :hosts
+  end
+
+  def test_has_asg
+    assert_equal 'esuilogs-razor-d0prod-r01-v000', @drain.asg.asg
+  end
+
+  def test_has_region
+    assert_equal 'us-west-2', @drain.region
+  end
+
+  def test_has_cluster
+    assert_respond_to @drain, :cluster
+  end
+
+  def test_cluster_health_is_valid
+    assert (@drain.cluster.healthy? || %w(red yellow green).include?(@drain.cluster.health['status']))
+  end
+
+  def test_es_cluster_hosts_match
+    assert_equal 'localhost:9250', @drain.hosts
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,26 @@
+require 'vcr'
+require 'ipaddr'
+require 'webmock'
+require 'simplecov'
+require 'minitest/spec'
+require 'minitest/autorun'
+require 'elasticsearch/extensions/test/cluster'
+
+gem 'minitest'
+SimpleCov.start
+
+require_relative '../lib/elasticsearch/drain'
+
+VCR.configure do |c|
+  c.cassette_library_dir = 'test/cassettes'
+  c.hook_into :webmock
+#  c.debug_logger = $stderr
+#  c.default_cassette_options = { allow_playback_repeats: true }
+end
+
+def private_ipaddress?(ip)
+  class_a = IPAddr.new('10.0.0.0/8')
+  class_b = IPAddr.new('172.16.0.0/12')
+  class_c = IPAddr.new('192.168.0.0/16')
+  [class_a, class_b, class_c].any? { |i| i.include?(ip) }
+end

metadata

@@ -0,0 +1,160 @@
+--- !ruby/object:Gem::Specification
+name: elasticsearch-drain
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Andrew Thompson
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-11-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: elasticsearch
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2'
+- !ruby/object:Gem::Dependency
+  name: thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.19'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.19'
+description: The purpose of this utility is to drain documents from Elasticsearch
+  nodes in an AutoScaling Group
+email:
+- Andrew_Thompson@rapid7.com
+executables:
+- drain
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rubocop.yml"
+- ".rubocop_todo.yml"
+- ".travis.yml"
+- Gemfile
+- Gemfile.lock
+- Guardfile
+- LICENSE
+- README.md
+- Rakefile
+- Thorfile
+- bin/drain
+- elasticsearch-drain.gemspec
+- lib/elasticsearch/drain.rb
+- lib/elasticsearch/drain/autoscaling.rb
+- lib/elasticsearch/drain/base.rb
+- lib/elasticsearch/drain/cli.rb
+- lib/elasticsearch/drain/cluster.rb
+- lib/elasticsearch/drain/node.rb
+- lib/elasticsearch/drain/nodes.rb
+- lib/elasticsearch/drain/version.rb
+- test/cassettes/autoscaling.yml
+- test/cassettes/cluster.yml
+- test/cassettes/drain.yml
+- test/cassettes/node.yml
+- test/cassettes/nodes.yml
+- test/elasticsearch/drain/test_autoscaling.rb
+- test/elasticsearch/drain/test_cluster.rb
+- test/elasticsearch/drain/test_node.rb
+- test/elasticsearch/drain/test_nodes.rb
+- test/elasticsearch/test_drain.rb
+- test/test_helper.rb
+homepage: ''
+licenses:
+- MIT
+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.4.4
+signing_key: 
+specification_version: 4
+summary: Elasticsearch node replacement utility that tries to keep the cluster healthy
+test_files:
+- test/cassettes/autoscaling.yml
+- test/cassettes/cluster.yml
+- test/cassettes/drain.yml
+- test/cassettes/node.yml
+- test/cassettes/nodes.yml
+- test/elasticsearch/drain/test_autoscaling.rb
+- test/elasticsearch/drain/test_cluster.rb
+- test/elasticsearch/drain/test_node.rb
+- test/elasticsearch/drain/test_nodes.rb
+- test/elasticsearch/test_drain.rb
+- test/test_helper.rb
+has_rdoc: