25 lines
733 B
Python
25 lines
733 B
Python
|
# 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())
|