2008-07-15 00:50:32 +00:00
|
|
|
require 'abstract_unit'
|
|
|
|
|
|
|
|
uses_mocha 'Memoizable' do
|
|
|
|
class MemoizableTest < Test::Unit::TestCase
|
|
|
|
class Person
|
|
|
|
include ActiveSupport::Memoizable
|
|
|
|
|
|
|
|
def name
|
|
|
|
fetch_name_from_floppy
|
|
|
|
end
|
|
|
|
|
|
|
|
def age
|
|
|
|
nil
|
|
|
|
end
|
2008-07-18 16:18:16 +00:00
|
|
|
|
|
|
|
def random
|
|
|
|
rand(0)
|
|
|
|
end
|
|
|
|
|
|
|
|
memoize :name, :age, :random
|
2008-07-15 00:50:32 +00:00
|
|
|
|
|
|
|
private
|
|
|
|
def fetch_name_from_floppy
|
|
|
|
"Josh"
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2008-07-18 16:18:16 +00:00
|
|
|
def setup
|
|
|
|
@person = Person.new
|
|
|
|
end
|
|
|
|
|
2008-07-15 00:50:32 +00:00
|
|
|
def test_memoization
|
2008-07-18 16:18:16 +00:00
|
|
|
assert_equal "Josh", @person.name
|
|
|
|
|
|
|
|
@person.expects(:fetch_name_from_floppy).never
|
|
|
|
2.times { assert_equal "Josh", @person.name }
|
|
|
|
end
|
2008-07-15 00:50:32 +00:00
|
|
|
|
2008-07-18 16:18:16 +00:00
|
|
|
def test_reloadable
|
|
|
|
random = @person.random
|
|
|
|
assert_equal random, @person.random
|
|
|
|
assert_not_equal random, @person.random(:reload)
|
2008-07-15 00:50:32 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def test_memoized_methods_are_frozen
|
2008-07-18 16:18:16 +00:00
|
|
|
assert_equal true, @person.name.frozen?
|
|
|
|
|
|
|
|
@person.freeze
|
|
|
|
assert_equal "Josh", @person.name
|
|
|
|
assert_equal true, @person.name.frozen?
|
2008-07-15 00:50:32 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def test_memoization_frozen_with_nil_value
|
2008-07-18 16:18:16 +00:00
|
|
|
@person.freeze
|
|
|
|
assert_equal nil, @person.age
|
2008-07-15 00:50:32 +00:00
|
|
|
end
|
2008-07-15 01:23:23 +00:00
|
|
|
|
|
|
|
def test_double_memoization
|
|
|
|
assert_raise(RuntimeError) { Person.memoize :name }
|
|
|
|
end
|
2008-07-15 00:50:32 +00:00
|
|
|
end
|
|
|
|
end
|