2021-06-21 13:05:33 -04:00
|
|
|
# Copyright 2021 (c) Anna Schumaker.
|
|
|
|
from . import tagstore
|
|
|
|
import unittest
|
|
|
|
|
|
|
|
class TestTagStore(unittest.TestCase):
|
|
|
|
def test_tag_store(self):
|
|
|
|
store = tagstore.TagStore()
|
|
|
|
self.assertEqual(store.store, { })
|
|
|
|
|
|
|
|
def test_tag_store_add_remove(self):
|
|
|
|
store = tagstore.TagStore()
|
|
|
|
tag = store.add("test", 1)
|
|
|
|
self.assertEqual(tag.tracks, [ 1 ])
|
|
|
|
|
|
|
|
self.assertEqual(store.add("test ", 2), tag)
|
|
|
|
self.assertEqual(tag.tracks, [ 1, 2 ])
|
|
|
|
|
|
|
|
store.remove(tag, 1)
|
|
|
|
self.assertEqual(tag.tracks, [ 2 ])
|
|
|
|
self.assertIn("test", store.store.keys())
|
|
|
|
|
|
|
|
store.remove(tag, 2)
|
|
|
|
self.assertEqual(tag.tracks, [ ])
|
|
|
|
self.assertNotIn("test", store.store.keys())
|
2021-06-24 15:16:56 -04:00
|
|
|
|
|
|
|
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())
|