lbspec 0.2.8 → 0.2.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: bc6f025cf48f9f28b899219aee4782724b75affe
-  data.tar.gz: bfaaad1d0182f19d0210f329d0a981a19151dad9
+  metadata.gz: 662a6f0ca37d759561d0596c7cc3a48626feb60e
+  data.tar.gz: 78286a3b3aa79cc0a819e15ba4e0b69e410b2e36
 SHA512:
-  metadata.gz: be891f0b85609b354e73c05fdc87bad1b03fa5e099744a25815d9eb2399e7eef34e083227fd00aa09dcc1178a91e0625475ff4fdfd1c3c5d1b5b61ba59b827bb
-  data.tar.gz: bef0f7f51636c4978f297957ce1853462c81e8dc33fa80db2be973ba89c4ca8515b12ab5b8c21a8869571f55bb724db5f2bfd3508521625d4a33f48de8077f87
+  metadata.gz: e1c1fac74410b25045a7c465c33edc1a9549e95ed3fa33dba38388f9439965694cf56e15dd57f0648d1eb8135e2cf6d6b3ae2e3a0b9b07a65525ab5a581b924d
+  data.tar.gz: 87da405d23608e193cdc930e269ff4db39e45f99d71fc3d55206cda1f2e85e4ef94096aea2cda68bca7a492209083b7ba2d86f14431070bb0d9be4f0fe6ee6e0

data/README.md

@@ -88,6 +88,27 @@ Lbspec::Util.logger = Logger.new('test.log')
 ```
 
 The default Logger show Logger::ERROR level on STDOUT.
+=======
+### #haelthcheck
+`#healthcheck` tests if a load balancer healthchecks to target nodes.
+
+#### chains
+You can use following chains with `#healthcheck`.
+
+- include
+  - Tests if healthchecks include string.
+- from
+  - Tests if healthchecks are from specific address.
+- interval
+  - Tests if healthchecks are at intervals of specific time.
+- port
+  - Tests if healthchecks are on specified port on target nodes.
+- icmp
+  - Tests if healthchecks are icmp.
+- tcp
+  - Tests if healthchecks are tcp.
+- udp
+  - Tests if healthchecks are udp.
 
 ## Requires
 * Users need to be able to login with ssh to the target nodes.
@@ -156,6 +177,14 @@ describe 'vhost_c:80/test/' do
   it { should transfer('node_c').http.from('node_a') }
 end
 
