from unittest import TestCase import urllib3 import genius import http_pool TITLE = 'A Line In The Sand' ARTIST = 'Linkin Park' URL = 'https://genius.com/Linkin-park-a-line-in-the-sand-lyrics' LYR1 = '''[Intro: Mike Shinoda] Today, we stood on the wall''' LYR2 = '''little did we know [Instrumental Break] [Verse 1:''' LYR3 = '''you are gonna get yours [Chorus: Chester Bennington] Another day''' class TestGenius(TestCase): def setUp(self) -> None: http_pool.get() def test_search_success(self) -> None: url = genius.search(TITLE, ARTIST) self.assertEqual(url, URL) def test_lyrics_parsing(self) -> None: lyrics = genius.parse(URL) self.assertTrue(lyrics.startswith(LYR1)) self.assertTrue(LYR2 in lyrics) self.assertTrue(LYR3 in lyrics) def tearDown(self) -> None: http_pool.get().clear()