def test_task_computed(self):
task = Task(Mock(), Mock(), Mock())
result = [{"data": True}, 123]
task.header.node_name = self.node
task.header.task_id = self.name
task.root_path = self.path
task.after_test = lambda res, tmp_dir: {}
task.query_extra_data_for_test_task = Mock()
tt = TaskTester(task, self.path, Mock(), Mock())
tt.tmp_dir = self.path
task_thread = TaskThread(result)
tt.task_computed(task_thread)
task_thread = MemTaskThread(None, 30210, "Some error")
with self.assertLogs(logger, level='WARNING'):
tt.task_computed(task_thread)
tt.error_callback.assert_called_with("Some error")
task_thread = MemTaskThread("result", 2010, "Another error")
self.assertIsNone(tt.get_progress())
tt.tt = task_thread
self.assertEqual(tt.get_progress(), "30%")
task_thread.error = True
self.assertEqual(tt.get_progress(), 0)
tt.task_computed(task_thread)
tt.error_callback.assert_called_with("Another error")
self.message = ""
def success_callback(res, est_mem, time_spent, after_test_data):
self.message = "Success " + after_test_data["warnings"]
task.header.node_name = self.node
task.header.task_id = self.name
task.root_path = self.path
task.after_test = lambda res, tmp_dir: {"warnings": "bla ble"}
task.query_extra_data_for_test_task = Mock()
tt = TaskTester(task, self.path, success_callback, None)
tt.tmp_dir = self.path
task_thread = TaskThread(result)
tt.task_computed(task_thread)
self.assertTrue("bla" in self.message)
self.assertTrue("ble" in self.message)
def test_is_success(self):
task = Task(Mock(), Mock(), Mock())
task.query_extra_data_for_test_task = Mock()
tt = TaskTester(task, self.path, Mock(), Mock())
task_thread = Mock()
# Proper task
task_thread.error = None
task_thread.result = ({"data": True}, 123)
assert tt.is_success(task_thread)
# Task thead result first arg is not tuple
task_thread.result = {"data": True}
assert not tt.is_success(task_thread)
assert task_thread.error == "Wrong result format"
def test_init(self):
task = Task(Mock(), Mock(), Mock())
task.query_extra_data_for_test_task = Mock()
self.assertIsNotNone(TaskTester(task, self.path, None, None))