elasticsearch-drain 0.0.2 → 0.0.4

data/test/elasticsearch/drain/test_autoscaling.rb

@@ -3,7 +3,7 @@ require 'pp'
 
 class TestAutoScaling < Minitest::Test
   def setup
-    VCR.insert_cassette 'autoscaling', record: :once
+    VCR.insert_cassette 'autoscaling', record: :new_episodes
     @asg = ::Elasticsearch::Drain::AutoScaling.new(
       'esuilogs-razor-d0prod-r01-v000',
       'us-east-1'
@@ -22,6 +22,12 @@ class TestAutoScaling < Minitest::Test
     assert_respond_to @asg.find_instances_in_asg, :each
   end
 
+  def test_missing_instance
+    assert_raises(::Elasticsearch::Drain::Errors::NodeNotFound) do
+      @asg.instance('1.1.1.1')
+    end
+  end
+
   def test_find_instances_matches_instance_pattern
     assert_match /i-[a-z0-9]{8}/, @asg.find_instances_in_asg.first
   end
@@ -34,4 +40,36 @@ class TestAutoScaling < Minitest::Test
     ip = @asg.instances.first
     assert private_ipaddress?(ip)
   end
+
+  # TODO: Figure out how to recode this...
+  def test_describe_asg_has_desired_capacity
+    disable_vcr do
+      asg = @asg.describe_autoscaling_group
+      assert_respond_to asg, :desired_capacity
+    end
+  end
+
+  # TODO: Figure out how to recode this...
+  def test_describe_asg_desired_capacity_equals
+    disable_vcr do
+      asg = @asg.describe_autoscaling_group
+      assert_equal 16, asg.desired_capacity
+    end
+  end
+
+  # TODO: Figure out how to recode this...
+  def test_describe_asg_has_min_size
+    disable_vcr do
+      asg = @asg.describe_autoscaling_group
+      assert_respond_to asg, :min_size
+    end
+  end
+
+  # TODO: Figure out how to recode this...
+  def test_describe_asg_has_min_size_equals
+    disable_vcr do
+      asg = @asg.describe_autoscaling_group
+      assert_equal 16, asg.min_size
+    end
+  end
 end

data/test/test_helper.rb

@@ -14,8 +14,8 @@ 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 }
+  # c.debug_logger = $stderr
+  # c.default_cassette_options = { allow_playback_repeats: true }
 end
 
 def private_ipaddress?(ip)
@@ -24,3 +24,14 @@ def private_ipaddress?(ip)
   class_c = IPAddr.new('192.168.0.0/16')
   [class_a, class_b, class_c].any? { |i| i.include?(ip) }
 end
+
+def disable_vcr(&block)
+  return skip unless ENV['ALLOW_DISABLED_VCR'] == 'true'
+  puts 'WARNING: Running with VCR disabled!'
+  VCR.eject_cassette
+  WebMock.allow_net_connect!
+  VCR.turned_off do
+    block.call
+  end
+  WebMock.disable_net_connect!
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: elasticsearch-drain
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
 platform: ruby
 authors:
 - Andrew Thompson
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-11-09 00:00:00.000000000 Z
+date: 2015-11-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -109,6 +109,7 @@ files:
 - lib/elasticsearch/drain/cluster.rb
 - lib/elasticsearch/drain/node.rb
 - lib/elasticsearch/drain/nodes.rb
+- lib/elasticsearch/drain/util.rb
 - lib/elasticsearch/drain/version.rb
 - test/cassettes/autoscaling.yml
 - test/cassettes/cluster.yml