tlb-testunit 0.3.0 → 0.3.1

data/.emacs_project.template

@@ -0,0 +1,21 @@
+;; copy me as .emacs_project in project root if using .emacs.d = git://github.com/janmejay/emacs.git
+;; after copying it across, modify the file to make it relevant to local rvm setup
+(setq rspec-executable "rspec")
+(setq rb_platform "jruby")
+(setq rb_platform "mri")
+
+(unless (getenv "ORIG_PATH")
+  (setenv "ORIG_PATH" (getenv "PATH")))
+
+(defun add-gem-bin (gem-bin)
+  (setenv "PATH" (concat gem-bin ":" (getenv "ORIG_PATH"))))
+
+(if (equal rb_platform "jruby")
+    (progn 
+      (setenv "GEM_HOME" "/home/janmejay/.rvm/gems/jruby-1.5.6@tlb")
+      (setenv "GEM_PATH" "/home/janmejay/.rvm/gems/jruby-1.5.6@tlb:/home/janmejay/.rvm/gems/jruby-1.5.6@global")
+      (add-gem-bin "/home/janmejay/.rvm/gems/jruby-1.5.6@tlb/bin"))
+    (progn
+      (setenv "GEM_HOME" "/home/janmejay/.rvm/gems/ruby-1.8.7-p334@tlb")
+      (setenv "GEM_PATH" "/home/janmejay/.rvm/gems/ruby-1.8.7-p334@tlb:/home/janmejay/.rvm/gems/ruby-1.8.7-p334@global")
+      (add-gem-bin "/home/janmejay/.rvm/gems/ruby-1.8.7-p334@tlb/bin")))

data/.gitignore

@@ -4,3 +4,5 @@ tlb_store
 tlb-*.jar
 .yardoc
 doc
+*.gem
+.emacs_project

data/Rakefile

@@ -23,4 +23,5 @@ end
 task :package do
   `gem build tlb-rspec2.gemspec`
   `gem build tlb-testunit.gemspec`
+  `gem build tlb-cucumber.gemspec`
 end

data/lib/tlb.rb

@@ -23,7 +23,7 @@ module Tlb
     end
 
     def self.port
-      ENV[TLB_BALANCER_PORT]
+      ENV[TLB_BALANCER_PORT] || '8019'
     end
 
     def self.send path, data
@@ -44,10 +44,13 @@ module Tlb
       false
     end
 
+    def self.terminate
+      get("/control/suicide")
+    end
+
     def self.wait_for_start
       loop do
         begin
-          TCPSocket.new(host, port)
           break if running?
         rescue
           #ignore