+describe 'loadbalancer' do
+  it { should healthcheck('node_c').include('/test/healthcheck').from('192.168.1.1') }
+end
+
+describe 'loadbalancer' do
+  it { should healthcheck('node_c').include('/test/healthcheck').interval(5) }
+end
+
 ```
 ## How it works
 ### #transfer

data/lib/lbspec/capture.rb

@@ -9,27 +9,24 @@ module Lbspec
 
     attr_reader :result, :output
 
-    def initialize(nodes, port, prove, include_str = nil)
+    def initialize(nodes, bpf = nil, prove = nil, include_str = nil,
+                   term_sec = nil)
       @nodes = nodes.respond_to?(:each) ? nodes : [nodes]
-      @port = port ? port : 0
-      @prove = prove
+      @bpf = bpf ? bpf : ''
+      @prove = prove ? prove : ''
       @include_str = include_str
-      @threads = []
-      @ssh = []
-      @nodes_connected = []
-      @result = false
-      @output = []
+      @term_sec = term_sec ? term_sec : -1
+      set_initial_value
       Util.log.debug("#{self.class} initialized #{inspect}")
     end
 
     def open
-      @nodes.each do |node|
-        open_node(node)
-      end
+      @nodes.each { |node| open_node(node) }
       wait_connected
     end
 
     def close
+      sleep 0.5 until capture_done?
       @threads.each do |t|
         t.kill
       end
@@ -38,8 +35,28 @@ module Lbspec
       end
     end
 
+    def self.bpf(options = {})
+      is_first = true
+      filter = ''
+
+      options.each do |k, v|
+        if k && v
+          filter << ' and ' unless is_first
+          filter << k.to_s.gsub('_', ' ') + ' ' + v.to_s
+          is_first = false
+        end
+      end
+      filter
+    end
+
     private
 
+    def set_initial_value
+      @threads, @ssh, @nodes_connected = [], [], []
+      @result = false
+      @output = []
+    end
+
     def open_node(node)
       @threads << Thread.new do
         Net::SSH.start(node, nil, config: true) do |ssh|
@@ -73,18 +90,25 @@ module Lbspec
 
     def exec_capture_command(channel, command)
       whole_data = ''
-      channel.exec command do |ch, stream, data|
-        ch.on_data do |c, d|
+      @start_sec = Time.now.to_i + 1
+      channel.exec command do |ch, _stream , _data|
+        ch.on_data do |_c, d|
           whole_data << d
-          patterns = [@prove]
-          patterns << @include_str if @include_str
-          @result = match_all?(whole_data, patterns)
+          @result = match_all?(whole_data)
         end
+        break if capture_done?
       end
       whole_data
     end
 
-    def match_all?(string, patterns)
+    def capture_done?
+      now_sec = Time.now.to_i
+      (@term_sec > 0 && now_sec - @start_sec > @term_sec) ? true : @result
+    end
+
+    def match_all?(string)
+      patterns = [@prove]
+      patterns << @include_str if @include_str
       num_patterns, num_match = 0, 0
       patterns.each do |pat|
         num_patterns += 1
@@ -94,8 +118,7 @@ module Lbspec
     end
 
     def capture_command
-      port_str = @port > 0 ? "port #{@port}" : ''
-      "sudo ngrep -W byline #{@prove} #{port_str} | grep -v \"match:\""
+      "sudo ngrep -W byline #{@prove} #{@bpf} | grep -v \"match:\""
     end
   end
 end

data/lib/lbspec/healthcheck.rb

@@ -0,0 +1,90 @@
+# -*- encoding: utf-8 -*-
+require 'lbspec'
+
+RSpec::Matchers.define :healthcheck do |nodes|
+  match do |lb|
+    @lb = lb
+    @nodes = nodes
+    bpf = Lbspec::Capture.bpf(port:     @port,
+                              src_host: @from,
+                              protocol: @protocol)
+    capture =
+      Lbspec::Capture.new(nodes, bpf, nil, @include_str, @interval)
+    capture.open
+    @output_capture = capture.output
+    capture.close
+    capture.result
+  end
+
+  chain :include do |str|
+    @include_str = str
+    @chain_str = Lbspec::Util.add_string(@chain_str, " including #{str}")
+  end
+
+  chain :from do |from|
+    @from = from
+    @chain_str = Lbspec::Util.add_string(@chain_str, " from #{from}")
+  end
+
+  chain :interval do |second|
+    @interval = second
+    @chain_str = Lbspec::Util.add_string(@chain_str,
+                                         " at interval of #{second} sec")
+  end
+
+  chain :port do |port|
+    @port = port
+    @chain_str = Lbspec::Util.add_string(@chain_str, " port #{port}")
+  end
+
+  chain :icmp do
+    @protocol = :icmp
+    @chain_str = Lbspec::Util.add_string(@chain_str, ' icmp')
+  end
+
+  chain :tcp do
+    @protocol = :tcp
+    @chain_str = Lbspec::Util.add_string(@chain_str, ' tcp')
+  end
+
+  chain :udp do
+    @protocol = :udp
+    @chain_str = Lbspec::Util.add_string(@chain_str, ' udp')
+  end
+
+  chain :options do |options|
+    @options = options
+  end
+
+  description do
+    "healthcheck #{@nodes}#{@chain_str}."
+  end
+
+  failure_message_for_should do
+    result_string
+  end
+
+  failure_message_for_should_not do
+    negative = true
+    result_string(negative)
+  end
+
+  def result_string(negative = false)
+    negation = negative ? '' : ' not'
+    result = "expected #{@lb} to healthcheck to"
+    result << @nodes.to_s + @chain_str
+    result << ", but did#{negation}.\n"
+    result << "\ncaptured:\n"
+    result << result_capture
+    result
+  end
+
+  def result_capture
+    result = ''
+    @output_capture.each do |o|
+      result << o[:node].gsub(/^/, '  ') + "\n"
+      result << o[:output].gsub(/^/, '    ') + "\n"
+    end
+    result
+  end
+end

data/lib/lbspec/transfer.rb

@@ -10,8 +10,9 @@ RSpec::Matchers.define :transfer do |nodes|
     log.debug("#transfer(#{nodes.inspect}) is called")
     @vhost = vhost
     prove = Lbspec::Util.create_prove
+    bpf = Lbspec::Capture.bpf(port: @port)
     capture =
-      Lbspec::Capture.new(nodes, @port, prove, @include_str)
+      Lbspec::Capture.new(nodes, bpf, prove, @include_str)
     capture.open
     request =
       Lbspec::Request.new(vhost, @from,

data/lib/lbspec/version.rb

@@ -1,5 +1,5 @@
 # -*- encoding: utf-8 -*-
 # Lbspec is an RSpec plugin for easy Loadbalancer testing.
 module Lbspec
-  VERSION = '0.2.8'
+  VERSION = '0.2.9'
 end

data/lib/lbspec.rb

@@ -2,6 +2,7 @@
 require 'lbspec/version'
 require 'lbspec/transfer'
 require 'lbspec/respond'
+require 'lbspec/healthcheck'
 require 'lbspec/request'
 require 'lbspec/capture'
 require 'lbspec/util'

data/spec/lbspec_healthcheck_spec.rb

@@ -0,0 +1,48 @@
+# -*- encoding: utf-8 -*-
+require 'spec_helper'
+require 'net/ssh'
+
+describe '#healthcheck' do
+  before(:each) do
+    include_str = 'X-Test: 1'
+    channel_connected = double('channel_connected')
+    channel_connected.stub(:request_pty).and_yield(channel_connected, true)
+    channel_connected.stub(:exec).and_yield(channel_connected, nil, nil)
+    channel_connected.stub(:on_data)
+      .and_yield(nil, include_str)
+      .and_yield(nil, include_str)
+    ssh_connected = double('ssh_connected')
+    ssh_connected.stub(:open_channel).and_yield(channel_connected)
+    ssh_connected.stub(:closed?).and_return(false)
+    ssh_connected.stub(:close)
+    ssh_connected.stub(:exec!).and_return(true)
+    Net::SSH.stub(:start).and_yield(ssh_connected).and_return(ssh_connected)
+    Kernel.stub(:system).and_return true
+    Kernel.stub(:`).and_return(include_str) # `
+  end
+
+  it 'should test loadbalancer healthchecks' do
+    'loadbalancer'.should healthcheck('node_a')
+  end
+  it 'should test loadbalancer healthchecks include string' do
+    'loadbalancer'.should healthcheck('node_a').include('X-Test')
+  end
+  it 'should test loadbalancer healthchecks from specified host' do
+    'loadbalancer'.should healthcheck('node_a').from('X-Test')
+  end
+  it 'should test loadbalancer healthchecks at specified interval' do
+    'loadbalancer'.should healthcheck('node_a').interval(5)
+  end
+  it 'should test loadbalancer healthchecks specified port' do
+    'loadbalancer'.should healthcheck('node_a').port(80)
+  end
+  it 'should test loadbalancer healthchecks icmp' do
+    'loadbalancer'.should healthcheck('node_a').icmp
+  end
+  it 'should test loadbalancer healthchecks tcp' do
+    'loadbalancer'.should healthcheck('node_a').tcp
+  end
+  it 'should test loadbalancer healthchecks udp' do
+    'loadbalancer'.should healthcheck('node_a').udp
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lbspec
 version: !ruby/object:Gem::Version
-  version: 0.2.8
+  version: 0.2.9
 platform: ruby
 authors:
 - OTA Hiroshi
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-04-24 00:00:00.000000000 Z
+date: 2014-05-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -144,6 +144,7 @@ files:
 - lbspec.gemspec
 - lib/lbspec.rb
 - lib/lbspec/capture.rb
+- lib/lbspec/healthcheck.rb
 - lib/lbspec/request.rb
 - lib/lbspec/respond.rb
 - lib/lbspec/transfer.rb
@@ -152,6 +153,7 @@ files:
 - rubocop-todo.yml
 - sample/lb_spec.rb
 - sample/spec_helper.rb
+- spec/lbspec_healthcheck_spec.rb
 - spec/lbspec_respond_spec.rb
 - spec/lbspec_spec.rb
 - spec/lbspec_transfer_spec.rb
@@ -177,11 +179,12 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.2.2
+rubygems_version: 2.0.14
 signing_key: 
 specification_version: 4
 summary: Easily test your Loadbalancers with RSpec.
 test_files:
+- spec/lbspec_healthcheck_spec.rb
 - spec/lbspec_respond_spec.rb
 - spec/lbspec_spec.rb
 - spec/lbspec_transfer_spec.rb