deep_test_pre 2.0

data/lib/deep_test/options.rb

@@ -0,0 +1,113 @@
+module DeepTest
+  Option = Struct.new :name, :default unless defined?(Option)
+
+  class Options
+    unless defined?(VALID_OPTIONS)
+      VALID_OPTIONS = [
+        Option.new(:distributed_hosts, nil),
+        Option.new(:number_of_agents,  nil),
+        Option.new(:metrics_file,      nil),
+        Option.new(:pattern,           nil),
+        Option.new(:server_port,       nil),
+        Option.new(:sync_options,      {}),
+        Option.new(:ui,                "DeepTest::UI::Console"),
+        Option.new(:listener,          "DeepTest::NullListener"),
+      ]
+    end
+
+    attr_accessor *VALID_OPTIONS.map {|o| o.name}
+    attr_accessor :ssh_client_connection_info, :environment_log_level
+
+    def number_of_agents
+      return CpuInfo.new.count unless @number_of_agents
+      @number_of_agents
+    end
+
+    def ui=(value)
+      @ui = value.to_s
+    end
+
+    def listener=(value)
+      @listener = value.to_s
+    end
+
+    def self.from_command_line(command_line)
+      return new({}) if command_line.nil? || command_line.empty?
+      Marshal.load Base64.decode64(command_line)
+    end
+
+    def initialize(hash)
+      @origin_hostname = Socket.gethostname
+      check_option_keys(hash)
+      VALID_OPTIONS.each do |option|
+        send("#{option.name}=", hash[option.name] || hash[option.name.to_s] || option.default)
+      end
+      self.environment_log_level = ENV['DEEP_TEST_LOG_LEVEL']
+    end
+
+    def gathering_metrics?
+      !@metrics_file.nil?
+    end
+
+    def new_listener_list
+      listeners = listener.split(',').map do |listener|
+        eval(listener).new
+      end
+      ListenerList.new(listeners)
+    end
+
+    def origin_hostname
+      (Socket.gethostname == @origin_hostname) ? 'localhost' : @origin_hostname
+    end
+
+    def connect_to_central_command
+      address = ssh_client_connection_info ? ssh_client_connection_info.address : "localhost"
+      Telegraph::Wire.connect(address, server_port) do |wire|
+        yield wire
+      end
+    end
+
+    # Don't store UI instances in the options instance, which will
+    # need to be dumped over Telegraph since UI instances may not be dumpable.
+    #
+    UI_INSTANCES = {} unless defined?(UI_INSTANCES)
+    def ui_instance
+      UI_INSTANCES[self] ||= eval(ui).new(self)
+    end
+
+    def to_command_line
+      Base64.encode64(Marshal.dump(self)).gsub("\n","")
+    end
+
+    def mirror_path
+      raise "No source directory specified in sync_options" unless sync_options[:source]
+      relative_mirror_path = @origin_hostname + sync_options[:source].gsub('/','_')
+      "#{sync_options[:remote_base_dir] || '/tmp'}/#{relative_mirror_path}"
+    end
+
+    def new_deployment
+      if distributed_hosts.nil?
+        LocalDeployment.new self
+      else
+        Distributed::RemoteDeployment.new self, new_landing_fleet, LocalDeployment.new(self)
+      end
+    end
+
+    def new_landing_fleet
+      landing_ships = distributed_hosts.map do |host|
+        Distributed::LandingShip.new :address => host
+      end
+      Distributed::LandingFleet.new self, landing_ships
+    end
+
+    protected
+
+    def check_option_keys(hash)
+      hash.keys.each do |key|
+        raise InvalidOptionError.new("#{key} is not a valid option") unless VALID_OPTIONS.any? {|o| o.name == key.to_sym}
+      end
+    end
+
+    class InvalidOptionError < StandardError; end
+  end
+end

data/lib/deep_test/proxy_io.rb

