1
0
mirror of https://github.com/arsenetar/send2trash.git synced 2024-12-08 21:49:03 +00:00
send2trash/tests/test_script_main.py
Matthew D. Scholefield 9ede898c3e
Create __main__.py (Fixes #15) (#38)
This adds a main method that mimics the behavior of `rm`. It can be called via `python -m send2trash somefile`.
2020-05-27 07:52:00 -04:00

33 lines
1004 B
Python

# encoding: utf-8
import os
import unittest
from tempfile import NamedTemporaryFile
from os import path as op
from send2trash.__main__ import main as trash_main
from tests.test_plat_other import HOMETRASH
class TestMainTrash(unittest.TestCase):
def setUp(self):
self.file = NamedTemporaryFile(dir=op.expanduser('~'), prefix='send2trash_test', delete=False)
def test_trash(self):
trash_main(['-v', self.file.name])
self.assertFalse(op.exists(self.file.name))
def test_no_args(self):
self.assertRaises(SystemExit, trash_main, [])
self.assertRaises(SystemExit, trash_main, ['-v'])
self.assertTrue(op.exists(self.file.name))
trash_main([self.file.name]) # Trash the file so tearDown runs properly
def tearDown(self):
name = op.basename(self.file.name)
os.remove(op.join(HOMETRASH, 'files', name))
os.remove(op.join(HOMETRASH, 'info', name + '.trashinfo'))
if __name__ == '__main__':
unittest.main()