class TestListingInRestoreCmd:
def setUp(self):
self.cmd = RestoreCmd(None, None, None, None, None)
self.cmd.curdir = lambda: "dir"
self.cmd.handle_trashed_files = self.capture_trashed_files
def test_with_no_args_and_files_in_trashcan(self):
def some_files():
yield FakeTrashedFile('<date>', 'dir/location')
yield FakeTrashedFile('<date>', 'dir/location')
yield FakeTrashedFile('<date>', 'anotherdir/location')
self.cmd.all_trashed_files = some_files
self.cmd.run(['trash-restore'])
assert_equals(['dir/location', 'dir/location'],
self.original_locations)
def test_with_no_args_and_files_in_trashcan(self):
def some_files():
yield FakeTrashedFile('<date>', 'dir/location')
yield FakeTrashedFile('<date>', 'dir/location')
yield FakeTrashedFile('<date>', 'specific/path')
self.cmd.all_trashed_files = some_files
self.cmd.run(['trash-restore', 'specific/path'])
assert_equals(['specific/path'], self.original_locations)
def capture_trashed_files(self, arg):
self.original_locations = []
for trashed_file in arg:
self.original_locations.append(trashed_file.original_location)
class TestListingInRestoreCmd:
def setUp(self):
trash_directories = make_trash_directories()
trashed_files = TrashedFiles(trash_directories, None, contents_of)
self.cmd = RestoreCmd(None, None,
exit=None,
input=None,
curdir=lambda: "dir",
trashed_files=trashed_files,
mount_points=os_mount_points)
self.cmd.handle_trashed_files = self.capture_trashed_files
self.trashed_files = Mock(spec=['all_trashed_files'])
self.cmd.trashed_files = self.trashed_files
def test_with_no_args_and_files_in_trashcan(self):
self.trashed_files.all_trashed_files.return_value = [
FakeTrashedFile('<date>', 'dir/location'),
FakeTrashedFile('<date>', 'dir/location'),
FakeTrashedFile('<date>', 'anotherdir/location')
]
self.cmd.run(['trash-restore'])
assert_equal([
'dir/location'
, 'dir/location'
] ,self.original_locations)
def test_with_no_args_and_files_in_trashcan_2(self):
self.trashed_files.all_trashed_files.return_value = [
FakeTrashedFile('<date>', 'dir/location'),
FakeTrashedFile('<date>', 'dir/location'),
FakeTrashedFile('<date>', 'specific/path'),
]
self.cmd.run(['trash-restore', 'specific/path'])
assert_equal([
'specific/path'
] ,self.original_locations)
def capture_trashed_files(self,arg):
self.original_locations = []
for trashed_file in arg:
self.original_locations.append(trashed_file.original_location)
class TestListingInRestoreCmd(unittest.TestCase):
def setUp(self):
trash_directories = make_trash_directories()
self.logger = Mock(spec=[])
trashed_files = TrashedFiles(self.logger,
trash_directories,
None,
contents_of)
self.cmd = RestoreCmd(None, None,
exit=None,
input=None,
curdir=lambda: "dir",
trashed_files=trashed_files,
mount_points=os_mount_points,
fs=restore.FileSystem())
self.cmd.handle_trashed_files = self.capture_trashed_files
self.trashed_files = Mock(spec=['all_trashed_files'])
self.cmd.trashed_files = self.trashed_files
def test_with_no_args_and_files_in_trashcan(self):
self.trashed_files.all_trashed_files.return_value = [
FakeTrashedFile('<date>', 'dir/location'),
FakeTrashedFile('<date>', 'dir/location'),
FakeTrashedFile('<date>', 'anotherdir/location')
]
self.cmd.run(['trash-restore'])
assert [
'dir/location'
, 'dir/location'
] ==self.original_locations
def test_with_no_args_and_files_in_trashcan_2(self):
self.trashed_files.all_trashed_files.return_value = [
FakeTrashedFile('<date>', '/dir/location'),
FakeTrashedFile('<date>', '/dir/location'),
FakeTrashedFile('<date>', '/specific/path'),
]
self.cmd.run(['trash-restore', '/specific/path'])
assert self.original_locations == ['/specific/path']
def test_with_with_path_prefix_bug(self):
self.trashed_files.all_trashed_files.return_value = [
FakeTrashedFile('<date>', '/prefix'),
FakeTrashedFile('<date>', '/prefix-with-other'),
]
self.cmd.run(['trash-restore', '/prefix'])
assert self.original_locations == ['/prefix']
def capture_trashed_files(self,arg):
self.original_locations = []
for trashed_file in arg:
self.original_locations.append(trashed_file.original_location)
class TestTrashRestoreCmd:
def setUp(self):
self.stdout = StringIO()
self.stderr = StringIO()
self.cmd = RestoreCmd(stdout=self.stdout,
stderr=self.stderr,
environ={},
exit=self.capture_exit_status,
input=lambda x: self.user_reply,
version=None)
def capture_exit_status(self, exit_status):
self.exit_status = exit_status
def test_should_print_version(self):
self.cmd.version = '1.2.3'
self.cmd.run(['trash-restore', '--version'])
assert_equals('trash-restore 1.2.3\n', self.stdout.getvalue())
def test_with_no_args_and_no_files_in_trashcan(self):
self.cmd.curdir = lambda: "cwd"
self.cmd.run(['trash-restore'])
assert_equals("No files trashed from current dir ('cwd')\n",
self.stdout.getvalue())
def test_until_the_restore_intgration(self):
from trashcli.fs import remove_file
from trashcli.fs import contents_of
self.user_reply = '0'
open('orig_file', 'w').write('original')
open('info_file', 'w').write('')
remove_file('parent/path')
remove_file('parent')
trashed_file = TrashedFile('parent/path', None, 'info_file',
'orig_file')
self.cmd.restore_asking_the_user([trashed_file])
assert_equals('', self.stdout.getvalue())
assert_equals('', self.stderr.getvalue())
assert_true(not os.path.exists('info_file'))
assert_true(not os.path.exists('orig_file'))
assert_true(os.path.exists('parent/path'))
assert_equals('original', contents_of('parent/path'))
def test_until_the_restore_unit(self):
trashed_file = TrashedFile('parent/path', None, 'info_file',
'orig_file')
fs = Mock()
self.cmd.fs = fs
self.cmd.path_exists = lambda _: False
self.user_reply = '0'
self.cmd.restore_asking_the_user([trashed_file])
assert_equals('', self.stdout.getvalue())
assert_equals('', self.stderr.getvalue())
assert_equals([
call.mkdirs('parent'),
call.move('orig_file', 'parent/path'),
call.remove_file('info_file')
], fs.mock_calls)
def test_when_user_reply_with_empty_string(self):
self.user_reply = ''
self.cmd.restore_asking_the_user([])
assert_equals('Exiting\n', self.stdout.getvalue())
def test_when_user_reply_with_not_number(self):
self.user_reply = 'non numeric'
self.cmd.restore_asking_the_user([])
assert_equals('Invalid entry\n', self.stderr.getvalue())
assert_equals('', self.stdout.getvalue())
assert_equals(1, self.exit_status)
def test_when_user_reply_with_an_out_of_range_number(self):
self.user_reply = '100'
self.cmd.restore_asking_the_user([])
assert_equals('Invalid entry\n', self.stderr.getvalue())
assert_equals('', self.stdout.getvalue())
assert_equals(1, self.exit_status)
class TestTrashRestoreCmd:
def setUp(self):
self.stdout = StringIO()
self.stderr = StringIO()
self.cmd = RestoreCmd(stdout=self.stdout,
stderr=self.stderr,
environ={},
exit = self.capture_exit_status,
input =lambda x: self.user_reply,
version = None)
def capture_exit_status(self, exit_status):
self.exit_status = exit_status
def test_should_print_version(self):
self.cmd.version = '1.2.3'
self.cmd.run(['trash-restore', '--version'])
assert_equals('trash-restore 1.2.3\n', self.stdout.getvalue())
def test_with_no_args_and_no_files_in_trashcan(self):
self.cmd.curdir = lambda: "cwd"
self.cmd.run(['trash-restore'])
assert_equals("No files trashed from current dir ('cwd')\n",
self.stdout.getvalue())
def test_until_the_restore(self):
from trashcli.fs import remove_file
from trashcli.fs import contents_of
self.user_reply = '0'
file('orig_file', 'w').write('original')
file('info_file', 'w').write('')
remove_file('parent/path')
remove_file('parent')
trashed_file = TrashedFile(
'parent/path',
None,
'info_file',
'orig_file')
self.cmd.restore_asking_the_user([trashed_file])
assert_equals('', self.stdout.getvalue())
assert_equals('', self.stderr.getvalue())
assert_true(not os.path.exists('info_file'))
assert_true(not os.path.exists('orig_file'))
assert_true(os.path.exists('parent/path'))
assert_equals('original', contents_of('parent/path'))
def test_until_the_restore_unit(self):
from mock import Mock, call
trashed_file = TrashedFile(
'parent/path',
None,
'info_file',
'orig_file')
fs = Mock()
self.cmd.fs = fs
self.cmd.path_exists = lambda _: False
self.user_reply = '0'
self.cmd.restore_asking_the_user([trashed_file])
assert_equals('', self.stdout.getvalue())
assert_equals('', self.stderr.getvalue())
assert_equals([
call.mkdirs('parent')
, call.move('orig_file', 'parent/path')
, call.remove_file('info_file')
], fs.mock_calls)
def test_with_no_args_and_files_in_trashcan(self):
class thing(object):
pass
def some_files(append, _):
t = thing()
t.deletion_date = None
t.path = None
append(t)
self.cmd.curdir = lambda: "cwd"
self.cmd.for_all_trashed_file_in_dir = some_files
self.cmd.input = lambda _ : ""
self.cmd.run(['trash-restore'])
assert_equals(" 0 None None\n"
"Exiting\n",
self.stdout.getvalue())
def test_when_user_reply_with_empty_string(self):
self.user_reply = ''
self.cmd.restore_asking_the_user([])
assert_equals('Exiting\n', self.stdout.getvalue())
def test_when_user_reply_with_empty_string(self):
self.user_reply = 'non numeric'
self.cmd.restore_asking_the_user([])
assert_equals('Invalid entry\n', self.stderr.getvalue())
assert_equals('', self.stdout.getvalue())
assert_equals(1, self.exit_status)
def test_when_user_reply_with_out_of_range_number(self):
self.user_reply = '100'
self.cmd.restore_asking_the_user([])
assert_equals('Invalid entry\n', self.stderr.getvalue())
assert_equals('', self.stdout.getvalue())
assert_equals(1, self.exit_status)
class TestTrashRestoreCmd(unittest.TestCase):
def setUp(self):
self.stdout = StringIO()
self.stderr = StringIO()
trash_directories = make_trash_directories()
self.logger = Mock(spec=[])
trashed_files = TrashedFiles(self.logger,
trash_directories,
TrashDirectory(),
contents_of)
self.fs = Mock(spec=restore.FileSystem)
self.cmd = RestoreCmd(stdout=self.stdout,
stderr=self.stderr,
exit = self.capture_exit_status,
input =lambda x: self.user_reply,
version=None,
trashed_files=trashed_files,
mount_points=os_mount_points,
fs=self.fs)
def capture_exit_status(self, exit_status):
self.exit_status = exit_status
def test_should_print_version(self):
self.cmd.version = '1.2.3'
self.cmd.run(['trash-restore', '--version'])
assert 'trash-restore 1.2.3\n' == self.stdout.getvalue()
def test_with_no_args_and_no_files_in_trashcan(self):
self.cmd.curdir = lambda: "cwd"
self.cmd.run(['trash-restore'])
assert ("No files trashed from current dir ('cwd')\n" ==
self.stdout.getvalue())
def test_until_the_restore_unit(self):
self.fs.path_exists.return_value = False
trashed_file = TrashedFile(
'parent/path',
None,
'info_file',
'orig_file')
self.user_reply = '0'
self.cmd.restore_asking_the_user([trashed_file])
assert '' == self.stdout.getvalue()
assert '' == self.stderr.getvalue()
assert [call.path_exists('parent/path'),
call.mkdirs('parent'),
call.move('orig_file', 'parent/path'),
call.remove_file('info_file')] == self.fs.mock_calls
def test_when_user_reply_with_empty_string(self):
self.user_reply = ''
self.cmd.restore_asking_the_user([])
assert 'Exiting\n' == self.stdout.getvalue()
def test_when_user_reply_with_not_number(self):
self.user_reply = 'non numeric'
self.cmd.restore_asking_the_user([])
assert 'Invalid entry: not an index: non numeric\n' == \
self.stderr.getvalue()
assert '' == self.stdout.getvalue()
assert 1 == self.exit_status
