2021-06-03 11:43:42 +02:00
|
|
|
#!/usr/bin/env python3
|
2022-06-24 13:06:16 +02:00
|
|
|
|
2016-02-20 16:10:39 +01:00
|
|
|
# Allow direct execution
|
|
|
|
import os
|
|
|
|
import sys
|
|
|
|
import unittest
|
2022-04-12 00:32:57 +02:00
|
|
|
|
2016-02-20 16:10:39 +01:00
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
|
2022-04-12 00:32:57 +02:00
|
|
|
|
2022-06-24 13:06:16 +02:00
|
|
|
from test.helper import FakeYDL, is_download_test
|
2021-02-24 19:45:56 +01:00
|
|
|
from yt_dlp.extractor import IqiyiIE
|
2016-02-20 16:10:39 +01:00
|
|
|
|
|
|
|
|
2022-04-11 17:10:28 +02:00
|
|
|
class WarningLogger:
|
2016-02-20 16:10:39 +01:00
|
|
|
def __init__(self):
|
|
|
|
self.messages = []
|
|
|
|
|
|
|
|
def warning(self, msg):
|
|
|
|
self.messages.append(msg)
|
|
|
|
|
|
|
|
def debug(self, msg):
|
|
|
|
pass
|
|
|
|
|
|
|
|
def error(self, msg):
|
|
|
|
pass
|
|
|
|
|
|
|
|
|
2021-07-23 16:48:15 +02:00
|
|
|
@is_download_test
|
2016-02-20 16:10:39 +01:00
|
|
|
class TestIqiyiSDKInterpreter(unittest.TestCase):
|
|
|
|
def test_iqiyi_sdk_interpreter(self):
|
2024-06-12 01:09:58 +02:00
|
|
|
"""
|
2016-02-20 16:10:39 +01:00
|
|
|
Test the functionality of IqiyiSDKInterpreter by trying to log in
|
|
|
|
|
|
|
|
If `sign` is incorrect, /validate call throws an HTTP 556 error
|
2024-06-12 01:09:58 +02:00
|
|
|
"""
|
2016-02-20 16:10:39 +01:00
|
|
|
logger = WarningLogger()
|
2022-03-18 21:53:33 +01:00
|
|
|
ie = IqiyiIE(FakeYDL({'logger': logger}))
|
|
|
|
ie._perform_login('foo', 'bar')
|
2016-02-20 16:10:39 +01:00
|
|
|
self.assertTrue('unable to log in:' in logger.messages[0])
|
|
|
|
|
2016-11-17 12:42:56 +01:00
|
|
|
|
2016-02-20 16:10:39 +01:00
|
|
|
if __name__ == '__main__':
|
|
|
|
unittest.main()
|