lbspec 0.2.11 → 0.2.12

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 48b23c45d1f0d0399f34543470a376b5fb6b85bd
-  data.tar.gz: e24a0f6d1576dcdd37678c55184b7a9c09838d6e
+  metadata.gz: 43048b897beb3ba1a36100e000c80c834e583e6b
+  data.tar.gz: 2dfab8428ce339d379cf35e0b5bcacb6b30b6acf
 SHA512:
-  metadata.gz: 52b6e18a76563962c7d2a11bf0de40aea25901a707a2bca1bf86731c0693e79ea31ddfc46ca4a92a32239abb145d9bc38dc9170dc38e9eedcbdb9c58abbcd734
-  data.tar.gz: 2040d851d1c251d52964ee3f1a819d3df5146f987018bd312286ed3f9a9f16414e39219b1e65d7b83cc39d4f2385c367399057a7448abbb9cc1e6a6bc81356d0
+  metadata.gz: f14e286e526b7a04d013d40dc879c18290e6faacb37d0b96865d72defdf40b09c05ca391861426e0070f801f2bcc1ffeee58cd6f0590a9446379ee885f6dac7e
+  data.tar.gz: b8ccc5ae96580c571872900149ec3885a505e0f13d95cfed0241ff0d51240c2f79d6d7befdfdd87aa0e8dc35b78011ce2dcbfb28d7b3c3793709d0e9ee2040a4

data/README.md

