Implemented --count, #1098 (#1102)

2023-06-24 10:50:34 -07:00
parent bb8e164f21
commit 2190628e82
2 changed files with 46 additions and 0 deletions
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -8176,6 +8176,46 @@ def test_query_added_in_last():
    assert results.exit_code == 0


+def test_query_count():
+    """Test query --count"""
+
+    runner = CliRunner()
+    cwd = os.getcwd()
+    # pylint: disable=not-context-manager
+    results = runner.invoke(
+        query,
+        [
+            "--library",
+            os.path.join(cwd, PHOTOS_DB_15_7),
+            "--added-before",
+            "2019-07-28",
+            "--count",
+        ],
+    )
+    assert results.exit_code == 0
+    assert results.output.strip() == "7"
+
+
+def test_query_count_0():
+    """Test query --count with zero results"""
+
+    runner = CliRunner()
+    cwd = os.getcwd()
+    # pylint: disable=not-context-manager
+    results = runner.invoke(
+        query,
+        [
+            "--library",
+            os.path.join(cwd, PHOTOS_DB_15_7),
+            "--added-before",
+            "1900-01-01",
+            "--count",
+        ],
+    )
+    assert results.exit_code == 0
+    assert results.output.strip() == "0"
+
+
 def test_export_export_dir_template():
    """Test {export_dir} template"""