diff --git a/tests/test_network.py b/tests/test_network.py index fafe66f..ab5dad2 100755 --- a/tests/test_network.py +++ b/tests/test_network.py @@ -315,6 +315,26 @@ class TestPyLastNetwork(PyLastTestCase): self.assertIsInstance(results, list) self.assertIsInstance(results[0], pylast.Album) + def test_album_search_images(self): + # Arrange + album = "Nevermind" + search = self.network.search_for_album(album) + + # Act + results = search.get_next_page() + images = results[0].images + + # Assert + self.assertEqual(len(images), 4) + + self.assertTrue(images[pylast.SIZE_SMALL].startswith("https://")) + self.assertTrue(images[pylast.SIZE_SMALL].endswith(".png")) + self.assertIn("/34s/", images[pylast.SIZE_SMALL]) + + self.assertTrue(images[pylast.SIZE_EXTRA_LARGE].startswith("https://")) + self.assertTrue(images[pylast.SIZE_EXTRA_LARGE].endswith(".png")) + self.assertIn("/300x300/", images[pylast.SIZE_EXTRA_LARGE]) + def test_artist_search(self): # Arrange artist = "Nirvana"