def test_downloadSubtitle_should_call_transformSubtitleXMLtoSRT(self): player = YouTubeSubtitleControl() sys.modules["__main__"].core._fetchPage = Mock() sys.modules["__main__"].core._fetchPage.return_value = { "status": 200, "content": "nothingness" } subtitlesettings = ["true", "2", "true"] sys.modules["__main__"].settings.getSetting = Mock() sys.modules[ "__main__"].settings.getSetting.side_effect = lambda x: subtitlesettings.pop( ) player.transformAnnotationToSSA = Mock() player.transformAnnotationToSSA.return_value = ("", "style") player.getSubtitleUrl = Mock() player.getSubtitleUrl.return_value = "" player.getTranscriptionUrl = Mock() player.getTranscriptionUrl.return_value = "something" player.transformSubtitleXMLtoSRT = Mock() player.transformSubtitleXMLtoSRT.return_value = "" player.downloadSubtitle() player.transformSubtitleXMLtoSRT.assert_called_with("nothingness")
def test_getTranscriptionUrl_should_call_return_correct_url(self): player = YouTubeSubtitleControl() sys.modules["__main__"].core._fetchPage = Mock() sys.modules["__main__"].core._fetchPage.return_value = {"status": 303, "content": ""} sys.modules["__main__"].settings.getSetting = Mock() sys.modules["__main__"].settings.getSetting.return_value = "1" ret = player.getTranscriptionUrl({"videoid": "some_id", "ttsurl": "http://some.url/transcript"}) print ret assert(ret == "http://some.url/transcript&type=trackformat=1&lang=en&kind=asr&name=&v=some_id&tlang=en")
def test_getTranscriptionUrl_should_call_return_correct_url(self): player = YouTubeSubtitleControl() sys.modules["__main__"].core._fetchPage = Mock() sys.modules["__main__"].core._fetchPage.return_value = {"status": 303, "content": ""} sys.modules["__main__"].settings.getSetting = Mock() sys.modules["__main__"].settings.getSetting.return_value = "1" ret = player.getTranscriptionUrl({"videoid": "some_id", "ttsurl": "http://some.url/transcript"}) print ret assert(ret == "http://some.url/transcript&type=trackformat=1&lang=en&kind=asr&name=&v=some_id&tlang=en")
def test_downloadSubtitle_should_call_getTranscriptionUrl(self): player = YouTubeSubtitleControl() sys.modules["__main__"].core._fetchPage = Mock() sys.modules["__main__"].core._fetchPage.return_value = {"status": 200, "content": "nothingness"} subtitlesettings = ["true", "2", "true"] sys.modules["__main__"].settings.getSetting = Mock() sys.modules["__main__"].settings.getSetting.side_effect = lambda x: subtitlesettings.pop() player.transformAnnotationToSSA = Mock() player.transformAnnotationToSSA.return_value = ("", "style") player.getSubtitleUrl = Mock() player.getSubtitleUrl.return_value = "" player.getTranscriptionUrl = Mock() player.getTranscriptionUrl.return_value = "" player.downloadSubtitle() player.getTranscriptionUrl.assert_called_with({})