Add tests for square art

Author: JonnyWong16

tests/test_mixins.py

@@ -223,11 +223,16 @@ def lock_poster(obj):
     _test_mixins_lock_image(obj, "posters")
 
 
+def lock_square_art(obj):
+    _test_mixins_lock_image(obj, "squareArts")
+
+
 def _test_mixins_edit_image(obj, attr):
     cap_attr = attr[:-1].capitalize()
     get_img_method = getattr(obj, attr)
     set_img_method = getattr(obj, "set" + cap_attr)
     upload_img_method = getattr(obj, "upload" + cap_attr)
+    delete_img_method = getattr(obj, "delete" + cap_attr)
     images = get_img_method()
     if images:
         default_image = images[0]
@@ -270,6 +275,12 @@ def _test_mixins_edit_image(obj, attr):
         ]
         assert file_image
 
+    # Test delete image
+    delete_img_method()
+    images = get_img_method()
+    selected_image = next((i for i in images if i.selected), None)
+    assert selected_image is None
+
     # Reset to default image
     if default_image:
         set_img_method(default_image)
@@ -287,6 +298,10 @@ def edit_poster(obj):
     _test_mixins_edit_image(obj, "posters")
 
 
+def edit_square_art(obj):
+    _test_mixins_edit_image(obj, "squareArts")
+
+
 def _test_mixins_imageUrl(obj, attr):
     url = getattr(obj, attr + "Url")
     if getattr(obj, attr):
@@ -307,6 +322,10 @@ def attr_posterUrl(obj):
     _test_mixins_imageUrl(obj, "thumb")
 
 
+def attr_squareArtUrl(obj):
+    _test_mixins_imageUrl(obj, "squareArt")
+
+
 def _test_mixins_edit_theme(obj):
     _fields = lambda: [f.name for f in obj.fields]
 

tests/test_video.py

@@ -695,8 +695,10 @@ def test_video_Movie_mixins_edit_advanced_settings(movie):
 def test_video_Movie_mixins_images(movie):
     test_mixins.lock_art(movie)
     test_mixins.lock_poster(movie)
+    test_mixins.lock_square_art(movie)
     test_mixins.edit_art(movie)
     test_mixins.edit_poster(movie)
+    test_mixins.edit_square_art(movie)
 
 
 def test_video_Movie_mixins_themes(movie):
@@ -962,10 +964,13 @@ def test_video_Show_mixins_edit_advanced_settings(show):
 def test_video_Show_mixins_images(show):
     test_mixins.lock_art(show)
     test_mixins.lock_poster(show)
+    test_mixins.lock_square_art(show)
     test_mixins.edit_art(show)
     test_mixins.edit_poster(show)
+    test_mixins.edit_square_art(show)
     test_mixins.attr_artUrl(show)
     test_mixins.attr_posterUrl(show)
+    test_mixins.attr_squareArtUrl(show)
 
 
 def test_video_Show_mixins_themes(show):
@@ -1119,10 +1124,13 @@ def test_video_Season_mixins_images(show):
     season = show.season(season=1)
     test_mixins.lock_art(season)
     test_mixins.lock_poster(season)
+    test_mixins.lock_square_art(season)
     test_mixins.edit_art(season)
     test_mixins.edit_poster(season)
+    test_mixins.edit_square_art(season)
     test_mixins.attr_artUrl(season)
     test_mixins.attr_posterUrl(season)
+    test_mixins.attr_squareArtUrl(season)
 
 
 def test_video_Season_mixins_themes(show):
@@ -1337,10 +1345,13 @@ def test_video_Episode_unwatched(tvshows):
 def test_video_Episode_mixins_images(episode):
     test_mixins.lock_art(episode)
     test_mixins.lock_poster(episode)
+    test_mixins.lock_square_art(episode)
     # test_mixins.edit_art(episode)  # Uploading episode artwork is broken in Plex
     test_mixins.edit_poster(episode)
+    test_mixins.edit_square_art(episode)
     test_mixins.attr_artUrl(episode)
     test_mixins.attr_posterUrl(episode)
+    test_mixins.attr_squareArtUrl(episode)
 
 
 def test_video_Episode_mixins_themes(episode):