osxphotos/tests/test_cli_add_locations.py
Rhet Turnbull 83d14ac191
Feature add locations 929 (#932)
* Initial implementation of add-locations command #929

* Updated help for add-locations

* Fixed handling of shared albums in sync

* Added test for add-locations
2023-01-16 21:26:02 -08:00

53 lines
1.7 KiB
Python

"""Test osxphotos add-locations command"""
import photoscript
import pytest
from click.testing import CliRunner
from osxphotos.cli.add_locations import add_locations
UUID_TEST_PHOTO_1 = "F12384F6-CD17-4151-ACBA-AE0E3688539E" # Pumkins1.jpg
UUID_TEST_PHOTO_LOCATION = "D79B8D77-BFFC-460B-9312-034F2877D35B" # Pumkins2.jpg
TEST_LOCATION = (41.26067, -95.94056) # Omaha, NE
@pytest.mark.test_add_locations
def test_add_locations():
"""Test add-locations command"""
with CliRunner().isolated_filesystem():
# need to clear location data from test photo
test_photo = photoscript.Photo(UUID_TEST_PHOTO_1)
test_photo.location = None
source_photo = photoscript.Photo(UUID_TEST_PHOTO_LOCATION)
source_photo.location = TEST_LOCATION
# now run add-locations
result = CliRunner().invoke(
add_locations,
[
"--window",
"2 hours",
"--dry-run",
],
)
assert result.exit_code == 0
# should find 3 matching locations: Pumkins1.jpg, Pumpkins3.jpg, Pumpkins4.jpg
assert "found location: 3" in result.output
# verify location not really added
assert test_photo.location == (None, None)
# run again without dry-run
result = CliRunner().invoke(
add_locations,
[
"--window",
"2 hours",
],
)
assert result.exit_code == 0
# should find 3 matching locations: Pumkins1.jpg, Pumpkins3.jpg, Pumpkins4.jpg
assert "found location: 3" in result.output
assert test_photo.location == TEST_LOCATION