root/lang/ruby/memoize/test/test_memoize.rb @ 27246

require File.expand_path(File.join(File.dirname(__FILE__), "helper"))
require "memoize"

class TestMemoize < Test::Unit::TestCase
  class InstanceMethod
    extend Memoize
    
    memoize
    def refrector n; n; end
    
    memoize [0] => 100, 1 => 101
    def refrector_with_default n; n; end
  end
  
  class ClassMethod
    class << self
      extend Memoize
      
      memoize
      def refrector n; n; end
      
      memoize [0] => 100, 1 => 101
      def refrector_with_default n; n; end
    end
  end
  
  module ModuleFunction
    extend Memoize
    
    memoize
    def refrector n; n; end
    module_function :refrector, :refrector_without_memoize
    
    memoize [0] => 100, 1 => 101
    def refrector_with_default n; n; end
    module_function :refrector_with_default, :refrector_with_default_without_memoize
  end
  
  
  def test_memoize_classes
    [
      InstanceMethod.new,
      ClassMethod,
      ModuleFunction,
    ].each do |target|
      assert_memoized target, :refrector
      assert_memoized target, :refrector_with_default, 0 => 100, 1 => 101, 2 => 2
    end
  end
  
  
  def assert_memoized target, method, defaults = {}
    string1 = "string"
    string2 = String.new(string1)
    
    refrect = target.__send__ method, string1
    assert_equal true, refrect.equal?(string1), target
    
    refrect = target.__send__ method, string1
    assert_equal true, refrect.equal?(string1), target
    
    defaults.each do |arg, default|
      assert_equal default, target.refrector_with_default(arg), target
    end
  end
end