From e4d700fcff75c51a58f06a950f9426f00b64ecb4 Mon Sep 17 00:00:00 2001
From: Rhet Turnbull <rturnbull@gmail.com>
Date: Fri, 27 Dec 2019 16:36:15 -0800
Subject: [PATCH] Added tests for PhotoInfo.__repr__

---
 tests/test_catalina_10_15_1.py | 18 +++++++++++++++++-
 tests/test_mojave_10_14_6.py   | 18 ++++++++++++++++--
 2 files changed, 33 insertions(+), 3 deletions(-)

diff --git a/tests/test_catalina_10_15_1.py b/tests/test_catalina_10_15_1.py
index a96cc8d9..b42dc7cf 100644
--- a/tests/test_catalina_10_15_1.py
+++ b/tests/test_catalina_10_15_1.py
@@ -760,7 +760,6 @@ def test_photosdb_repr():
     import osxphotos
 
     photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
-    print(repr(photosdb))
     photosdb2 = eval(repr(photosdb))
 
     ignore_keys = ["_tmp_db", "_tmp_files"]
@@ -768,3 +767,20 @@ def test_photosdb_repr():
         k: v for k, v in photosdb2.__dict__.items() if k not in ignore_keys
     }
 
+
+def test_photosinfo_repr():
+    import osxphotos
+    import datetime
+
+    photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
+    photos = photosdb.photos(uuid=[UUID_DICT["favorite"]])
+    photo = photos[0]
+    photo2 = eval(repr(photo))
+
+    assert {
+        k: str(v).encode("utf-8")
+        for k, v in photo.__dict__.items()
+    } == {
+        k: str(v).encode("utf-8")
+        for k, v in photo2.__dict__.items()
+    }
diff --git a/tests/test_mojave_10_14_6.py b/tests/test_mojave_10_14_6.py
index 55fa9955..2f807401 100644
--- a/tests/test_mojave_10_14_6.py
+++ b/tests/test_mojave_10_14_6.py
@@ -33,6 +33,8 @@ KEYWORDS_DICT = {
 PERSONS_DICT = {"Katie": 3, "Suzy": 2, "Maria": 1}
 ALBUM_DICT = {"Pumpkin Farm": 3, "Test Album": 1, "Test Album (1)": 1}
 
+UUID_DICT = {"favorite": "6bxcNnzRQKGnK4uPrCJ9UQ"}
+
 
 def test_init():
     import osxphotos
@@ -149,7 +151,7 @@ def test_favorite():
     import osxphotos
 
     photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
-    photos = photosdb.photos(uuid=["6bxcNnzRQKGnK4uPrCJ9UQ"])
+    photos = photosdb.photos(uuid=[UUID_DICT["favorite"]])
     assert len(photos) == 1
     p = photos[0]
     assert p.favorite == True
@@ -330,7 +332,6 @@ def test_photosdb_repr():
     import osxphotos
 
     photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
-    print(repr(photosdb))
     photosdb2 = eval(repr(photosdb))
 
     ignore_keys = ["_tmp_db", "_tmp_files"]
@@ -338,3 +339,16 @@ def test_photosdb_repr():
         k: v for k, v in photosdb2.__dict__.items() if k not in ignore_keys
     }
 
+
+def test_photosinfo_repr():
+    import osxphotos
+    import datetime
+
+    photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
+    photos = photosdb.photos(uuid=[UUID_DICT["favorite"]])
+    photo = photos[0]
+    photo2 = eval(repr(photo))
+
+    assert {k: str(v).encode("utf-8") for k, v in photo.__dict__.items()} == {
+        k: str(v).encode("utf-8") for k, v in photo2.__dict__.items()
+    }