# encoding: utf-8 import pytest import codecs import os import sys from os import path as op from send2trash.compat import PY3 from send2trash import TrashPermissionError try: from configparser import ConfigParser except ImportError: # py2 from ConfigParser import ConfigParser # noqa: F401 from tempfile import mkdtemp, NamedTemporaryFile, mktemp import shutil import stat if sys.platform != "win32": import send2trash.plat_other from send2trash.plat_other import send2trash as s2t HOMETRASH = send2trash.plat_other.HOMETRASH else: pytest.skip("Skipping non-windows tests", allow_module_level=True) @pytest.fixture def testfile(): file = NamedTemporaryFile(dir=op.expanduser("~"), prefix="send2trash_test", delete=False) file.close() assert op.exists(file.name) is True yield file # Cleanup trash files on supported platforms if sys.platform != "win32": name = op.basename(file.name) # Remove trash files if they exist if op.exists(op.join(HOMETRASH, "files", name)): os.remove(op.join(HOMETRASH, "files", name)) os.remove(op.join(HOMETRASH, "info", name + ".trashinfo")) if op.exists(file.name): os.remove(file.name) @pytest.fixture def testfiles(): files = list( map( lambda index: NamedTemporaryFile( dir=op.expanduser("~"), prefix="send2trash_test{}".format(index), delete=False, ), range(10), ) ) [file.close() for file in files] assert all([op.exists(file.name) for file in files]) is True yield files filenames = [op.basename(file.name) for file in files] [os.remove(op.join(HOMETRASH, "files", filename)) for filename in filenames] [os.remove(op.join(HOMETRASH, "info", filename + ".trashinfo")) for filename in filenames] def test_trash(testfile): s2t(testfile.name) assert op.exists(testfile.name) is False def test_multitrash(testfiles): filenames = [file.name for file in testfiles] s2t(filenames) assert any([op.exists(filename) for filename in filenames]) is False def touch(path): with open(path, "a"): os.utime(path, None) def _filesys_enc(): enc = sys.getfilesystemencoding() # Get canonical name of codec return codecs.lookup(enc).name @pytest.fixture def testUnicodefile(): name = u"send2trash_tést1" file = op.join(op.expanduser(b"~"), name.encode("utf-8")) touch(file) assert op.exists(file) is True yield file # Cleanup trash files on supported platforms if sys.platform != "win32": # Remove trash files if they exist if op.exists(op.join(HOMETRASH, "files", name)): os.remove(op.join(HOMETRASH, "files", name)) os.remove(op.join(HOMETRASH, "info", name + ".trashinfo")) if op.exists(file): os.remove(file) @pytest.mark.skipif(_filesys_enc() == "ascii", reason="Requires Unicode filesystem") def test_trash_bytes(testUnicodefile): s2t(testUnicodefile) assert not op.exists(testUnicodefile) @pytest.mark.skipif(_filesys_enc() == "ascii", reason="Requires Unicode filesystem") def test_trash_unicode(testUnicodefile): s2t(testUnicodefile.decode(sys.getfilesystemencoding())) assert not op.exists(testUnicodefile) class ExtVol: def __init__(self, path): self.trashTopdir = path if PY3: self.trashTopdir_b = os.fsencode(self.trashTopdir) else: self.trashTopdir_b = self.trashTopdir def s_getdev(path): from send2trash.plat_other import is_parent st = os.lstat(path) if is_parent(self.trashTopdir, path): return "dev" return st.st_dev def s_ismount(path): if op.realpath(path) in (op.realpath(self.trashTopdir), op.realpath(self.trashTopdir_b),): return True return old_ismount(path) self.old_ismount = old_ismount = op.ismount self.old_getdev = send2trash.plat_other.get_dev send2trash.plat_other.os.path.ismount = s_ismount send2trash.plat_other.get_dev = s_getdev def cleanup(self): send2trash.plat_other.get_dev = self.old_getdev send2trash.plat_other.os.path.ismount = self.old_ismount shutil.rmtree(self.trashTopdir) @pytest.fixture def testExtVol(): trashTopdir = mkdtemp(prefix="s2t") volume = ExtVol(trashTopdir) fileName = "test.txt" filePath = op.join(volume.trashTopdir, fileName) touch(filePath) assert op.exists(filePath) is True yield volume, fileName, filePath volume.cleanup() def test_trash_topdir(testExtVol): trashDir = op.join(testExtVol[0].trashTopdir, ".Trash") os.mkdir(trashDir, 0o777 | stat.S_ISVTX) s2t(testExtVol[2]) assert op.exists(testExtVol[2]) is False assert op.exists(op.join(trashDir, str(os.getuid()), "files", testExtVol[1])) is True assert op.exists(op.join(trashDir, str(os.getuid()), "info", testExtVol[1] + ".trashinfo",)) is True # info relative path (if another test is added, with the same fileName/Path, # then it gets renamed etc.) cfg = ConfigParser() cfg.read(op.join(trashDir, str(os.getuid()), "info", testExtVol[1] + ".trashinfo")) assert (testExtVol[1] == cfg.get("Trash Info", "Path", raw=True)) is True def test_trash_topdir_fallback(testExtVol): s2t(testExtVol[2]) assert op.exists(testExtVol[2]) is False assert op.exists(op.join(testExtVol[0].trashTopdir, ".Trash-" + str(os.getuid()), "files", testExtVol[1],)) is True def test_trash_topdir_failure(testExtVol): os.chmod(testExtVol[0].trashTopdir, 0o500) # not writable to induce the exception pytest.raises(TrashPermissionError, s2t, [testExtVol[2]]) os.chmod(testExtVol[0].trashTopdir, 0o700) # writable to allow deletion def test_trash_symlink(testExtVol): # Use mktemp (race conditioney but no symlink equivalent) # Since is_parent uses realpath(), and our getdev uses is_parent, # this should work slDir = mktemp(prefix="s2t", dir=op.expanduser("~")) os.mkdir(op.join(testExtVol[0].trashTopdir, "subdir"), 0o700) filePath = op.join(testExtVol[0].trashTopdir, "subdir", testExtVol[1]) touch(filePath) os.symlink(op.join(testExtVol[0].trashTopdir, "subdir"), slDir) s2t(op.join(slDir, testExtVol[1])) assert op.exists(filePath) is False assert op.exists(op.join(testExtVol[0].trashTopdir, ".Trash-" + str(os.getuid()), "files", testExtVol[1],)) is True os.remove(slDir)