Add tests

Author: hugovk

tests/test_run_release.py

@@ -88,7 +88,7 @@ def test_check_doc_unreleased_version_no_file_alpha(tmp_path: Path) -> None:
     run_release.check_doc_unreleased_version(cast(ReleaseShelf, db))
 
 
-def test_check_doc_unreleased_version_ok(monkeypatch, tmp_path: Path) -> None:
+def test_check_doc_unreleased_version_ok(tmp_path: Path) -> None:
     prepare_fake_docs(
         tmp_path,
         "<div>New in 3.13</div>",
@@ -124,3 +124,55 @@ def test_check_doc_unreleased_version_waived(monkeypatch, tmp_path: Path) -> Non
     }
     with fake_answers(monkeypatch, ["yes"]):
         run_release.check_doc_unreleased_version(cast(ReleaseShelf, db))
+
+
+def test_modify_the_docs_by_version_page_prerelease(
+    capsys: pytest.LogCaptureFixture,
+) -> None:
+    # Arrange
+    db = {"release": Tag("3.14.0a7")}
+
+    # Act
+    run_release.modify_the_docs_by_version_page(cast(ReleaseShelf, db))
+
+    # Assert
+    captured = capsys.readouterr()
+    assert captured.out == ""
+    assert captured.err == ""
+
+
+def test_modify_the_docs_by_version_page_final_no(capsys, monkeypatch) -> None:
+    # Arrange
+    db = {"release": Tag("3.13.3")}
+
+    # Act
+    with (
+        fake_answers(monkeypatch, ["no"]),
+        pytest.raises(run_release.ReleaseException),
+    ):
+        run_release.modify_the_docs_by_version_page(cast(ReleaseShelf, db))
+
+    # Assert
+    captured = capsys.readouterr()
+    assert (
+        "* `Python 3.13.3 <https://docs.python.org/release/3.13.3/>`_, documentation released on"
+        in captured.out
+    )
+    assert captured.err == ""
+
+
+def test_modify_the_docs_by_version_page_final_yes(capsys, monkeypatch) -> None:
+    # Arrange
+    db = {"release": Tag("3.13.3")}
+
+    # Act
+    with fake_answers(monkeypatch, ["yes"]):
+        run_release.modify_the_docs_by_version_page(cast(ReleaseShelf, db))
+
+    # Assert
+    captured = capsys.readouterr()
+    assert (
+        "* `Python 3.13.3 <https://docs.python.org/release/3.13.3/>`_, documentation released on"
+        in captured.out
+    )
+    assert captured.err == ""