Docserver: Fix the TOC rendering for API references by fixing the haphazard way

chrome/common/extensions/docs/server2/document_parser_test.py

 #!/usr/bin/env python
 # Copyright 2013 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import unittest
 
 from document_parser import ParseDocument, RemoveTitle
 
 
 _WHOLE_DOCUMENT = '''
 Preamble before heading.
 
 <h1 id='main' class='header'>Main header</h1>
 Some intro to the content.
 
-<h2 id='banana' class='header'>Bananas</h2>
+<h2 id='banana' class='header' title=''>Bananas</h2>
 Something about bananas.
 
-<h2 id='orange'>Oranges</h2>
+<h2 id='orange' title='hello'>Oranges</h2>
 Something about oranges.
 
 <h3 id='valencia'>Valencia Oranges</h3>
 A description of valencia oranges.
 
 <h3 id='seville'>Seville Oranges</h3>
 A description of seville oranges.
 
 <h2>Grapefruit</h3>
 Grapefruit closed a h2 with a h3. This should be a warning.
@@ skipping 17 matching lines @@
 Header 5s are not parsed.
 '''
 
 
 _WHOLE_DOCUMENT_WITHOUT_TITLE = '''
 Preamble before heading.
 
 
 Some intro to the content.
 
-<h2 id='banana' class='header'>Bananas</h2>
+<h2 id='banana' class='header' title=''>Bananas</h2>
 Something about bananas.
 
-<h2 id='orange'>Oranges</h2>
+<h2 id='orange' title='hello'>Oranges</h2>
 Something about oranges.
 
 <h3 id='valencia'>Valencia Oranges</h3>
 A description of valencia oranges.
 
 <h3 id='seville'>Seville Oranges</h3>
 A description of seville oranges.
 
 <h2>Grapefruit</h3>
 Grapefruit closed a h2 with a h3. This should be a warning.
@@ skipping 86 matching lines @@
       'Found <h3> in the middle of processing a <h2> (line 25, column 9)',
       # TODO(kalman): Re-enable this warning once the reference pages have
       # their references fixed.
       #'Found <h4> without any preceding <h3> (line 28, column 1)',
     ], result.warnings)
 
     # The non-trivial table of contents assertions...
     self.assertEqual(1, len(result.sections))
     entries = result.sections[0].structure
 
-    self.assertEqual(5, len(entries), entries)
-    entry0, entry1, entry2, entry3, entry4 = entries
+    self.assertEqual(4, len(entries), entries)
+    entry0, entry1, entry2, entry3 = entries
 
-    self.assertEqual('Bananas', entry0.name)
-    self.assertEqual({'id': 'banana', 'class': 'header'}, entry0.attributes)
-    self.assertEqual([], entry0.entries)
+    self.assertEqual('hello', entry0.name)
+    self.assertEqual({'id': 'orange'}, entry0.attributes)
+    self.assertEqual(2, len(entry0.entries))
+    entry0_0, entry0_1 = entry0.entries
 
-    self.assertEqual('Oranges', entry1.name)
-    self.assertEqual({'id': 'orange'}, entry1.attributes)
-    self.assertEqual(2, len(entry1.entries))
-    entry1_0, entry1_1 = entry1.entries
+    self.assertEqual('Valencia Oranges', entry0_0.name)
+    self.assertEqual({'id': 'valencia'}, entry0_0.attributes)
+    self.assertEqual([], entry0_0.entries)
+    self.assertEqual('Seville Oranges', entry0_1.name)
+    self.assertEqual({'id': 'seville'}, entry0_1.attributes)
+    self.assertEqual([], entry0_1.entries)
 
-    self.assertEqual('Valencia Oranges', entry1_0.name)
-    self.assertEqual({'id': 'valencia'}, entry1_0.attributes)
-    self.assertEqual([], entry1_0.entries)
-    self.assertEqual('Seville Oranges', entry1_1.name)
-    self.assertEqual({'id': 'seville'}, entry1_1.attributes)
-    self.assertEqual([], entry1_1.entries)
+    self.assertEqual('Grapefruit', entry1.name)
+    self.assertEqual({}, entry1.attributes)
+    self.assertEqual([], entry1.entries)
 
-    self.assertEqual('Grapefruit', entry2.name)
-    self.assertEqual({}, entry2.attributes)
+    self.assertEqual('Not the main header', entry2.name)
+    self.assertEqual({'id': 'not-main'}, entry2.attributes)
     self.assertEqual([], entry2.entries)
 
-    self.assertEqual('Not the main header', entry3.name)
-    self.assertEqual({'id': 'not-main'}, entry3.attributes)
-    self.assertEqual([], entry3.entries)
+    self.assertEqual('Not a banana', entry3.name)
+    self.assertEqual({}, entry3.attributes)
+    self.assertEqual(2, len(entry3.entries))
+    entry3_1, entry3_2 = entry3.entries
 
-    self.assertEqual('Not a banana', entry4.name)
-    self.assertEqual({}, entry4.attributes)
-    self.assertEqual(2, len(entry4.entries))
-    entry4_1, entry4_2 = entry4.entries
+    self.assertEqual('It\'s a h4', entry3_1.name)
+    self.assertEqual({}, entry3_1.attributes)
+    self.assertEqual([], entry3_1.entries)
 
-    self.assertEqual('It\'s a h4', entry4_1.name)
-    self.assertEqual({}, entry4_1.attributes)
-    self.assertEqual([], entry4_1.entries)
+    self.assertEqual('Plantains', entry3_2.name)
+    self.assertEqual({}, entry3_2.attributes)
+    self.assertEqual(1, len(entry3_2.entries))
+    entry3_2_1, = entry3_2.entries
 
-    self.assertEqual('Plantains', entry4_2.name)
-    self.assertEqual({}, entry4_2.attributes)
-    self.assertEqual(1, len(entry4_2.entries))
-    entry4_2_1, = entry4_2.entries
-
-    self.assertEqual('Another h4', entry4_2_1.name)
-    self.assertEqual({}, entry4_2_1.attributes)
-    self.assertEqual([], entry4_2_1.entries)
+    self.assertEqual('Another h4', entry3_2_1.name)
+    self.assertEqual({}, entry3_2_1.attributes)
+    self.assertEqual([], entry3_2_1.entries)
 
   def testSingleExplicitSection(self):
     def test(document):
       result = ParseDocument(document, expect_title=True)
       self.assertEqual([], result.warnings)
       self.assertEqual('Header', result.title)
       self.assertEqual(1, len(result.sections))
       section0, = result.sections
       entry0, = section0.structure
       self.assertEqual('An inner header', entry0.name)
@@ skipping 30 matching lines @@
     def assert_single_header(section, name):
       self.assertEqual(1, len(section.structure))
       self.assertEqual(name, section.structure[0].name)
     assert_single_header(section0, 'First header')
     assert_single_header(section1, 'Second header')
     assert_single_header(section2, 'Third header')
 
 
 if __name__ == '__main__':
   unittest.main()