@@ -0,0 +1,77 @@
+module DeepTest
+  class ProxyIO < StringIO
+    def initialize(output_module, wire)
+      @output_module = output_module
+      @wire = wire
+      super("")
+    end
+
+    def write(*args)
+      super
+      @wire.send_message @output_module::Output.new(string)
+      self.string = ""
+    end
+
+    def flush
+      @wire.send_message @output_module::Flush.new
+    end
+
+    def self.replace_stdout_stderr!(wire)
+      old_stdout_const, old_stdout_global = STDOUT, $stdout
+      old_stderr_const, old_stderr_global = STDERR, $stderr
+
+      supress_warnings { Object.const_set :STDOUT, ProxyIO.new(Stdout, wire) }
+      $stdout = STDOUT
+
+      supress_warnings { Object.const_set :STDERR, ProxyIO.new(Stderr, wire) }
+      $stderr = STDERR
+
+      DeepTest.logger = nil
+
+      yield
+    ensure
+      $stdout = old_stdout_global
+      supress_warnings { Object.const_set :STDOUT, old_stdout_const }
+
+      $stderr = old_stderr_global
+      supress_warnings { Object.const_set :STDERR, old_stderr_const }
+    end
+
+    def self.supress_warnings
+      old_verbose, $VERBOSE = $VERBOSE, nil
+      yield
+    ensure
+      $VERBOSE = old_verbose
+    end
+
+    class AbstractOutput
+      include CentralCommand::Operation
+      attr_reader :s
+      def initialize(s); @s = s; end
+      def execute; stream.write s; end
+      def ==(other); self.class == other.class && s == other.s; end
+    end
+
+    module Stdout
+      class Output < AbstractOutput
+        def stream; $stdout; end
+      end
+
+      class Flush
+        include CentralCommand::Operation
+        def execute; $stdout.flush; end
+      end
+    end
+
+    module Stderr
+      class Output < AbstractOutput
+        def stream; $stderr; end
+      end
+
+      class Flush
+        include CentralCommand::Operation
+        def execute; $stderr.flush; end
+      end
+    end
+  end
+end

data/lib/deep_test/rake_tasks.rb

@@ -0,0 +1,13 @@
+require "socket"
+require "base64"
+
+require File.dirname(__FILE__) + "/options"
+require File.dirname(__FILE__) + "/cpu_info"
+require File.dirname(__FILE__) + "/test_task"
+require File.dirname(__FILE__) + "/rspec_detector"
+
+DeepTest::RSpecDetector.if_rspec_available do
+  require 'spec/rake/spectask'
+  require File.dirname(__FILE__) + "/spec/extensions/spec_task"
+end
+

data/lib/deep_test/result_reader.rb

@@ -0,0 +1,40 @@
+module DeepTest
+  class ResultReader
+    def initialize(central_command)
+      @central_command = central_command
+    end
+
+    def read(original_work_units_by_id)
+      work_units_by_id = original_work_units_by_id.dup
+      errors = 0
+
+      begin
+        until errors == work_units_by_id.size
+          Thread.pass
+          result = @central_command.take_result
+          next if result.nil?
+
+          if Agent::Error === result
+            puts result
+            errors += 1
+          else
+            if result.respond_to?(:output) && (output = result.output)
+              print output
+            end
+
+            work_unit = work_units_by_id.delete(result.identifier)
+            yield [work_unit, result]
+          end
+        end
+      rescue CentralCommand::NoAgentsRunningError
+        FailureMessage.show "DeepTest Agents Are Not Running", <<-end_msg
+          DeepTest's test running agents have not contacted the 
+          server to indicate they are still running.
+          Shutting down the test run on the assumption that they have died. 
+        end_msg
+      end
+
+      work_units_by_id
+    end
+  end
+end

data/lib/deep_test/rspec_detector.rb

@@ -0,0 +1,21 @@
+module DeepTest
+  class RSpecDetector
+    def self.if_rspec_available
+      if defined?(::Spec)
+        require 'spec/version'
+        if ::Spec::VERSION::MAJOR == 1 &&
+           ::Spec::VERSION::MINOR == 1 &&
+           ::Spec::VERSION::TINY  >= 8
+          yield
+        else
+          require 'spec/rake/spectask'
+          ::Spec::Rake::SpecTask.class_eval do
+            def deep_test(options)
+              raise "* DeepTest RSpec support requires RSpec 1.1.8"
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/deep_test/spec.rb

@@ -0,0 +1,17 @@
+require 'rubygems'
+require 'spec/runner/example_group_runner'
+if ::Spec::VERSION::MAJOR == 1 &&
+   ::Spec::VERSION::MINOR == 1 &&
+   ::Spec::VERSION::TINY  >= 12
+  require 'spec/example/before_and_after_hooks'
+end
+require 'spec/example/example_group_methods'
+require 'spec/rake/spectask'
+
+require File.dirname(__FILE__) + "/spec/extensions/example_group_methods"
+require File.dirname(__FILE__) + "/spec/extensions/example_methods"
+require File.dirname(__FILE__) + "/spec/extensions/spec_task"
+require File.dirname(__FILE__) + "/spec/extensions/options"
+require File.dirname(__FILE__) + "/spec/runner"
+require File.dirname(__FILE__) + "/spec/work_unit"
+require File.dirname(__FILE__) + "/spec/work_result"

