def test_ask_rename(self):
with patch("__builtin__.raw_input") as _raw_input:
file_item = PacketRenamer(u"file1.com", files=[u"/asd/file.com"])
_raw_input.return_value = u'y'
res1 = file_item.ask_rename(u"/asd/file.com", u"/asd/file1.com")
_raw_input.return_value = u'asd'
res2 = file_item.ask_rename(u"/asd/file.com", u"/asd/file2.com")
welcome_msg1 = u"Move /asd/file.com -> /asd/file1.com ? (Y/N) "
welcome_msg2 = u"Move /asd/file.com -> /asd/file2.com ? (Y/N) "
self.assertEqual(_raw_input.call_args_list, [call(welcome_msg1), call(welcome_msg2)])
self.assertEqual((True, True), res1)
self.assertEqual((False, True), res2)
def test_rename(self):
file_item = PacketRenamer(u"qwe", files=[u"/asd/file.com"])
with patch('os.rename') as os_rename, \
patch('os.path.exists') as os_exists, \
patch('__builtin__.raw_input') as _raw_input:
os_exists.return_value = True
_raw_input.return_value = u'y'
file_item.rename()
file_item = PacketRenamer(u"zxc", files=[u"/asd/file.com"])
os_exists.return_value = False
file_item.rename()
self.assertEqual(os_rename.call_args_list, [
call(u"/asd/file.com", u"/asd/qwe"),
call(u"/asd/file.com", u"/asd/zxc"),
])
self.assertEqual(os_exists.call_args_list, [call(u"/asd/qwe",), call(u"/asd/zxc")])
self.assertEqual(_raw_input.call_args_list, [
# call ask_if_exists(...)
call(u"Move /asd/file.com -> /asd/qwe Dest already exists, replace anyway ? (Y/N) "),
# call ask_rename(...)
call(u"Move /asd/file.com -> /asd/zxc ? (Y/N) "),
])
