X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=tests%2Ftest_remove_old_files.py;h=93f8669eebe0dd7bbbb35169ae40cc53fc6ea069;hb=e78080336c7f7b636cbc6d168bac1d54cec41a09;hp=95dd98cb46111219b7f7df24f43d9a39bdd09e64;hpb=c6e98694274a9c3e09fa3192a5f247607081c1ae;p=ppu.git diff --git a/tests/test_remove_old_files.py b/tests/test_remove_old_files.py index 95dd98c..93f8669 100755 --- a/tests/test_remove_old_files.py +++ b/tests/test_remove_old_files.py @@ -2,11 +2,10 @@ from time import time import os -import shutil import subprocess import sys -from tempfile import mkdtemp -from find_in_path import find_in_path +from ppu_tu import setup, teardown, find_in_path # noqa +from ppu_tu import create_files, assert_files_exist, assert_files_not_exist tmp_dir = None @@ -17,41 +16,6 @@ if not test_prog_path: sys.exit("Cannot find remove-old-files.py in %s" % os.environ["PATH"]) -def setup(): - global tmp_dir - tmp_dir = mkdtemp() - os.chdir(tmp_dir) - - -def teardown(): - os.chdir(os.sep) # To the root of the FS - shutil.rmtree(tmp_dir) - - -def create_files(files, subdirectory=None): - if subdirectory: - os.makedirs(subdirectory) - else: - subdirectory = '' - for fname in files: - with open(os.path.join(subdirectory, fname), 'w'): - pass - - -def assert_files_exist(files): - if isinstance(files, str): - files = [files] - for fname in files: - assert os.path.exists(fname) - - -def assert_files_not_exist(files): - if isinstance(files, str): - files = [files] - for fname in files: - assert not os.path.exists(fname) - - def test_remove_old_files(): create_files(['test1', 'test2']) assert_files_exist(['test1', 'test2'])