Added test cases and documentation for shared photos and shared albums

This commit is contained in:
Rhet Turnbull
2019-12-26 22:13:25 -08:00
parent 10284d6589
commit 6d20e9e361
125 changed files with 1138 additions and 32 deletions

View File

@@ -0,0 +1,68 @@
import pytest
# TODO: put some of this code into a pre-function
PHOTOS_DB = "./tests/Test-10.14.6.photoslibrary/database/photos.db"
PHOTOS_DB_PATH = "/Test-10.14.6.photoslibrary/database/Photos.sqlite"
PHOTOS_LIBRARY_PATH = "/Test-10.14.6.photoslibrary"
ALBUMS = ["Pumpkin Farm", "Test Album", "Test Album (1)"]
ALBUM_DICT = {"Pumpkin Farm": 3, "Test Album": 1, "Test Album (1)": 1}
def test_albums():
import osxphotos
photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
albums = photosdb.albums
assert len(albums) == len(ALBUMS)
for album in albums:
assert album in ALBUMS
def test_albums_shared():
import osxphotos
photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
albums_shared = photosdb.albums_shared
assert len(albums_shared) == 0
def test_albums_as_dict():
import osxphotos
photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
albums_as_dict = photosdb.albums_as_dict
for album in albums_as_dict:
assert album in ALBUM_DICT
assert albums_as_dict[album] == ALBUM_DICT[album]
def test_albums_shared_as_dict():
import osxphotos
photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
albums_shared_as_dict = photosdb.albums_shared_as_dict
assert albums_shared_as_dict == {}
def test_shared():
import osxphotos
photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
photos = [p for p in photosdb.photos() if p.shared]
assert len(photos) == 0
def test_not_shared():
import osxphotos
photosdb = osxphotos.PhotosDB(dbfile=PHOTOS_DB)
photos = [p for p in photosdb.photos() if not p.shared]
assert len(photos) == 7