Merge ed30e8a into 9ae506d

evolve75 · Jan 28, 2015 · dde5383 · dde5383
2 parents 9ae506d + ed30e8a
commit dde5383
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/lib/tree.rb b/lib/tree.rb
@@ -230,7 +230,7 @@ def initialize(name, content = nil)
     #
     # @return [Tree::TreeNode] A copy of this node.
     def detached_copy
-      Tree::TreeNode.new(@name, @content ? @content.clone : nil)
+      self.class.new(@name, @content ? @content.clone : nil)
     end
 
     # Returns a copy of entire (sub-)tree from this node.

diff --git a/test/test_subclassed_node.rb b/test/test_subclassed_node.rb
@@ -64,6 +64,11 @@ def test_subclassed_camelcase_methods
 
     end
 
+    def test_subclassed_detached_copy_is_same_class
+      root = MyNode.new("Root")
+      assert_equal(MyNode, root.detached_copy.class)
+    end
+
   end
 end