rdoc 3.12.2 → 4.0.0.preview2

data/test/test_rdoc_class_module.rb

@@ -10,9 +10,9 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_add_comment
-    tl1 = RDoc::TopLevel.new 'one.rb'
-    tl2 = RDoc::TopLevel.new 'two.rb'
-    tl3 = RDoc::TopLevel.new 'three.rb'
+    tl1 = @store.add_file 'one.rb'
+    tl2 = @store.add_file 'two.rb'
+    tl3 = @store.add_file 'three.rb'
 
     cm = RDoc::ClassModule.new 'Klass'
     cm.add_comment '# comment 1', tl1
@@ -42,7 +42,7 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_add_comment_stopdoc
-    tl = RDoc::TopLevel.new 'file.rb'
+    tl = @store.add_file 'file.rb'
 
     cm = RDoc::ClassModule.new 'Klass'
     cm.stop_doc
@@ -53,7 +53,7 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_ancestors
-    assert_equal [@parent], @child.ancestors
+    assert_equal [@parent, "Object"], @child.ancestors
   end
 
   def test_comment_equals
@@ -96,13 +96,16 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_each_ancestor
-    ancestors = []
+    assert_equal [@parent], @child.each_ancestor.to_a
+  end
 
-    @child.each_ancestor do |mod|
-      ancestors << mod
-    end
+  def test_each_ancestor_cycle
+    m_incl = RDoc::Include.new 'M', nil
+
+    m = @top_level.add_module RDoc::NormalModule, 'M'
+    m.add_include m_incl
 
-    assert_equal [@parent], ancestors
+    assert_empty m.each_ancestor.to_a
   end
 
   # handle making a short module alias of yourself
@@ -114,7 +117,7 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_from_module_comment
-    tl = RDoc::TopLevel.new 'file.rb'
+    tl = @store.add_file 'file.rb'
     klass = tl.add_class RDoc::NormalModule, 'Klass'
     klass.add_comment 'really a class', tl
 
@@ -124,7 +127,8 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_marshal_dump
-    tl = RDoc::TopLevel.new 'file.rb'
+    @store.path = Dir.tmpdir
+    tl = @store.add_file 'file.rb'
 
     ns = tl.add_module RDoc::NormalModule, 'Namespace'
 
@@ -145,14 +149,26 @@ class TestRDocClassModule < XrefTestCase
     i1 = RDoc::Include.new 'I1', ''
     i1.record_location tl
 
+    e1 = RDoc::Extend.new 'E1', ''
+    e1.record_location tl
+
+    section_comment = RDoc::Comment.new('section comment')
+    section_comment.location = tl
+
+    assert_equal 1, cm.sections.length, 'sanity, default section only'
+    s0 = cm.sections.first
+    s1 = cm.add_section 'section', section_comment
+
     cm.add_attribute a1
     cm.add_attribute a2
     cm.add_method m1
     cm.add_constant c1
     cm.add_include i1
+    cm.add_extend e1
     cm.add_comment 'this is a comment', tl
 
     loaded = Marshal.load Marshal.dump cm
+    loaded.store = @store
 
     assert_equal cm, loaded
 
@@ -167,9 +183,15 @@ class TestRDocClassModule < XrefTestCase
     assert_equal [c1],               loaded.constants
     assert_equal 'Namespace::Klass', loaded.full_name
     assert_equal [i1],               loaded.includes
+    assert_equal [e1],               loaded.extends
     assert_equal [m1],               loaded.method_list
     assert_equal 'Klass',            loaded.name
     assert_equal 'Super',            loaded.superclass
+    assert_equal [tl],               loaded.in_files
+    assert_equal 'Namespace',        loaded.parent.name
+
+    expected = { nil => s0, 'section' => s1 }
+    assert_equal expected, loaded.sections_hash
 
     assert_equal tl, loaded.attributes.first.file
 
@@ -177,11 +199,13 @@ class TestRDocClassModule < XrefTestCase
 
     assert_equal tl, loaded.includes.first.file
 
+    assert_equal tl, loaded.extends.first.file
+
     assert_equal tl, loaded.method_list.first.file
   end
 
   def test_marshal_load_version_0
-    tl = RDoc::TopLevel.new 'file.rb'
+    tl = @store.add_file 'file.rb'
     ns = tl.add_module RDoc::NormalModule, 'Namespace'
     cm = ns.add_class RDoc::NormalClass, 'Klass', 'Super'
 
@@ -190,25 +214,29 @@ class TestRDocClassModule < XrefTestCase
     c = RDoc::Constant.new('C1', nil, '')
     i = RDoc::Include.new('I1', '')
 
+    s0 = cm.sections.first
+
     cm.add_attribute a
     cm.add_method m
     cm.add_constant c
     cm.add_include i
     cm.add_comment 'this is a comment', tl
 
