lbspec 0.2.0 → 0.2.1

data/README.md

@@ -48,6 +48,31 @@ You can use following chains with `#transfer`.
   - Options which can be used in http or https request commands.
   - You can use `options` if you configure request commands or capture commands.
 
+### #respond
+`#respond` tests if a virtual host on a load balancer respond as same as expected.
+
+#### chains
+You can use following chains with `#respond`.
+
+- tcp
+  - Tests with tcp packets for the virtual host.
+- udp
+  - Tests with udp packets for the virtual host.
+- http
+  - Tests with an http request for the virtual host.
+- https
+  - Tests with an https request for the virtual host.
+- from
+  - Specifies which host sends to the virtual host.
+- path
+  - Specifies a path for http or https requests.
+- with
+  - Specifies a string included in requests.
+- options
+  - Options which can be used in http or https request commands.
+  - You can use `options` if you configure request commands or capture commands.
+
+
 ## Requires
 * Users need to be able to login with ssh to the target nodes.
 * Users need to be able to `sudo` on the target nodes.
@@ -74,14 +99,17 @@ require_relative 'spec_helper'
 
 describe 'vhost_a' do
   it { should transfer('node_a') }
+  it { should respond('200 OK') }
 end
 
 describe 'vhost_b' do
   it { should transfer(['node_b','node_c']) }
+  it { should respond('200 OK') }
 end
 
 describe 'vhost_c:80' do
   it { should transfer(['node_b','node_c']).port(80) }
+  it { should respond('404') }
 end
 
 describe 'vhost_c:80' do
@@ -90,10 +118,12 @@ end
 
 describe 'vhost_c:80/test/' do
   it { should transfer('node_c').http }
