mirror of
				https://github.com/yt-dlp/yt-dlp.git
				synced 2025-10-30 22:25:19 +00:00 
			
		
		
		
	[test_iqiyi_sdk_interpreter] Add test for iQiyi login
This commit is contained in:
		
							
								
								
									
										47
									
								
								test/test_iqiyi_sdk_interpreter.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										47
									
								
								test/test_iqiyi_sdk_interpreter.py
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,47 @@ | ||||
| #!/usr/bin/env python | ||||
|  | ||||
| from __future__ import unicode_literals | ||||
|  | ||||
| # Allow direct execution | ||||
| import os | ||||
| import sys | ||||
| import unittest | ||||
| sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) | ||||
|  | ||||
| from test.helper import FakeYDL | ||||
| from youtube_dl.extractor import IqiyiIE | ||||
|  | ||||
|  | ||||
| class IqiyiIEWithCredentials(IqiyiIE): | ||||
|     def _get_login_info(self): | ||||
|         return 'foo', 'bar' | ||||
|  | ||||
|  | ||||
| class WarningLogger(object): | ||||
|     def __init__(self): | ||||
|         self.messages = [] | ||||
|  | ||||
|     def warning(self, msg): | ||||
|         self.messages.append(msg) | ||||
|  | ||||
|     def debug(self, msg): | ||||
|         pass | ||||
|  | ||||
|     def error(self, msg): | ||||
|         pass | ||||
|  | ||||
|  | ||||
| class TestIqiyiSDKInterpreter(unittest.TestCase): | ||||
|     def test_iqiyi_sdk_interpreter(self): | ||||
|         ''' | ||||
|         Test the functionality of IqiyiSDKInterpreter by trying to log in | ||||
|  | ||||
|         If `sign` is incorrect, /validate call throws an HTTP 556 error | ||||
|         ''' | ||||
|         logger = WarningLogger() | ||||
|         ie = IqiyiIEWithCredentials(FakeYDL({'logger': logger})) | ||||
|         ie._login() | ||||
|         self.assertTrue('unable to log in:' in logger.messages[0]) | ||||
|  | ||||
| if __name__ == '__main__': | ||||
|     unittest.main() | ||||
		Reference in New Issue
	
	Block a user
	 Yen Chi Hsuan
					Yen Chi Hsuan