@@ -67,6 +70,10 @@ module Tlb
     rel_file_name = abs_file_name.sub(/^#{Dir.pwd}/, '.')
   end
 
+  def self.relative_file_paths file_names
+    file_names.map { |file_name| relative_file_path(file_name) }
+  end
+
   def self.balance_and_order file_set
     ensure_server_running
     Balancer.send(Balancer::BALANCE_PATH, file_set.join("\n")).split("\n")
@@ -106,37 +113,126 @@ module Tlb
     "java -jar #{tlb_jar}"
   end
 
-  def self.write_to_file file_var, clob
-    File.open(ENV[file_var], 'a') do |h|
-      h.write(clob)
+  def self.can_fork?
+    RUBY_PLATFORM != 'java'
+  end
+
+  class BalancerProcess
+    class StreamPumper
+      def initialize stream, file
+        @stream, @file = stream, file
+        @thd = Thread.new { pump }
+      end
+
+      def pump
+        loop do
+          data_available? && flush_stream
+          Thread.current[:stop_pumping] && break
+          sleep 0.1
+        end
+      end
+
+      def flush_stream
+        File.open(ENV[@file], 'a') do |h|
+          h.write(read)
+        end
+      end
+
+      def stop_pumping!
+        @thd[:stop_pumping] = true
+        @thd.join
+      end
+    end
+
+    def initialize server_command
+      pumper_type, out, err  = start(server_command)
+      @out_pumper = pumper_type.new(out, TLB_OUT_FILE)
+      @err_pumper = pumper_type.new(err, TLB_ERR_FILE)
+    end
+
+    def stop_pumping
+      @out_pumper.stop_pumping!
+      @err_pumper.stop_pumping!
+    end
+
+    def die
+      Balancer.terminate
+      stop_pumping
     end
   end
 
-  def self.start_server
-    ENV[TLB_APP] = 'tlb.balancer.BalancerInitializer'
-    @pid, input, out, err = Open4.popen4(server_command)
-    @out_pumper = stream_pumper_for(out, TLB_OUT_FILE)
-    @err_pumper = stream_pumper_for(err, TLB_ERR_FILE)
-    Balancer.wait_for_start
+  class ForkBalancerProcess < BalancerProcess
+    def start server_command
+      @pid, input, out, err = Open4.popen4(server_command)
+      unless (out)
+        raise "out was nil"
+      end
+      return Class.new(StreamPumper) do
+        def data_available?
+          not @stream.eof?
+        end
+
+        def read
+          @stream.read
+        end
+      end, out, err
+    end
+
+    def die
+      super
+      @pid = nil
+      Process.wait
+    end
   end
 
-  def self.stream_pumper_for stream, dump_file
-    Thread.new do
-      loop do
-        stream.eof? || write_to_file(dump_file, stream.read)
-        Thread.current[:stop_pumping] && break
-        sleep 1
+  class JavaBalancerProcess < BalancerProcess
+    def start server_command
+      require 'java'
+      pb = java.lang.ProcessBuilder.new(server_command.split)
+      ENV.each do |key, val|
+        pb.environment[key] = val
       end
+      @process = pb.start()
+      return Class.new(StreamPumper) do
+        def data_available?
+          @stream.ready
+        end
+
+        def read
+          @stream.read_line
+        end
+
+        def stop_pumping!
+          super
+          @stream.close
+        end
+      end, buf_reader(@process.input_stream), buf_reader(@process.error_stream)
+    end
+
+    def buf_reader stream
+      java.io.BufferedReader.new(java.io.InputStreamReader.new(stream))
     end
+
+    def die
+      super
+      @process.destroy
+      @process.waitFor
+      @process = nil
+    end
+  end
+
+  def self.balancer_process_type
+    can_fork? ? ForkBalancerProcess : JavaBalancerProcess
+  end
+
+  def self.start_server
+    ENV[TLB_APP] = 'tlb.balancer.BalancerInitializer'
+    bal_klass = balancer_process_type
+    @balancer_process = bal_klass.new(server_command)
+    Balancer.wait_for_start
   end
 
   def self.stop_server
-    Process.kill(Signal.list["TERM"], @pid)
-    @pid = nil
-    @out_pumper[:stop_pumping] = true
-    @err_pumper[:stop_pumping] = true
-    @out_pumper.join
-    @err_pumper.join
-    Process.wait
+    @balancer_process.die
   end
 end

data/spike/Rakefile

@@ -0,0 +1,12 @@
+require 'rubygems'
+require 'cucumber'
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'lib', 'tlb'))
+require 'tlb/cucumber/rake/cucumber_task'
+
+Tlb::Cucumber::Rake::CucumberTask.new(:cucumber_tests)
+
+load 'tasks/tlb.rake'
+desc "Run Cucumber features in a load-balanced fashion (based on environment variables)"
+task :bal => ['tlb:start', :cucumber_tests]
+
+

data/spike/features/step_definitions/addition.rb

@@ -0,0 +1,18 @@
+require ('test/unit/assertions')
+
+World(Test::Unit::Assertions)
+
+Given /There are two numbers (\d+) and (\d+)/ do |one, two|
+  instance_variable_set("@first", one)
+  instance_variable_set("@second", two)
+end
+
+When /I add them together/ do
+  one = instance_variable_get("@first")
+  two = instance_variable_get("@second")
+  instance_variable_set("@sum", one.to_i + two.to_i)
+end
+
+Then /I should get (\d+)/ do |expected_sum|
+  assert_equal expected_sum.to_i, instance_variable_get("@sum")
+end

data/spike/features/step_definitions/answer.rb

@@ -0,0 +1,13 @@
+require ('test/unit/assertions')
+
+World(Test::Unit::Assertions)
+
+Given /I am wondering what the hell is the answer/ do
+end
+
+When /I ask Deep thought/ do
+end
+
+Then /Deep Thought should say "(\d+)"/ do |answer|
+   assert_equal 42, answer.to_i
+end

data/spike/features/step_definitions/sub.rb

@@ -0,0 +1,18 @@
+require ('test/unit/assertions')
+
+World(Test::Unit::Assertions)
+
+Given /There are two numbers for subtraction (\d+) and (\d+)/ do |one, two|
+  instance_variable_set("@first", one)
+  instance_variable_set("@second", two)
+end
+
+When /I subtract second from the first/ do
+  one = instance_variable_get("@first")
+  two = instance_variable_get("@second")
+  instance_variable_set("@diff", one.to_i - two.to_i)
+end
+
+Then /I should get the difference (\d+)/ do |expected_difference|
+  assert_equal expected_difference.to_i, instance_variable_get("@diff")
+end

