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