def test_rename_file(self):
backup = Config.storedir
Config.storedir = Utils.new_tmp_dir()
relative_path = 'baz/file.txt'
new_path = 'baz/txt.elif'
StorageManager.save_file(relative_path, 'foobar'.encode())
StorageManager.rename_file(relative_path, new_path)
with open(StorageManager.get_absolute_path(new_path), 'r') as file:
lines = file.readlines()
self.assertEqual('foobar', lines[0])
Config.storedir = backup
def get_input(task_name, attempt):
"""
Fetch an input from the queue and properly rename it
:param task_name: Name of the task
:param attempt: Number of the attempt for the user
:return: A pair, the first element is the id of the input file,
the second the path
"""
if ContestManager.input_queue[task_name].empty():
Logger.warning("TASK", "Empty queue for task %s!" % task_name)
input = ContestManager.input_queue[task_name].get()
path = StorageManager.new_input_file(input["id"], task_name, attempt)
StorageManager.rename_file(input["path"], path)
return input["id"], path