Test locking/unlocking fields when adding/removing tags

Author: JonnyWong16

plexapi/utils.py

@@ -335,6 +335,15 @@ def download(url, token, filename=None, savepath=None, session=None, chunksize=4
     return fullpath
 
 
+def tag_singular(tag):
+    if tag == 'countries':
+        return 'country'
+    elif tag == 'similar':
+        return 'similar'
+    else:
+        return tag[:-1]
+
+
 def tag_plural(tag):
     if tag == 'country':
         return 'countries'

tests/test_mixins.py

@@ -1,17 +1,27 @@
 # -*- coding: utf-8 -*-
+from plexapi.utils import tag_singular
+
 TEST_MIXIN_TAG = "Test Tag"
 
 
 def _test_mixins_tag(obj, attr, tag_method):
     add_tag_method = getattr(obj, "add" + tag_method)
     remove_tag_method = getattr(obj, "remove" + tag_method)
+    field_name = tag_singular(attr)
+    # Check tag is not present to begin with
     assert TEST_MIXIN_TAG not in [tag.tag for tag in getattr(obj, attr)]
+    # Add tag and lock the field
     add_tag_method(TEST_MIXIN_TAG)
     obj.reload()
+    field = [f for f in obj.fields if f.name == field_name]
     assert TEST_MIXIN_TAG in [tag.tag for tag in getattr(obj, attr)]
-    remove_tag_method(TEST_MIXIN_TAG)
+    assert field and field[0].locked
+    # Remove tag and unlock to field to restore the clean state
+    remove_tag_method(TEST_MIXIN_TAG, locked=False)
     obj.reload()
+    field = [f for f in obj.fields if f.name == field_name]
     assert TEST_MIXIN_TAG not in [tag.tag for tag in getattr(obj, attr)]
+    assert not field
 
 
 def edit_collection(obj):