data/lib/deep_test/spec/extensions/example_group_methods.rb

@@ -0,0 +1,64 @@
+module Spec
+  module Example
+    module ExampleGroupMethods
+      class << self
+        def assign_instance_method_to_constant(proposed_constant)
+          method_sym = proposed_constant.to_s.downcase
+
+          unless const_defined?(proposed_constant)
+            const_set(proposed_constant, instance_method(method_sym))
+          end
+        end
+        
+        private :assign_instance_method_to_constant
+      end
+
+      assign_instance_method_to_constant :PREPEND_BEFORE
+      assign_instance_method_to_constant :APPEND_BEFORE
+      assign_instance_method_to_constant :PREPEND_AFTER
+      assign_instance_method_to_constant :APPEND_AFTER
+
+      def prepend_before(*args, &block)
+        check_filter_args(args)
+        call_regular_instance_method :prepend_before, *args, &block
+      end
+
+      def append_before(*args, &block)
+        check_filter_args(args)
+        call_regular_instance_method :append_before, *args, &block
+      end
+      
+      alias_method :before, :append_before
+
+      def prepend_after(*args, &block)
+        check_filter_args(args)
+        call_regular_instance_method :prepend_after, *args, &block
+      end
+
+      def append_after(*args, &block)
+        check_filter_args(args)
+        call_regular_instance_method :append_after, *args, &block
+      end
+      
+      alias_method :after, :append_after
+
+    private
+    
+      DeepTestAllBlockWarning = 
+        "Warning: DeepTest will run before(:all) and after(:all) blocks for *every* test that is run.  To remove this warning either convert all before/after blocks to each blocks or set $show_deep_test_all_block_warning to false" unless defined?(DeepTestAllBlockWarning)
+      
+      $show_deep_test_all_block_warning = true
+
+      def check_filter_args(args)
+        if args.first == :all && $show_deep_test_all_block_warning
+          $show_deep_test_all_block_warning = false
+          $stderr.puts DeepTestAllBlockWarning 
+        end
+      end
+    
+      def call_regular_instance_method(sym, *args, &block)
+        ExampleGroupMethods.const_get(sym.to_s.upcase).bind(self).call(*args, &block)
+      end
+    end
+  end
+end

data/lib/deep_test/spec/extensions/example_methods.rb

@@ -0,0 +1,52 @@
+module Spec
+  module Example
+    module ExampleMethods
+      def identifier
+        if ::Spec::VERSION::MAJOR == 1 &&
+           ::Spec::VERSION::MINOR == 1 &&
+           ::Spec::VERSION::TINY  >= 12
+          file, line = eval("caller", @_implementation).first.split(/:/)
+        else
+          file, line = implementation_backtrace.first.split(/:/)
+        end
+        Identifier.new(file, line.to_i, self.class.description, description)
+      end
+
+      class Identifier
+        attr_reader :file, :line, :group_description, :description
+        def initialize(file, line, group_description, description)
+          @file, @line, @group_description, @description = 
+           file,  line,  group_description,  description
+        end
+
+        def ==(other)
+          eql?(other)
+        end
+
+        def eql?(other)
+          File.basename(file) == File.basename(other.file) && 
+                         line == other.line &&
+            group_description == other.group_description &&
+                  description == other.description
+        end
+
+        def hash
+          description.hash
+        end
+
+        def locate(groups)
+          groups.each do |group|
+            group.examples.each do |example|
+              return example if example.identifier == self
+            end
+          end
+          raise "Unable to locate example #{self}"
+        end
+
+        def to_s
+          "#{group_description} #{description}"
+        end
+      end
+    end
+  end
+end

data/lib/deep_test/spec/extensions/options.rb

@@ -0,0 +1,43 @@
+module Spec
+  module Runner
+    class Options
+      def run_one_example(identifier)
+        example = identifier.locate(example_groups)
+        SingleExampleRunner.new(self, example).run
+      end
+
+      class SingleExampleRunner < ExampleGroupRunner
+        def initialize(options, example)
+          super(options)
+          @example = example
+          example_group.extend ExampleGroupHelper
+        end
+
+        def example_group
+          @example.class
+        end
+
+        def example_groups
+          [example_group]
+        end
+
+        def run
+          example_group.with_example_objects([@example]) do
+            super
+          end
+        end
+
+        module ExampleGroupHelper
+          def with_example_objects(example_objects)
+            original_example_objects = @example_objects
+            @example_objects = example_objects
+            yield
+          ensure
+            @example_objects = original_example_objects
+          end
+        end
+      end
+    end
+  end
+end
+