1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
|
# Redmine - project management software
# Copyright (C) 2006-2012 Jean-Philippe Lang
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
require File.expand_path('../../test_helper', __FILE__)
class ProjectNestedSetTest < ActiveSupport::TestCase
def setup
Project.delete_all
@a = Project.create!(:name => 'A', :identifier => 'projecta')
@a1 = Project.create!(:name => 'A1', :identifier => 'projecta1')
@a1.set_parent!(@a)
@a2 = Project.create!(:name => 'A2', :identifier => 'projecta2')
@a2.set_parent!(@a)
@b = Project.create!(:name => 'B', :identifier => 'projectb')
@b1 = Project.create!(:name => 'B1', :identifier => 'projectb1')
@b1.set_parent!(@b)
@b11 = Project.create!(:name => 'B11', :identifier => 'projectb11')
@b11.set_parent!(@b1)
@b2 = Project.create!(:name => 'B2', :identifier => 'projectb2')
@b2.set_parent!(@b)
@c = Project.create!(:name => 'C', :identifier => 'projectc')
@c1 = Project.create!(:name => 'C1', :identifier => 'projectc1')
@c1.set_parent!(@c)
@a, @a1, @a2, @b, @b1, @b11, @b2, @c, @c1 = *(Project.all.sort_by(&:name))
end
def test_valid_tree
assert_valid_nested_set
end
def test_moving_a_child_to_a_different_parent_should_keep_valid_tree
assert_no_difference 'Project.count' do
Project.find_by_name('B1').set_parent!(Project.find_by_name('A2'))
end
assert_valid_nested_set
end
def test_renaming_a_root_to_first_position_should_update_nested_set_order
@c.name = '1'
@c.save!
assert_valid_nested_set
end
def test_renaming_a_root_to_middle_position_should_update_nested_set_order
@a.name = 'BA'
@a.save!
assert_valid_nested_set
end
def test_renaming_a_root_to_last_position_should_update_nested_set_order
@a.name = 'D'
@a.save!
assert_valid_nested_set
end
def test_renaming_a_root_to_same_position_should_update_nested_set_order
@c.name = 'D'
@c.save!
assert_valid_nested_set
end
def test_renaming_a_child_should_update_nested_set_order
@a1.name = 'A3'
@a1.save!
assert_valid_nested_set
end
def test_renaming_a_child_with_child_should_update_nested_set_order
@b1.name = 'B3'
@b1.save!
assert_valid_nested_set
end
def test_adding_a_root_to_first_position_should_update_nested_set_order
project = Project.create!(:name => '1', :identifier => 'projectba')
assert_valid_nested_set
end
def test_adding_a_root_to_middle_position_should_update_nested_set_order
project = Project.create!(:name => 'BA', :identifier => 'projectba')
assert_valid_nested_set
end
def test_adding_a_root_to_last_position_should_update_nested_set_order
project = Project.create!(:name => 'Z', :identifier => 'projectba')
assert_valid_nested_set
end
def test_destroying_a_root_with_children_should_keep_valid_tree
assert_difference 'Project.count', -4 do
Project.find_by_name('B').destroy
end
assert_valid_nested_set
end
def test_destroying_a_child_with_children_should_keep_valid_tree
assert_difference 'Project.count', -2 do
Project.find_by_name('B1').destroy
end
assert_valid_nested_set
end
private
def assert_nested_set_values(h)
assert Project.valid?
h.each do |project, expected|
project.reload
assert_equal expected, [project.parent_id, project.lft, project.rgt], "Unexpected nested set values for #{project.name}"
end
end
def assert_valid_nested_set
projects = Project.all
lft_rgt = projects.map {|p| [p.lft, p.rgt]}.flatten
assert_equal projects.size * 2, lft_rgt.uniq.size
assert_equal 1, lft_rgt.min
assert_equal projects.size * 2, lft_rgt.max
projects.each do |project|
# lft should always be < rgt
assert project.lft < project.rgt, "lft=#{project.lft} was not < rgt=#{project.rgt} for project #{project.name}"
if project.parent_id
# child lft/rgt values must be greater/lower
assert_not_nil project.parent, "parent was nil for project #{project.name}"
assert project.lft > project.parent.lft, "lft=#{project.lft} was not > parent.lft=#{project.parent.lft} for project #{project.name}"
assert project.rgt < project.parent.rgt, "rgt=#{project.rgt} was not < parent.rgt=#{project.parent.rgt} for project #{project.name}"
end
# no overlapping lft/rgt values
overlapping = projects.detect {|other|
other != project && (
(other.lft > project.lft && other.lft < project.rgt && other.rgt > project.rgt) ||
(other.rgt > project.lft && other.rgt < project.rgt && other.lft < project.lft)
)
}
assert_nil overlapping, (overlapping && "Project #{overlapping.name} (#{overlapping.lft}/#{overlapping.rgt}) overlapped #{project.name} (#{project.lft}/#{project.rgt})")
end
# root projects sorted alphabetically
assert_equal Project.roots.map(&:name).sort, Project.roots.sort_by(&:lft).map(&:name), "Root projects were not properly sorted"
projects.each do |project|
if project.children.any?
# sibling projects sorted alphabetically
assert_equal project.children.map(&:name).sort, project.children.order('lft').map(&:name), "Project #{project.name}'s children were not properly sorted"
end
end
end
end
|
