lib: Allow passing None for the TagStore add and remove track

Signed-off-by: Anna Schumaker <Anna@NoWheyCreamery.com>

lib/tagstore.py

@@ -5,15 +5,17 @@ class TagStore:
     def __init__(self):
         self.store = dict()
 
-    def add(self, name, track):
+    def add(self, name, track=None):
         name = name.strip()
         if (t := self.store.get(name)) == None:
             t = tag.Tag(name)
             self.store[name] = t
-        t.add_track(track)
+        if track:
+            t.add_track(track)
         return t
 
-    def remove(self, t, track):
-        t.remove_track(track)
-        if len(t) == 0 and t in self.store.values():
+    def remove(self, t, track=None):
+        if track:
+            t.remove_track(track)
+        if (track==None or len(t) == 0) and t in self.store.values():
             self.store.pop(t.name)

lib/test_tagstore.py

@@ -22,3 +22,14 @@ class TestTagStore(unittest.TestCase):
         store.remove(tag, 2)
         self.assertEqual(tag.tracks, [ ])
         self.assertNotIn("test", store.store.keys())
+
+    def test_tag_store_add_remove_none(self):
+        store = tagstore.TagStore()
+        tag = store.add("test")
+        self.assertEqual(tag.tracks, [ ])
+
+        tag.add_track(1)
+        tag.add_track(2)
+
+        store.remove(tag)
+        self.assertNotIn("test", store.store.keys())