send2trash/tests/test_plat_win.py

# 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))