mirrors/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2025-06-27 17:08:32 +00:00
Code Issues Projects Releases Wiki Activity

add test

Browse Source
This commit is contained in:
doe1080 2025-05-10 12:43:44 +09:00
parent b497cf7123
commit 8057c858ba
1 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
test/test_utils.py
View File

@ -1786,6 +1786,9 @@ def test_get_element_html_by_class(self):
GET_ELEMENT_BY_ATTRIBUTE_TEST_STRING = '''
<div itemprop="author" itemscope>foo</div>
'''
VOID_ELEMENT_TEST_STRING = '''
<img alt="foo" src="bar.png"><img alt="foobar" src="baz.jpg"><img alt="foo"/>
'''
def test_get_element_by_attribute(self):
html = self.GET_ELEMENT_BY_CLASS_TEST_STRING
@ -1798,6 +1801,10 @@ def test_get_element_by_attribute(self):
self.assertEqual(get_element_by_attribute('itemprop', 'author', html), 'foo')
html = self.VOID_ELEMENT_TEST_STRING
self.assertEqual(get_element_by_attribute('alt', 'foo', html), '')
def test_get_element_html_by_attribute(self):
html = self.GET_ELEMENT_BY_CLASS_TEST_STRING
@ -1809,6 +1816,10 @@ def test_get_element_html_by_attribute(self):
self.assertEqual(get_element_html_by_attribute('itemprop', 'author', html), html.strip())
html = self.VOID_ELEMENT_TEST_STRING
self.assertEqual(get_element_html_by_attribute('alt', 'foo', html), '<img alt="foo" src="bar.png">')
GET_ELEMENTS_BY_CLASS_TEST_STRING = '''
<span class="foo bar">nice</span><span class="foo bar">also nice</span>
'''
@ -1833,6 +1844,10 @@ def test_get_elements_by_attribute(self):
self.assertEqual(get_elements_by_attribute('class', 'foo', html), [])
self.assertEqual(get_elements_by_attribute('class', 'no-such-foo', html), [])
html = self.VOID_ELEMENT_TEST_STRING
self.assertEqual(get_elements_by_attribute('alt', 'foo', html), ['', ''])
def test_get_elements_html_by_attribute(self):
html = self.GET_ELEMENTS_BY_CLASS_TEST_STRING
@ -1840,6 +1855,11 @@ def test_get_elements_html_by_attribute(self):
self.assertEqual(get_elements_html_by_attribute('class', 'foo', html), [])
self.assertEqual(get_elements_html_by_attribute('class', 'no-such-foo', html), [])
html = self.VOID_ELEMENT_TEST_STRING
self.assertEqual(get_elements_html_by_attribute(
'alt', 'foo', html), ['<img alt="foo" src="bar.png">', '<img alt="foo"/>'])
def test_get_elements_text_and_html_by_attribute(self):
html = self.GET_ELEMENTS_BY_CLASS_TEST_STRING
@ -1852,6 +1872,11 @@ def test_get_elements_text_and_html_by_attribute(self):
self.assertEqual(list(get_elements_text_and_html_by_attribute(
'class', 'foo', '<a class="foo">nice</a><span class="foo">nice</span>', tag='a')), [('nice', '<a class="foo">nice</a>')])
html = self.VOID_ELEMENT_TEST_STRING
self.assertEqual(list(get_elements_text_and_html_by_attribute(
'alt', 'foo', html, tag='img')), [('', '<img alt="foo" src="bar.png">'), ('', '<img alt="foo"/>')])
GET_ELEMENT_BY_TAG_TEST_STRING = '''
random text lorem ipsum</p>
<div>
@ -1880,6 +1905,10 @@ def test_get_element_text_and_html_by_tag(self):
(self.GET_ELEMENT_BY_TAG_RES_INNERSPAN_TEXT, self.GET_ELEMENT_BY_TAG_RES_INNERSPAN_HTML))
self.assertRaises(compat_HTMLParseError, get_element_text_and_html_by_tag, 'article', html)
html = self.VOID_ELEMENT_TEST_STRING
self.assertEqual(get_element_text_and_html_by_tag('img', html), ('', '<img alt="foo" src="bar.png">'))
def test_iri_to_uri(self):
self.assertEqual(
iri_to_uri('https://www.google.com/search?q=foo&ie=utf-8&oe=utf-8&client=firefox-b'),