70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
# coding: utf-8
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import unittest
|
|
from os import path as op
|
|
from tempfile import gettempdir
|
|
|
|
from send2trash import send2trash as s2t
|
|
|
|
|
|
@unittest.skipIf(sys.platform != 'win32', 'Windows only')
|
|
class TestNormal(unittest.TestCase):
|
|
def setUp(self):
|
|
self.dirname = '\\\\?\\' + op.join(gettempdir(), 'python.send2trash')
|
|
self.file = op.join(self.dirname, 'testfile.txt')
|
|
self._create_tree(self.file)
|
|
|
|
def tearDown(self):
|
|
shutil.rmtree(self.dirname, ignore_errors=True)
|
|
|
|
def _create_tree(self, path):
|
|
dirname = op.dirname(path)
|
|
if not op.isdir(dirname):
|
|
os.makedirs(dirname)
|
|
with open(path, 'w') as writer:
|
|
writer.write('send2trash test')
|
|
|
|
def test_trash_file(self):
|
|
s2t(self.file)
|
|
self.assertFalse(op.exists(self.file))
|
|
|
|
def test_file_not_found(self):
|
|
file = op.join(self.dirname, 'otherfile.txt')
|
|
self.assertRaises(WindowsError, s2t, file)
|
|
|
|
@unittest.skipIf(sys.platform != 'win32', 'Windows only')
|
|
class TestLongPath(unittest.TestCase):
|
|
def setUp(self):
|
|
filename = 'A' * 100
|
|
self.dirname = '\\\\?\\' + op.join(gettempdir(), filename)
|
|
self.file = op.join(
|
|
self.dirname,
|
|
filename,
|
|
filename, # From there, the path is not trashable from Explorer
|
|
filename,
|
|
filename + '.txt')
|
|
self._create_tree(self.file)
|
|
|
|
def tearDown(self):
|
|
shutil.rmtree(self.dirname, ignore_errors=True)
|
|
|
|
def _create_tree(self, path):
|
|
dirname = op.dirname(path)
|
|
if not op.isdir(dirname):
|
|
os.makedirs(dirname)
|
|
with open(path, 'w') as writer:
|
|
writer.write('Looong filename!')
|
|
|
|
def test_trash_file(self):
|
|
s2t(self.file)
|
|
self.assertFalse(op.exists(self.file))
|
|
|
|
@unittest.skipIf(
|
|
op.splitdrive(os.getcwd())[0] != op.splitdrive(gettempdir())[0],
|
|
'Cannot trash long path from other drive')
|
|
def test_trash_folder(self):
|
|
s2t(self.dirname)
|
|
self.assertFalse(op.exists(self.dirname))
|