+  it { should respond('200 OK').http }
 end
 
 describe 'vhost_c:443' do
   it { should transfer(['node_b','node_c']).port(80).https.path('/test/' }
+  it { should respond('200 OK').https.path('/test/') }
 end
 
 describe 'vhost_c:443/test/' do
@@ -101,6 +131,9 @@ describe 'vhost_c:443/test/' do
     should transfer(['node_b','node_c']).port(80).https
       .options(ignore_valid_ssl: true)
   end
+  it do should respond('200 OK').path('/test/').https
+      .options(ignore_valid_ssl: true)
+  end
 end
 
 describe 'vhost_c:80/test/' do

data/lib/lbspec/request.rb

@@ -15,7 +15,7 @@ module Lbspec
       @options = options[:options] ? options[:options] : {}
     end
 
-    def send(prove)
+    def send(prove = nil)
       if @application
         send_application(prove)
       else
@@ -54,11 +54,11 @@ module Lbspec
       opt << (options[:ignore_valid_ssl] ? ' -k' : '')
       opt << (options[:proxy] ? %Q( -x "#{options[:proxy]}") : '')
       if options[:noproxy]
-        env << %Q( no_proxy="#{options[:noproxy]}")
-        env << %Q( NO_PROXY="#{options[:noproxy]}")
+        env << %Q(no_proxy="#{options[:noproxy]}" )
+        env << %Q(NO_PROXY="#{options[:noproxy]}" )
       end
       opt << header_option(options[:header])
-      %Q(#{env} curl -o /dev/null -s #{opt} '#{uri}')
+      %Q(#{env}curl -i -s #{opt} '#{uri}')
     end
 
     def header_option(header)

data/lib/lbspec/respond.rb

@@ -0,0 +1,88 @@
+# -*- encoding: utf-8 -*-
+require 'net/ssh'
+require 'rspec/core'
+require 'rspec/expectations'
+require 'lbspec'
+
+RSpec::Matchers.define :respond do |expect|
+
+  @expect = expect
+  @protocol = :tcp
+  @application = nil
+  @path = nil
+  @string = nil
+  @chain_str = ''
+  @options = {}
+  @output_request = ''
+
+  match do |vhost|
+    request =
+      Lbspec::Request.new(vhost, @from,
+                          protocol: @protocol, application: @application,
+                          path: @path, options: @options)
+    @output_request = request.send(@string)
+    @output_request.include?(expect)
+  end
+
+  chain :tcp do
+    @protocol = :tcp
+    @application = nil
+    @chain_str << ' tcp'
+  end
+
+  chain :udp do
+    @protocol = :udp
+    @application = nil
+    @chain_str << ' udp'
+  end
+
+  chain :http do
+    @protocol = :tcp
+    @application = :http
+    @chain_str << ' http'
+  end
+
+  chain :https do
+    @protocol = :tcp
+    @application = :https
+    @chain_str << ' https'
+  end
+
+  chain :from do |from|
+    @from = from
+    @chain_str << " from #{from}"
+  end
+
+  chain :path do |path|
+    @path = path
+    @chain_str << " via #{path}"
+  end
+
+  chain :with do |string|
+    @string = string
+    @chain_str << " via #{string}"
+  end
+
+  chain :options do |options|
+    @options = options
+  end
+
+  description do
+    "respond #{@expect} #{@chain_str}"
+  end
+
+  failure_message_for_should do |vhost|
+    result =  "expected #{vhost} to respond #{@expect}"
+    result << result_string
+  end
+
+  failure_message_for_should_not do |vhost|
+    result =  "expected #{vhost} not to respond #{@expect}"
+    result << result_string
+  end
+
+  def result_string
+    result =  ", but did.\n" + "requested:\n"
+    result << request_str.gsub(/^/, '  ')
+  end
+end

data/lib/lbspec/util.rb

@@ -17,7 +17,7 @@ module Lbspec
     end
 
     def self.exec_command(command, node = nil)
-      output = command
+      output = command + "\n"
       if node
         Net::SSH.start(node, nil, config: true) do |ssh|
           output << ssh.exec!(command).to_s

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.0'
+  VERSION = '0.2.1'
 end

data/lib/lbspec.rb

@@ -1,6 +1,7 @@
 # -*- encoding: utf-8 -*-
 require 'lbspec/version'
 require 'lbspec/transfer'
+require 'lbspec/respond'
 require 'lbspec/request'
 require 'lbspec/capture'
 require 'lbspec/util'

data/spec/lbspec_respond_spec.rb

@@ -0,0 +1,61 @@
+# -*- encoding: utf-8 -*-
+require 'spec_helper'
+require 'net/ssh'
+
+describe '#respond' do
+
+  context 'http/https' do
+    it 'should test vhost_a responds with 200 OK' do
+      Lbspec::Util.stub(:exec_command).and_return '200 OK'
+      'vhost_a'.should respond('200 OK')
+    end
+    it 'should test vhost:80 responds with 200 OK by request with options' do
+      Lbspec::Util.stub(:exec_command).and_return '200 OK'
+      'vhost_a:443'.should respond('200 OK').http.path('/test')
+    end
+    it 'should test vhost:443 responds with 404 by request with options' do
+      Lbspec::Util.stub(:exec_command).and_return '404 Not Found'
+      'vhost_a:443'.should respond('404').https.path('/test')
+        .options(ignore_valid_ssl: false, timeout: 5)
+    end
+    it 'should test vhost:443 responds by requests from specified host' do
+      Lbspec::Util.stub(:exec_command).and_return '200 OK'
+      'vhost_a:443'.should respond('200 OK')
+        .https.path('/test').from('node_a')
+    end
+    it 'should test vhost:443 does not respond 404' do
+      Lbspec::Util.stub(:exec_command).and_return '200 OK'
+      'vhost_a:443'.should_not respond('404')
+        .https.path('/test').from('node_a')
+    end
+    it 'should test vhost:443/test does not respond 404' do
+      Lbspec::Util.stub(:exec_command).and_return '200 OK'
+      'vhost_a:443/test'.should_not respond('404')
+        .https.from('node_a')
+    end
+    it 'should test vhost:443/test does not respond 404' do
+      Lbspec::Util.stub(:exec_command).and_return '200 OK'
+      'vhost_a:443/test'.should_not respond('404')
+        .https.from('node_a')
+    end
+  end
+  context 'tcp/udp' do
+    it 'should test vhost:25/tcp respond 220' do
+      Lbspec::Util.stub(:exec_command).and_return '220'
+      'vhost_a:25'.should respond('220')
+        .tcp.with('HELO test.example.com')
+    end
+    it 'should test vhost:53/udp respond ' do
+      Lbspec::Util.stub(:exec_command).and_return '220'
+      'vhost_a:25'.should respond('220')
+        .udp.with('HELO test.example.com')
+    end
+  end
+  context 'description works with 200 OK' do
+    subject { 'vhost_a' }
+    it  do
+      Lbspec::Util.stub(:exec_command).and_return '200 OK'
+      should respond('200 OK')
+    end
+  end
+end

data/spec/lbspec_transfer_spec.rb

@@ -2,105 +2,103 @@
 require 'spec_helper'
 require 'net/ssh'
 
-describe Lbspec do
-  describe '#transfer' do
-    before(:each) do
-      key = Lbspec::Util.create_prove
-      include_str = 'X-Test: 1'
-      Lbspec::Util.stub(:create_prove).and_return(key)
-      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, key + include_str)
-        .and_yield(nil, key + 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(key + include_str) # `
-    end
+describe '#transfer' do
+  before(:each) do
+    key = Lbspec::Util.create_prove
+    include_str = 'X-Test: 1'
+    Lbspec::Util.stub(:create_prove).and_return(key)
+    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, key + include_str)
+      .and_yield(nil, key + 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(key + include_str) # `
+  end
 
-    it 'should test transfer a node' do
-      'vhost_a'.should transfer('node_a')
-    end
-    it 'should test transfer nodes' do
-      'vhost_a'.should transfer(%w(node_a node_b))
-    end
-    it 'should test transfer a node on port 80' do
-      'vhost_a'.should transfer('node_a').port(80)
-    end
-    it 'should test transfer vhost:80 and a node on port 80' do
-      'vhost_a:80'.should transfer('node_a').port(80)
-    end
-    it 'should test transfer vhost:80 and a node on port 80/tcp' do
-      'vhost_a:80'.should transfer('node_a').port(80).tcp
-    end
-    it 'should test transfer vhost:80 and a node on port 53/tcp' do
-      'vhost_a:80'.should transfer('node_a').port(53).udp
-    end
-    it 'should test transfer vhost:80 and a node with http' do
-      'vhost_a:80'.should transfer('node_a').http
-    end
-    it 'should test transfer vhost:443 and a node with https' do
-      'vhost_a:443'.should transfer('node_a').port(80).https
-    end
-    it 'should test transfer vhost:443 and a node:80 with https' do
-      'vhost_a:443'.should transfer('node_a').port(80).https
-    end
-    it 'should test transfer vhost:443 and a node with https /test' do
-      'vhost_a:443/test'.should transfer('node_a').https
-    end
-    it 'should test transfer vhost:443 with options for requests' do
-      'vhost_a:443'.should transfer('node_a').https.path('/test')
-        .options(ignore_valid_ssl: true)
-      'vhost_a:443/test'.should transfer('node_a').https
-        .options(ignore_valid_ssl: false, timeout: 5)
+  it 'should test transfer a node' do
+    'vhost_a'.should transfer('node_a')
+  end
+  it 'should test transfer nodes' do
+    'vhost_a'.should transfer(%w(node_a node_b))
+  end
+  it 'should test transfer a node on port 80' do
+    'vhost_a'.should transfer('node_a').port(80)
+  end
+  it 'should test transfer vhost:80 and a node on port 80' do
+    'vhost_a:80'.should transfer('node_a').port(80)
+  end
+  it 'should test transfer vhost:80 and a node on port 80/tcp' do
+    'vhost_a:80'.should transfer('node_a').port(80).tcp
+  end
+  it 'should test transfer vhost:80 and a node on port 53/tcp' do
+    'vhost_a:80'.should transfer('node_a').port(53).udp
+  end
+  it 'should test transfer vhost:80 and a node with http' do
+    'vhost_a:80'.should transfer('node_a').http
+  end
+  it 'should test transfer vhost:443 and a node with https' do
+    'vhost_a:443'.should transfer('node_a').port(80).https
+  end
+  it 'should test transfer vhost:443 and a node:80 with https' do
+    'vhost_a:443'.should transfer('node_a').port(80).https
+  end
+  it 'should test transfer vhost:443 and a node with https /test' do
+    'vhost_a:443/test'.should transfer('node_a').https
+  end
+  it 'should test transfer vhost:443 with options for requests' do
+    'vhost_a:443'.should transfer('node_a').https.path('/test')
+      .options(ignore_valid_ssl: true)
+    'vhost_a:443/test'.should transfer('node_a').https
+      .options(ignore_valid_ssl: false, timeout: 5)
+  end
+  it 'should test transfer vhost:443 requests from specified host' do
+    'vhost_a:443/test'.should transfer('node_a')
+      .https.from('node_a')
+  end
+  it 'should test transfer vhost:80 and a node with http includes ' do
+    'vhost_a:80'.should transfer('node_a').http
+      .include('X-Test: 1')
+    'vhost_a:80'.should transfer('node_a').http
+      .include(/Test:/)
+  end
+
+  describe 'request_command' do
+    it 'should create single header options for http' do
+      Lbspec::Util.should_receive(:exec_command)
+        .with(/ -H.*/, nil)
+      Lbspec::Util.should_not_receive(:exec_command)
+        .with(/ -H.* -H/, nil)
+      'vhost_a:443'.should transfer('node_a').http
+        .options(header: 'X-Test1:1')
     end
-    it 'should test transfer vhost:443 requests from specified host' do
-      'vhost_a:443/test'.should transfer('node_a')
-        .https.from('node_a')
+    it 'should create single header options for https' do
+      Lbspec::Util.should_receive(:exec_command)
+        .with(/ -H.*/, nil)
+      Lbspec::Util.should_not_receive(:exec_command)
+        .with(/ -H.* -H/, nil)
+      'vhost_a:443'.should transfer('node_a').https
+        .options(header: 'X-Test1:1')
     end
-    it 'should test transfer vhost:80 and a node with http includes ' do
-      'vhost_a:80'.should transfer('node_a').http
-        .include('X-Test: 1')
-      'vhost_a:80'.should transfer('node_a').http
-        .include(/Test:/)
+    it 'should create multi header options for http' do
+      Lbspec::Util.should_receive(:exec_command)
+        .with(/ -H.* -H/, nil)
+      'vhost_a:443'.should transfer('node_a').http
+        .options(header: %w(X-Test1:1 X-Test2:2))
     end
-
-    describe 'request_command' do
-      it 'should create single header options for http' do
-        Lbspec::Util.should_receive(:exec_command)
-          .with(/ -H.*/, nil)
-        Lbspec::Util.should_not_receive(:exec_command)
-          .with(/ -H.* -H/, nil)
-        'vhost_a:443'.should transfer('node_a').http
-          .options(header: 'X-Test1:1')
-      end
-      it 'should create single header options for https' do
-        Lbspec::Util.should_receive(:exec_command)
-          .with(/ -H.*/, nil)
-        Lbspec::Util.should_not_receive(:exec_command)
-          .with(/ -H.* -H/, nil)
-        'vhost_a:443'.should transfer('node_a').https
-          .options(header: 'X-Test1:1')
-      end
-      it 'should create multi header options for http' do
-        Lbspec::Util.should_receive(:exec_command)
-          .with(/ -H.* -H/, nil)
-        'vhost_a:443'.should transfer('node_a').http
-          .options(header: %w(X-Test1:1 X-Test2:2))
-      end
-      it 'should create multi header options for https' do
-        Lbspec::Util.should_receive(:exec_command)
-          .with(/ -H.* -H/, nil)
-        'vhost_a:443'.should transfer('node_a').https
-          .options(header: %w(X-Test1:1 X-Test2:2))
-      end
+    it 'should create multi header options for https' do
+      Lbspec::Util.should_receive(:exec_command)
+        .with(/ -H.* -H/, nil)
+      'vhost_a:443'.should transfer('node_a').https
+        .options(header: %w(X-Test1:1 X-Test2:2))
     end
-
   end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lbspec
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-03-30 00:00:00.000000000 Z
+date: 2014-04-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -162,12 +162,14 @@ files:
 - lib/lbspec.rb
 - lib/lbspec/capture.rb
 - lib/lbspec/request.rb
+- lib/lbspec/respond.rb
 - lib/lbspec/transfer.rb
 - lib/lbspec/util.rb
 - lib/lbspec/version.rb
 - rubocop-todo.yml
 - sample/lb_spec.rb
 - sample/spec_helper.rb
+- spec/lbspec_respond_spec.rb
 - spec/lbspec_spec.rb
 - spec/lbspec_transfer_spec.rb
 - spec/spec_helper.rb
@@ -186,7 +188,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -1301553932859158891
+      hash: -4421484859760496342
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -195,7 +197,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -1301553932859158891
+      hash: -4421484859760496342
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.21
@@ -203,6 +205,7 @@ signing_key:
 specification_version: 3
 summary: Easily test your Loadbalancers with RSpec.
 test_files:
+- spec/lbspec_respond_spec.rb
 - spec/lbspec_spec.rb
 - spec/lbspec_transfer_spec.rb
 - spec/spec_helper.rb