[test_utils] Add tests for determine_ext

2015-11-22 06:33:52 +06:00 · 2015-11-22 06:33:52 +06:00 · 5035536e3f
parent 3e12bc583a
commit 5035536e3f
1 changed files with 5 additions and 0 deletions
--- a/test/test_utils.py
+++ b/test/test_utils.py
@ -21,6 +21,7 @@ from youtube_dl.utils import (
    clean_html,
    DateRange,
    detect_exe_version,
+    determine_ext,
    encodeFilename,
    escape_rfc3986,
    escape_url,
@ -238,6 +239,10 @@ class TestUtil(unittest.TestCase):
        self.assertEqual(unified_strdate('25-09-2014'), '20140925')
        self.assertEqual(unified_strdate('UNKNOWN DATE FORMAT'), None)

+    def test_determine_ext(self):
+        self.assertEqual(determine_ext('http://example.com/foo/bar.mp4/?download'), 'mp4')
+        self.assertEqual(determine_ext('http://example.com/foo/bar/?download', None), None)
+
    def test_find_xpath_attr(self):
        testxml = '''<root>
            <node/>