X-Git-Url: https://jfr.im/git/yt-dlp.git/blobdiff_plain/0c3d0f51778b153f65c21906031c2e091fcfb641..bef9a9e5361fd7a72e21d0f1a8c8afb70d89e8c5:/test/test_overwrites.py diff --git a/test/test_overwrites.py b/test/test_overwrites.py index d5c866c83..6954c07f9 100644 --- a/test/test_overwrites.py +++ b/test/test_overwrites.py @@ -1,20 +1,22 @@ -#!/usr/bin/env python -from __future__ import unicode_literals +#!/usr/bin/env python3 +# Allow direct execution import os -from os.path import join -import subprocess import sys import unittest + sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) -from test.helper import try_rm +import subprocess + +from test.helper import is_download_test, try_rm root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) -download_file = join(root_dir, 'test.webm') +download_file = os.path.join(root_dir, 'test.webm') +@is_download_test class TestOverwrites(unittest.TestCase): def setUp(self): # create an empty file @@ -23,7 +25,7 @@ def setUp(self): def test_default_overwrites(self): outp = subprocess.Popen( [ - sys.executable, 'youtube_dlc/__main__.py', + sys.executable, 'yt_dlp/__main__.py', '-o', 'test.webm', 'https://www.youtube.com/watch?v=jNQXAC9IVRw' ], cwd=root_dir, stdout=subprocess.PIPE, stderr=subprocess.PIPE) @@ -35,7 +37,7 @@ def test_default_overwrites(self): def test_yes_overwrites(self): outp = subprocess.Popen( [ - sys.executable, 'youtube_dlc/__main__.py', '--yes-overwrites', + sys.executable, 'yt_dlp/__main__.py', '--yes-overwrites', '-o', 'test.webm', 'https://www.youtube.com/watch?v=jNQXAC9IVRw' ], cwd=root_dir, stdout=subprocess.PIPE, stderr=subprocess.PIPE) @@ -45,7 +47,7 @@ def test_yes_overwrites(self): self.assertTrue(os.path.getsize(download_file) > 1) def tearDown(self): - try_rm(join(root_dir, 'test.webm')) + try_rm(os.path.join(root_dir, 'test.webm')) if __name__ == '__main__':