From 2c6d751816396a8b0ee09248d8b7d937f6ec45d9 Mon Sep 17 00:00:00 2001 From: James Clarke Date: Sun, 12 Mar 2023 18:02:40 +0000 Subject: [PATCH] Fixed Album test case --- src/tests/test_objects.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/src/tests/test_objects.py b/src/tests/test_objects.py index 3104d0f..8b27ac9 100644 --- a/src/tests/test_objects.py +++ b/src/tests/test_objects.py @@ -1,4 +1,4 @@ -from music_kraken import objects + import pycountry import unittest import sys @@ -7,6 +7,7 @@ import os # Add the parent directory of the src package to the Python module search path sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) +from music_kraken import objects class TestSong(unittest.TestCase): @@ -134,13 +135,15 @@ class TestAlbum(unittest.TestCase): self.assertEqual(self.album.language.alpha_2, "en") def test_album_label(self): - self.assertEqual(self.album.label_list[0].name, "an album label") + self.assertEqual(self.album.label_collection[0].name, "an album label") def test_album_source(self): + sp = self.album.source_collection.get_sources_from_page(objects.SourcePages.ENCYCLOPAEDIA_METALLUM)[0] + self.assertEqual( - self.album.source_list[0].page, objects.SourcePages.ENCYCLOPAEDIA_METALLUM) + sp.page_enum, objects.SourcePages.ENCYCLOPAEDIA_METALLUM) self.assertEqual( - self.album.source_list[0].url, "https://www.metal-archives.com/albums/I%27m_in_a_Coffin/One_Final_Action/207614") + sp.url, "https://www.metal-archives.com/albums/I%27m_in_a_Coffin/One_Final_Action/207614") class TestCollection(unittest.TestCase):