def testCreateOutput(self):
"""Test that output directories are created."""
writer = output_manager.LocalOutputWriter(
base_output_dir=self.base_output_dir, unique_dir='unique_dir')
output_dir = writer.create_output_dir()
self.remove_dirs.append(output_dir)
self.assertTrue(os.path.exists(output_dir))
self.assertTrue('unique_dir' in output_dir)
def testNoFileWrite(self):
"""Test that write fails when no source file exists."""
test_file = 'test.txt'
writer = output_manager.LocalOutputWriter(
base_output_dir=self.base_output_dir, unique_dir='unique_dir')
output_dir = writer.create_output_dir()
self.remove_dirs.append(output_dir)
src = os.path.join(self.base_output_dir, test_file)
dst = os.path.join(output_dir, test_file)
self.assertFalse(writer.copy_to(src))
self.assertFalse(os.path.exists(dst))
def testWrite(self):
"""Test that file contents are written."""
contents = 'test contents'
test_file = 'test.txt'
writer = output_manager.LocalOutputWriter(
base_output_dir=self.base_output_dir, unique_dir='unique_dir')
output_dir = writer.create_output_dir()
self.remove_dirs.append(output_dir)
src = os.path.join(self.base_output_dir, test_file)
dst = os.path.join(output_dir, test_file)
self.remove_files.append(src)
self.remove_files.append(dst)
with open(src, 'w') as file_handle:
file_handle.write(contents)
self.assertTrue(writer.copy_to(src))
self.assertTrue(os.path.exists(dst))
self.assertEqual(contents, open(dst).read())
def testFileExistsWrite(self):
"""Test that file is not overwritten when it exists."""
contents = 'test contents'
other_contents = 'other test contents'
test_file = 'test.txt'
writer = output_manager.LocalOutputWriter(
base_output_dir=self.base_output_dir, unique_dir='unique_dir')
output_dir = writer.create_output_dir()
self.remove_dirs.append(output_dir)
src = os.path.join(self.base_output_dir, test_file)
dst = os.path.join(output_dir, test_file)
self.remove_files.append(src)
self.remove_files.append(dst)
with open(src, 'w') as file_handle:
file_handle.write(contents)
with open(dst, 'w') as file_handle:
file_handle.write(other_contents)
self.assertFalse(writer.copy_to(src))
self.assertEqual(other_contents, open(dst).read())
