[test_utils] Fix expand_path tests

This commit is contained in:
Sergey M․ 2017-03-26 03:07:56 +07:00
parent d212c93d16
commit 41c5e60dd5
No known key found for this signature in database
GPG Key ID: 2C393E0F18A9236D

View File

@ -218,11 +218,16 @@ class TestUtil(unittest.TestCase):
self.assertEqual(sanitize_path('./../abc'), '..\\abc')
def test_expand_path(self):
def env(var):
return '%{0}%'.format(var) if sys.platform == 'win32' else '${0}'.format(var)
compat_setenv('YOUTUBE-DL-EXPATH-PATH', 'expanded')
self.assertEqual(expand_path('%YOUTUBE-DL-EXPATH-PATH%'), 'expanded')
self.assertEqual(expand_path('%HOMEPATH%'), compat_getenv('HOMEPATH'))
self.assertEqual(expand_path(env('YOUTUBE-DL-EXPATH-PATH')), 'expanded')
self.assertEqual(expand_path(env('HOMEPATH')), compat_getenv('HOMEPATH'))
self.assertEqual(expand_path('~'), compat_getenv('HOME'))
self.assertEqual(expand_path('~/%YOUTUBE-DL-EXPATH-PATH%'), '%s/expanded' % compat_getenv('HOME'))
self.assertEqual(
expand_path('~/%s' % env('YOUTUBE-DL-EXPATH-PATH')),
'%s/expanded' % compat_getenv('HOME'))
def test_prepend_extension(self):
self.assertEqual(prepend_extension('abc.ext', 'temp'), 'abc.temp.ext')