From d6fc8fc3b1d276fd6b22550e50ec1bdeeb3acf6f Mon Sep 17 00:00:00 2001
From: Rhet Turnbull <rturnbull@gmail.com>
Date: Sat, 19 Nov 2022 18:09:47 -0800
Subject: [PATCH] Added test for #848

---
 tests/test_cli_add_to_album.py | 29 +++++++++++++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/tests/test_cli_add_to_album.py b/tests/test_cli_add_to_album.py
index 394a7f9c..aa58c92c 100644
--- a/tests/test_cli_add_to_album.py
+++ b/tests/test_cli_add_to_album.py
@@ -1,15 +1,20 @@
 """ Test --add-exported-to-album """
 
-import pytest
 import os
-from click.testing import CliRunner
+
 import photoscript
+import pytest
+from click.testing import CliRunner
 
 UUID_EXPORT = {"3DD2C897-F19E-4CA6-8C22-B027D5A71907": {"filename": "IMG_4547.jpg"}}
 UUID_MISSING = {
     "8E1D7BC9-9321-44F9-8CFB-4083F6B9232A": {"filename": "IMG_2000.JPGssss"}
 }
 
+# photos with matching names
+QUERY_NAME = "AAF035"
+QUERY_COUNT = 4
+
 
 @pytest.mark.addalbum
 def test_export_add_to_album(addalbum_library):
@@ -113,3 +118,23 @@ def test_query_add_to_album(addalbum_library):
         got_uuids = [p.uuid for p in album.photos()]
         assert sorted(got_uuids) == sorted(list(UUID_EXPORT.keys()))
 
+
+@pytest.mark.addalbum
+def test_query_add_to_album_multiple_results(addalbum_library):
+    """Test osxphotos query --add-to-album with multiple results, see #848"""
+    from osxphotos.cli import query
+
+    runner = CliRunner()
+    cwd = os.getcwd()
+    with runner.isolated_filesystem():
+        QUERY_ALBUM = "OSXPhotos Query"
+
+        result = runner.invoke(
+            query, ["--add-to-album", QUERY_ALBUM, "--name", QUERY_NAME]
+        )
+        assert result.exit_code == 0
+
+        photoslib = photoscript.PhotosLibrary()
+        album = photoslib.album(QUERY_ALBUM)
+        assert album is not None
+        assert len(album) == QUERY_COUNT