From 1c701717d10879d492d8b55fb8ca2f1c53d7e13f Mon Sep 17 00:00:00 2001 From: Dimid Duchovny Date: Tue, 23 Jan 2018 22:34:22 +0200 Subject: add callback to Msgthr#add The motivation is to allow the client to have a custom code executed, whenever a child is added. Signed-off-by: Dimid Duchovny --- test/test_msgthr.rb | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) (limited to 'test/test_msgthr.rb') diff --git a/test/test_msgthr.rb b/test/test_msgthr.rb index b14e135..0f31762 100644 --- a/test/test_msgthr.rb +++ b/test/test_msgthr.rb @@ -6,8 +6,15 @@ require 'msgthr' class TestMsgthr < Test::Unit::TestCase def test_msgthr thr = Msgthr.new + parent_child = '' + # Note that C is added after B, + # hence it's message will be empty after adding B + expected_parent_child = '->B' thr.add('a', %w(c b), 'abc') - thr.add('b', %w(c), 'B') + thr.add('b', %w(c), 'B') do |parent, child| + parent_child = "#{parent.msg}->#{child.msg}" + end + assert_equal parent_child, expected_parent_child thr.add('c', nil, 'c') thr.add('D', nil, 'D') thr.add('d', %w(missing), 'd') -- cgit v1.2.3-24-ge0c7