data/spike/features/sub.feature

@@ -0,0 +1,6 @@
+Feature: Subtraction
+
+  Scenario: Make sure I can subtract 2 numbers
+    Given There are two numbers for subtraction 2 and 1
+    When I subtract second from the first
+    Then I should get the difference 0

data/spike/features/sum.feature

@@ -0,0 +1,6 @@
+Feature: Addition
+
+  Scenario: Make sure I can add 2 numbers
+    Given There are two numbers 1 and 2
+    When I add them together
+    Then I should get 3

data/spike/features/the_answer.feature

@@ -0,0 +1,6 @@
+Feature: The Answer mate!
+
+         Scenario: What is the answer mate?
+         Given I am wondering what the hell is the answer
+         When I ask Deep thought
+         Then Deep Thought should say "42"

data/test.sh

@@ -0,0 +1,42 @@
+#!/bin/bash
+
+# this is a smart hack strictly meant for developer convinience, not for CI. -janmejay
+
+[[ -s "$HOME/.rvm/scripts/rvm" ]] && source "$HOME/.rvm/scripts/rvm"
+
+sep() {
+    in_red "-----------------------------------------------------------------"
+    echo
+}
+
+function in_red {
+    tput sgr0
+    tput setaf 1; 
+    tput setab 7; 
+    echo -n $1; 
+    tput sgr0
+}
+
+function show_running_with {
+    echo -e "\n"
+    sep
+    in_red "|"
+    echo -n " "
+    tput setaf 4
+    echo -n "Running tests with: "
+    tput bold; 
+    tput setaf 0
+    ruby --version
+    sep
+    echo 
+}
+
+run_tests_with() {
+    rvm use $1
+    show_running_with  
+    rake test
+}
+
+run_tests_with ruby-1.8.7-p334@tlb
+
+run_tests_with jruby-1.5.6@tlb

data/tlb-cucumber.gemspec

@@ -0,0 +1,9 @@
+$name="tlb-cucumber"
+$framework='cucumber'
+require File.join(File.dirname(__FILE__), 'gem_common')
+
+Gem::Specification.new do |s|
+  configure_tlb(s)
+  s.files = files('tests', File.join('lib', 'tlb', 'rspec'), File.join('lib', 'tlb', 'test_unit'))
+  s.add_runtime_dependency 'cucumber', '>= 0.10.2'
+end

data/tlb-rspec2.gemspec

@@ -4,8 +4,7 @@ require File.join(File.dirname(__FILE__), 'gem_common')
 
 Gem::Specification.new do |s|
   configure_tlb(s)
-
-  s.files            = files('tests', File.join('lib', 'tlb', 'test_unit'))
-
+  s.files = files('tests', File.join('lib', 'tlb', 'test_unit'), File.join('lib', 'tlb', 'cucumber'))
   s.add_runtime_dependency 'rspec', '>= 2.3.0'
 end
+

data/tlb-testunit.gemspec

@@ -4,6 +4,5 @@ require File.join(File.dirname(__FILE__), 'gem_common')
 
 Gem::Specification.new do |s|
   configure_tlb(s)
-
-  s.files            = files('tests', File.join('lib', 'tlb', 'rspec'))
+  s.files = files('tests', File.join('lib', 'tlb', 'rspec'), File.join('lib', 'tlb', 'cucumber'))
 end

metadata

@@ -1,54 +1,49 @@
 --- !ruby/object:Gem::Specification 
 name: tlb-testunit
 version: !ruby/object:Gem::Version 
-  hash: 19
-  prerelease: 
+  prerelease: false
   segments: 
-  - 0
-  - 3
-  - 0
-  version: 0.3.0
+    - 0
+    - 3
+    - 1
+  version: 0.3.1
 platform: ruby
 authors: 
-- Janmejay Singh
-- Pavan KS
+  - Janmejay Singh
+  - Pavan KS
 autorequire: 
 bindir: bin
 cert_chain: []
 
-date: 2011-03-05 00:00:00 +05:30
+date: 2011-05-11 00:00:00 +05:30
 default_executable: 
 dependencies: 
