def test_next_task_removed_file(self):
# cleaning tasks_dirs
tasks_dir = os.path.join(self.c_dir, 'test_next_task_removed_file')
to_proc_dir = os.path.join(tasks_dir, 'to_proc')
ready_dir = os.path.join(tasks_dir, 'ready')
shutil.rmtree(tasks_dir, ignore_errors=True)
# creating tasks
os.makedirs(to_proc_dir)
os.makedirs(ready_dir)
task_name = '101'
with open(os.path.join(to_proc_dir, task_name), 'w') as f:
f.write('one\n')
f.write('two\n')
tm = TaskManager(to_proc_dir, ready_dir)
self.assertTrue(task_name in os.listdir(to_proc_dir))
self.assertTrue(task_name not in os.listdir(ready_dir))
tm.next_task()
# cleaning tasks_dirs
shutil.rmtree(tasks_dir, ignore_errors=True)
# checking task removed from tasks sets
self.assertRaises(TaskFromFileLoadingError, tm.next_task)
self.assertTrue(task_name not in tm.tasks)
self.assertTrue(task_name not in tm.tasks_in_process)
def test_receive_task(self):
# cleaning tasks_dirs
tasks_dir = os.path.join(self.c_dir, 'test_receive_task')
to_proc_dir = os.path.join(tasks_dir, 'to_proc')
ready_dir = os.path.join(tasks_dir, 'ready')
shutil.rmtree(tasks_dir, ignore_errors=True)
# creating tasks
os.makedirs(to_proc_dir)
os.makedirs(ready_dir)
task_name = '100'
with open(os.path.join(to_proc_dir, task_name), 'w') as f:
f.write('one\n')
f.write('two\n')
# testing task receive
tm = TaskManager(to_proc_dir, ready_dir)
self.assertTrue(task_name in os.listdir(to_proc_dir))
self.assertTrue(task_name not in os.listdir(ready_dir))
task = tm.next_task()
self.assertEquals(1, len(tm.tasks_in_process))
task.result = 'ready'
tm.receive_task(task)
self.assertTrue(task_name in os.listdir(to_proc_dir))
self.assertTrue(task_name in os.listdir(ready_dir))
self.assertEquals(0, len(tm.tasks_in_process))
with open(os.path.join(ready_dir, task_name)) as f:
self.assertEquals(task.result, f.read())
# testing duplicate receive_task call
task.result = 'new_%s' % task.result
tm.receive_task(task)
self.assertTrue(task_name in os.listdir(to_proc_dir))
self.assertTrue(task_name in os.listdir(ready_dir))
self.assertEquals(0, len(tm.tasks_in_process))
with open(os.path.join(ready_dir, task_name)) as f:
self.assertEquals(task.result, f.read())
# cleaning tasks_dirs
shutil.rmtree(tasks_dir, ignore_errors=True)
