mirror of
				https://github.com/ytdl-org/youtube-dl.git
				synced 2025-10-29 09:26:20 -07:00 
			
		
		
		
	adding TestCase class and corresponding test
This commit is contained in:
		| @@ -2,22 +2,47 @@ | ||||
|  | ||||
| # Various small unit tests | ||||
|  | ||||
| import unittest | ||||
|  | ||||
| #from youtube_dl.utils import htmlentity_transform | ||||
| from youtube_dl.utils import timeconvert | ||||
| from youtube_dl.utils import sanitize_filename | ||||
| from youtube_dl.utils import unescapeHTML | ||||
| from youtube_dl.utils import orderedSet | ||||
|  | ||||
| def test_sanitize_filename(): | ||||
| 	assert sanitize_filename(u'abc') == u'abc' | ||||
| 	assert sanitize_filename(u'abc_d-e') == u'abc_d-e' | ||||
|  | ||||
| 	assert sanitize_filename(u'123') == u'123' | ||||
| class TestUtil(unittest.TestCase): | ||||
|     def test_timeconvert(self): | ||||
|         self.assertIsNone(timeconvert('')) | ||||
|         self.assertIsNone(timeconvert('bougrg')) | ||||
|  | ||||
| 	assert u'/' not in sanitize_filename(u'abc/de') | ||||
| 	assert u'abc' in sanitize_filename(u'abc/de') | ||||
| 	assert u'de' in sanitize_filename(u'abc/de') | ||||
| 	assert u'/' not in sanitize_filename(u'abc/de///') | ||||
|     def test_sanitize_filename(self): | ||||
|         self.assertEqual(sanitize_filename(u'abc'), u'abc') | ||||
|         self.assertEqual(sanitize_filename(u'abc_d-e'), u'abc_d-e') | ||||
|  | ||||
|         self.assertEqual(sanitize_filename(u'123'), u'123') | ||||
|  | ||||
|         self.assertNotIn(u'/', sanitize_filename(u'abc/de')) | ||||
|         self.assertNotIn(u'abc', sanitize_filename(u'abc/de')) | ||||
|         self.assertIn(u'de', sanitize_filename(u'abc/de')) | ||||
|         self.assertNotIn(u'/', sanitize_filename(u'abc/de///')) | ||||
|  | ||||
|         self.assertIn(u'\\', sanitize_filename(u'abc\\de')) | ||||
|         self.assertIn(u'abc', sanitize_filename(u'abc\\de')) | ||||
|         self.assertIn(u'de', sanitize_filename(u'abc\\de')) | ||||
|  | ||||
|         self.assertEqual(sanitize_filename(u'ä'), u'ä') | ||||
|         self.assertEqual(sanitize_filename(u'кириллица'), u'кириллица') | ||||
|  | ||||
|     def test_ordered_set(self): | ||||
|         self.assertEqual(orderedSet([1,1,2,3,4,4,5,6,7,3,5]), [1,2,3,4,5,6,7]) | ||||
|         self.assertEqual(orderedSet([]), []) | ||||
|         self.assertEqual(orderedSet([1]), [1]) | ||||
|         #keep the list ordered | ||||
|         self.assertEqual(orderedSet([135,1,1,1]), [135,1]) | ||||
|  | ||||
|     def test_unescape_html(self): | ||||
|         self.assertEqual(unescapeHTML(u"%20;"), u"%20;") | ||||
|         self.assertEqual(unescapeHTML(u"gre&tre&yre"), [u'gre', u'tre', u'yre']) | ||||
|  | ||||
| 	assert u'\\' not in sanitize_filename(u'abc\\de') | ||||
| 	assert u'abc' in sanitize_filename(u'abc\\de') | ||||
| 	assert u'de' in sanitize_filename(u'abc\\de') | ||||
|  | ||||
| 	assert sanitize_filename(u'ä') == u'ä' | ||||
| 	assert sanitize_filename(u'кириллица') == u'кириллица' | ||||
|   | ||||
		Reference in New Issue
	
	Block a user