puppet 0.18.4 → 0.22.4

data/test/language/transportable.rb

@@ -1,19 +1,15 @@
-#!/usr/bin/ruby
+#!/usr/bin/env ruby
 
-if __FILE__ == $0
-    $:.unshift '../../lib'
-    $:.unshift '..'
-    $puppetbase = "../.."
-end
+$:.unshift("../lib").unshift("../../lib") if __FILE__ =~ /\.rb$/
 
 require 'puppet'
 require 'puppet/transportable'
-require 'test/unit'
 require 'puppettest'
+require 'puppettest/parsertesting'
 require 'yaml'
 
 class TestTransportable < Test::Unit::TestCase
-	include ParserTesting
+    include PuppetTest::ParserTesting
 
     def test_yamldumpobject
         obj = mk_transobject
@@ -107,3 +103,5 @@ class TestTransportable < Test::Unit::TestCase
             "Found incorrect number of collectable objects")
     end
 end
+
+# $Id: transportable.rb 1793 2006-10-16 22:01:40Z luke $

data/test/lib/mocha.rb

@@ -0,0 +1,19 @@
+require 'mocha_standalone'
+require 'mocha/test_case_adapter'
+
+require 'test/unit/testcase'
+
+module Test
+
+  module Unit
+
+    class TestCase
+  
+      include Mocha::Standalone
+      include Mocha::TestCaseAdapter
+      
+    end
+
+  end
+
+end

data/test/lib/mocha/any_instance_method.rb

@@ -0,0 +1,35 @@
+require 'mocha/class_method'
+
+module Mocha
+
+  class AnyInstanceMethod < ClassMethod
+  
+    def unstub
+      remove_new_method
+      restore_original_method
+      stubbee.any_instance.reset_mocha
+    end
+    
+    def mock
+      stubbee.any_instance.mocha
+    end
+   
+    def hide_original_method
+      stubbee.class_eval "alias_method :#{hidden_method}, :#{method}" if stubbee.method_defined?(method)
+    end
+
+    def define_new_method
+      stubbee.class_eval "def #{method}(*args, &block); self.class.any_instance.mocha.method_missing(:#{method}, *args, &block); end"
+    end
+
+    def remove_new_method
+      stubbee.class_eval "remove_method :#{method}"
+    end
+
+    def restore_original_method
+      stubbee.class_eval "alias_method :#{method}, :#{hidden_method}; remove_method :#{hidden_method}" if stubbee.method_defined?(hidden_method)
+    end
+
+  end
+  
+end

data/test/lib/mocha/auto_verify.rb

@@ -0,0 +1,113 @@
+require 'mocha/mock'
+
+# Methods added to TestCase allowing creation of mock objects.
+#
+# Mocks created this way will have their expectations automatically verified at the end of the test.
+#
+# See Mocha::MockMethods for methods on mock objects.
+module Mocha
+  
+  module AutoVerify
+  
+    def mocks # :nodoc:
+      @mocks ||= []
+    end
+  
+    def reset_mocks # :nodoc:
+      @mocks = nil
+    end
+    
+    # :call-seq: mock(name) -> mock object
+    #            mock(expected_methods = {}) -> mock object
+    #            mock(name, expected_methods = {}) -> mock object
+    #
+    # Creates a mock object.
+    #
+    # +name+ is a +String+ identifier for the mock object.
+    #
+    # +expected_methods+ is a +Hash+ with expected method name symbols as keys and corresponding return values as values.
+    #
+    # Note that (contrary to expectations set up by #stub) these expectations <b>must</b> be fulfilled during the test.
+    #   def test_product
+    #     product = mock('ipod_product', :manufacturer => 'ipod', :price => 100)
+    #     assert_equal 'ipod', product.manufacturer
+    #     assert_equal 100, product.price
+    #     # an error will be raised unless both Product#manufacturer and Product#price have been called
+    #   end 
+    def mock(*args)
+      name, expectations = name_and_expectations_from_args(args)
+      build_mock_with_expectations(:expects, expectations, name)
+    end
+  
+    # :call-seq: stub(name) -> mock object
+    #            stub(stubbed_methods = {}) -> mock object
+    #            stub(name, stubbed_methods = {}) -> mock object
+    #
+    # Creates a mock object.
+    #
+    # +name+ is a +String+ identifier for the mock object.
+    #
+    # +stubbed_methods+ is a +Hash+ with stubbed method name symbols as keys and corresponding return values as values.
+    #
+    # Note that (contrary to expectations set up by #mock) these expectations <b>need not</b> be fulfilled during the test.
+    #   def test_product
+    #     product = stub('ipod_product', :manufacturer => 'ipod', :price => 100)
+    #     assert_equal 'ipod', product.manufacturer
+    #     assert_equal 100, product.price
+    #     # an error will not be raised even if Product#manufacturer and Product#price have not been called
+    #   end
+    def stub(*args)
+      name, expectations = name_and_expectations_from_args(args)
+      build_mock_with_expectations(:stubs, expectations, name)
+    end
+  
+    # :call-seq: stub_everything(name) -> mock object
+    #            stub_everything(stubbed_methods = {}) -> mock object
+    #            stub_everything(name, stubbed_methods = {}) -> mock object
+    #
+    # Creates a mock object that accepts calls to any method.
+    #
+    # By default it will return +nil+ for any method call.
+    #
+    # +name+ and +stubbed_methods+ work in the same way as for #stub.
+    #   def test_product
+    #     product = stub_everything('ipod_product', :price => 100)
+    #     assert_nil product.manufacturer
+    #     assert_nil product.any_old_method
+    #     assert_equal 100, product.price
+    #   end
+    def stub_everything(*args)
+      name, expectations = name_and_expectations_from_args(args)
+      build_mock_with_expectations(:stub_everything, expectations, name)
+    end
+    
+    def verify_mocks # :nodoc:
+      mocks.each { |mock| mock.verify { yield if block_given? } }
+    end
+
+    def teardown_mocks # :nodoc:
+      reset_mocks
+    end
+  
+    def build_mock_with_expectations(expectation_type = :expects, expectations = {}, name = nil) # :nodoc:
+      stub_everything = (expectation_type == :stub_everything)
+      expectation_type = :stubs if expectation_type == :stub_everything
+      mock = Mocha::Mock.new(stub_everything, name)
+      expectations.each do |method, result|
+        mock.__send__(expectation_type, method).returns(result)
+      end
+      mocks << mock
+      mock
+    end
+    
+  private
+
+    def name_and_expectations_from_args(args) # :nodoc:
+      name = args.first.is_a?(String) ? args.delete_at(0) : nil
+      expectations = args.first || {}
+      [name, expectations]
+    end
+  
+  end
+  
+end

