##// END OF EJS Templates
jro_apps » redmine
RSS
Additional unit tests for project tree....
Jean-Philippe Lang -
r4530:306a47023ce1
parent child
Show More
Side by Side Unified
Context file:
r4530:306a47023ce1 Loading diff...:
Show file before | Show file after | Hide comments
@@ -19,40 +19,100 require File.expand_path('../../test_helper', __FILE__)
19
19
20 class ProjectNestedSetTest < ActiveSupport::TestCase
20 class ProjectNestedSetTest < ActiveSupport::TestCase
21
21
22 def setup
22 context "nested set" do
23 Project.delete_all
23 setup do
24 end
24 Project.delete_all
25
25
26 def test_destroy_root_and_chldren_should_not_mess_up_the_tree
26 @a = Project.create!(:name => 'Project A', :identifier => 'projecta')
27 a = Project.create!(:name => 'Project A', :identifier => 'projecta')
27 @a1 = Project.create!(:name => 'Project A1', :identifier => 'projecta1')
28 a1 = Project.create!(:name => 'Project A1', :identifier => 'projecta1')
28 @a1.set_parent!(@a)
29 a2 = Project.create!(:name => 'Project A2', :identifier => 'projecta2')
29 @a2 = Project.create!(:name => 'Project A2', :identifier => 'projecta2')
30 a1.set_parent!(a)
30 @a2.set_parent!(@a)
31 a2.set_parent!(a)
31
32 b = Project.create!(:name => 'Project B', :identifier => 'projectb')
32 @b = Project.create!(:name => 'Project B', :identifier => 'projectb')
33 b1 = Project.create!(:name => 'Project B1', :identifier => 'projectb1')
33 @b1 = Project.create!(:name => 'Project B1', :identifier => 'projectb1')
34 b1.set_parent!(b)
34 @b1.set_parent!(@b)
35
35 @b11 = Project.create!(:name => 'Project B11', :identifier => 'projectb11')
36 a.reload
36 @b11.set_parent!(@b1)
37 a1.reload
37 @b2 = Project.create!(:name => 'Project B2', :identifier => 'projectb2')
38 a2.reload
38 @b2.set_parent!(@b)
39 b.reload
39
40 b1.reload
40 @c = Project.create!(:name => 'Project C', :identifier => 'projectc')
41
41 @c1 = Project.create!(:name => 'Project C1', :identifier => 'projectc1')
42 assert_equal [nil, 1, 6], [a.parent_id, a.lft, a.rgt]
42 @c1.set_parent!(@c)
43 assert_equal [a.id, 2, 3], [a1.parent_id, a1.lft, a1.rgt]
43
44 assert_equal [a.id, 4, 5], [a2.parent_id, a2.lft, a2.rgt]
44 [@a, @a1, @a2, @b, @b1, @b11, @b2, @c, @c1].each(&:reload)
45 assert_equal [nil, 7, 10], [b.parent_id, b.lft, b.rgt]
45 end
46 assert_equal [b.id, 8, 9], [b1.parent_id, b1.lft, b1.rgt]
47
46
48 assert_difference 'Project.count', -3 do
47 context "#create" do
49 a.destroy
48 should "build valid tree" do
49 assert_nested_set_values({
50 @a => [nil, 1, 6],
51 @a1 => [@a.id, 2, 3],
52 @a2 => [@a.id, 4, 5],
53 @b => [nil, 7, 14],
54 @b1 => [@b.id, 8, 11],
55 @b11 => [@b1.id,9, 10],
56 @b2 => [@b.id,12, 13],
57 @c => [nil, 15, 18],
58 @c1 => [@c.id,16, 17]
59 })
60 end
50 end
61 end
51
62
52 b.reload
63 context "#set_parent!" do
53 b1.reload
64 should "keep valid tree" do
54
65 assert_no_difference 'Project.count' do
55 assert_equal [nil, 1, 4], [b.parent_id, b.lft, b.rgt]
66 Project.find_by_name('Project B1').set_parent!(Project.find_by_name('Project A2'))
56 assert_equal [b.id, 2, 3], [b1.parent_id, b1.lft, b1.rgt]
67 end
68 assert_nested_set_values({
69 @a => [nil, 1, 10],
70 @a2 => [@a.id, 4, 9],
71 @b1 => [@a2.id,5, 8],
72 @b11 => [@b1.id,6, 7],
73 @b => [nil, 11, 14],
74 @c => [nil, 15, 18]
75 })
76 end
77 end
78
79 context "#destroy" do
80 context "a root with children" do
81 should "not mess up the tree" do
82 assert_difference 'Project.count', -4 do
83 Project.find_by_name('Project B').destroy
84 end
85 assert_nested_set_values({
86 @a => [nil, 1, 6],
87 @a1 => [@a.id, 2, 3],
88 @a2 => [@a.id, 4, 5],
89 @c => [nil, 7, 10],
90 @c1 => [@c.id, 8, 9]
91 })
92 end
93 end
94
95 context "a child with children" do
96 should "not mess up the tree" do
97 assert_difference 'Project.count', -2 do
98 Project.find_by_name('Project B1').destroy
99 end
100 assert_nested_set_values({
101 @a => [nil, 1, 6],
102 @b => [nil, 7, 10],
103 @b2 => [@b.id, 8, 9],
104 @c => [nil, 11, 14]
105 })
106 end
107 end
108 end
109 end
110
111 def assert_nested_set_values(h)
112 assert Project.valid?
113 h.each do |project, expected|
114 project.reload
115 assert_equal expected, [project.parent_id, project.lft, project.rgt], "Unexpected nested set values for #{project.name}"
116 end
57 end
117 end
58 end No newline at end of file
118 end
General Comments 0
You need to be logged in to leave comments. Login now