|
|
@ -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 |
|
|
|
|
|
|
|
|
|
|
|
class TestUtil(unittest.TestCase): |
|
|
|
def test_timeconvert(self): |
|
|
|
self.assertIsNone(timeconvert('')) |
|
|
|
self.assertIsNone(timeconvert('bougrg')) |
|
|
|
|
|
|
|
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')) |
|
|
|
|
|
|
|
def test_sanitize_filename(): |
|
|
|
assert sanitize_filename(u'abc') == u'abc' |
|
|
|
assert sanitize_filename(u'abc_d-e') == u'abc_d-e' |
|
|
|
self.assertEqual(sanitize_filename(u'ä'), u'ä') |
|
|
|
self.assertEqual(sanitize_filename(u'кириллица'), u'кириллица') |
|
|
|
|
|
|
|
assert sanitize_filename(u'123') == u'123' |
|
|
|
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]) |
|
|
|
|
|
|
|
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_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'кириллица' |