added some tests

This commit is contained in:
Hellow 2023-03-24 19:43:33 +01:00
parent 6d2a7469fc
commit 6b2d2e0bfc

View File

@ -168,21 +168,9 @@ class TestCollection(unittest.TestCase):
for song in self.collection: for song in self.collection:
self.assertIn(song.unified_title, self.unified_titels) self.assertIn(song.unified_title, self.unified_titels)
class TestCollectionAppending(unittest.TestCase):
def setUp(self):
self.song_list: objects.song = [
objects.Song(title="hasskrank"),
objects.Song(title="HaSSkrank"),
objects.Song(title="Suicideseason", isrc="uniqueID"),
objects.Song(title="same isrc different title", isrc="uniqueID")
]
self.unified_titels = set(song.unified_title for song in self.song_list)
def test_appending(self): def test_appending(self):
collection = objects.Collection( collection = objects.Collection(
element_type=objects.Song, element_type=objects.Song
data=self.song_list
) )
res = collection.append(self.song_list[0]) res = collection.append(self.song_list[0])
@ -197,9 +185,11 @@ class TestCollectionAppending(unittest.TestCase):
self.assertEqual(res.was_in_collection, False) self.assertEqual(res.was_in_collection, False)
self.assertEqual(res.current_element, self.song_list[2]) self.assertEqual(res.current_element, self.song_list[2])
res = collection.append(self.song_list[3]) res = collection.append(self.song_list[3], merge_into_existing=False)
self.assertEqual(res.was_in_collection, True) self.assertEqual(res.was_in_collection, True)
self.assertEqual(res.current_element, self.song_list[2]) self.assertEqual(res.current_element, self.song_list[3])