data/test/lib/mocha/central.rb

@@ -0,0 +1,35 @@
+module Mocha
+  
+  class Central
+  
+    attr_accessor :stubba_methods
+  
+    def initialize
+      self.stubba_methods = []
+    end
+   
+    def stub(method)
+      unless stubba_methods.include?(method)
+        method.stub 
+        stubba_methods.push method
+      end
+    end
+    
+    def verify_all(&block)
+      unique_mocks.each { |mock| mock.verify(&block) }
+    end
+    
+    def unique_mocks
+      stubba_methods.inject({}) { |mocks, method| mocks[method.mock.__id__] = method.mock; mocks }.values
+    end
+  
+    def unstub_all
+      while stubba_methods.size > 0
+        method = stubba_methods.pop
+        method.unstub
+      end
+    end
+
+  end
+
+end

data/test/lib/mocha/class_method.rb

@@ -0,0 +1,62 @@
+require 'mocha/metaclass'
+
+module Mocha
+
+  class ClassMethod
+  
+    attr_reader :stubbee, :method
+   
+    def initialize(stubbee, method)
+      @stubbee, @method = stubbee, method
+    end
+  
+    def stub
+      hide_original_method
+      define_new_method
+    end
+  
+    def unstub
+      remove_new_method
+      restore_original_method
+      stubbee.reset_mocha
+    end
+    
+    def mock
+      stubbee.mocha
+    end
+  
+    def hide_original_method
+      stubbee.__metaclass__.class_eval "alias_method :#{hidden_method}, :#{method}" if stubbee.__metaclass__.method_defined?(method)
+    end
+  
+    def define_new_method
+      stubbee.__metaclass__.class_eval "def #{method}(*args, &block); mocha.method_missing(:#{method}, *args, &block); end"
+    end
+  
+    def remove_new_method
+      stubbee.__metaclass__.class_eval "remove_method :#{method}"
+    end
+  
+    def restore_original_method
+      stubbee.__metaclass__.class_eval "alias_method :#{method}, :#{hidden_method}; remove_method :#{hidden_method}" if stubbee.__metaclass__.method_defined?(hidden_method)
+    end
+  
+    def hidden_method
+      method_name = method.to_s.gsub(/\W/) {|s| "_substituted_character_#{s[0]}_" }
+      "__stubba__#{method_name}__stubba__"
+    end  
+  
+    def eql?(other)
+      return false unless (other.class == self.class)
+      (stubbee == other.stubbee) and (method == other.method)
+    end
+  
+    alias_method :==, :eql?
+  
+    def to_s
+      "#{stubbee}.#{method}"
+    end
+
+  end
+  
+end

data/test/lib/mocha/expectation.rb

