Add tests for media tags

tests/test_audio.py

@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 from . import conftest as utils
-from . import test_mixins
+from . import test_media, test_mixins
 
 
 def test_audio_Artist_attr(artist):
@@ -87,6 +87,16 @@ def test_audio_Artist_mixins_tags(artist):
     test_mixins.edit_style(artist)
 
 
+def test_audio_Artist_media_tags(artist):
+    artist.reload()
+    test_media.tag_collection(artist)
+    test_media.tag_country(artist)
+    test_media.tag_genre(artist)
+    test_media.tag_mood(artist)
+    test_media.tag_similar(artist)
+    test_media.tag_style(artist)
+
+
 def test_audio_Album_attrs(album):
     assert utils.is_datetime(album.addedAt)
     if album.art:
@@ -165,6 +175,15 @@ def test_audio_Album_mixins_tags(album):
     test_mixins.edit_style(album)
 
 
+def test_audio_Album_media_tags(album):
+    album.reload()
+    test_media.tag_collection(album)
+    test_media.tag_genre(album)
+    test_media.tag_label(album)
+    test_media.tag_mood(album)
+    test_media.tag_style(album)
+
+
 def test_audio_Track_attrs(album):
     track = album.get("As Colourful As Ever").reload()
     assert utils.is_datetime(track.addedAt)
@@ -294,6 +313,12 @@ def test_audio_Track_mixins_tags(track):
     test_mixins.edit_mood(track)
 
 
+def test_audio_Track_media_tags(track):
+    track.reload()
+    test_media.tag_collection(track)
+    test_media.tag_mood(track)
+
+
 def test_audio_Audio_section(artist, album, track):
     assert artist.section()
     assert album.section()

tests/test_collection.py

@@ -33,7 +33,7 @@ def test_Collection_attrs(collection):
     assert collection.summary == ""
     assert collection.thumb.startswith("/library/collections/%s/composite" % collection.ratingKey)
     assert collection.thumbBlurHash is None
-    assert collection.title == "marvel"
+    assert collection.title == "Marvel"
     assert collection.titleSort == collection.title
     assert collection.type == "collection"
     assert utils.is_datetime(collection.updatedAt)

tests/test_media.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+
+def _test_media_tag(obj, attr):
+    tags = getattr(obj, attr)
+    if tags:
+        assert obj in tags[0].items()
+
+
+def tag_collection(obj):
+    _test_media_tag(obj, "collections")
+
+
+def tag_country(obj):
+    _test_media_tag(obj, "countries")
+
+
+def tag_director(obj):
+    _test_media_tag(obj, "directors")
+
+
+def tag_genre(obj):
+    _test_media_tag(obj, "genres")
+
+
+def tag_label(obj):
+    _test_media_tag(obj, "labels")
+
+
+def tag_mood(obj):
+    _test_media_tag(obj, "moods")
+
+
+def tag_producer(obj):
+    _test_media_tag(obj, "producers")
+
+
+def tag_role(obj):
+    _test_media_tag(obj, "roles")
+
+
+def tag_similar(obj):
+    _test_media_tag(obj, "similar")
+
+
+def tag_style(obj):
+    _test_media_tag(obj, "styles")
+
+
+def tag_tag(obj):
+    _test_media_tag(obj, "tags")
+
+
+def tag_writer(obj):
+    _test_media_tag(obj, "writers")

tests/test_photo.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-from . import test_mixins
+from . import test_media, test_mixins
 
 
 def test_photo_Photoalbum(photoalbum):
@@ -20,3 +20,8 @@ def test_photo_Photoalbum_mixins_images(photoalbum):
 
 def test_photo_Photo_mixins_tags(photo):
     test_mixins.edit_tag(photo)
+
+
+def test_photo_Photo_media_tags(photo):
+    photo.reload()
+    test_media.tag_tag(photo)

tests/test_video.py

@@ -8,7 +8,7 @@ import pytest
 from plexapi.exceptions import BadRequest, NotFound
 
 from . import conftest as utils
-from . import test_mixins
+from . import test_media, test_mixins
 
 
 def test_video_Movie(movies, movie):
@@ -59,6 +59,30 @@ def test_video_Movie_mixins_tags(movie):
     test_mixins.edit_writer(movie)
 
 
+def test_video_Movie_media_tags(movie):
+    movie.reload()
+    test_media.tag_collection(movie)
+    test_media.tag_country(movie)
+    test_media.tag_director(movie)
+    test_media.tag_genre(movie)
+    test_media.tag_label(movie)
+    test_media.tag_producer(movie)
+    test_media.tag_role(movie)
+    test_media.tag_similar(movie)
+    test_media.tag_writer(movie)
+
+
+def test_video_Movie_media_tags_Exception(movie):
+    with pytest.raises(BadRequest):
+        movie.genres[0].items()
+
+
+def test_video_Movie_media_tags_collection(movie, collection):
+    movie.reload()
+    collection_tag = next(c for c in movie.collections if c.tag == "Marvel")
+    assert collection == collection_tag.collection()
+
+
 def test_video_Movie_getStreamURL(movie, account):
     key = movie.ratingKey
     assert movie.getStreamURL() == (
@@ -741,6 +765,15 @@ def test_video_Show_mixins_tags(show):
     test_mixins.edit_label(show)
 
 
+def test_video_Show_media_tags(show):
+    show.reload()
+    test_media.tag_collection(show)
+    test_media.tag_genre(show)
+    test_media.tag_label(show)
+    test_media.tag_role(show)
+    test_media.tag_similar(show)
+
+
 def test_video_Episode(show):
     episode = show.episode("Winter Is Coming")
     assert episode == show.episode(season=1, episode=1)
@@ -900,6 +933,13 @@ def test_video_Episode_mixins_tags(episode):
     test_mixins.edit_writer(episode)
 
 
+def test_video_Episode_media_tags(episode):
+    episode.reload()
+    test_media.tag_collection(episode)
+    test_media.tag_director(episode)
+    test_media.tag_writer(episode)
+
+
 def test_video_Season(show):
     seasons = show.seasons()
     assert len(seasons) == 2
@@ -998,6 +1038,12 @@ def test_video_Season_mixins_tags(show):
     test_mixins.edit_collection(season)
 
 
+def test_video_Season_media_tags(show):
+    season = show.season(season=1)
+    season.reload()
+    test_media.tag_collection(season)
+
+
 def test_that_reload_return_the_same_object(plex):
     # we want to check this that all the urls are correct
     movie_library_search = plex.library.section("Movies").search("Elephants Dream")[0]