Test textwrap_body, current_date and sortable_datetime

python · hugovk · Jan 15, 2025 · Jan 7, 2025 · Jan 7, 2025 · Jan 7, 2025
commit 156988c6d9366ef750e3416f23a3512b25497c49
diff --git a/pyproject.toml b/pyproject.toml
@@ -34,6 +34,7 @@ optional-dependencies.tests = [
   "pyfakefs",
   "pytest",
   "pytest-cov",
+  "time-machine",
 ]
 urls.Changelog = "https://github.com/python/blurb/blob/main/CHANGELOG.md"
 urls.Homepage = "https://github.com/python/blurb"

diff --git a/tests/test_blurb.py b/tests/test_blurb.py
@@ -1,4 +1,5 @@
 import pytest
+import time_machine
 
 from blurb import blurb
 
@@ -44,6 +45,66 @@ def test_unsanitize_section_changed(section, expected):
     assert unsanitized == expected
 
 
+@pytest.mark.parametrize(
+    "body, subsequent_indent, expected",
+    (
+        (
+            "This is a test of the textwrap_body function with a string. It should wrap the text to 79 characters.",
+            "",
+            (
+                "This is a test of the textwrap_body function with a string. It should wrap\n"
+                "the text to 79 characters.\n"
+            ),
+        ),
+        (
+            [
+                "This is a test of the textwrap_body function",
+                "with an iterable of strings.",
+                "It should wrap the text to 79 characters.",
+            ],
+            "",
+            (
+                "This is a test of the textwrap_body function with an iterable of strings. It\n"
+                "should wrap the text to 79 characters.\n"
+            ),
+        ),
+        (
+            "This is a test of the textwrap_body function with a string and subsequent indent.",
+            "    ",
+            (
+                "This is a test of the textwrap_body function with a string and subsequent\n"
+                "    indent.\n"
+            ),
+        ),
+        (
+            "This is a test of the textwrap_body function with a bullet list and subsequent indent. The list should not be wrapped.\n"
+            "\n"
+            "* Item 1\n"
+            "* Item 2\n",
+            "    ",
+            (
+                "This is a test of the textwrap_body function with a bullet list and\n"
+                "    subsequent indent. The list should not be wrapped.\n"
+                "\n"
+                "    * Item 1\n"
+                "    * Item 2\n"
+            ),
+        ),
+    ),
+)
+def test_textwrap_body(body, subsequent_indent, expected):
+    assert blurb.textwrap_body(body, subsequent_indent=subsequent_indent) == expected
+
+
+@time_machine.travel("2025-01-07")
+def test_current_date():
+    assert blurb.current_date() == "2025-01-07"
+
+
+@time_machine.travel("2025-01-07 16:28:41")
+def test_sortable_datetime():
+    assert blurb.sortable_datetime() == "2025-01-07-16-28-41"
+
 @pytest.mark.parametrize(
     "version1, version2",
     (