From: Sergey M․ <dstftw@gmail.com>
Date: Sun, 22 Nov 2015 00:33:52 +0000 (+0600)
Subject: [test_utils] Add tests for determine_ext
X-Git-Url: http://git.bitcoin.ninja/?a=commitdiff_plain;h=5035536e3f32d4c47b2d3067c12e074cb9a4a199;p=youtube-dl

[test_utils] Add tests for determine_ext
---

diff --git a/test/test_utils.py b/test/test_utils.py
index ea1ff0547..4f0d9e481 100644
--- 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/>