Added support for parent_id setter with blank value

saks · Jul 5, 2010 · 64d11f0 · 64d11f0
1 parent 7e2a125
commit 64d11f0
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/lib/mongoid_acts_as_tree.rb b/lib/mongoid_acts_as_tree.rb
@@ -31,8 +31,14 @@ def acts_as_tree(options = {})
 
 					self.class_eval do
 						define_method "#{parent_id_field}=" do | new_parent_id |
-							new_parent = self.class.find new_parent_id
-							new_parent.children.push self, false
+						  if new_parent_id.present?
+  							new_parent = self.class.find new_parent_id
+  							new_parent.children.push self, false
+  						else
+    						self.write_attribute parent_id_field, nil
+    						self[path_field] = []
+    						self[depth_field] = 0
+						  end
 						end
 					end
 

diff --git a/test/test_tree.rb b/test/test_tree.rb
@@ -89,6 +89,15 @@ class TestMongoidActsAsTree < Test::Unit::TestCase
 			assert more_deep_child.ancestors.include? child
 			assert more_deep_child.ancestors.include? parent
 		end
+
+		should "assign blank parent_id" do
+		  @child_1.parent_id = ''
+		  @child_1.save
+
+		  assert_nil @child_1.reload.parent_id
+		  assert_equal 0, @child_1.depth
+		  assert_equal [], @child_1.path
+	  end
 
     context "node" do
       should "have a root" do