async def test_task_register_watched():
conn = Mock()
parent, child = MockTask(), MockTask()
mgr = TaskManager(parent)
mgr[child.id] = child
mgr.register(conn, child.id, {
'id': str(child.id),
'name': 'child',
'image': 'test',
})
# connection should be stored
assert conn in mgr.conns
# task connection reference should be set
assert mgr[child.id].conn == conn
async def test_version_check_accept():
parent = MockTask()
conn = Mock()
mgr = TaskManager(parent)
mgr.reject = AsyncMock()
mgr.register = Mock()
await mgr.on_child_init(conn, id='abc', task={}, version=version)
mgr.reject.assert_not_called()
mgr.register.assert_called()
async def test_task_register_virtual():
conn = Mock()
parent, child = MockTask(), MockTask()
mgr = TaskManager(parent)
assert child.id not in mgr
mgr.register(conn, child.id, {
'id': child.id,
'name': 'child',
'image': 'test',
'parent': parent.id,
})
# connection should be stored
assert conn in mgr.conns
# a remote task instance should be created
assert child.id in mgr
assert mgr[child.id].conn == conn