lbspec 0.0.6 → 0.1.0

data/README.md

@@ -19,6 +19,33 @@ Or install it yourself as:
 
     $ gem install lbspec
 
+## Functions
+You can use lbspec to test load balancers.
+
+### #transfer
+`#transfer` tests if a virtual host on a load balancer transfer requests to target nodes.
+
+#### chains
+You can use following chains with `#transfer`.
+
+- port
+ - Tests if a virtual host transfers requests to specified port on target nodes.
+- 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.
+- path
+  - Specifies a path for http or https requests.
+- from
+  - Specifies which host sends to the virtual host.
+- 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.
@@ -73,6 +100,10 @@ describe 'vhost_c:443' do
   end
 end
 
+describe 'vhost_c:80' do
+  it { should transfer('node_c').http.path('/test/').from('node_a') }
+end
+
 ```
 ## How it works
 ### #transfer
@@ -84,9 +115,10 @@ end
 
 ![#tranfer works][1]
 
+
 ## Configuration
 ### #transfer
-You can change how to capture probes and access to `vhost` with probes. You can replace default procedures to your procedures in spec_helpers or .spec files as follows.
+You can change how to capture probes and access to `vhost` with probes. You can replace default procedures to your procedures in spec_helpers or .spec files as follows. If you use `Lbspec::Util.exec_command()`, you can specify the node which generate request.
 ```ruby
 RSpec.configuration.lbspec_capture_command =
   lambda do |port, prove|
@@ -97,7 +129,7 @@ end
 RSpec.configuration.lbspec_https_request_command =
   lambda do |addr, port, path, prove|
   uri = 'https://' + "#{addr}:#{port}#{path}?#{prove}"
-  system("curl -o /dev/null -sk #{uri}")
+  Lbspec::Util.exec_command("curl -o /dev/null -sk #{uri}", @request_node)
 end
 ```
 You can also use the procedures with `options` with a chain `options`.
@@ -106,7 +138,7 @@ RSpec.configuration.lbspec_https_request_command =
   lambda do |addr, port, path, prove|
   opt =  @options[:timeout] ? " -m #{@options[:timeout]}" : ''
   opt << (@options[:ignore_valid_ssl] ? ' -k' : '')
-  system("curl -o /dev/null -s #{opt} #{uri}")
+  Lbspec::Util.exec_command("curl -o /dev/null -s #{opt} #{uri}", @request_node)
 end
 ```
 

data/lib/lbspec/transfer.rb

@@ -22,6 +22,7 @@ RSpec::Matchers.define :transfer do |nodes|
   @http_path = '/'
   @vhost_port = 80
   @node_port = 0
+  @request_node = nil
   @options = {}
 
   @capture_command = lambda do |port, prove|
@@ -30,21 +31,25 @@ RSpec::Matchers.define :transfer do |nodes|
   end
 
   @udp_request_command = lambda do |addr, port, prove|
-    system("echo #{prove} | nc -u #{addr} #{port}")
+    Lbspec::Util
+      .exec_command("echo #{prove} | nc -u #{addr} #{port}", @request_node)
   end
   @tcp_request_command = lambda do |addr, port, prove|
-    system("echo #{prove} | nc #{addr} #{port}")
+    Lbspec::Util
+      .exec_command("echo #{prove} | nc #{addr} #{port}", @request_node)
   end
   @http_request_command = lambda do |addr, port, path, prove|
     opt =  @options[:timeout] ? " -m #{@options[:timeout]}" : ''
     uri = 'http://' + "#{addr}:#{port}#{path}?#{prove}"
-    system("curl -o /dev/null -s #{opt} #{uri}")
+    Lbspec::Util
+      .exec_command("curl -o /dev/null -s #{opt} #{uri}", @request_node)
   end
   @https_request_command = lambda do |addr, port, path, prove|
     opt =  @options[:timeout] ? " -m #{@options[:timeout]}" : ''
     opt << (@options[:ignore_valid_ssl] ? ' -k' : '')
     uri = 'https://' + "#{addr}:#{port}#{path}?#{prove}"
-    system("curl -o /dev/null -sk #{opt} #{uri}")
+    Lbspec::Util
+      .exec_command("curl -o /dev/null -sk #{opt} #{uri}", @request_node)
   end
 
   @result = false
@@ -92,6 +97,11 @@ RSpec::Matchers.define :transfer do |nodes|
     @chain_str << " via #{path}"
   end
 
+  chain :from do |from|
+    @request_node = from
+    @chain_str << " from #{from}"
+  end
+
   chain :options do |options|
     @options = options
   end

data/lib/lbspec/util.rb

@@ -14,5 +14,14 @@ module Lbspec
       port = splits.last.to_i if /\d+/ =~ splits.last
       { addr: addr, port: port }
     end
+    def self.exec_command(command, node = nil)
+      if node
+        Net::SSH.start(node, nil, config: true) do |ssh|
+          ssh.exec!(command)
+        end
+      else
+        system(command)
+      end
+    end
   end
 end

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.0.6'
+  VERSION = '0.1.0'
 end

data/spec/lbspec_transfer_spec.rb

@@ -15,6 +15,7 @@ describe Lbspec do
       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
     end
@@ -49,19 +50,15 @@ describe Lbspec do
     it 'should test transfer vhost:443 and a node with https /test' do
       'vhost_a:443'.should transfer('node_a').https.path('/test')
     end
-    describe 'vhost_a:443' do
-      it { should transfer('node_a').https.path('/test') }
-      it { should transfer('node_a').port(80).tcp.https.path('/test') }
+    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'.should transfer('node_a').https.path('/test')
+        .options(ignore_valid_ssl: false, timeout: 5)
     end
-    describe 'vhost_a:443' do
-      it do
-        should transfer('node_a').https.path('/test')
-          .options(ignore_valid_ssl: true)
-      end
-      it do
-        should transfer('node_a').https.path('/test')
-          .options(ignore_valid_ssl: false, timeout: 5)
-      end
+    it 'should test transfer vhost:443 requests from specified host' do
+      'vhost_a:443'.should transfer('node_a')
+        .https.path('/test').from('node_a')
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lbspec
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -200,7 +200,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -3960315955505799530
+      hash: -1732435815710576022
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -209,7 +209,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -3960315955505799530
+      hash: -1732435815710576022
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.21