97 lines
2.3 KiB
Ruby
97 lines
2.3 KiB
Ruby
require File.expand_path(File.dirname(__FILE__) + '/neo')
|
|
|
|
class AboutBlocks < Neo::Koan
|
|
def method_with_block
|
|
result = yield
|
|
result
|
|
end
|
|
|
|
def test_methods_can_take_blocks
|
|
yielded_result = method_with_block { 1 + 2 }
|
|
assert_equal __, yielded_result
|
|
end
|
|
|
|
def test_blocks_can_be_defined_with_do_end_too
|
|
yielded_result = method_with_block do 1 + 2 end
|
|
assert_equal __, yielded_result
|
|
end
|
|
|
|
# ------------------------------------------------------------------
|
|
|
|
def method_with_block_arguments
|
|
yield("Jim")
|
|
end
|
|
|
|
def test_blocks_can_take_arguments
|
|
method_with_block_arguments do |argument|
|
|
assert_equal __, argument
|
|
end
|
|
end
|
|
|
|
# ------------------------------------------------------------------
|
|
|
|
def many_yields
|
|
yield(:peanut)
|
|
yield(:butter)
|
|
yield(:and)
|
|
yield(:jelly)
|
|
end
|
|
|
|
def test_methods_can_call_yield_many_times
|
|
result = []
|
|
many_yields { |item| result << item }
|
|
assert_equal __, result
|
|
end
|
|
|
|
# ------------------------------------------------------------------
|
|
|
|
def yield_tester
|
|
if block_given?
|
|
yield
|
|
else
|
|
:no_block
|
|
end
|
|
end
|
|
|
|
def test_methods_can_see_if_they_have_been_called_with_a_block
|
|
assert_equal __, yield_tester { :with_block }
|
|
assert_equal __, yield_tester
|
|
end
|
|
|
|
# ------------------------------------------------------------------
|
|
|
|
def test_block_can_affect_variables_in_the_code_where_they_are_created
|
|
value = :initial_value
|
|
method_with_block { value = :modified_in_a_block }
|
|
assert_equal __, value
|
|
end
|
|
|
|
def test_blocks_can_be_assigned_to_variables_and_called_explicitly
|
|
add_one = lambda { |n| n + 1 }
|
|
assert_equal __, add_one.call(10)
|
|
|
|
# Alternative calling syntax
|
|
assert_equal __, add_one[10]
|
|
end
|
|
|
|
def test_stand_alone_blocks_can_be_passed_to_methods_expecting_blocks
|
|
make_upper = lambda { |n| n.upcase }
|
|
result = method_with_block_arguments(&make_upper)
|
|
assert_equal __, result
|
|
end
|
|
|
|
# ------------------------------------------------------------------
|
|
|
|
def method_with_explicit_block(&block)
|
|
block.call(10)
|
|
end
|
|
|
|
def test_methods_can_take_an_explicit_block_argument
|
|
assert_equal __, method_with_explicit_block { |n| n * 2 }
|
|
|
|
add_one = lambda { |n| n + 1 }
|
|
assert_equal __, method_with_explicit_block(&add_one)
|
|
end
|
|
|
|
end
|