-    loaded = Marshal.load "\x04\bU:\x16RDoc::NormalClass[\x0Ei\x00\"\nKlass" \
-                          "\"\x15Namespace::KlassI\"\nSuper\x06:\x06EF" \
-                          "o:\eRDoc::Markup::Document\x06:\v@parts[\x06" \
-                          "o:\x1CRDoc::Markup::Paragraph\x06;\b[\x06I" \
-                          "\"\x16this is a comment\x06;\x06F[\x06[\aI" \
-                          "\"\aa1\x06;\x06FI\"\aRW\x06;\x06F[\x06[\aI" \
-                          "\"\aC1\x06;\x06Fo;\a\x06;\b[\x00[\x06[\aI" \
-                          "\"\aI1\x06;\x06Fo;\a\x06;\b[\x00[\a[\aI" \
-                          "\"\nclass\x06;\x06F[\b[\a:\vpublic[\x00[\a" \
-                          ":\x0Eprotected[\x00[\a:\fprivate[\x00[\aI" \
-                          "\"\rinstance\x06;\x06F[\b[\a;\n[\x06I" \
+    loaded = Marshal.load "\x04\bU:\x16RDoc::NormalClass[\x0Ei\x00\"\nKlass" +
+                          "\"\x15Namespace::KlassI\"\nSuper\x06:\x06EF" +
+                          "o:\eRDoc::Markup::Document\x06:\v@parts[\x06" +
+                          "o:\x1CRDoc::Markup::Paragraph\x06;\b[\x06I" +
+                          "\"\x16this is a comment\x06;\x06F[\x06[\aI" +
+                          "\"\aa1\x06;\x06FI\"\aRW\x06;\x06F[\x06[\aI" +
+                          "\"\aC1\x06;\x06Fo;\a\x06;\b[\x00[\x06[\aI" +
+                          "\"\aI1\x06;\x06Fo;\a\x06;\b[\x00[\a[\aI" +
+                          "\"\nclass\x06;\x06F[\b[\a:\vpublic[\x00[\a" +
+                          ":\x0Eprotected[\x00[\a:\fprivate[\x00[\aI" +
+                          "\"\rinstance\x06;\x06F[\b[\a;\n[\x06I" +
                           "\"\am1\x06;\x06F[\a;\v[\x00[\a;\f[\x00"
 