-- !ruby/object:Gem::Dependency 
-  name: open4
-  prerelease: false
-  requirement: &id001 !ruby/object:Gem::Requirement 
-    none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 21
-        segments: 
-        - 1
-        - 0
-        - 1
-        version: 1.0.1
-  type: :runtime
-  version_requirements: *id001
-- !ruby/object:Gem::Dependency 
-  name: rake
-  prerelease: false
-  requirement: &id002 !ruby/object:Gem::Requirement 
-    none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        hash: 3
-        segments: 
-        - 0
-        version: "0"
-  type: :runtime
-  version_requirements: *id002
+  - !ruby/object:Gem::Dependency 
+    name: open4
+    prerelease: false
+    requirement: &id001 !ruby/object:Gem::Requirement 
+      requirements: 
+        - - ">="
+          - !ruby/object:Gem::Version 
+            segments: 
+              - 1
+              - 0
+              - 1
+            version: 1.0.1
+    type: :runtime
+    version_requirements: *id001
+  - !ruby/object:Gem::Dependency 
+    name: rake
+    prerelease: false
+    requirement: &id002 !ruby/object:Gem::Requirement 
+      requirements: 
+        - - ">="
+          - !ruby/object:Gem::Version 
+            segments: 
+              - 0
+            version: "0"
+    type: :runtime
+    version_requirements: *id002
 description: |
   TLB ruby implementation base, which provides support for load balancing tests written in test::unit.
   TLB.rb test suite is not bundled, please check http://github.com/test-load-balancer/tlb.rb for tests.
@@ -60,25 +55,34 @@ executables: []
 extensions: []
 
 extra_rdoc_files: 
-- README.markdown
+  - README.markdown
 files: 
-- .emacs_project
-- .gitignore
-- .gitmodules
-- README.markdown
-- Rakefile
-- gem_common.rb
-- lib/tasks/tlb.rake
-- lib/tlb.rb
-- lib/tlb/run_data.rb
-- lib/tlb/test_unit/mediator_inflection.rb
-- lib/tlb/test_unit/test_observer.rb
-- lib/tlb/test_unit/test_splitter.rb
-- lib/tlb/test_unit/test_task.rb
-- lib/tlb/util.rb
-- tlb-rspec2.gemspec
-- tlb-testunit.gemspec
-- tlb-alien-g0.3.0.jar
+  - .emacs_project.template
+  - .gitignore
+  - .gitmodules
+  - README.markdown
+  - Rakefile
+  - gem_common.rb
+  - lib/tasks/tlb.rake
+  - lib/tlb.rb
+  - lib/tlb/run_data.rb
+  - lib/tlb/test_unit/mediator_inflection.rb
+  - lib/tlb/test_unit/test_observer.rb
+  - lib/tlb/test_unit/test_splitter.rb
+  - lib/tlb/test_unit/test_task.rb
+  - lib/tlb/util.rb
+  - spike/Rakefile
+  - spike/features/step_definitions/addition.rb
+  - spike/features/step_definitions/answer.rb
+  - spike/features/step_definitions/sub.rb
+  - spike/features/sub.feature
+  - spike/features/sum.feature
+  - spike/features/the_answer.feature
+  - test.sh
+  - tlb-cucumber.gemspec
+  - tlb-rspec2.gemspec
+  - tlb-testunit.gemspec
+  - tlb-alien-g0.3.0-4-g1077d1b.jar
 has_rdoc: true
 homepage: http://github.com/test-load-balancer/tlb.rb
 licenses: []
@@ -102,33 +106,29 @@ post_install_message: |
   -------------------------------------------------------------------------
 
 rdoc_options: 
-- --charset=UTF-8
+  - --charset=UTF-8
 require_paths: 
-- lib
+  - lib
 required_ruby_version: !ruby/object:Gem::Requirement 
-  none: false
   requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      hash: 3
-      segments: 
-      - 0
-      version: "0"
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+          - 0
+        version: "0"
 required_rubygems_version: !ruby/object:Gem::Requirement 
-  none: false
   requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      hash: 3
-      segments: 
-      - 0
-      version: "0"
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+          - 0
+        version: "0"
 requirements: []
 
 rubyforge_project: tlb-rb
-rubygems_version: 1.5.2
+rubygems_version: 1.3.6
 signing_key: 
 specification_version: 3
-summary: tlb-testunit-0.3.0
+summary: tlb-testunit-0.3.1
 test_files: []
 

data/.emacs_project

@@ -1,7 +0,0 @@
-(setq rspec-executable "rspec")
-(setenv "GEM_HOME" "/home/janmejay/.rvm/gems/ruby-1.8.7-head@vanilla")
-(setenv "GEM_PATH" "/home/janmejay/.rvm/gems/ruby-1.8.7-head@vanilla:/home/janmejay/.rvm/gems/ruby-1.8.7-head@global")
-(let ((gem-bin "/home/janmejay/.rvm/gems/ruby-1.8.7-head@vanilla/bin")
-      (path (getenv "PATH")))
-  (unless (string-match-p  gem-bin path)
-    (setenv "PATH" (concat gem-bin ":" path))))