def testPrependAllowed(self):
temp_handle, temp_name = tempfile.mkstemp()
chromium_step.Prepend(temp_name, '111', self.output_dir,
self.perf_output_dir)
self.assertTrue(_readData(temp_name), '111')
os.close(temp_handle)
os.unlink(temp_name)
def testPrependAllowedSubDir(self):
# Use mkdtemp() just to generate a unique filename.
temp_name = self._generateTempName(self.output_dir)
# Create a path containing a sub-directory of the output_dir. Make sure it
# doesn't exist so we can verify it's created.
subdir = os.path.join(self.output_dir, temp_name)
subdir_file = os.path.join(subdir, temp_name)
self.assertFalse(os.path.exists(subdir_file))
try:
chromium_step.Prepend(subdir_file, '222', self.output_dir,
self.perf_output_dir)
# Verify the subdir was also created, in addition to the file.
self.assertTrue(os.path.exists(subdir))
self.assertTrue(os.path.exists(subdir_file))
self.assertTrue(_readData(subdir_file), '222')
finally:
shutil.rmtree(subdir)
def dummy_func():
chromium_step.Prepend(invalid_filename, '333', self.output_dir,
self.perf_output_dir)
def testPrependAllowed(self):
with tempfile.NamedTemporaryFile(dir=self.output_dir) as temp_file:
chromium_step.Prepend(temp_file.name, '111', self.output_dir,
self.perf_output_dir)
self.assertTrue(_readData(temp_file.name), '111')