Allow different headers in check

draios · Jan 22, 2024 · 35f2e45 · 35f2e45
1 parent 87812d3
commit 35f2e45
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 2 deletions.
diff --git a/mark2confluence/main.py b/mark2confluence/main.py
@@ -100,7 +100,7 @@ def publish(path: str)-> tuple:
 
 
 def has_mark_headers(path: str) -> bool:
-  space_re = re.compile("<!--.?[Ss]pace:.*-->", re.MULTILINE)
+  space_re = re.compile("<!--.?(space|parent|title):.*-->", re.IGNORECASE)
   with open(path, 'r+') as f:
     data = f.read().split("\n")
     for line in data:

diff --git a/tests/resources/markdown/test_has_mark_headers/with_mark_parent_header.md b/tests/resources/markdown/test_has_mark_headers/with_mark_parent_header.md
@@ -0,0 +1,5 @@
+<!-- Parent: Football -->
+<!-- Parent: New Team -->
+<!-- Title: I have Mark Lenders -->
+
+And you don't.
diff --git a/...est_has_mark_headers/with_mark_headers.md → ...as_mark_headers/with_mark_space_header.md b/...est_has_mark_headers/with_mark_headers.md → ...as_mark_headers/with_mark_space_header.md
diff --git a/tests/resources/markdown/test_has_mark_headers/with_mark_title_header.md b/tests/resources/markdown/test_has_mark_headers/with_mark_title_header.md
@@ -0,0 +1,3 @@
+<!-- Title: I have Mark Lenders -->
+
+And you don't.
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -34,7 +34,9 @@ def test_load_env_prefixes():
 
 def test_has_mark_headers():
     resource_directory = f"{RESOURCE_DIR}/markdown/test_has_mark_headers"
-    assert main.has_mark_headers(f"{resource_directory}/with_mark_headers.md")
+    assert main.has_mark_headers(f"{resource_directory}/with_mark_space_header.md")
+    assert main.has_mark_headers(f"{resource_directory}/with_mark_parent_header.md")
+    assert main.has_mark_headers(f"{resource_directory}/with_mark_title_header.md")
     assert not main.has_mark_headers(f"{resource_directory}/without_mark_headers.md")
 
 def test_check_header_template():