@@ -5,6 +5,102 @@ Lbspec is an RSpec plugin for easy Loadbalancer testing.
 [![Build Status](https://travis-ci.org/otahi/lbspec.png?branch=master)](https://travis-ci.org/otahi/lbspec)
 [![Coverage Status](https://coveralls.io/repos/otahi/lbspec/badge.png?branch=master)](https://coveralls.io/r/otahi/lbspec?branch=master)
 [![Gem Version](https://badge.fury.io/rb/lbspec.png)](http://badge.fury.io/rb/lbspec)
+
+## Usage
+
+Lbspec is best described by example. First, require `lbspec` in your `spec_helper.rb`:
+
+```ruby
+# spec/spec_helper.rb
+require 'lbspec'
+```
+
+Then, create a spec like this:
+
+```ruby
+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(%w(node_b node_c)) }
+  it { should respond('200 OK') }
+end
+
+describe 'vhost_c:80' do
+  it { should transfer(%w(node_b node_c)).port(80) }
+  it { should respond('404') }
+end
+
+describe 'vhost_c:80' do
+  it { should transfer(%w(node_b node_c)).port(53).udp }
+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(%w(node_b node_c)).port(80).https.path('/test/') }
+  it { should respond('200 OK').https.path('/test/') }
+end
+
+describe 'vhost_c:443/test/' do
+  it do
+    should transfer(%w(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
+  it { should transfer('node_c').http.from('node_a') }
+end
+
+describe 'loadbalancer' do
+  it do should healthcheck('node_c')
+      .include('/test/healthcheck').from('192.168.1.1')
+  end
+end
+
+describe 'loadbalancer' do
+  it { should healthcheck('node_c').include('/test/healthcheck').interval(5) }
+end
+```
+
+You will get a result:
+```
+$ bundle exec rspec spec.rb
+
+vhost_a
+  should transfer requests to node_a.
+  should respond 200 OK.
+vhost_b
+  should transfer requests to ["node_b", "node_c"].
+  should respond 200 OK.
+vhost_c:80
+  should transfer requests to ["node_b", "node_c"] port 80.
+  should respond 404.
+vhost_c:80
+  should transfer requests to ["node_b", "node_c"] port 53 udp.
+vhost_c:80/test/
+  should transfer requests to node_c http.
+  should respond 200 OK http.
+vhost_c:443
+  should transfer requests to ["node_b", "node_c"] port 80 https via /test/.
+  should respond 200 OK https via /test/.
+vhost_c:443/test/
+  should transfer requests to ["node_b", "node_c"] port 80 https.
+  should respond 200 OK
+```
+
+
 ## Installation
 
 Add this line to your application's Gemfile:
@@ -119,73 +215,6 @@ You can use following chains with `#healthcheck`.
 ## Limitations
 * Lbspec uses only ssh configuration in ~/.ssh/config
 
-## Usage
-
-Lbspec is best described by example. First, require `lbspec` in your `spec_helper.rb`:
-
-```ruby
-# spec/spec_helper.rb
-require 'rspec'
-require 'lbspec'
-```
-
-Then, create a spec like this:
-
-```ruby
-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
-  it { should transfer(['node_b','node_c']).port(53).udp }
-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
-  it 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
-  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

@@ -51,7 +51,7 @@ module Lbspec
     private
 
     def set_initial_value
-      @threads, @ssh, @nodes_connected = [], [], []
+      @threads, @ssh, @nodes_connected = [], [], {}
       @result = false
       @output = []
     end
@@ -82,21 +82,22 @@ module Lbspec
     end
 
     def exec_capture(channel)
-      exec_capture_command(channel, capture_command)
-      @nodes_connected.push(true)
-    end
-
-    def exec_capture_command(channel, command)
       whole_data = ''
       @start_sec = Time.now.to_i + 1
-      channel.exec command do |ch, _stream, _data|
-        ch.on_data do |_c, d|
-          whole_data << d
-          @result = match_all?(whole_data)
-        end
+      channel.exec(capture_command) do |ch, _stream, _data|
+        receive_data(ch, whole_data)
         break if capture_done?
       end
-      whole_data
+      capture_command + "\n" + whole_data
+    end
+
+    def receive_data(channel, data)
+      channel.on_data do |_c, d|
+        @nodes_connected.merge!(Thread.current.to_s => true)
+        data << d
+        @result = match_all?(data)
+      end
+      data
     end
 
     def capture_done?

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.11'
+  VERSION = '0.2.12'
 end

data/sample/lb_spec.rb

@@ -1,12 +1,81 @@
 require_relative 'spec_helper'
+require 'net/ssh'
 
-describe 'vhost_a' do
-  it { should transfer('node_a') }
-  it { should transfer(['node_c','node_b']) }
-  it 'should test transfer a node on port 80' do
-    'vhost_a'.should transfer('node_a:80')
+describe 'sample spec' do
+  before(:each) do
+    key = Lbspec::Util.create_prove
+    include_str = 'X-Test: 1  /test/healthcheck'
+    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)
+    Lbspec::Util.stub(:`).and_return(key + include_str) # `
+    Lbspec::Util.stub(:exec_command).and_return '200 OK 404'
+    Kernel.stub(:system).and_return true
+    Kernel.stub(:`).and_return(include_str) # `
   end
-  it 'should test transfer a node on port 8080' do
-    'vhost_a'.should_not transfer('node_a:8080')
+
+  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(%w(node_b node_c)) }
+    it { should respond('200 OK') }
+  end
+
+  describe 'vhost_c:80' do
+    it { should transfer(%w(node_b node_c)).port(80) }
+    it { should respond('404') }
+  end
+
+  describe 'vhost_c:80' do
+    it { should transfer(%w(node_b node_c)).port(53).udp }
+  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(%w(node_b node_c)).port(80).https.path('/test/') }
+    it { should respond('200 OK').https.path('/test/') }
+  end
+
+  describe 'vhost_c:443/test/' do
+    it do
+      should transfer(%w(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
+    it { should transfer('node_c').http.from('node_a') }
+  end
+
+  describe 'loadbalancer' do
+    it do should healthcheck('node_c')
+        .include('/test/healthcheck').from('192.168.1.1')
+    end
+  end
+
+  describe 'loadbalancer' do
+    it { should healthcheck('node_c').include('/test/healthcheck').interval(5) }
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: lbspec
 version: !ruby/object:Gem::Version
-  version: 0.2.11
+  version: 0.2.12
 platform: ruby
 authors:
 - OTA Hiroshi
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-08-23 00:00:00.000000000 Z
+date: 2014-08-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler