def setUp(self):
ploader = PluginLoaderMock()
self.task_manager = TaskManager(ploader, {})
assert ploader.provides_ident == 'TaskManager'
assert isinstance(ploader.provides_obj, TaskManager)
class TaskManagerTest(TestCase):
def setUp(self):
ploader = PluginLoaderMock()
self.task_manager = TaskManager(ploader, {})
assert ploader.provides_ident == 'TaskManager'
assert isinstance(ploader.provides_obj, TaskManager)
def test_run_task(self):
def some_task(level):
if level <= 0:
raise StopIteration('We are done!', level)
# one event name
event, data = yield 'some_event'
self.assertEqual('some_event', event)
self.assertDictEqual({'some': 'data'}, data)
# multiple event names
event, data = yield 'other_event', 'never_emitted', 'this_neither'
self.assertEqual('other_event', event)
self.assertDictEqual({'other': 'data'}, data)
# event name + check
event, data = yield 'cool_event', lambda e, d: True
self.assertEqual('cool_event', event)
self.assertDictEqual({'more': 'data'}, data)
# subtask
event, data = yield some_task(level - 1)
self.assertEqual(None, event)
self.assertTupleEqual(('We are done!', level - 1), data)
raise StopIteration('We are done!', level)
def emit_them(level):
"""Recursively emit and check the events waited for by the task"""
if level <= 0:
return
self.assertIn('some_event', self.task_manager.event.handlers,
'Event not registered')
self.task_manager.event.emit('some_event', {'some': 'data'})
self.assertFalse(self.task_manager.event.handlers['some_event'],
'Emitted event not unregistered')
self.assertIn('other_event', self.task_manager.event.handlers,
'Event not registered')
self.assertIn('never_emitted', self.task_manager.event.handlers,
'Event not registered')
self.assertIn('this_neither', self.task_manager.event.handlers,
'Event not registered')
self.task_manager.event.emit('other_event', {'other': 'data'})
self.assertFalse(self.task_manager.event.handlers['other_Event'],
'Emitted event not unregistered')
self.assertIn('cool_event', self.task_manager.event.handlers,
'Event not registered')
self.task_manager.event.emit('cool_event', {'more': 'data'})
self.assertFalse(self.task_manager.event.handlers['cool_event'],
'Emitted event not unregistered')
emit_them(level - 1)
done = [False] # because we can't use nonlocal in Python 2.7
test_case = self # to make flake8 happy
class SomeParent(object):
def on_success(self, data):
done[0] = True
def on_error(self, e):
test_case.fail('Task failed with %s: %s' %
(e.__class__.__name__, e.args))
levels = 3
task = some_task(levels)
parent = SomeParent()
ret = self.task_manager.run_task(task, parent)
self.assertIsInstance(ret, Task)
emit_them(levels)
assert done[0]
def test_failure(self):
last_data = ['started']
def task_with_failure():
last_data[0] = yield 'bbbb'
raise TaskFailed('Some error!')
def task_deep_failure():
def task_deepest_failure():
last_data[0] = yield 'cccc'
raise TaskFailed('Low level error!')
def task_deeper_failure():
yield task_deepest_failure() # error falls through
try:
yield task_deeper_failure()
except TaskFailed as error:
raise TaskFailed('High level error!').with_error(error)
def task_with_exception():
last_data[0] = yield 'dddd'
1 / 0
def top_task():
try:
yield task_with_failure()
except TaskFailed as e:
self.assertEqual('Some error!', e.message)
self.assertEqual(None, e.prev_error)
self.assertEqual(1, len(e.tasktrace))
self.assertEqual('task_with_failure', e.tasktrace[0].name)
self.assertEqual(1, len(e.full_tasktrace))
self.assertEqual('task_with_failure', e.full_tasktrace[0].name)
else:
self.fail('TaskFailed not passed into task')
try:
yield task_deep_failure()
except TaskFailed as e:
self.assertEqual('High level error!', e.message)
self.assertEqual('Low level error!', e.prev_error.message)
self.assertEqual(None, e.prev_error.prev_error)
self.assertEqual(1, len(e.tasktrace))
self.assertEqual('task_deep_failure', e.tasktrace[0].name)
self.assertEqual(3, len(e.full_tasktrace))
self.assertEqual('task_deepest_failure',
e.full_tasktrace[0].name)
self.assertEqual('task_deeper_failure',
e.full_tasktrace[1].name)
self.assertEqual('task_deep_failure', e.full_tasktrace[2].name)
else:
self.fail('TaskFailed not passed through tasks')
try:
yield task_with_exception()
except ZeroDivisionError as e:
pass # everything OK
else:
self.fail('Exception not passed into task')
task = top_task()
ret = self.task_manager.run_task(task)
self.assertIsInstance(ret, Task)
self.assertEqual('started', last_data[0])
self.emit_and_check('bbbb', last_data)
self.emit_and_check('cccc', last_data)
self.emit_and_check('dddd', last_data)
def emit_and_check(self, event, last_data):
data = random.random()
self.task_manager.event.emit(event, data)
self.assertEqual(event, last_data[0][0])
self.assertEqual(data, last_data[0][1])
class TaskManagerTest(TestCase):
def setUp(self):
ploader = PluginLoaderMock()
self.task_manager = TaskManager(ploader, {})
assert ploader.provides_ident == "TaskManager"
assert isinstance(ploader.provides_obj, TaskManager)
def test_run_task(self):
def some_task(level):
if level <= 0:
raise StopIteration("We are done!", level)
# one event name
event, data = yield "some_event"
self.assertEqual("some_event", event)
self.assertDictEqual({"some": "data"}, data)
# multiple event names
event, data = yield "other_event", "never_emitted", "this_neither"
self.assertEqual("other_event", event)
self.assertDictEqual({"other": "data"}, data)
# event name + check
event, data = yield "cool_event", lambda e, d: True
self.assertEqual("cool_event", event)
self.assertDictEqual({"more": "data"}, data)
# subtask
event, data = yield some_task(level - 1)
self.assertEqual(None, event)
self.assertTupleEqual(("We are done!", level - 1), data)
raise StopIteration("We are done!", level)
def emit_them(level):
"""Recursively emit and check the events waited for by the task"""
if level <= 0:
return
self.assertIn("some_event", self.task_manager.event.handlers, "Event not registered")
self.task_manager.event.emit("some_event", {"some": "data"})
self.assertFalse(self.task_manager.event.handlers["some_event"], "Emitted event not unregistered")
self.assertIn("other_event", self.task_manager.event.handlers, "Event not registered")
self.assertIn("never_emitted", self.task_manager.event.handlers, "Event not registered")
self.assertIn("this_neither", self.task_manager.event.handlers, "Event not registered")
self.task_manager.event.emit("other_event", {"other": "data"})
self.assertFalse(self.task_manager.event.handlers["other_Event"], "Emitted event not unregistered")
self.assertIn("cool_event", self.task_manager.event.handlers, "Event not registered")
self.task_manager.event.emit("cool_event", {"more": "data"})
self.assertFalse(self.task_manager.event.handlers["cool_event"], "Emitted event not unregistered")
emit_them(level - 1)
done = [False] # because we can't use nonlocal in Python 2.7
test_case = self # to make flake8 happy
class SomeParent(object):
def on_success(self, data):
done[0] = True
def on_error(self, e):
test_case.fail("Task failed with %s: %s" % (e.__class__.__name__, e.args))
levels = 3
task = some_task(levels)
parent = SomeParent()
ret = self.task_manager.run_task(task, parent)
self.assertIsInstance(ret, Task)
emit_them(levels)
assert done[0]
def test_failure(self):
last_data = ["started"]
def task_with_failure():
last_data[0] = yield "bbbb"
raise TaskFailed("Some error!")
def task_deep_failure():
def task_deepest_failure():
last_data[0] = yield "cccc"
raise TaskFailed("Low level error!")
def task_deeper_failure():
yield task_deepest_failure() # error falls through
try:
yield task_deeper_failure()
except TaskFailed as error:
raise TaskFailed("High level error!").with_error(error)
def task_with_exception():
last_data[0] = yield "dddd"
1 / 0
def top_task():
try:
yield task_with_failure()
except TaskFailed as e:
self.assertEqual("Some error!", e.message)
self.assertEqual(None, e.prev_error)
self.assertEqual(1, len(e.tasktrace))
self.assertEqual("task_with_failure", e.tasktrace[0].name)
self.assertEqual(1, len(e.full_tasktrace))
self.assertEqual("task_with_failure", e.full_tasktrace[0].name)
else:
self.fail("TaskFailed not passed into task")
try:
yield task_deep_failure()
except TaskFailed as e:
self.assertEqual("High level error!", e.message)
self.assertEqual("Low level error!", e.prev_error.message)
self.assertEqual(None, e.prev_error.prev_error)
self.assertEqual(1, len(e.tasktrace))
self.assertEqual("task_deep_failure", e.tasktrace[0].name)
self.assertEqual(3, len(e.full_tasktrace))
self.assertEqual("task_deepest_failure", e.full_tasktrace[0].name)
self.assertEqual("task_deeper_failure", e.full_tasktrace[1].name)
self.assertEqual("task_deep_failure", e.full_tasktrace[2].name)
else:
self.fail("TaskFailed not passed through tasks")
try:
yield task_with_exception()
except ZeroDivisionError as e:
pass # everything OK
else:
self.fail("Exception not passed into task")
def emit_and_check(event):
data = random.random()
self.task_manager.event.emit(event, data)
self.assertEqual(event, last_data[0][0])
self.assertEqual(data, last_data[0][1])
task = top_task()
ret = self.task_manager.run_task(task)
self.assertIsInstance(ret, Task)
self.assertEqual("started", last_data[0])
emit_and_check("bbbb")
emit_and_check("cccc")
emit_and_check("dddd")
def setUp(self):
ploader = PluginLoaderMock()
self.task_manager = TaskManager(ploader, {})
assert ploader.provides_ident == "TaskManager"
assert isinstance(ploader.provides_obj, TaskManager)
class TaskManagerTest(TestCase):
def setUp(self):
ploader = PluginLoaderMock()
self.task_manager = TaskManager(ploader, {})
assert ploader.provides_ident == 'TaskManager'
assert isinstance(ploader.provides_obj, TaskManager)
def test_run_task(self):
def some_task(level):
if level <= 0:
raise StopIteration('We are done!', level)
# one event name
event, data = yield 'some_event'
self.assertEqual('some_event', event)
self.assertDictEqual({'some': 'data'}, data)
# multiple event names
event, data = yield 'other_event', 'never_emitted', 'this_neither'
self.assertEqual('other_event', event)
self.assertDictEqual({'other': 'data'}, data)
# event name + check
event, data = yield 'cool_event', lambda e, d: True
self.assertEqual('cool_event', event)
self.assertDictEqual({'more': 'data'}, data)
# subtask
event, data = yield some_task(level - 1)
self.assertEqual(None, event)
self.assertTupleEqual(('We are done!', level - 1), data)
raise StopIteration('We are done!', level)
def emit_them(level):
"""Recursively emit and check the events waited for by the task"""
if level <= 0:
return
self.assertIn('some_event', self.task_manager.event.handlers,
'Event not registered')
self.task_manager.event.emit('some_event', {'some': 'data'})
self.assertFalse(self.task_manager.event.handlers['some_event'],
'Emitted event not unregistered')
self.assertIn('other_event', self.task_manager.event.handlers,
'Event not registered')
self.assertIn('never_emitted', self.task_manager.event.handlers,
'Event not registered')
self.assertIn('this_neither', self.task_manager.event.handlers,
'Event not registered')
self.task_manager.event.emit('other_event', {'other': 'data'})
self.assertFalse(self.task_manager.event.handlers['other_Event'],
'Emitted event not unregistered')
self.assertIn('cool_event', self.task_manager.event.handlers,
'Event not registered')
self.task_manager.event.emit('cool_event', {'more': 'data'})
self.assertFalse(self.task_manager.event.handlers['cool_event'],
'Emitted event not unregistered')
emit_them(level - 1)
done = [False] # because we can't use nonlocal in Python 2.7
test_case = self # to make flake8 happy
class SomeParent(object):
def on_success(self, data):
done[0] = True
def on_error(self, e):
test_case.fail('Task failed with %s: %s'
% (e.__class__.__name__, e.args))
levels = 3
task = some_task(levels)
parent = SomeParent()
ret = self.task_manager.run_task(task, parent)
self.assertIsInstance(ret, Task)
emit_them(levels)
assert done[0]
def test_failure(self):
last_data = ['started']
def task_with_failure():
last_data[0] = yield 'bbbb'
raise TaskFailed('Some error!')
def task_deep_failure():
def task_deepest_failure():
last_data[0] = yield 'cccc'
raise TaskFailed('Low level error!')
def task_deeper_failure():
yield task_deepest_failure() # error falls through
try:
yield task_deeper_failure()
except TaskFailed as error:
raise TaskFailed('High level error!').with_error(error)
def task_with_exception():
last_data[0] = yield 'dddd'
1/0
def top_task():
try:
yield task_with_failure()
except TaskFailed as e:
self.assertEqual('Some error!', e.message)
self.assertEqual(None, e.prev_error)
self.assertEqual(1, len(e.tasktrace))
self.assertEqual('task_with_failure', e.tasktrace[0].name)
self.assertEqual(1, len(e.full_tasktrace))
self.assertEqual('task_with_failure',
e.full_tasktrace[0].name)
else:
self.fail('TaskFailed not passed into task')
try:
yield task_deep_failure()
except TaskFailed as e:
self.assertEqual('High level error!', e.message)
self.assertEqual('Low level error!', e.prev_error.message)
self.assertEqual(None, e.prev_error.prev_error)
self.assertEqual(1, len(e.tasktrace))
self.assertEqual('task_deep_failure', e.tasktrace[0].name)
self.assertEqual(3, len(e.full_tasktrace))
self.assertEqual('task_deepest_failure',
e.full_tasktrace[0].name)
self.assertEqual('task_deeper_failure',
e.full_tasktrace[1].name)
self.assertEqual('task_deep_failure',
e.full_tasktrace[2].name)
else:
self.fail('TaskFailed not passed through tasks')
try:
yield task_with_exception()
except ZeroDivisionError as e:
pass # everything OK
else:
self.fail('Exception not passed into task')
task = top_task()
ret = self.task_manager.run_task(task)
self.assertIsInstance(ret, Task)
self.assertEqual('started', last_data[0])
self.emit_and_check('bbbb', last_data)
self.emit_and_check('cccc', last_data)
self.emit_and_check('dddd', last_data)
def emit_and_check(self, event, last_data):
data = random.random()
self.task_manager.event.emit(event, data)
self.assertEqual(event, last_data[0][0])
self.assertEqual(data, last_data[0][1])
