def test_database_update_with_exception():
"""Test that replacing the database_entries members refreshes the database.
"""
root = RootTask()
task1 = ComplexTask(name='task1',
database_entries={'val1': 2.0})
task2 = SimpleTask(name='task2',
database_entries={'val2': 1},
access_exs={'val2': 1})
task3 = ComplexTask(name='task3')
task1.add_child_task(0, task2)
root.add_child_task(0, task1)
root.add_child_task(1, task3)
assert task3.get_from_database('task2_val2')
entries = task2.database_entries.copy()
del entries['val2']
task2.database_entries = entries
with pytest.raises(KeyError):
task1.get_from_database('task2_val2')
with pytest.raises(KeyError):
task3.get_from_database('task2_val2')
def test_task_renaming():
"""Test renaming simple and complex task.
"""
root = RootTask()
task1 = ComplexTask(name='task1',
database_entries={'val1': 2.0})
task2 = ComplexTask(name='task2')
task3 = SimpleTask(name='task3',
database_entries={'val2': 1},
access_exs={'val2': 2})
task2.add_child_task(0, task3)
task1.add_child_task(0, task2)
root.add_child_task(0, task1)
task3.name = 'worker3'
with pytest.raises(KeyError):
root.get_from_database('task3_val2')
assert root.get_from_database('worker3_val2') == 1
task1.name = 'worker1'
with pytest.raises(KeyError):
root.get_from_database('task1_val1')
assert root.get_from_database('worker1_val1') == 2.0
assert root.get_from_database('worker3_val2') == 1
def test_deleting_child():
"""Test deleting a child.
"""
root = RootTask()
task1 = ComplexTask(name='task1',
database_entries={'val1': 2.0})
task2 = SimpleTask(name='task2',
database_entries={'val2': 1},
access_exs={'val2': 2})
task3 = ComplexTask(name='task3')
task4 = ComplexTask(name='task4')
task1.add_child_task(0, task2)
task1.add_child_task(1, task4)
root.add_child_task(0, task1)
root.add_child_task(1, task3)
listener = SignalListener()
task1.observe('children_changed', listener.listen)
task1.remove_child_task(0)
assert listener.counter == 1
assert listener.signals[0].removed
assert task1.preferences['children_0']['name'] == 'task4'
assert 'task2_val2' not in task3.list_accessible_database_entries()
root.remove_child_task(0)
assert len(root.children) == 1
with pytest.raises(KeyError):
root.get_from_database('task1_val1')
def test_moving_child():
"""Test moving a child.
"""
root = RootTask()
task1 = ComplexTask(name='task1',
database_entries={'val1': 2.0})
task2 = SimpleTask(name='task2',
database_entries={'val2': 1},
access_exs={'val2': 2})
task3 = ComplexTask(name='task3')
task4 = ComplexTask(name='task4')
task1.add_child_task(0, task2)
task1.add_child_task(1, task4)
root.add_child_task(0, task1)
root.add_child_task(1, task3)
listener = SignalListener()
task1.observe('children_changed', listener.listen)
assert task1.preferences['children_0']['name'] == 'task2'
assert task1.preferences['children_1']['name'] == 'task4'
task1.move_child_task(0, 1)
assert listener.counter == 1
assert listener.signals[0].moved
assert task1.preferences['children_0']['name'] == 'task4'
assert task1.preferences['children_1']['name'] == 'task2'
assert task3.get_from_database('task2_val2') == 1
def setup(self):
self.root = RootTask()
database = self.root.database
database.set_value('root', 'val1', 1)
database.create_node('root', 'node1')
database.set_value('root/node1', 'val2', 10.0)
database.add_access_exception('root', 'root/node1', 'val2')
def setup(self):
root = RootTask()
root.should_pause = Event()
root.should_stop = Event()
root.paused = Event()
root.resumed = Event()
root.default_path = 'toto'
self.root = root
def test_database_operation():
"""Test setting, getting, deleting a value from the database.
"""
root = RootTask()
root.write_in_database('test', 1)
assert root.get_from_database('test') == 1
root.remove_from_database('test')
with pytest.raises(KeyError):
root.get_from_database('test')
def test_root_registering():
"""Check that the root task does write its default entries in the database
when instantiated.
"""
root = RootTask()
assert root.get_from_database('default_path') == ''
root.children = [SimpleTask(name='task2',
database_entries={'val2': 1},
root=root, parent=root,
database=root.database)]
root.register_in_database()
assert root.get_from_database('task2_val2') == 1
def test_database_update():
"""Test that replacing the database_entries members refreshes the database.
"""
root = RootTask()
entries = root.database_entries.copy()
del entries['default_path']
entries['name'] = 'Test'
root.database_entries = entries
assert root.get_from_database('name') == 'Test'
with pytest.raises(KeyError):
root.get_from_database('default_path')
def test_build_from_config1(self):
"""Test building a interfaceable task with no interface from a config.
"""
aux = RootTask()
aux.add_child_task(0, IMixin())
bis = RootTask.build_from_config(
aux.preferences,
{'ecpy.task': {
'tests.IMixin': IMixin,
'ecpy.RootTask': RootTask
}})
assert type(bis.children[0]).__name__ == 'IMixin'
def test_update_preferences_from_members():
"""Test updating the preferences.
Only operation on the children cause re-registering to ensure the children
ordering.
"""
root = RootTask()
task1 = SimpleTask(name='task1')
root.add_child_task(0, task1)
assert root.preferences['children_0']['name'] == 'task1'
task1.name = 'worker1'
assert root.preferences['children_0']['name'] == 'task1'
root.update_preferences_from_members()
assert root.preferences['children_0']['name'] == 'worker1'
def test_build_from_config1(self):
"""Test building a interfaceable interface with no interface from a
config.
"""
aux = RootTask()
mixin = Mixin()
mixin.interface = InterfaceTest3()
aux.add_child_task(0, mixin)
deps = {
'ecpy.task': {
'tests.Mixin': Mixin,
'ecpy.RootTask': RootTask
},
'ecpy.tasks.interface': {
('InterfaceTest3', ('tests.Mixin', )): InterfaceTest3
}
}
bis = RootTask.build_from_config(aux.preferences, deps)
assert type(bis.children[0].interface).__name__ == 'InterfaceTest3'
def test_traverse():
"""Test traversing a task hierarchy to collect infos.
"""
root = RootTask()
task1 = ComplexTask(name='task1',
database_entries={'val1': 2.0})
task2 = SimpleTask(name='task2',
database_entries={'val2': 1},
access_exs={'val2': 2})
task3 = ComplexTask(name='task3')
task1.add_child_task(0, task2)
root.add_child_task(0, task1)
root.add_child_task(1, task3)
flat = list(root.traverse())
assert flat == [root, task1, task2, task3]
flat = list(root.traverse(0))
assert flat == [root, task1, task3]
def test_adding_child():
"""Test adding children.
This test adding a child with and without access_exs to a task which is not
root and then to the root. This makes sure that giving the root afterwards
does trigger the right updates.
"""
root = RootTask()
listener = SignalListener()
root.observe('children_changed', listener.listen)
task1 = ComplexTask(name='task1',
database_entries={'val1': 2.0})
task2 = SimpleTask(name='task2',
database_entries={'val2': 1},
access_exs={'val2': 2})
task3 = ComplexTask(name='task3')
task1.add_child_task(0, task2)
root.add_child_task(0, task1)
root.add_child_task(1, task3)
assert task1.depth == 1
assert task1.path == 'root'
assert task1.database is root.database
assert task1.root is root
assert task1.parent is root
assert task2.depth == 2
assert task2.path == 'root/task1'
assert task2.database is root.database
assert task2.root is root
assert task2.parent is task1
assert task1.get_from_database('task1_val1') == 2.0
assert root.get_from_database('task1_val1') == 2.0
assert task3.get_from_database('task2_val2') == 1
assert listener.counter == 2
assert all([bool(c.added) for c in listener.signals])
def test_access_exceptions():
"""Test adding, modifying and removing an access exception after creation.
"""
root = RootTask()
listener = SignalListener()
root.observe('children_changed', listener.listen)
task1 = ComplexTask(name='task1',
database_entries={'val1': 2.0})
task2 = ComplexTask(name='task2')
task3 = SimpleTask(name='task3',
database_entries={'val2': 1, 'val3': 2},
)
task2.add_child_task(0, task3)
task1.add_child_task(0, task2)
root.add_child_task(0, task1)
with pytest.raises(KeyError):
task2.get_from_database('task3_val2')
task3.add_access_exception('val2', 1)
task3.add_access_exception('val3', 1)
assert task2.get_from_database('task3_val2') == 1
assert task2.get_from_database('task3_val2') == 1
with pytest.raises(KeyError):
task1.get_from_database('task3_val2')
task3.modify_access_exception('val2', 2)
task3.modify_access_exception('val3', -1)
assert task1.get_from_database('task3_val2') == 1
with pytest.raises(KeyError):
task2.get_from_database('task3_val3')
task3.remove_access_exception('val2')
with pytest.raises(KeyError):
task2.get_from_database('task3_val2')
def setup(self):
self.root = RootTask()
self.mixin = InterfaceTest3()
self.root.add_child_task(0, Mixin(name='Simple', interface=self.mixin))
def setup(self):
self.root = RootTask()
self.mixin = Mixin(name='Simple')
self.root.add_child_task(0, self.mixin)
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = ConditionalTask(name='Test')
self.root.add_child_task(0, self.task)
self.check = CheckTask(name='check')
self.task.add_child_task(0, self.check)
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
self.task = LoopTask(name='Test')
self.root.add_child_task(0, self.task)
def setup(self):
self.root = RootTask(should_stop=Event(), should_pause=Event())
