Adds support to update/fix headings with wrong capitalization

vulder · vulder · commit ef2332c97659 · 2021-02-12T19:48:35.000+01:00
diff --git a/tools/tests/topic_updater_test.py b/tools/tests/topic_updater_test.py
@@ -107,6 +107,28 @@ def test_heading_lookup(self):
             missing_mid.meta_text[0].rstrip(),
             "_ This section could be missing in the middle of the topics _")
 
+    def test_heading_lookup_wrong_capitalization(self):
+        """
+        Checks if we can lookup different section headings in the skeleton,
+        even if the original text is wrongly capitialized.
+        """
+        wrong_upper_case_heading = self.test_skeleton.lookup_heading(
+            "## Section with User content:")
+        wrong_lower_case_heading = self.test_skeleton.lookup_heading(
+            "## missing mid section")
+
+        self.assertEqual(wrong_upper_case_heading.header_text,
+                         "## Section with user content: provided by the user")
+        self.assertEqual(
+            wrong_upper_case_heading.meta_text[0].rstrip(),
+            "_ Example section where user text is in the header _")
+
+        self.assertEqual(wrong_lower_case_heading.header_text,
+                         "## Missing mid section")
+        self.assertEqual(
+            wrong_lower_case_heading.meta_text[0].rstrip(),
+            "_ This section could be missing in the middle of the topics _")
+
     def test_title_heading_lookup(self):
         """
         Checks if we get the correct title heading from the test skeleton.
diff --git a/topic_updater.py b/topic_updater.py
@@ -102,8 +102,8 @@ def lookup_heading(self, start_heading_line: str) -> SectionHeading:
         Returns: the section heading that starts like the heading line
         """
         for heading in self.headings:
-            if start_heading_line.startswith(
-                    heading.header_text.split(":")[0]):
+            if start_heading_line.lower().startswith(
+                    heading.header_text.split(":")[0].lower()):
                 return heading
         raise LookupError(
             f"Could not find heading that starts with: {start_heading_line}")
