class TestMoveTo(unittest.TestCase):
def setUp(self):
# Make a temp directory and test file
self.source_dir = tempfile.mkdtemp()
self.dest_dir = os.path.join(self.source_dir, 'dest/')
os.mkdir(self.dest_dir)
self.test_filename = "test.txt"
self.source_file = os.path.join(self.source_dir, self.test_filename)
self.dest_file = os.path.join(self.dest_dir, self.test_filename)
with open(self.source_file, 'w') as output:
output.write("This is a test file.")
self.action = MoveTo(self.dest_dir)
def test_moveto(self):
# Test that moveto works with the simple case - source file exists, dest file does not exist,
# and source file gets moved successfully
new_filename = self.action.do_action(self.source_file)
self.assertEqual(new_filename, self.dest_file)
self.assertTrue(os.path.isfile(self.dest_file), "Destination file does not exist")
self.assertFalse(os.path.isfile(self.source_file), "Source file still exists")
def test_moveto_dry_run(self):
# Test that file is NOT moved when dryrun=True
new_filename = self.action.do_action(self.source_file, dry_run=True)
self.assertEqual(new_filename, self.source_file)
self.assertFalse(os.path.isfile(self.dest_file), "Destination file does not exist")
self.assertTrue(os.path.isfile(self.source_file), "Source file still exists")
def test_moveto_destination_exists(self):
# Test that if the file already exists in the destination directory, add _001
# to the filename and try again.
shutil.copy(self.source_file, self.dest_dir)
new_filename = self.action.do_action(self.source_file)
self.assertFalse(os.path.isfile(self.source_file), "Source file exists and should have been moved")
self.assertTrue(os.path.isfile(new_filename), "Destination file does not exist and should")
self.assertTrue(new_filename[-8:-4] == '_001', 'New file does not end in "_001"')
def test_moveto_destination_exists_twice(self):
# Test that if the file already exists in the destination directory, and a file with the _001 suffix
# in the filename exists, that the file will be renamed to <filename>_002.extension
shutil.copy(self.source_file, self.dest_dir)
filename, extension = os.path.splitext(self.dest_file)
file1 = os.path.join(self.dest_dir, (filename + '_001' + extension))
file2 = os.path.join(self.dest_dir, (filename + '_002' + extension))
shutil.copy(self.source_file, file1)
new_filename = self.action.do_action(self.source_file)
self.assertFalse(os.path.isfile(self.source_file), "Source file exists and should have been moved")
self.assertTrue(os.path.isfile(new_filename), "Destination file does not exist and should")
self.assertTrue(new_filename == file2, 'New file does not have the correct name')
def test_moveto_source_does_not_exist(self):
self.assertRaises(IOError, self.action.do_action, 'thisfiledoesnotexist.')
def tearDown(self):
# Clean up temp files
shutil.rmtree(self.source_dir)
def setUp(self):
# Make a temp directory and test file
self.source_dir = tempfile.mkdtemp()
self.dest_dir = os.path.join(self.source_dir, 'dest/')
os.mkdir(self.dest_dir)
self.test_filename = "test.txt"
self.source_file = os.path.join(self.source_dir, self.test_filename)
self.dest_file = os.path.join(self.dest_dir, self.test_filename)
with open(self.source_file, 'w') as output:
output.write("This is a test file.")
self.action = MoveTo(self.dest_dir)
def setUp(self):
# Make a temp directory and test file
self.source_dir = tempfile.mkdtemp()
self.dest_dir = os.path.join(self.source_dir, 'dest/')
os.mkdir(self.dest_dir)
self.test_filename = "test.txt"
self.source_file = os.path.join(self.source_dir, self.test_filename)
self.dest_file = os.path.join(self.dest_dir, self.test_filename)
with open(self.source_file, 'w') as output:
output.write("This is a test file.")
self.action = MoveTo(self.dest_dir)
class TestMoveTo(unittest.TestCase):
def setUp(self):
# Make a temp directory and test file
self.source_dir = tempfile.mkdtemp()
self.dest_dir = os.path.join(self.source_dir, 'dest/')
os.mkdir(self.dest_dir)
self.test_filename = "test.txt"
self.source_file = os.path.join(self.source_dir, self.test_filename)
self.dest_file = os.path.join(self.dest_dir, self.test_filename)
with open(self.source_file, 'w') as output:
output.write("This is a test file.")
self.action = MoveTo(self.dest_dir)
def test_moveto(self):
# Test that moveto works with the simple case - source file exists, dest file does not exist,
# and source file gets moved successfully
new_filename = self.action.do_action(self.source_file)
self.assertEqual(new_filename, self.dest_file)
self.assertTrue(os.path.isfile(self.dest_file),
"Destination file does not exist")
self.assertFalse(os.path.isfile(self.source_file),
"Source file still exists")
def test_moveto_dry_run(self):
# Test that file is NOT moved when dryrun=True
new_filename = self.action.do_action(self.source_file, dry_run=True)
self.assertEqual(new_filename, self.source_file)
self.assertFalse(os.path.isfile(self.dest_file),
"Destination file does not exist")
self.assertTrue(os.path.isfile(self.source_file),
"Source file still exists")
def test_moveto_destination_exists(self):
# Test that if the file already exists in the destination directory, add _001
# to the filename and try again.
shutil.copy(self.source_file, self.dest_dir)
new_filename = self.action.do_action(self.source_file)
self.assertFalse(os.path.isfile(self.source_file),
"Source file exists and should have been moved")
self.assertTrue(os.path.isfile(new_filename),
"Destination file does not exist and should")
self.assertTrue(new_filename[-8:-4] == '_001',
'New file does not end in "_001"')
def test_moveto_destination_exists_twice(self):
# Test that if the file already exists in the destination directory, and a file with the _001 suffix
# in the filename exists, that the file will be renamed to <filename>_002.extension
shutil.copy(self.source_file, self.dest_dir)
filename, extension = os.path.splitext(self.dest_file)
file1 = os.path.join(self.dest_dir, (filename + '_001' + extension))
file2 = os.path.join(self.dest_dir, (filename + '_002' + extension))
shutil.copy(self.source_file, file1)
new_filename = self.action.do_action(self.source_file)
self.assertFalse(os.path.isfile(self.source_file),
"Source file exists and should have been moved")
self.assertTrue(os.path.isfile(new_filename),
"Destination file does not exist and should")
self.assertTrue(new_filename == file2,
'New file does not have the correct name')
def test_moveto_source_does_not_exist(self):
self.assertRaises(IOError, self.action.do_action,
'thisfiledoesnotexist.')
def tearDown(self):
# Clean up temp files
shutil.rmtree(self.source_dir)