From 9a38b0853bd23a4910d2459187fc7d8b46bdc42b Mon Sep 17 00:00:00 2001
From: Nick Gauthier <nick@smartlogicsolutions.com>
Date: Thu, 28 Jan 2010 16:52:11 -0500
Subject: [PATCH] flipped worker test to test in foreground and background.
 more accurate rcov results

---
 test/runner_test.rb | 20 ++++++++---------
 test/worker_test.rb | 53 ++++++++++++++++++++++++++++++++-------------
 2 files changed, 48 insertions(+), 25 deletions(-)

diff --git a/test/runner_test.rb b/test/runner_test.rb
index 3d0959d..44f4314 100644
--- a/test/runner_test.rb
+++ b/test/runner_test.rb
@@ -18,23 +18,23 @@ class RunnerTest < Test::Unit::TestCase
       # flip it around to the parent is in the fork, this gives
       # us more direct control over the runner and proper test
       # coverage output
-      @pipe = Hydra::Pipe.new
-      @parent = Process.fork do
-        request_a_file_and_verify_completion(@pipe)
+      pipe = Hydra::Pipe.new
+      parent = Process.fork do
+        request_a_file_and_verify_completion(pipe)
       end
-      run_the_runner(@pipe)
-      Process.wait(@parent)
+      run_the_runner(pipe)
+      Process.wait(parent)
     end
 
     # this flips the above test, so that the main process runs a bit of the parent
     # code, but only with minimal assertion
     should "be able to tell a runner to run a test" do
-      @pipe = Hydra::Pipe.new
-      @child = Process.fork do
-        run_the_runner(@pipe)
+      pipe = Hydra::Pipe.new
+      child = Process.fork do
+        run_the_runner(pipe)
       end
-      request_a_file_and_verify_completion(@pipe)
-      Process.wait(@child)
+      request_a_file_and_verify_completion(pipe)
+      Process.wait(child)
     end
   end
 
diff --git a/test/worker_test.rb b/test/worker_test.rb
index 242c30c..eb1a7cc 100644
--- a/test/worker_test.rb
+++ b/test/worker_test.rb
@@ -13,30 +13,53 @@ class WorkerTest < Test::Unit::TestCase
       FileUtils.rm_f(TARGET)
     end
 
+    # run the worker in the foreground and the requests in the background
     should "run a test" do
       num_runners = 4
-      @pipe = Hydra::Pipe.new
-      @child = Process.fork do
-        @pipe.identify_as_child
-        Hydra::Worker.new(@pipe, num_runners)
-        @pipe.close
+      pipe = Hydra::Pipe.new
+      child = Process.fork do
+        request_a_file_and_verify_completion(pipe, num_runners)
+        pipe.close
       end
-      @pipe.identify_as_parent
-      num_runners.times do
-        assert @pipe.gets.is_a?(Hydra::Messages::Worker::RequestFile)
-      end
-      @pipe.write(Hydra::Messages::Worker::RunFile.new(:file => TESTFILE))
+      run_the_worker(pipe, num_runners)
+      Process.wait(child)
+    end
 
-      response = @pipe.gets
+    # inverse of the above test to run the worker in the background
+    should "be able to tell a worker to run a test" do
+      num_runners = 4
+      pipe = Hydra::Pipe.new
+      child = Process.fork do
+        run_the_worker(pipe, num_runners)
+      end
+      request_a_file_and_verify_completion(pipe, num_runners)
+      Process.wait(child)
+      pipe.close
+    end
+  end
+
+  module WorkerTestHelper
+    def run_the_worker(pipe, num_runners)
+      pipe.identify_as_child
+      Hydra::Worker.new(pipe, num_runners)
+      pipe.close
+    end
+
+    def request_a_file_and_verify_completion(pipe, num_runners)
+      pipe.identify_as_parent
+      num_runners.times do
+        assert pipe.gets.is_a?(Hydra::Messages::Worker::RequestFile)
+      end
+      pipe.write(Hydra::Messages::Worker::RunFile.new(:file => TESTFILE))
+
+      response = pipe.gets
       assert response.is_a?(Hydra::Messages::Worker::Results)
 
-      @pipe.write(Hydra::Messages::Worker::Shutdown.new)
+      pipe.write(Hydra::Messages::Worker::Shutdown.new)
 
       assert File.exists?(TARGET)
       assert_equal "HYDRA", File.read(TARGET)
-
-      Process.wait(@child)
-      @pipe.close
     end
   end
+  include WorkerTestHelper
 end