lbspec 0.0.1 → 0.0.2

data/.coveralls.yml

@@ -0,0 +1 @@
+service_name: travis-ci

data/.rubocop.yml

@@ -0,0 +1 @@
+inherit_from: rubocop-todo.yml

data/.travis.yml

@@ -1,3 +1,7 @@
 language: ruby
+
+script: "rake travis"
+
 rvm:
   - 1.9.3
+  - 2.0.0

data/README.md

@@ -2,6 +2,9 @@
 
 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)
 ## Installation
 
 Add this line to your application's Gemfile:
@@ -46,6 +49,11 @@ end
 describe 'vhost_b' do
   it { should transfer(['node_b','node_c']) }
 end
+
+describe 'vhost_c:80' do
+  it { should transfer(['node_b','node_c']).port(80) }
+end
+
 ```
 
 ## Contributing

data/Rakefile

@@ -1,6 +1,21 @@
-require "bundler/gem_tasks"
-require "rspec/core/rake_task"
+#!/usr/bin/env rake
+# -*- coding: utf-8 -*-
 
-RSpec::Core::RakeTask.new(:spec)
+require 'rake'
 
-task :default => :spec
+task :default => :travis
+task :travis => [:spec, :rubocop, 'coveralls:push']
+
+require 'rspec/core/rake_task'
+require 'rubocop/rake_task'
+require 'coveralls/rake/task'
+
+Coveralls::RakeTask.new
+
+RSpec::Core::RakeTask.new(:spec) do |t|
+  t.pattern = 'spec/**/*_spec.rb'
+end
+
+Rubocop::RakeTask.new(:rubocop) do |task|
+  task.patterns = %w(lib/**/*.rb spec/**/*.rb)
+end

data/lbspec.gemspec

@@ -21,6 +21,9 @@ Gem::Specification.new do |spec|
   spec.add_development_dependency "bundler", "~> 1.5"
   spec.add_development_dependency "rake"
   spec.add_development_dependency "rspec"
+  spec.add_development_dependency "rubocop"
+  spec.add_development_dependency "coveralls"
+  spec.add_development_dependency "debugger"
 
   spec.add_runtime_dependency "rspec"
   spec.add_runtime_dependency "net-ssh"

data/lib/lbspec/transfer.rb

@@ -1,74 +1,107 @@
+# -*- encoding: utf-8 -*-
 require 'net/ssh'
 require 'rspec/expectations'
+require 'lbspec'
 
 RSpec::Matchers.define :transfer do |nodes|
   @ssh = []
   @threads = []
   @nodes_connected = []
+  @vhost_port = 80
+  @node_port = 0
   @result = false
+  Thread.abort_on_exception = true
 
   match do |vhost|
-    gen_keyword
-    connect_nodes nodes
+    @keyword = gen_keyword
+    capture_on_nodes nodes
     wait_nodes_connected nodes
-    send_request(vhost, 80)
+    send_request vhost
     disconnect_nodes
-    is_ok?
+    @result
+  end
+
+  chain :port do |port|
+    @node_port = port
   end
 
   def gen_keyword
-    t = Time.now
-    @keyword = t.to_i.to_s + t.nsec.to_s
+    Lbspec::Util.gen_keyword
   end
 
   def wait_nodes_connected(nodes)
-    nodes_length = 1
-    nodes_length = nodes.length if nodes.respond_to? :each
+    nodes_length = (nodes.respond_to? :each) ? nodes.length : 1
+    sleep 0.5 until @nodes_connected.length == nodes_length
+  end
 
-    until (@nodes_connected.length == nodes_length) do
-      sleep 0.5
+  def capture_on_nodes(nodes)
+    if nodes.respond_to? :each
+      nodes.each { |node| capture_on_node(node) }
+    else
+      capture_on_node(nodes)
     end
   end
 
-  def connect_nodes(nodes)
-    if nodes.respond_to? :each
-      nodes.each do |node|
-        @threads << Thread.new { connect_node(node) }
+  def capture_on_node(node)
+    @threads << Thread.new do
+      Net::SSH.start(node, nil, :config => true) do |ssh|
+        @ssh << ssh
+        ssh.open_channel { |channel| run_check channel }
       end
-    else
-      @threads << Thread.new { connect_node(nodes) }
     end
   end
 
-  def connect_node(node)
-    @ssh.push = Net::SSH.start(node, nil, :config => true) do |ssh|
-      ssh.open_channel do |ch|
-        ch.request_pty do |ch, success|
-          raise "Could not obtain pty " if !success
-          @nodes_connected.push(true)
-          ch.exec "sudo ngrep #{@keyword}" do |ch, stream, data|
-            ch.on_data do |c, data|
-              @result = true if /#{@keyword}/ === data
-            end
-          end
-        end
+  def run_check(channel)
+    channel.request_pty do |chan, success|
+      fail 'Could not obtain pty' unless success
+      @nodes_connected.push(true)
+      exec_capture(chan, capture_cmd)
+    end
+  end
+
+  def exec_capture(channel, command)
+    channel.exec capture_cmd do |ch, stream, data|
+      num_match = 0
+      ch.on_data do |c, d|
+        # because of ngrep output condition first
+        # to avoid incorrect match, count 2
+        num_match += 1 if /#{@keyword}/ =~ d
+        @result = true if num_match > 1
       end
     end
   end
 
+  def capture_cmd
+    port_str = @node_port > 0 ? "port #{@node_port}" : ''
+    "sudo ngrep #{@keyword} #{port_str}"
+  end
+
   def disconnect_nodes
+    sleep 1
     @threads.each do |t|
       t.kill
     end
     @ssh.each do |ssh|
-      ssh.close if ! ssh.closed?
+      ssh.close unless ssh.closed?
     end
   end
 
-  def send_request(vhost, port)
-    system("echo #{@keyword} | nc #{vhost} #{port}")
+  def send_request(vhost)
+    addr_port = Lbspec::Util.split_addr_port(vhost.to_s)
+    vhost_addr, vhost_port = addr_port[:addr], addr_port[:port]
+    @vhost_port = vhost_port if vhost_port > 0
+    system("echo #{@keyword} | nc #{vhost_addr} #{@vhost_port}")
   end
-  def is_ok?
-    @result
+
+  description do
+    "transfer requests to #{nodes}."
+  end
+
+  failure_message_for_should do |vhost|
+    "expected #{vhost} to transfer requests to #{nodes}, but did not."
+  end
+
+  failure_message_for_should_not do |vhost|
+    "expected #{vhost} not to transfer requests to #{nodes}, but it did."
   end
 end

data/lib/lbspec/util.rb

@@ -0,0 +1,18 @@
+# -*- encoding: utf-8 -*-
+# Lbspec is an RSpec plugin for easy Loadbalancer testing.
+module Lbspec
+  # Lbspec::Util provides some utilities
+  class Util
+    def self.gen_keyword
+      t = Time.now
+      t.to_i.to_s + t.nsec.to_s
+    end
+    def self.split_addr_port(addr_port_str)
+      port = 0
+      splits = addr_port_str.split(':')
+      addr = splits.first
+      port = splits.last.to_i if /\d+/ =~ splits.last
+      { :addr => addr, :port => port }
+    end
+  end
+end

data/lib/lbspec/version.rb

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

data/lib/lbspec.rb

@@ -1,5 +1,8 @@
+# -*- encoding: utf-8 -*-
 require 'lbspec/version'
 require 'lbspec/transfer'
+require 'lbspec/util'
 
+# Lbspec is an RSpec plugin for easy Loadbalancer testing.
 module Lbspec
 end

data/rubocop-todo.yml

@@ -0,0 +1,2 @@
+HashSyntax:
+  EnforcedStyle: hash_rockets

data/sample/lb_spec.rb

@@ -3,4 +3,10 @@ require_relative 'spec_helper'
 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')
+  end
+  it 'should test transfer a node on port 8080' do
+    'vhost_a'.should_not transfer('node_a:8080')
+  end
 end

data/spec/lbspec_spec.rb

@@ -1,11 +1,8 @@
+# -*- encoding: utf-8 -*-
 require 'spec_helper'
 
 describe Lbspec do
   it 'should have a version number' do
     Lbspec::VERSION.should_not be_nil
   end
-
-  it 'should do something useful' do
-    false.should eq(true)
-  end
 end

data/spec/lbspec_transfer_spec.rb

@@ -1,12 +1,35 @@
+# -*- encoding: utf-8 -*-
 require 'spec_helper'
+require 'net/ssh'
 
 describe Lbspec do
   describe '#transfer' do
+    before(:each) do
+      key = Lbspec::Util.gen_keyword
+      Lbspec::Util.stub(:gen_keyword).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).and_yield(nil, key)
+      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)
+      Net::SSH.stub(:start).and_yield(ssh_connected).and_return(ssh_connected)
+      Kernel.stub(:system).and_return true
+    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(['node_a','node_b'])
+      '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
   end
 end

data/spec/spec_helper.rb

@@ -1,2 +1,6 @@
+# -*- encoding: utf-8 -*-
+require 'coveralls'
+Coveralls.wear!
+
 $LOAD_PATH.unshift File.expand_path('../../lib', __FILE__)
 require 'lbspec'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: lbspec
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-03-07 00:00:00.000000000 Z
+date: 2014-03-08 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -59,6 +59,54 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: coveralls
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: debugger
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
@@ -98,8 +146,10 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- .coveralls.yml
 - .gitignore
 - .rspec
+- .rubocop.yml
 - .travis.yml
 - Gemfile
 - LICENSE
@@ -109,7 +159,9 @@ files:
 - lbspec.gemspec
 - lib/lbspec.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_spec.rb