@@ -0,0 +1,295 @@
+require 'mocha/infinite_range'
+require 'mocha/pretty_parameters'
+require 'mocha/expectation_error'
+
+class Object
+
+  alias_method :__is_a__, :is_a?
+
+end
+
+module Mocha
+  # Methods on expectations returned from Mocha::MockMethods#expects and Mocha::MockMethods#stubs
+  class Expectation
+  
+    # :stopdoc:
+    
+    class InvalidExpectation < Exception; end
+    
+    class AlwaysEqual
+      def ==(other)
+        true
+      end
+    end
+  
+    attr_reader :method_name, :backtrace
+
+    def initialize(mock, method_name, backtrace = nil)
+      @mock, @method_name = mock, method_name
+      @count = 1
+      @parameters, @parameter_block = AlwaysEqual.new, nil
+      @invoked, @return_value = 0, nil
+      @backtrace = backtrace || caller
+      @yield = nil
+    end
+    
+    def yield?
+      @yield
+    end
+
+    def match?(method_name, *arguments)
+      if @parameter_block then
+        @parameter_block.call(*arguments)
+      else
+        (@method_name == method_name) and (@parameters == arguments)
+      end
+    end
+
+    # :startdoc:
+    
+    # :call-seq: times(range) -> expectation
+    #
+    # Modifies expectation so that the number of calls to the expected method must be within a specific +range+.
+    #
+    # +range+ can be specified as an exact integer or as a range of integers
+    #   object = mock()
+    #   object.expects(:expected_method).times(3)
+    #   3.times { object.expected_method } # => verify succeeds
+    #
+    #   object = mock()
+    #   object.expects(:expected_method).times(3)
+    #   2.times { object.expected_method } # => verify fails
+    #
+    #   object = mock()
+    #   object.expects(:expected_method).times(2..4)
+    #   3.times { object.expected_method } # => verify succeeds
+    #
+    #   object = mock()
+    #   object.expects(:expected_method).times(2..4)
+    #   object.expected_method # => verify fails
+    def times(range)
+      @count = range
+      self
+    end
+  
+    # :call-seq: never -> expectation
+    #
+    # Modifies expectation so that the expected method must never be called.
+    #   object = mock()
+    #   object.expects(:expected_method).never
+    #   object.expected_method # => verify fails
+    #
+    #   object = mock()
+    #   object.expects(:expected_method).never
+    #   object.expected_method # => verify succeeds
+    def never
+      times(0)
+      self
+    end
+  
+    # :call-seq: at_least(minimum_number_of_times) -> expectation
+    #
+    # Modifies expectation so that the expected method must be called at least a +minimum_number_of_times+.
+    #   object = mock()
+    #   object.expects(:expected_method).at_least(2)
+    #   3.times { object.expected_method } # => verify succeeds
+    #
+    #   object = mock()
+    #   object.expects(:expected_method).at_least(2)
+    #   object.expected_method # => verify fails
+    def at_least(minimum_number_of_times)
+      times(Range.at_least(minimum_number_of_times))
+      self
+    end
+  
+    # :call-seq: at_least_once() -> expectation
+    #
+    # Modifies expectation so that the expected method must be called at least once.
+    #   object = mock()
+    #   object.expects(:expected_method).at_least_once
+    #   object.expected_method # => verify succeeds
+    #
+    #   object = mock()
+    #   object.expects(:expected_method).at_least_once
+    #   # => verify fails
+    def at_least_once()
+      at_least(1)
+      self
+    end
+  
+    # :call-seq: at_most(maximum_number_of_times) -> expectation
+    #
+    # Modifies expectation so that the expected method must be called at most a +maximum_number_of_times+.
+    #   object = mock()
+    #   object.expects(:expected_method).at_most(2)
+    #   2.times { object.expected_method } # => verify succeeds
+    #
+    #   object = mock()
+    #   object.expects(:expected_method).at_most(2)
+    #   3.times { object.expected_method } # => verify fails
+    def at_most(maximum_number_of_times)
+      times(Range.at_most(maximum_number_of_times))
+      self
+    end
+  
+    # :call-seq: at_most_once() -> expectation
+    #
+    # Modifies expectation so that the expected method must be called at most once.
+    #   object = mock()
+    #   object.expects(:expected_method).at_most_once
+    #   object.expected_method # => verify succeeds
+    #
+    #   object = mock()
+    #   object.expects(:expected_method).at_most_once
+    #   2.times { object.expected_method } # => verify fails
+    def at_most_once()
+      at_most(1)
+      self
+    end
+  
+    # :call-seq: with(*arguments, &parameter_block) -> expectation
+    #
+    # Modifies expectation so that the expected method must be called with specified +arguments+.
+    #   object = mock()
+    #   object.expects(:expected_method).with(:param1, :param2)
+    #   object.expected_method(:param1, :param2) # => verify succeeds
+    #
+    #   object = mock()
+    #   object.expects(:expected_method).with(:param1, :param2)
+    #   object.expected_method(:param3) # => verify fails
+    # If a +parameter_block+ is given, the block is called with the parameters passed to the expected method.
+    # The expectation is matched if the block evaluates to +true+.
+    #   object = mock()
+    #   object.expects(:expected_method).with() { |value| value % 4 == 0 }
+    #   object.expected_method(16) # => verify succeeds
+    #
+    #   object = mock()
+    #   object.expects(:expected_method).with() { |value| value % 4 == 0 }
+    #   object.expected_method(17) # => verify fails
+    def with(*arguments, &parameter_block)
+      @parameters, @parameter_block = arguments, parameter_block
+      class << @parameters; def to_s; join(', '); end; end
+      self
+    end
+  
+    # :call-seq: yields(*parameters) -> expectation
+    #
+    # Modifies expectation so that when the expected method is called, it yields with the specified +parameters+.
+    #   object = mock()
+    #   object.expects(:expected_method).yields('result')
+    #   yielded_value = nil
+    #   object.expected_method { |value| yielded_value = value }
+    #   yielded_value # => 'result'
+    def yields(*parameters)
+      @yield = true
+      @parameters_to_yield = parameters
+      self
+    end
+
+    # :call-seq: returns(value) -> expectation
+    # :call-seq: returns(*values) -> expectation
+    #
+    # Modifies expectation so that when the expected method is called, it returns the specified +value+.
+    #   object = mock()
+    #   object.stubs(:stubbed_method).returns('result')
+    #   object.stubbed_method # => 'result'
+    #   object.stubbed_method # => 'result'
+    # If multiple +values+ are given, these are returned in turn on consecutive calls to the method.
+    #   object = mock()
+    #   object.stubs(:stubbed_method).returns(1, 2)
+    #   object.stubbed_method # => 1
+    #   object.stubbed_method # => 2
+    # If +value+ is a Proc, then expected method will return result of calling Proc.
+    #   object = mock()
+    #   object.stubs(:stubbed_method).returns(lambda { rand(100) })
+    #   object.stubbed_method # => 41
+    #   object.stubbed_method # => 77
+    def returns(*values)
+      @return_value = (values.size > 1) ? lambda { values.shift } : @return_value = values.first
+      self
+    end
+  
+    # :call-seq: raises(exception = RuntimeError, message = nil) -> expectation
+    #
+    # Modifies expectation so that when the expected method is called, it raises the specified +exception+ with the specified +message+.
+    #   object = mock()
+    #   object.expects(:expected_method).raises(Exception, 'message')
+    #   object.expected_method # => raises exception of class Exception and with message 'message'
+    def raises(exception = RuntimeError, message = nil)
+      @return_value = message ? lambda { raise exception, message } : lambda { raise exception }
+      self
+    end
+
+    # :stopdoc:
+    
+    def invoke
+      @invoked += 1
+      yield(*@parameters_to_yield) if yield? and block_given?
+      @return_value.__is_a__(Proc) ? @return_value.call : @return_value
+    end
+
+    def verify
+      yield(self) if block_given?
+      unless (@count === @invoked) then
+        error = ExpectationError.new(error_message(@count, @invoked))
+        error.set_backtrace(filtered_backtrace)
+        raise error
+      end
+    end
+    
+    def mocha_lib_directory
+      File.expand_path(File.join(File.dirname(__FILE__), "..")) + File::SEPARATOR
+    end
+    
+    def filtered_backtrace
+      backtrace.reject { |location| Regexp.new(mocha_lib_directory).match(File.expand_path(location)) }
+    end
+  
+    def method_signature
+      return "#{method_name}" if @parameters.__is_a__(AlwaysEqual)
+      "#{@method_name}(#{PrettyParameters.new(@parameters).pretty})"
+    end
+    
+    def error_message(expected_count, actual_count)
+      "#{@mock.mocha_inspect}.#{method_signature} - expected calls: #{expected_count}, actual calls: #{actual_count}"
+    end
+  
+    # :startdoc:
+    
+  end
+
+  # :stopdoc:
+  
+  class Stub < Expectation
+  
+    def verify
+      true
+    end
+  
+  end
+
+  class MissingExpectation < Expectation
+  
+    def initialize(mock, method_name)
+      super
+      @invoked = true
+    end
+  
+    def verify
+      msg = error_message(0, 1)
+      similar_expectations_list = similar_expectations.collect { |expectation| expectation.method_signature }.join("\n")
+      msg << "\nSimilar expectations:\n#{similar_expectations_list}" unless similar_expectations.empty?
+      error = ExpectationError.new(msg)
+      error.set_backtrace(filtered_backtrace)
+      raise error if @invoked
+    end
+  
+    def similar_expectations
+      @mock.expectations.select { |expectation| expectation.method_name == self.method_name }
+    end
+  
+  end
+
+  # :startdoc:
+  
+end