def test_getInfo_should_use_cache_when_possible(self):
sys.modules["__main__"].cache.get.return_value = '["something"]'
player = YouTubePlayer()
player.getInfo({"videoid": "some_id"})
sys.modules["__main__"].cache.get.assert_called_with("videoidcachesome_id")
def test_selectVideoQuality_should_add_user_agent_when_not_called_by_download_function(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
player = YouTubePlayer()
url = player.selectVideoQuality({},{35: "SD", 22: "720p", 37: "1080p"})
assert(url.find("| Mozilla/5.0 (MOCK)") > 0)
def test_selectVideoQuality_should_choose_highest_sd_quality_if_only_multiple_sd_qualities_are_available(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
player = YouTubePlayer()
url = player.selectVideoQuality({},{5: "1", 33: "2", 18: "3", 26: "4", 43: "5", 34: "6", 78: "7", 44: "8", 59: "9", 35: "10"})
assert(url[:url.find("|")].strip() == "10")
def test_selectVideoQuality_should_limit_to_720p_if_user_has_selected_that_option(self):
sys.modules["__main__"].settings.getSetting.return_value = "2"
player = YouTubePlayer()
url = player.selectVideoQuality({},{35: "SD", 22: "720p", 37: "1080p"})
assert(url == "720p | Mozilla/5.0 (MOCK)")
def test_selectVideoQuality_should_prefer_720p_if_asked_to(self):
sys.modules["__main__"].settings.getSetting.return_value = "2"
player = YouTubePlayer()
url = player.selectVideoQuality({"quality": "720p"},{37: "1080p", 22: "720p", 35: "SD"})
assert(url == "720p | Mozilla/5.0 (MOCK)")
def test_selectVideoQuality_should_choose_highest_sd_quality_if_only_multiple_sd_qualities_are_available(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
player = YouTubePlayer()
url = player.selectVideoQuality({},{5: "1", 33: "2", 18: "3", 26: "4", 43: "5", 34: "6", 78: "7", 44: "8", 59: "9", 35: "10"})
assert(url[:url.find("|")].strip() == "10")
def test_selectVideoQuality_should_prefer_SD_if_asked_to(self):
sys.modules["__main__"].settings.getSetting.return_value = "2"
player = YouTubePlayer()
url = player.selectVideoQuality({"quality": "SD"},{37: "1080p", 22: "720p", 35: "SD"})
assert(url[:url.find("|")].strip() == "SD")
def test_selectVideoQuality_should_not_add_user_agent_when_called_by_download_function(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
player = YouTubePlayer()
url = player.selectVideoQuality({"action": "download"},{35: "SD", 22: "720p", 37: "1080p"})
assert(url.find("|" + urllib.urlencode({'User-Agent':"Mozilla/5.0 (MOCK)"})) < 0)
def test_selectVideoQuality_should_limit_to_sd_if_user_has_selected_that_option(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
player = YouTubePlayer()
url = player.selectVideoQuality({},{35: "SD", 22: "720p", 37: "1080p"})
assert(url[:url.find("|")].strip() == "SD")
def test_playVideo_should_call_getVideoObject(self):
player = YouTubePlayer()
player.buildVideoObject = Mock(return_value=[{"apierror": "some error"}, 303])
player.playVideo()
player.buildVideoObject.assert_called_with({})
def test_getInfo_should_use_cache_when_possible(self):
sys.modules["__main__"].cache.get.return_value = '["something"]'
player = YouTubePlayer()
player.getInfo({"videoid": "some_id"})
sys.modules["__main__"].cache.get.assert_called_with("videoidcachesome_id")
def test_selectVideoQuality_should_limit_to_sd_if_user_has_selected_that_option(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
player = YouTubePlayer()
url = player.selectVideoQuality({},{35: "SD", 22: "720p", 37: "1080p"})
assert(url[:url.find("|")].strip() == "SD")
def test_selectVideoQuality_should_prefer_SD_if_asked_to(self):
sys.modules["__main__"].settings.getSetting.return_value = "2"
player = YouTubePlayer()
url = player.selectVideoQuality({"quality": "SD"},{37: "1080p", 22: "720p", 35: "SD"})
assert(url[:url.find("|")].strip() == "SD")
def test_playVideo_should_call_getVideoObject(self):
player = YouTubePlayer()
player.buildVideoObject = Mock(return_value=[{"apierror": "some error"}, 303])
player.playVideo()
player.buildVideoObject.assert_called_with({})
def test_selectVideoQuality_should_not_add_user_agent_when_called_by_download_function(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
player = YouTubePlayer()
url = player.selectVideoQuality({"action": "download"},{35: "SD", 22: "720p", 37: "1080p"})
assert(url.find("|" + urllib.urlencode({'User-Agent':"Mozilla/5.0 (MOCK)"})) < 0)
def test_selectVideoQuality_should_call_userSelectsVideoQuality_if_user_selected_that_option(self):
sys.modules["__main__"].settings.getSetting.return_value = "0"
player = YouTubePlayer()
player.userSelectsVideoQuality = Mock()
player.selectVideoQuality({},{35: "SD", 22: "720p", 37: "1080p"})
player.userSelectsVideoQuality.assert_called_with({}, {35: 'SD', 37: '1080p', 22: '720p'})
def test_selectVideoQuality_should_prefer_h264_over_vp8_for_720p_as_appletv2_cant_handle_vp8_properly(self):
sys.modules["__main__"].settings.getSetting.return_value = "2"
player = YouTubePlayer()
url = player.selectVideoQuality({},{22: "h264", 45: "vp8"})
print "url: " + repr(url)
assert(url[:url.find("|")].strip() == "h264")
def test__getVideoLinks_should_return_empty_dictionary_on_missing_map(self):
sys.modules["__main__"].core._fetchPage.return_value = {"status":303, "content": ""}
sys.modules["__main__"].common.parseDOM.return_value = []
player = YouTubePlayer()
(result, status) = player.extractVideoLinksFromYoutube({},{"videoid":"123"})
assert (result == {})
def test_getInfo_should_call_fetchPage_with_correct_url(self):
sys.modules["__main__"].cache.get.return_value = {}
sys.modules["__main__"].core._fetchPage.return_value = {"status": 303, "content": "something"}
player = YouTubePlayer()
player.getInfo({"videoid": "some_id"})
sys.modules["__main__"].core._fetchPage.assert_called_with({"api": "true", "link": player.urls["video_info"] % ("some_id")})
def test_selectVideoQuality_should_call_userSelectsVideoQuality_if_user_selected_that_option(self):
sys.modules["__main__"].settings.getSetting.return_value = "0"
player = YouTubePlayer()
player.userSelectsVideoQuality = Mock()
player.selectVideoQuality({},{35: "SD", 22: "720p", 37: "1080p"})
player.userSelectsVideoQuality.assert_called_with({}, {35: 'SD', 37: '1080p', 22: '720p'})
def test_getInfo_should_call_fetchPage_with_correct_url(self):
sys.modules["__main__"].cache.get.return_value = {}
sys.modules["__main__"].core._fetchPage.return_value = {"status": 303, "content": "something"}
player = YouTubePlayer()
player.getInfo({"videoid": "some_id"})
sys.modules["__main__"].core._fetchPage.assert_called_with({"api": "true", "link": player.urls["video_info"] % ("some_id")})
def test_buildVideoLinks_should_try_scraping_first(self):
sys.modules["__main__"].core._fetchPage.return_value = {"status":303, "content": ""}
sys.modules["__main__"].common.parseDOM.return_value = []
player = YouTubePlayer()
player.extractVideoLinksFromYoutube({},{"videoid":"123"})
sys.modules["__main__"].core._fetchPage.assert_any_call({"link":player.urls["video_stream"] % "123", "login":"******"})
def test_buildVideoLinks_should_try_scraping_first(self):
sys.modules["__main__"].core._fetchPage.return_value = {"status":303, "content": ""}
sys.modules["__main__"].common.parseDOM.return_value = []
player = YouTubePlayer()
player.extractVideoLinksFromYoutube({},{"videoid":"123"})
sys.modules["__main__"].core._fetchPage.assert_any_call({"link":player.urls["video_stream"] % "123", "login":"******"})
def test_selectVideoQuality_should_prefer_h264_over_vp8_for_720p_as_appletv2_cant_handle_vp8_properly(self):
sys.modules["__main__"].settings.getSetting.return_value = "2"
player = YouTubePlayer()
url = player.selectVideoQuality({},{22: "h264", 45: "vp8"})
print "url: " + repr(url)
assert(url[:url.find("|")].strip() == "h264")
def test__getVideoLinks_should_return_empty_dictionary_on_missing_map(self):
sys.modules["__main__"].core._fetchPage.return_value = {"status":303, "content": ""}
sys.modules["__main__"].common.parseDOM.return_value = []
player = YouTubePlayer()
(result, status) = player.extractVideoLinksFromYoutube({},{"videoid":"123"})
assert (result == {})
def test_getInfo_should_call_core_getVideoInfo_to_parse_youtube_xml(self):
sys.modules["__main__"].cache.get.return_value = {}
sys.modules["__main__"].core._fetchPage.return_value = {"status": 200, "content": "something"}
sys.modules["__main__"].core.getVideoInfo.return_value = [{"videoid": "some_id"}]
player = YouTubePlayer()
player.getInfo({"videoid": "some_id"})
sys.modules["__main__"].core.getVideoInfo.assert_called_with("something", {"videoid": "some_id"})
def test_userSelectsVideoQuality_should_prefer_h264_over_vp8_as_appletv2_cant_handle_vp8_properly(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
sys.modules["__main__"].xbmcgui.Dialog().select.return_value = 0
sys.modules["__main__"].language.return_value = ""
player = YouTubePlayer()
url = player.userSelectsVideoQuality({}, {22: "h264", 45: "vp8"})
assert(url == "h264")
def test_playVideo_should_log_and_fail_gracefully_on_apierror(self):
player = YouTubePlayer()
player.buildVideoObject = Mock()
player.buildVideoObject.return_value = [{"apierror": "some error"}, 303]
result = player.playVideo()
assert(result == False)
sys.modules["__main__" ].common.log.assert_called_with("construct video url failed contents of video item {'apierror': 'some error'}")
def test_getInfo_should_save_video_info_in_cache(self):
sys.modules["__main__"].cache.get.return_value = {}
sys.modules["__main__"].core._fetchPage.return_value = {"status": 200, "content": "something"}
sys.modules["__main__"].core.getVideoInfo.return_value = [{"videoid": "some_id"}]
player = YouTubePlayer()
player.getInfo({"videoid": "some_id"})
sys.modules["__main__"].cache.set.assert_called_with('videoidcachesome_id', "{'videoid': 'some_id'}")
def test_playVideo_should_log_and_fail_gracefully_on_apierror(self):
player = YouTubePlayer()
player.buildVideoObject = Mock()
player.buildVideoObject.return_value = [{"apierror": "some error"}, 303]
result = player.playVideo()
assert(result == False)
sys.modules["__main__" ].common.log.assert_called_with("construct video url failed contents of video item {'apierror': 'some error'}")
def test_userSelectsVideoQuality_should_prefer_h264_over_vp8_as_appletv2_cant_handle_vp8_properly(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
sys.modules["__main__"].xbmcgui.Dialog().select.return_value = 0
sys.modules["__main__"].language.return_value = ""
player = YouTubePlayer()
url = player.userSelectsVideoQuality({}, {22: "h264", 45: "vp8"})
assert(url == "h264")
def test_getInfo_should_call_core_getVideoInfo_to_parse_youtube_xml(self):
sys.modules["__main__"].cache.get.return_value = {}
sys.modules["__main__"].core._fetchPage.return_value = {"status": 200, "content": "something"}
sys.modules["__main__"].core.getVideoInfo.return_value = [{"videoid": "some_id"}]
player = YouTubePlayer()
player.getInfo({"videoid": "some_id"})
sys.modules["__main__"].core.getVideoInfo.assert_called_with("something", {"videoid": "some_id"})
def test_getInfo_should_save_video_info_in_cache(self):
sys.modules["__main__"].cache.get.return_value = {}
sys.modules["__main__"].core._fetchPage.return_value = {"status": 200, "content": "something"}
sys.modules["__main__"].core.getVideoInfo.return_value = [{"videoid": "some_id"}]
player = YouTubePlayer()
player.getInfo({"videoid": "some_id"})
sys.modules["__main__"].cache.set.assert_called_with('videoidcachesome_id', "{'videoid': 'some_id'}")
def test_playVideo_should_update_locally_stored_watched_status(self):
sys.modules["__main__"].settings.getSetting.return_value = "0"
sys.argv = ["test1", "1", "test2"]
player = YouTubePlayer()
player.buildVideoObject = Mock()
player.buildVideoObject.return_value = ({"Title": "someTitle", "videoid": "some_id", "thumbnail": "someThumbnail", "video_url": "someUrl"}, 200)
player.addSubtitles = Mock()
player.playVideo({"videoid": "some_id"})
sys.modules["__main__"].storage.storeValue.assert_called_with("vidstatus-some_id", "7" )
def test_userSelectsVideoQuality_should_append_list_of_known_qualities(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
sys.modules["__main__"].xbmcgui.Dialog().select.return_value = -1
sys.modules["__main__"].language.return_value = ""
player = YouTubePlayer()
url = player.userSelectsVideoQuality({}, {35: u"SD", 22: u"720p", 37: u"1080p", 35: u"480p", 18: u"380p", 34: u"360p", 5: u"240p", 17: u"144p"})
print repr(url)
sys.modules["__main__"].xbmcgui.Dialog().select.assert_any_call("", [u"1080p", u"720p", u"480p", u"380p", u"360p", u"240p", u"144p"])
def test_userSelectsVideoQuality_should_append_list_of_known_qualities(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
sys.modules["__main__"].xbmcgui.Dialog().select.return_value = -1
sys.modules["__main__"].language.return_value = ""
player = YouTubePlayer()
url = player.userSelectsVideoQuality({}, {35: u"SD", 22: u"720p", 37: u"1080p", 35: u"480p", 18: u"380p", 34: u"360p", 5: u"240p", 17: u"144p"})
print repr(url)
sys.modules["__main__"].xbmcgui.Dialog().select.assert_any_call("", [u"1080p", u"720p", u"480p", u"380p", u"360p", u"240p", u"144p"])
def test_userSelectsVideoQuality_should_call_xbmc_dialog_select_to_ask_for_user_input(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
sys.modules["__main__"].xbmcgui.Dialog().select.return_value = -1
sys.modules["__main__"].language.return_value = ""
player = YouTubePlayer()
url = player.userSelectsVideoQuality({}, {35: "SD", 22: "720p", 37: "1080p"})
print repr(url)
assert(sys.modules["__main__"].xbmcgui.Dialog().select.call_count > 0)
def test_getVideoLinks_should_parse_live_video_structure_on_webpage_correctly(self):
sys.modules["__main__"].core._fetchPage.return_value = {"status": 200, "content": self.readTestInput("live-video-page-e93MaEwrsfc.html", False)}
sys.modules["__main__"].common.parseDOM.return_value = []
player = YouTubePlayer()
sys.modules["__main__"].core._findErrors.return_value = "mock error"
result = player.extractVideoLinksFromYoutube({}, {"videoid": "some_id"})
print repr(result)
assert(result[0].has_key(35))
def test_userSelectsVideoQuality_should_select_proper_quality_based_on_user_input(self):
player = YouTubePlayer()
sys.modules["__main__"].settings.getSetting.return_value = "1"
sys.modules["__main__"].xbmcgui.Dialog().select.return_value = 0
sys.modules["__main__"].language.return_value = ""
sys.modules["__main__"].common.makeUTF8.return_value = "bla"
url = player.userSelectsVideoQuality({}, {35: u"SD", 22: u"720p", 37: u"1080p"})
sys.modules["__main__"].xbmcgui.Dialog().select.assert_called_with("", [u"1080p", u"720p", u"480p"])
assert(url == "1080p")
def test_playVideo_should_call_addSubtitles(self):
video = {"Title": "someTitle", "videoid": "some_id", "thumbnail": "someThumbnail", "video_url": "someUrl"}
sys.modules["__main__"].settings.getSetting.return_value = "1"
sys.argv = ["test1", "1", "test2"]
player = YouTubePlayer()
player.buildVideoObject = Mock()
player.buildVideoObject.return_value = (video, 200)
player.playVideo({"videoid": "some_id"})
sys.modules["__main__"].subtitles.addSubtitles.assert_called_with(video)
def test_selectVideoQuality_should_prefer_720p_if_asked_to(self):
sys.modules["__main__"].settings.getSetting.return_value = "2"
player = YouTubePlayer()
url = player.selectVideoQuality({"quality": "720p"}, {
37: "1080p",
22: "720p",
35: "SD"
})
assert (url == "720p | Mozilla/5.0 (MOCK)")
def test_buildVideoObject_should_use_local_file_for_playback_if_found(self):
sys.modules["__main__"].subtitles.getLocalFileSource.return_value = "somePath/someTitle.mp4"
params = {"videoid": "some_id"}
player = YouTubePlayer()
player.getInfo = Mock()
video = {"videoid": "some_id", "Title": "someTitle"}
player.getInfo.return_value = (video, 200)
(video, status) = player.buildVideoObject(params)
assert(video["video_url"] == "somePath/someTitle.mp4")
def test_buildVideoObject_should_use_local_file_for_playback_if_found(self):
sys.modules["__main__"].subtitles.getLocalFileSource.return_value = "somePath/someTitle.mp4"
params = {"videoid": "some_id"}
player = YouTubePlayer()
player.getInfo = Mock()
video = {"videoid": "some_id", "Title": "someTitle"}
player.getInfo.return_value = (video, 200)
(video, status) = player.buildVideoObject(params)
assert(video["video_url"] == "somePath/someTitle.mp4")
def test_getVideoLinks_should_parse_live_video_structure_on_webpage_correctly(self):
sys.modules["__main__"].core._fetchPage.return_value = {"status": 200, "content": self.readTestInput("live-video-page-e93MaEwrsfc.html", False)}
sys.modules["__main__"].common.parseDOM.return_value = []
player = YouTubePlayer()
sys.modules["__main__"].core._findErrors.return_value = "mock error"
result = player.extractVideoLinksFromYoutube({}, {"videoid": "some_id"})
print repr(result)
assert(result[0].has_key(35))
def test_buildVideoObject_should_get_video_information_from_getInfo(self):
sys.modules["__main__"].settings.getSetting.return_value = ""
player = YouTubePlayer()
player.getInfo = Mock()
player.getInfo.return_value = ({}, 303)
player.extractVideoLinksFromYoutube = Mock()
player.extractVideoLinksFromYoutube.return_value = ({}, {})
player.buildVideoObject({})
player.getInfo.assert_called_with({})
def test_userSelectsVideoQuality_should_call_xbmc_dialog_select_to_ask_for_user_input(self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
sys.modules["__main__"].xbmcgui.Dialog().select.return_value = -1
sys.modules["__main__"].language.return_value = ""
player = YouTubePlayer()
url = player.userSelectsVideoQuality({}, {35: "SD", 22: "720p", 37: "1080p"})
print repr(url)
assert(sys.modules["__main__"].xbmcgui.Dialog().select.call_count > 0)
def test_userSelectsVideoQuality_should_select_proper_quality_based_on_user_input(self):
player = YouTubePlayer()
sys.modules["__main__"].settings.getSetting.return_value = "1"
sys.modules["__main__"].xbmcgui.Dialog().select.return_value = 0
sys.modules["__main__"].language.return_value = ""
sys.modules["__main__"].common.makeUTF8.return_value = "bla"
url = player.userSelectsVideoQuality({}, {35: u"SD", 22: u"720p", 37: u"1080p"})
sys.modules["__main__"].xbmcgui.Dialog().select.assert_called_with("", [u"1080p", u"720p", u"480p"])
assert(url == "1080p")
def test_playVideo_should_call_addSubtitles(self):
video = {"Title": "someTitle", "videoid": "some_id", "thumbnail": "someThumbnail", "video_url": "someUrl"}
sys.modules["__main__"].settings.getSetting.return_value = "1"
sys.argv = ["test1", "1", "test2"]
player = YouTubePlayer()
player.buildVideoObject = Mock()
player.buildVideoObject.return_value = (video, 200)
player.playVideo({"videoid": "some_id"})
sys.modules["__main__"].subtitles.addSubtitles.assert_called_with(video)
def test_playVideo_should_call_xbmc_setResolvedUrl(self):
sys.modules["__main__"].settings.getSetting.return_value = "0"
player = YouTubePlayer()
player.addSubtitles = Mock()
player.buildVideoObject = Mock()
player.buildVideoObject.return_value = ({"Title": "someTitle", "videoid": "some_id", "thumbnail": "someThumbnail", "video_url": "someUrl"}, 200)
sys.argv = ["test1", "1", "test2"]
player.playVideo({"videoid": "some_id"})
assert(sys.modules["__main__"].xbmcplugin.setResolvedUrl.call_count > 0)
def test_buildVideoObject_should_check_for_local_file_before_scraping(self):
sys.modules["__main__"].subtitles.getLocalFileSource.return_value = "somePath/someTitle.mp4"
sys.modules["__main__"].settings.getSetting.return_value = "1"
params = {"videoid": "some_id"}
player = YouTubePlayer()
player.getInfo = Mock()
video = {"videoid": "some_id", "Title": "someTitle"}
player.getInfo.return_value = (video, 200)
(video, status) = player.buildVideoObject(params)
sys.modules["__main__"].subtitles.getLocalFileSource.assert_called_with(params, video)
def test_buildVideoObject_should_not_scrape_webpage_if_local_file_is_found(self):
sys.modules["__main__"].subtitles.getLocalFileSource.return_value = "somePath/someTitle.mp4"
params = {"videoid": "some_id"}
player = YouTubePlayer()
player.getInfo = Mock()
player.extractVideoLinksFromYoutube = Mock()
video = {"videoid": "some_id", "Title": "someTitle"}
player.getInfo.return_value = (video, 200)
player.buildVideoObject(params)
assert(player.extractVideoLinksFromYoutube.call_count == 0)
def test_selectVideoQuality_should_add_user_agent_when_not_called_by_download_function(
self):
sys.modules["__main__"].settings.getSetting.return_value = "1"
player = YouTubePlayer()
url = player.selectVideoQuality({}, {
35: "SD",
22: "720p",
37: "1080p"
})
assert (url.find("| Mozilla/5.0 (MOCK)") > 0)
def test_buildVideoObject_should_check_for_local_file_before_scraping(self):
sys.modules["__main__"].subtitles.getLocalFileSource.return_value = "somePath/someTitle.mp4"
sys.modules["__main__"].settings.getSetting.return_value = "1"
params = {"videoid": "some_id"}
player = YouTubePlayer()
player.getInfo = Mock()
video = {"videoid": "some_id", "Title": "someTitle"}
player.getInfo.return_value = (video, 200)
(video, status) = player.buildVideoObject(params)
sys.modules["__main__"].subtitles.getLocalFileSource.assert_called_with(params, video)
def test_selectVideoQuality_should_limit_to_720p_if_user_has_selected_that_option(
self):
sys.modules["__main__"].settings.getSetting.return_value = "2"
player = YouTubePlayer()
url = player.selectVideoQuality({}, {
35: "SD",
22: "720p",
37: "1080p"
})
assert (url == "720p | Mozilla/5.0 (MOCK)")
def test_getInfo_should_fail_correctly_if_api_is_unavailable(self):
sys.modules["__main__"].cache.get.return_value = {}
sys.modules["__main__"].core._fetchPage.return_value = {"status": 200, "content": "something"}
sys.modules["__main__"].core.getVideoInfo.return_value = []
sys.modules["__main__"].language.return_value = "some_string"
player = YouTubePlayer()
(video, status) = player.getInfo({"videoid": "some_id"})
sys.modules["__main__"].common.log.assert_called_with("- Couldn't parse API output, YouTube doesn't seem to know this video id?")
sys.modules["__main__"].language.assert_called_with(30608)
assert(video["apierror"] == "some_string")
def test_playVideo_should_call_remove_from_watch_later_if_viewing_video_from_watch_later_queue(self):
player = YouTubePlayer()
sys.modules["__main__"].settings.getSetting.return_value = "0"
sys.argv = ["test1", "1", "test2"]
player.buildVideoObject = Mock()
player.buildVideoObject.return_value = ({"Title": "someTitle", "videoid": "some_id", "thumbnail": "someThumbnail", "video_url": "someUrl"}, 200)
player.addSubtitles = Mock()
call_params = {"videoid": "some_id", "watch_later": "true", "playlist": "playlist_id", "playlist_entry_id": "entry_id"}
player.playVideo(call_params)
sys.modules["__main__"].core.remove_from_watch_later.assert_called_with(call_params)
def test_buildVideoObject_with_hls(self):
sys.modules["__main__"].core._fetchPage.return_value = {"status": 200, "content": self.readTestInput("live-video-hls-page-S_pMsJw-_oA.html", False)}
sys.modules["__main__"].subtitles.getLocalFileSource.return_value = ""
sys.modules["__main__"].common.parseDOM.return_value = []
params = {"videoid": "some_id"}
player = YouTubePlayer()
player.getInfo = Mock()
video = {"videoid": "some_id", "Title": "someTitle"}
player.getInfo.return_value = (video, 200)
(video, status) = player.buildVideoObject(params)
assert(video["video_url"] == "http://www.youtube.com/api/manifest/hls_variant/id/S_pMsJw-_oA.2/sparams/cp%2Cid%2Cip%2Cipbits%2Cmaudio%2Cplaylist_type%2Cpmbypass%2Csource%2Cexpire/expire/1364700604/playlist_type/DVR/ipbits/8/upn/Ftpj_wWBeuQ/signature/4B46E966BDE27C0772BA90DE72F0B083F7C44E5A.6F1348A5EE63E4167F1E03C1DE268DABA3EB836E/fexp/932000%2C906383%2C902000%2C919512%2C929903%2C931202%2C900821%2C900823%2C931203%2C931401%2C908529%2C919373%2C930803%2C906836%2C920201%2C929602%2C930101%2C930603%2C900824%2C910223/ip/205.178.10.177/key/yt1/maudio/1/sver/3/cp/U0hVSVdLTl9FTkNONV9PRVJHOmtkY0VJTkJQay02/pmbypass/yes/source/yt_live_broadcast/file/index.m3u8")
def test_playVideo_should_call_remove_from_watch_later_if_viewing_video_from_watch_later_queue(
self):
player = YouTubePlayer()
sys.modules["__main__"].settings.getSetting.return_value = "0"
sys.argv = ["test1", "1", "test2"]
player.buildVideoObject = Mock()
player.buildVideoObject.return_value = ({
"Title": "someTitle",
"videoid": "some_id",
"thumbnail": "someThumbnail",
"video_url": "someUrl"
}, 200)
player.addSubtitles = Mock()
call_params = {
"videoid": "some_id",
"watch_later": "true",
"playlist": "playlist_id",
"playlist_entry_id": "entry_id"
}
player.playVideo(call_params)
sys.modules[
"__main__"].core.remove_from_watch_later.assert_called_with(
call_params)
def test_buildVideoObject_should_use_pre_defined_error_messages_on_missing_url(self):
sys.modules["__main__"].settings.getSetting.return_value = ""
player = YouTubePlayer()
player.getInfo = Mock()
player.getInfo.return_value = ({}, 303)
player.getLocalFileSource = Mock(return_value="")
player.extractVideoLinksFromYoutube = Mock(return_value = ({}, {}))
player.buildVideoObject({})
player.getInfo.assert_called_with({})
sys.modules["__main__"].language.assert_called_with(30618)
def test_playVideo_should_update_locally_stored_watched_status(self):
sys.modules["__main__"].settings.getSetting.return_value = "0"
sys.argv = ["test1", "1", "test2"]
player = YouTubePlayer()
player.buildVideoObject = Mock()
player.buildVideoObject.return_value = ({"Title": "someTitle", "videoid": "some_id", "thumbnail": "someThumbnail", "video_url": "someUrl"}, 200)
player.addSubtitles = Mock()
player.playVideo({"videoid": "some_id"})
sys.modules["__main__"].storage.storeValue.assert_called_with("vidstatus-some_id", "7" )
