Add a test for Version.filter with default branch

python · Dec 8, 2024 · 67842fd · 67842fd
1 parent 978b8ec
commit 67842fd
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/tests/test_build_docs_version.py b/tests/test_build_docs_version.py
@@ -1,7 +1,29 @@
 from build_docs import Version
 
 
-def test_filter() -> None:
+def test_filter_default() -> None:
+    # Arrange
+    versions = [
+        Version("3.14", status="feature"),
+        Version("3.13", status="bugfix"),
+        Version("3.12", status="bugfix"),
+        Version("3.11", status="security"),
+        Version("3.10", status="security"),
+        Version("3.9", status="security"),
+    ]
+
+    # Act
+    filtered = Version.filter(versions)
+
+    # Assert
+    assert filtered == [
+        Version("3.14", status="feature"),
+        Version("3.13", status="bugfix"),
+        Version("3.12", status="bugfix"),
+    ]
+
+
+def test_filter_one() -> None:
     # Arrange
     versions = [
         Version("3.14", status="feature"),