def setUp(self):
super(self.__class__, self).setUp()
sys.modules["__main__"].common.parseDOM.return_value = [
"some_string", "some_string", "some_string"
]
sys.modules["__main__"].core._fetchPage.return_value = {
"content": "some_content",
"status": 200
}
sys.modules[
"__main__"].common.makeAscii.return_value = "some_ascii_string"
sys.modules[
"__main__"].common.replaceHTMLCodes.return_value = "some_html_free_string"
sys.modules["__main__"].utils.extractVID.return_value = [
"some_id_1", "some_id_2", "some_id_3"
]
sys.modules[
"__main__"].language.return_value = "some_language_string %s"
sys.modules[
"__main__"].common.stripTags.return_value = "some_tag_less_string"
sys.modules["__main__"].settings.getSetting.return_value = "1"
sys.modules["__main__"].cache.cacheFunction.return_value = ([
"some_cached_string"
], 200)
self.scraper = YouTubeScraper()
self.scraper.createUrl = Mock()
self.scraper.createUrl.return_value = "some_url"
def test_createUrl_should_return_proper_url_for_search_disco(self):
self.scraper = YouTubeScraper()
url = self.scraper.createUrl({
"scraper": "search_disco",
"search": "some_search"
})
assert (url == self.scraper.urls["disco_search"] % "some_search")
def test_createUrl_should_return_proper_url_for_music_top_100(self):
self.scraper = YouTubeScraper()
url = self.scraper.createUrl({"scraper": "music_top100"})
assert (url == self.scraper.urls["disco_main"])