+    loaded.store = @store
+
     assert_equal cm, loaded
 
     comment = RDoc::Markup::Document.new(
@@ -222,12 +250,300 @@ class TestRDocClassModule < XrefTestCase
     assert_equal [m],                loaded.method_list
     assert_equal 'Klass',            loaded.name
     assert_equal 'Super',            loaded.superclass
-    assert_equal nil,                loaded.file
+    assert_nil                       loaded.file
+    assert_empty                     loaded.in_files
+    assert_nil                       loaded.parent
+    assert                           loaded.current_section
+
+    expected = { nil => s0 }
+    assert_equal expected, loaded.sections_hash
+  end
+
+  def test_marshal_load_version_1
+    tl = @store.add_file 'file.rb'
+
+    ns = tl.add_module RDoc::NormalModule, 'Namespace'
+
+    cm = ns.add_class RDoc::NormalClass, 'Klass', 'Super'
+    cm.record_location tl
+
+    a1 = RDoc::Attr.new nil, 'a1', 'RW', ''
+    a1.record_location tl
+    a2 = RDoc::Attr.new nil, 'a2', 'RW', '', true
+    a2.record_location tl
+
+    m1 = RDoc::AnyMethod.new nil, 'm1'
+    m1.record_location tl
+
+    c1 = RDoc::Constant.new 'C1', nil, ''
+    c1.record_location tl
+
+    i1 = RDoc::Include.new 'I1', ''
+    i1.record_location tl
+
+    s0 = cm.sections.first
+
+    cm.add_attribute a1
+    cm.add_attribute a2
+    cm.add_method m1
+    cm.add_constant c1
+    cm.add_include i1
+    cm.add_comment 'this is a comment', tl
+
+    loaded = Marshal.load "\x04\bU:\x16RDoc::NormalClass[\x0Ei\x06I\"\nKlass" +
+                          "\x06:\x06EFI\"\x15Namespace::Klass\x06;\x06FI" +
+                          "\"\nSuper\x06;\x06Fo:\eRDoc::Markup::Document\a" +
+                          ":\v@parts[\x06o;\a\a;\b[\x06o" +
+                          ":\x1CRDoc::Markup::Paragraph\x06;\b" +
+                          "[\x06I\"\x16this is a comment\x06;\x06F" +
+                          ":\n@fileI\"\ffile.rb\x06;\x06F;\n0[\a[\nI" +
+                          "\"\aa2\x06;\x06FI\"\aRW\x06;\x06F:\vpublicT@\x11" +
+                          "[\nI\"\aa1\x06;\x06FI\"\aRW\x06;\x06F;\vF@\x11" +
+                          "[\x06[\bI\"\aC1\x06;\x06Fo;\a\a;\b[\x00;\n0@\x11" +
+                          "[\x06[\bI\"\aI1\x06;\x06Fo;\a\a;\b[\x00;\n0@\x11" +
+                          "[\a[\aI\"\nclass\x06;\x06F[\b[\a;\v[\x00" +
+                          "[\a:\x0Eprotected[\x00[\a:\fprivate[\x00[\aI" +
+                          "\"\rinstance\x06;\x06F[\b[\a;\v[\x06[\aI" +
+                          "\"\am1\x06;\x06F@\x11[\a;\f[\x00[\a;\r[\x00"
+
+    loaded.store = @store
+
+    assert_equal cm, loaded
+
+    inner = RDoc::Markup::Document.new(
+      RDoc::Markup::Paragraph.new('this is a comment'))
+    inner.file = tl
+
+    comment = RDoc::Markup::Document.new inner
+
+    assert_equal [a2, a1],           loaded.attributes.sort
+    assert_equal comment,            loaded.comment
+    assert_equal [c1],               loaded.constants
+    assert_equal 'Namespace::Klass', loaded.full_name
+    assert_equal [i1],               loaded.includes
+    assert_empty                     loaded.extends
+    assert_equal [m1],               loaded.method_list
+    assert_equal 'Klass',            loaded.name
+    assert_equal 'Super',            loaded.superclass
+    assert_empty                     loaded.in_files
+    assert_nil                       loaded.parent
+    assert                           loaded.current_section
+
+    assert_equal tl, loaded.attributes.first.file
+    assert_equal tl, loaded.constants.first.file
+    assert_equal tl, loaded.includes.first.file
+    assert_equal tl, loaded.method_list.first.file
+
+    expected = { nil => s0 }
+    assert_equal expected, loaded.sections_hash
+  end
+
+  def test_marshal_load_version_2
+    tl = @store.add_file 'file.rb'
+
+    ns = tl.add_module RDoc::NormalModule, 'Namespace'
+
+    cm = ns.add_class RDoc::NormalClass, 'Klass', 'Super'
+    cm.record_location tl
+
+    a1 = RDoc::Attr.new nil, 'a1', 'RW', ''
+    a1.record_location tl
+    a2 = RDoc::Attr.new nil, 'a2', 'RW', '', true
+    a2.record_location tl
+
+    m1 = RDoc::AnyMethod.new nil, 'm1'
+    m1.record_location tl
+
+    c1 = RDoc::Constant.new 'C1', nil, ''
+    c1.record_location tl
+
+    i1 = RDoc::Include.new 'I1', ''
+    i1.record_location tl
+
+    e1 = RDoc::Extend.new 'E1', ''
+    e1.record_location tl
+
+    s0 = cm.sections.first
+
+    cm.add_attribute a1
+    cm.add_attribute a2
+    cm.add_method m1
+    cm.add_constant c1
+    cm.add_include i1
+    cm.add_extend e1
+    cm.add_comment 'this is a comment', tl
+
+    loaded = Marshal.load "\x04\bU:\x16RDoc::NormalClass[\x0Fi\aI\"\nKlass" +
+                          "\x06:\x06EFI\"\x15Namespace::Klass\x06;\x06FI" +
+                          "\"\nSuper\x06;\x06Fo:\eRDoc::Markup::Document\a" +
+                          ":\v@parts[\x06o;\a\a;\b[\x06o" +
+                          ":\x1CRDoc::Markup::Paragraph\x06;\b" +
+                          "[\x06I\"\x16this is a comment\x06;\x06F" +
+                          ":\n@fileI\"\ffile.rb\x06;\x06F;\n0[\a[\nI" +
+                          "\"\aa2\x06;\x06FI\"\aRW\x06;\x06F:\vpublicT@\x11" +
+                          "[\nI\"\aa1\x06;\x06FI\"\aRW\x06;\x06F;\vF@\x11" +
+                          "[\x06[\bI\"\aC1\x06;\x06Fo;\a\a;\b[\x00;\n0@\x11" +
+                          "[\x06[\bI\"\aI1\x06;\x06Fo;\a\a;\b[\x00;\n0@\x11" +
+                          "[\a[\aI\"\nclass\x06;\x06F[\b[\a;\v[\x00" +
+                          "[\a:\x0Eprotected[\x00[\a:\fprivate[\x00[\aI" +
+                          "\"\rinstance\x06;\x06F[\b[\a;\v[\x06[\aI" +
+                          "\"\am1\x06;\x06F@\x11[\a;\f[\x00[\a;\r[\x00" +
+                          "[\x06[\bI\"\aE1\x06;\x06Fo;\a\a;\b[\x00;\n0@\x11"
+
+    loaded.store = @store
+
+    assert_equal cm, loaded
+
+    inner = RDoc::Markup::Document.new(
+      RDoc::Markup::Paragraph.new('this is a comment'))
+    inner.file = tl
+
+    comment = RDoc::Markup::Document.new inner
+
+    assert_equal [a2, a1],           loaded.attributes.sort
+    assert_equal comment,            loaded.comment
+    assert_equal [c1],               loaded.constants
+    assert_equal 'Namespace::Klass', loaded.full_name
+    assert_equal [i1],               loaded.includes
+    assert_equal [e1],               loaded.extends
+    assert_equal [m1],               loaded.method_list
+    assert_equal 'Klass',            loaded.name
+    assert_equal 'Super',            loaded.superclass
+    assert_empty                     loaded.in_files
+    assert_nil                       loaded.parent
+    assert                           loaded.current_section
+
+    assert_equal tl, loaded.attributes. first.file
+    assert_equal tl, loaded.constants.  first.file
+    assert_equal tl, loaded.includes.   first.file
+    assert_equal tl, loaded.extends.    first.file
+    assert_equal tl, loaded.method_list.first.file
+
+    expected = { nil => s0 }
+    assert_equal expected, loaded.sections_hash
+  end
+
+  def test_marshal_load_version_3
+    tl = @store.add_file 'file.rb'
+
+    ns = tl.add_module RDoc::NormalModule, 'Namespace'
+
+    cm = ns.add_class RDoc::NormalClass, 'Klass', 'Super'
+    cm.record_location tl
+
+    a1 = RDoc::Attr.new nil, 'a1', 'RW', ''
+    a1.record_location tl
+    a2 = RDoc::Attr.new nil, 'a2', 'RW', '', true
+    a2.record_location tl
+
+    m1 = RDoc::AnyMethod.new nil, 'm1'
+    m1.record_location tl
+
+    c1 = RDoc::Constant.new 'C1', nil, ''
+    c1.record_location tl
+
+    i1 = RDoc::Include.new 'I1', ''
+    i1.record_location tl
+
+    e1 = RDoc::Extend.new 'E1', ''
+    e1.record_location tl
+
+    section_comment = RDoc::Comment.new('section comment')
+    section_comment.location = tl
+
+    assert_equal 1, cm.sections.length, 'sanity, default section only'
+    s0 = cm.sections.first
+    s1 = cm.add_section 'section', section_comment
+
+    cm.add_attribute a1
+    cm.add_attribute a2
+    cm.add_method m1
+    cm.add_constant c1
+    cm.add_include i1
+    cm.add_extend e1
+    cm.add_comment 'this is a comment', tl
+
+    loaded = Marshal.load "\x04\bU:\x16RDoc::NormalClass[\x13i\bI\"\nKlass" +
+                          "\x06:\x06ETI\"\x15Namespace::Klass\x06;\x06TI" +
+                          "\"\nSuper\x06;\x06To:\eRDoc::Markup::Document\a" +
+                          ":\v@parts[\x06o;\a\a;\b[\x06o" +
+                          ":\x1CRDoc::Markup::Paragraph\x06;\b[\x06I" +
+                          "\"\x16this is a comment\x06;\x06T:\n@fileI" +
+                          "\"\ffile.rb\x06;\x06T;\n0[\a[\nI\"\aa2\x06;" +
+                          "\x06TI\"\aRW\x06;\x06T:\vpublicT@\x11[\nI" +
+                          "\"\aa1\x06;\x06TI\"\aRW\x06;\x06T;\vF@\x11" +
+                          "[\x06U:\x13RDoc::Constant[\x0Fi\x00I\"\aC1\x06" +
+                          ";\x06TI\"\x19Namespace::Klass::C1\x06;\x06T00o" +
+                          ";\a\a;\b[\x00;\n0@\x11@\ac\x16RDoc::NormalClass0" +
+                          "[\x06[\bI\"\aI1\x06;\x06To;\a\a;\b[\x00;\n0@\x11" +
+                          "[\a[\aI\"\nclass\x06;\x06T[\b[\a;\v[\x00[\a" +
+                          ":\x0Eprotected[\x00[\a:\fprivate[\x00[\aI" +
+                          "\"\rinstance\x06;\x06T[\b[\a;\v[\x06[\aI" +
+                          "\"\am1\x06;\x06T@\x11[\a;\r[\x00[\a;\x0E[\x00" +
+                          "[\x06[\bI\"\aE1\x06;\x06To;\a\a;\b[\x00;\n0@\x11" +
+                          "[\aU:\eRDoc::Context::Section[\bi\x000o;\a\a;\b" +
+                          "[\x00;\n0U;\x0F[\bi\x00I\"\fsection\x06;\x06To" +
+                          ";\a\a;\b[\x06o;\a\a;\b[\x06o;\t\x06;\b[\x06I" +
+                          "\"\x14section comment\x06;\x06T;\n@\x11;\n0" +
+                          "[\x06@\x11I\"\x0ENamespace\x06" +
+                          ";\x06Tc\x17RDoc::NormalModule"
+
+    loaded.store = @store
+
+    assert_equal cm, loaded
+
+    inner = RDoc::Markup::Document.new(
+      RDoc::Markup::Paragraph.new('this is a comment'))
+    inner.file = tl
+
+    comment = RDoc::Markup::Document.new inner
+
+    assert_equal [a2, a1],           loaded.attributes.sort
+    assert_equal comment,            loaded.comment
+    assert_equal [c1],               loaded.constants
+    assert_equal 'Namespace::Klass', loaded.full_name
+    assert_equal [i1],               loaded.includes
+    assert_equal [e1],               loaded.extends
+    assert_equal [m1],               loaded.method_list
+    assert_equal 'Klass',            loaded.name
+    assert_equal 'Super',            loaded.superclass
+    assert_equal 'Namespace',        loaded.parent.name
+    assert                           loaded.current_section
+
+    expected = {
+      nil       => s0,
+      'section' => s1,
+    }
+
+    assert_equal expected,           loaded.sections_hash
+    assert_equal [tl],               loaded.in_files
+
+    assert_equal tl, loaded.attributes. first.file
+    assert_equal tl, loaded.constants.  first.file
+    assert_equal tl, loaded.includes.   first.file
+    assert_equal tl, loaded.extends.    first.file
+    assert_equal tl, loaded.method_list.first.file
+  end
+
+  def test_merge
+    tl = @store.add_file 'one.rb'
+    p1  = tl.add_class RDoc::NormalClass, 'Parent'
+    c1  = p1.add_class RDoc::NormalClass, 'Klass'
+
+    c2 = RDoc::NormalClass.new 'Klass'
+
+    c2.merge c1
+
+    assert_equal 'Parent', c1.parent_name, 'original parent name'
+    assert_equal 'Parent', c2.parent_name, 'merged parent name'
+
+    assert c1.current_section, 'original current_section'
+    assert c2.current_section, 'merged current_section'
   end
 
   def test_merge_attributes
-    tl1 = RDoc::TopLevel.new 'one.rb'
-    tl2 = RDoc::TopLevel.new 'two.rb'
+    tl1 = @store.add_file 'one.rb'
+    tl2 = @store.add_file 'two.rb'
 
     cm1 = RDoc::ClassModule.new 'Klass'
 
@@ -262,7 +578,7 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_merge_attributes_version_0
-    tl1 = RDoc::TopLevel.new 'one.rb'
+    tl1 = @store.add_file 'one.rb'
 
     cm1 = RDoc::ClassModule.new 'Klass'
 
@@ -295,7 +611,7 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_merge_collections_drop
-    tl = RDoc::TopLevel.new 'file'
+    tl = @store.add_file 'file'
 
     cm1 = RDoc::ClassModule.new 'C'
     cm1.record_location tl
@@ -322,17 +638,21 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_merge_comment
-    tl1 = RDoc::TopLevel.new 'one.rb'
-    tl2 = RDoc::TopLevel.new 'two.rb'
+    tl1 = @store.add_file 'one.rb'
+    tl2 = @store.add_file 'two.rb'
 
-    cm1 = RDoc::ClassModule.new 'Klass'
+    cm1 = tl1.add_class RDoc::ClassModule, 'Klass'
     cm1.add_comment 'klass 1', tl1
+    cm1.record_location tl1
 
-    cm2 = RDoc::ClassModule.new 'Klass'
+    cm2 = tl1.add_class RDoc::NormalClass, 'Klass'
     cm2.add_comment 'klass 2', tl2
     cm2.add_comment 'klass 3', tl1
+    cm2.record_location tl1
+    cm2.record_location tl2
 
     cm2 = Marshal.load Marshal.dump cm2
+    cm2.store = @store
 
     cm1.merge cm2
 
@@ -347,7 +667,7 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_merge_comment_version_0
-    tl = RDoc::TopLevel.new 'file.rb'
+    tl = @store.add_file 'file.rb'
 
     cm1 = RDoc::ClassModule.new 'Klass'
     cm1.add_comment 'klass 1', tl
@@ -372,17 +692,19 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_merge_constants
-    tl1 = RDoc::TopLevel.new 'one.rb'
-    tl2 = RDoc::TopLevel.new 'two.rb'
+    tl1 = @store.add_file 'one.rb'
+    tl2 = @store.add_file 'two.rb'
 
-    cm1 = RDoc::ClassModule.new 'Klass'
+    cm1 = tl1.add_class RDoc::ClassModule, 'Klass'
 
     const = cm1.add_constant RDoc::Constant.new('C1', nil, 'one')
     const.record_location tl1
     const = cm1.add_constant RDoc::Constant.new('C3', nil, 'one')
     const.record_location tl1
 
-    cm2 = RDoc::ClassModule.new 'Klass'
+    store = RDoc::Store.new
+    tl = store.add_file 'one.rb'
+    cm2 = tl.add_class RDoc::ClassModule, 'Klass'
     cm2.instance_variable_set :@comment, @RM::Document.new
 
     const = cm2.add_constant RDoc::Constant.new('C2', nil, 'two')
@@ -406,16 +728,18 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_merge_constants_version_0
-    tl1 = RDoc::TopLevel.new 'one.rb'
+    tl1 = @store.add_file 'one.rb'
 
-    cm1 = RDoc::ClassModule.new 'Klass'
+    cm1 = tl1.add_class RDoc::ClassModule, 'Klass'
 
     const = cm1.add_constant RDoc::Constant.new('C1', nil, 'one')
     const.record_location tl1
     const = cm1.add_constant RDoc::Constant.new('C3', nil, 'one')
     const.record_location tl1
 
-    cm2 = RDoc::ClassModule.new 'Klass'
+    store = RDoc::Store.new
+    tl = store.add_file 'one.rb'
+    cm2 = tl.add_class RDoc::ClassModule, 'Klass'
     cm2.instance_variable_set :@comment, @RM::Document.new
 
     const = cm2.add_constant RDoc::Constant.new('C2', nil, 'two')
@@ -436,18 +760,55 @@ class TestRDocClassModule < XrefTestCase
     assert_equal expected, cm1.constants.sort
   end
 
+  def test_merge_extends
+    tl1 = @store.add_file 'one.rb'
+    cm1 = tl1.add_class RDoc::ClassModule, 'Klass'
+
+    ext = cm1.add_extend RDoc::Extend.new('I1', 'one')
+    ext.record_location tl1
+    ext = cm1.add_extend RDoc::Extend.new('I3', 'one')
+    ext.record_location tl1
+
+    tl2 = @store.add_file 'two.rb'
+    tl2.store = RDoc::Store.new
+
+    cm2 = tl2.add_class RDoc::ClassModule, 'Klass'
+    cm2.instance_variable_set :@comment, @RM::Document.new
+
+    ext = cm2.add_extend RDoc::Extend.new('I2', 'two')
+    ext.record_location tl2
+    ext = cm2.add_extend RDoc::Extend.new('I3', 'one')
+    ext.record_location tl1
+    ext = cm2.add_extend RDoc::Extend.new('I4', 'one')
+    ext.record_location tl1
+
+    cm1.merge cm2
+
+    expected = [
+      RDoc::Extend.new('I2', 'two'),
+      RDoc::Extend.new('I3', 'one'),
+      RDoc::Extend.new('I4', 'one'),
+    ]
+
+    expected.each do |a| a.parent = cm1 end
+
+    assert_equal expected, cm1.extends.sort
+  end
+
   def test_merge_includes
-    tl1 = RDoc::TopLevel.new 'one.rb'
-    tl2 = RDoc::TopLevel.new 'two.rb'
+    tl1 = @store.add_file 'one.rb'
 
-    cm1 = RDoc::ClassModule.new 'Klass'
+    cm1 = tl1.add_class RDoc::ClassModule, 'Klass'
 
     incl = cm1.add_include RDoc::Include.new('I1', 'one')
     incl.record_location tl1
     incl = cm1.add_include RDoc::Include.new('I3', 'one')
     incl.record_location tl1
 
-    cm2 = RDoc::ClassModule.new 'Klass'
+    tl2 = @store.add_file 'two.rb'
+    tl2.store = RDoc::Store.new
+
+    cm2 = tl2.add_class RDoc::ClassModule, 'Klass'
     cm2.instance_variable_set :@comment, @RM::Document.new
 
     incl = cm2.add_include RDoc::Include.new('I2', 'two')
@@ -471,16 +832,19 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_merge_includes_version_0
-    tl1 = RDoc::TopLevel.new 'one.rb'
+    tl1 = @store.add_file 'one.rb'
 
-    cm1 = RDoc::ClassModule.new 'Klass'
+    cm1 = tl1.add_class RDoc::ClassModule, 'Klass'
 
     incl = cm1.add_include RDoc::Include.new('I1', 'one')
     incl.record_location tl1
     incl = cm1.add_include RDoc::Include.new('I3', 'one')
     incl.record_location tl1
 
-    cm2 = RDoc::ClassModule.new 'Klass'
+    tl2 = @store.add_file 'one.rb'
+    tl2.store = RDoc::Store.new
+
+    cm2 = tl2.add_class RDoc::ClassModule, 'Klass'
     cm2.instance_variable_set :@comment, @RM::Document.new
 
     incl = cm2.add_include RDoc::Include.new('I2', 'two')
@@ -502,10 +866,10 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_merge_methods
-    tl1 = RDoc::TopLevel.new 'one.rb'
-    tl2 = RDoc::TopLevel.new 'two.rb'
+    tl1 = @store.add_file 'one.rb'
+    tl2 = @store.add_file 'two.rb'
 
-    cm1 = RDoc::ClassModule.new 'Klass'
+    cm1 = tl1.add_class RDoc::NormalClass, 'Klass'
 
     meth = cm1.add_method RDoc::AnyMethod.new(nil, 'm1')
     meth.record_location tl1
@@ -513,6 +877,7 @@ class TestRDocClassModule < XrefTestCase
     meth.record_location tl1
 
     cm2 = RDoc::ClassModule.new 'Klass'
+    cm2.store = @store
     cm2.instance_variable_set :@comment, @RM::Document.new
 
     meth = cm2.add_method RDoc::AnyMethod.new(nil, 'm2')
@@ -536,9 +901,9 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_merge_methods_version_0
-    tl1 = RDoc::TopLevel.new 'one.rb'
+    tl1 = @store.add_file 'one.rb'
 
-    cm1 = RDoc::ClassModule.new 'Klass'
+    cm1 = tl1.add_class RDoc::NormalClass, 'Klass'
 
     meth = cm1.add_method RDoc::AnyMethod.new(nil, 'm1')
     meth.record_location tl1
@@ -546,6 +911,7 @@ class TestRDocClassModule < XrefTestCase
     meth.record_location tl1
 
     cm2 = RDoc::ClassModule.new 'Klass'
+    cm2.store = @store
     cm2.instance_variable_set :@comment, @RM::Document.new
 
     meth = cm2.add_method RDoc::AnyMethod.new(nil, 'm2')
@@ -566,9 +932,111 @@ class TestRDocClassModule < XrefTestCase
     assert_equal expected, cm1.method_list.sort
   end
 
+  def test_merge_sections
+    store1 = @store
+
+    tl1_1 = store1.add_file 'one.rb'
+
+    cm1  = tl1_1.add_class RDoc::ClassModule, 'Klass'
+    cm1.record_location tl1_1
+
+    s1_0 = cm1.sections.first
+    s1_1 = cm1.add_section 'section 1', comment('comment 1',   tl1_1)
+           cm1.add_section 'section 2', comment('comment 2 a', tl1_1)
+           cm1.add_section 'section 4', comment('comment 4 a', tl1_1)
+
+    store2 = RDoc::Store.new
+    tl2_1 = store2.add_file 'one.rb'
+    tl2_2 = store2.add_file 'two.rb'
+
+    cm2  = tl2_1.add_class RDoc::ClassModule, 'Klass'
+    cm2.record_location tl2_1
+    cm2.record_location tl2_2
+
+           cm2.sections.first
+    s2_2 = cm2.add_section 'section 2', comment('comment 2 b', tl2_1)
+    s2_3 = cm2.add_section 'section 3', comment('comment 3',   tl2_2)
+           cm2.add_section 'section 4', comment('comment 4 b', tl2_2)
+
+    cm1.merge cm2
+
+    expected = [
+      s1_0,
+      s1_1,
+      s2_2,
+      s2_3,
+      RDoc::Context::Section.new(cm1, 'section 4', nil)
+    ]
+
+    merged_sections = cm1.sections.sort_by do |s|
+      s.title || ''
+    end
+
+    assert_equal expected, merged_sections
+
+    assert_equal [comment('comment 2 b', tl2_1)],
+                 cm1.sections_hash['section 2'].comments
+
+    expected_s4_comments = [
+      comment('comment 4 a', tl2_1),
+      comment('comment 4 b', tl2_2),
+    ]
+
+    assert_equal expected_s4_comments, cm1.sections_hash['section 4'].comments
+  end
+
+  def test_merge_sections_overlap
+    store1 = @store
+
+    tl1_1 = store1.add_file 'one.rb'
+    tl1_3 = store1.add_file 'three.rb'
+
+    cm1  = tl1_1.add_class RDoc::ClassModule, 'Klass'
+    cm1.record_location tl1_1
+
+    cm1.add_section 'section', comment('comment 1 a', tl1_1)
+    cm1.add_section 'section', comment('comment 3',   tl1_3)
+
+    store2 = RDoc::Store.new
+    tl2_1 = store2.add_file 'one.rb'
+    tl2_2 = store2.add_file 'two.rb'
+    tl2_3 = store2.add_file 'three.rb'
+
+    cm2  = tl2_1.add_class RDoc::ClassModule, 'Klass'
+    cm2.record_location tl2_1
+    cm2.record_location tl2_2
+
+    s2_0 = cm2.sections.first
+    s2_1 = cm2.add_section 'section', comment('comment 1 b', tl1_1)
+           cm2.add_section 'section', comment('comment 2',   tl2_2)
+
+    cm1.merge_sections cm2
+
+    expected = [
+      s2_0,
+      s2_1,
+    ]
+
+    merged_sections = cm1.sections.sort_by do |s|
+      s.title || ''
+    end
+
+    assert_equal expected, merged_sections
+
+    expected = [
+      comment('comment 1 b', tl2_1),
+      comment('comment 3',   tl2_3),
+      comment('comment 2',   tl2_2),
+    ]
+
+    comments = cm1.sections_hash['section'].comments
+
+    assert_equal expected, comments.sort_by { |c| c.file.name }
+  end
+
   def test_parse
-    tl1 = RDoc::TopLevel.new 'one.rb'
-    tl2 = RDoc::TopLevel.new 'two.rb'
+    tl1 = @store.add_file 'one.rb'
+    tl2 = @store.add_file 'two.rb'
 
     cm = RDoc::ClassModule.new 'Klass'
     cm.add_comment 'comment 1', tl1
@@ -585,7 +1053,7 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_parse_comment
-    tl1 = RDoc::TopLevel.new 'one.rb'
+    tl1 = @store.add_file 'one.rb'
 
     cm = RDoc::ClassModule.new 'Klass'
     cm.comment = comment 'comment 1', tl1
@@ -597,7 +1065,7 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_parse_comment_format
-    tl1 = RDoc::TopLevel.new 'one.rb'
+    tl1 = @store.add_file 'one.rb'
 
     cm = RDoc::ClassModule.new 'Klass'
     cm.comment = comment 'comment ((*1*))', tl1
@@ -610,10 +1078,10 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_parse_comment_location
-    tl1 = RDoc::TopLevel.new 'one.rb'
-    tl2 = RDoc::TopLevel.new 'two.rb'
+    tl1 = @store.add_file 'one.rb'
+    tl2 = @store.add_file 'two.rb'
 
-    cm = RDoc::ClassModule.new 'Klass'
+    cm = tl1.add_class RDoc::NormalClass, 'Klass'
     cm.add_comment 'comment 1', tl1
     cm.add_comment 'comment 2', tl2
 
@@ -628,12 +1096,12 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_remove_nodoc_children
-    parent = RDoc::ClassModule.new 'A'
+    parent = @top_level.add_class RDoc::ClassModule, 'A'
     parent.modules_hash.replace 'B' => true, 'C' => true
-    RDoc::TopLevel.all_modules_hash.replace 'A::B' => true
+    @store.modules_hash.replace 'A::B' => true
 
     parent.classes_hash.replace 'D' => true, 'E' => true
-    RDoc::TopLevel.all_classes_hash.replace 'A::D' => true
+    @store.classes_hash.replace 'A::D' => true
 
     parent.remove_nodoc_children
 
@@ -658,8 +1126,8 @@ class TestRDocClassModule < XrefTestCase
   end
 
   def test_search_record_merged
-    @c2_c3.add_comment 'comment A', RDoc::TopLevel.new('a.rb')
-    @c2_c3.add_comment 'comment B', RDoc::TopLevel.new('b.rb')
+    @c2_c3.add_comment 'comment A', @store.add_file('a.rb')
+    @c2_c3.add_comment 'comment B', @store.add_file('b.rb')
 
     expected = [
       'C3',
@@ -674,6 +1142,50 @@ class TestRDocClassModule < XrefTestCase
     assert_equal expected, @c2_c3.search_record
   end
 
+  def test_store_equals
+    # version 2
+    loaded = Marshal.load "\x04\bU:\x16RDoc::NormalClass[\x0Fi\aI\"\nKlass" +
+                          "\x06:\x06EFI\"\x15Namespace::Klass\x06;\x06FI" +
+                          "\"\nSuper\x06;\x06Fo:\eRDoc::Markup::Document\a" +
+                          ":\v@parts[\x06o;\a\a;\b[\x06o" +
+                          ":\x1CRDoc::Markup::Paragraph\x06;\b" +
+                          "[\x06I\"\x16this is a comment\x06;\x06F" +
+                          ":\n@fileI\"\ffile.rb\x06;\x06F;\n0[\a[\nI" +
+                          "\"\aa2\x06;\x06FI\"\aRW\x06;\x06F:\vpublicT@\x11" +
+                          "[\nI\"\aa1\x06;\x06FI\"\aRW\x06;\x06F;\vF@\x11" +
+                          "[\x06[\bI\"\aC1\x06;\x06Fo;\a\a;\b[\x00;\n0@\x11" +
+                          "[\x06[\bI\"\aI1\x06;\x06Fo;\a\a;\b[\x00;\n0@\x11" +
+                          "[\a[\aI\"\nclass\x06;\x06F[\b[\a;\v[\x00" +
+                          "[\a:\x0Eprotected[\x00[\a:\fprivate[\x00[\aI" +
+                          "\"\rinstance\x06;\x06F[\b[\a;\v[\x06[\aI" +
+                          "\"\am1\x06;\x06F@\x11[\a;\f[\x00[\a;\r[\x00" +
+                          "[\x06[\bI\"\aE1\x06;\x06Fo;\a\a;\b[\x00;\n0@\x11"
+
+    loaded.store = @store
+
+    assert_same @store, loaded.store
+
+    a = loaded.attributes.first
+    assert_same @store, a.store
+    assert_same @store, a.file.store
+
+    c = loaded.constants.first
+    assert_same @store, c.store
+    assert_same @store, c.file.store
+
+    i = loaded.includes.first
+    assert_same @store, i.store
+    assert_same @store, i.file.store
+
+    e = loaded.extends.first
+    assert_same @store, e.store
+    assert_same @store, e.file.store
+
+    m = loaded.method_list.first
+    assert_same @store, m.store
+    assert_same @store, m.file.store
+  end
+
   def test_superclass
     assert_equal @c3_h1, @c3_h2.superclass
   end
@@ -751,6 +1263,30 @@ class TestRDocClassModule < XrefTestCase
     assert_equal 'O1::A1', o1_a1_m.full_name
   end
 
+  def test_update_aliases_reparent_root
+    store = RDoc::Store.new
+
+    top_level = store.add_file 'file.rb'
+
+    klass  = top_level.add_class RDoc::NormalClass, 'Klass'
+    object = top_level.add_class RDoc::NormalClass, 'Object'
+
+    const = RDoc::Constant.new 'A', nil, ''
+    const.record_location top_level
+    const.is_alias_for = klass
+
+    top_level.add_module_alias klass, 'A', top_level
+
+    object.add_constant const
+
+    object.update_aliases
+
+    assert_equal %w[A Klass Object], store.classes_hash.keys.sort
+
+    assert_equal 'A',     store.classes_hash['A'].full_name
+    assert_equal 'Klass', store.classes_hash['Klass'].full_name
+  end
+
   def test_update_includes
     a = RDoc::Include.new 'M1', nil
     b = RDoc::Include.new 'M2', nil
@@ -764,10 +1300,11 @@ class TestRDocClassModule < XrefTestCase
     @m1_m2.document_self = nil
     assert @m1_m2.remove_from_documentation?
 
-    assert RDoc::TopLevel.all_modules_hash.key? @m1_m2.full_name
-    refute RDoc::TopLevel.all_modules_hash[@m1_m2.full_name].nil?
-    RDoc::TopLevel.remove_nodoc RDoc::TopLevel.all_modules_hash
-    refute RDoc::TopLevel.all_modules_hash.key? @m1_m2.full_name
+    assert @store.modules_hash.key? @m1_m2.full_name
+    refute @store.modules_hash[@m1_m2.full_name].nil?
+
+    @store.remove_nodoc @store.modules_hash
+    refute @store.modules_hash.key? @m1_m2.full_name
 
     @c1.update_includes
 
@@ -800,15 +1337,75 @@ class TestRDocClassModule < XrefTestCase
     @m1_m2.document_self = nil
     assert @m1_m2.remove_from_documentation?
 
-    assert RDoc::TopLevel.all_modules_hash.key? @m1_m2.full_name
-    refute RDoc::TopLevel.all_modules_hash[@m1_m2.full_name].nil?
-    RDoc::TopLevel.remove_nodoc RDoc::TopLevel.all_modules_hash
-    refute RDoc::TopLevel.all_modules_hash.key? @m1_m2.full_name
+    assert @store.modules_hash.key? @m1_m2.full_name
+    refute @store.modules_hash[@m1_m2.full_name].nil?
+    @store.remove_nodoc @store.modules_hash
+    refute @store.modules_hash.key? @m1_m2.full_name
 
     @c1.update_includes
 
     assert_equal [a, c], @c1.includes
   end
 
+  def test_update_extends
+    a = RDoc::Extend.new 'M1', nil
+    b = RDoc::Extend.new 'M2', nil
+    c = RDoc::Extend.new 'C', nil
+
+    @c1.add_extend a
+    @c1.add_extend b
+    @c1.add_extend c
+    @c1.each_extend do |extend| extend.module end # cache extended modules
+
+    @m1_m2.document_self = nil
+    assert @m1_m2.remove_from_documentation?
+
+    assert @store.modules_hash.key? @m1_m2.full_name
+    refute @store.modules_hash[@m1_m2.full_name].nil?
+    @store.remove_nodoc @store.modules_hash
+    refute @store.modules_hash.key? @m1_m2.full_name
+
+    @c1.update_extends
+
+    assert_equal [a, c], @c1.extends
+  end
+
+  def test_update_extends_trim
+    a = RDoc::Extend.new 'D::M', nil
+    b = RDoc::Extend.new 'D::M', nil
+
+    @c1.add_extend a
+    @c1.add_extend b
+    @c1.each_extend do |extend| extend.module end # cache extended modules
+
+    @c1.update_extends
+
+    assert_equal [a], @c1.extends
+  end
+
+  def test_update_extends_with_colons
+    a = RDoc::Extend.new 'M1', nil
+    b = RDoc::Extend.new 'M1::M2', nil
+    c = RDoc::Extend.new 'C', nil
+
+    @c1.add_extend a
+    @c1.add_extend b
+    @c1.add_extend c
+    @c1.each_extend do |extend| extend.module end # cache extended modules
+
+    @m1_m2.document_self = nil
+    assert @m1_m2.remove_from_documentation?
+
+    assert @store.modules_hash.key? @m1_m2.full_name
+    refute @store.modules_hash[@m1_m2.full_name].nil?
+
+    @store.remove_nodoc @store.modules_hash
+    refute @store.modules_hash.key? @m1_m2.full_name
+
+    @c1.update_extends
+
+    assert_equal [a, c], @c1.extends
+  end
+
 end