Commit e2a2033228c615b47c01ef18644704c5d65175d5
1 parent
0da926a2
Exists in
master
and in
29 other branches
ActionItem23: adding traversal method to acts_as_filesystem
git-svn-id: https://svn.colivre.coop.br/svn/noosfero/trunk@1120 3f533792-8f58-4932-b0fe-aaf55b0a4547
Showing
2 changed files
with
44 additions
and
0 deletions
Show diff stats
lib/acts_as_filesystem.rb
@@ -150,6 +150,21 @@ module ActsAsFileSystem | @@ -150,6 +150,21 @@ module ActsAsFileSystem | ||
150 | @hierarchy | 150 | @hierarchy |
151 | end | 151 | end |
152 | 152 | ||
153 | + def map_traversal(&block) | ||
154 | + stack = [] | ||
155 | + result = [] | ||
156 | + stack.push(self) | ||
157 | + while !stack.empty? | ||
158 | + element = stack.pop | ||
159 | + result.push(element) | ||
160 | + element.children.each do |item| | ||
161 | + stack.push(item) | ||
162 | + end | ||
163 | + end | ||
164 | + block ||= (lambda { |x| x }) | ||
165 | + result.map(&block) | ||
166 | + end | ||
167 | + | ||
153 | end | 168 | end |
154 | end | 169 | end |
155 | 170 |
test/unit/acts_as_filesystem_test.rb
@@ -21,4 +21,33 @@ class ActsAsFilesystemTest < Test::Unit::TestCase | @@ -21,4 +21,33 @@ class ActsAsFilesystemTest < Test::Unit::TestCase | ||
21 | assert_not_same list, a.hierarchy(true) | 21 | assert_not_same list, a.hierarchy(true) |
22 | end | 22 | end |
23 | 23 | ||
24 | + should 'list the full tree' do | ||
25 | + profile = create_user('testinguser').person | ||
26 | + | ||
27 | + a1 = profile.articles.build(:name => 'a1'); a1.save! | ||
28 | + | ||
29 | + a1_1 = profile.articles.build(:name => 'a1.1'); a1_1.parent = a1; a1_1.save! | ||
30 | + a1_2 = profile.articles.build(:name => 'a1.2'); a1_2.parent = a1; a1_2.save! | ||
31 | + | ||
32 | + a1_1_1 = profile.articles.build(:name => 'a1.1.1'); a1_1_1.parent = a1_1; a1_1_1.save! | ||
33 | + a1_1_2 = profile.articles.build(:name => 'a1.1.2'); a1_1_2.parent = a1_1; a1_1_2.save! | ||
34 | + | ||
35 | + assert_equivalent [a1, a1_1, a1_2, a1_1_1, a1_1_2], a1.map_traversal | ||
36 | + end | ||
37 | + | ||
38 | + should 'be able to traverse with a block' do | ||
39 | + profile = create_user('testinguser').person | ||
40 | + | ||
41 | + a1 = profile.articles.build(:name => 'a1'); a1.save! | ||
42 | + | ||
43 | + a1_1 = profile.articles.build(:name => 'a1.1'); a1_1.parent = a1; a1_1.save! | ||
44 | + a1_2 = profile.articles.build(:name => 'a1.2'); a1_2.parent = a1; a1_2.save! | ||
45 | + | ||
46 | + a1_1_1 = profile.articles.build(:name => 'a1.1.1'); a1_1_1.parent = a1_1; a1_1_1.save! | ||
47 | + a1_1_2 = profile.articles.build(:name => 'a1.1.2'); a1_1_2.parent = a1_1; a1_1_2.save! | ||
48 | + | ||
49 | + assert_equivalent ['a1', 'a1.1', 'a1.2', 'a1.1.1', 'a1.1.2'], a1.map_traversal { |item| item.name } | ||
50 | + | ||
51 | + end | ||
52 | + | ||
24 | end | 53 | end |