from concurrent.futures import ThreadPoolExecutor
import allure


def session_on_node(node_id, session):
    with allure.step(f"Session#{session} on Node#{node_id}"):
        pass


def parallel_task_for_specific_node(node_id):
    with allure.step(f"Node#{node_id}") as step:
        with ThreadPoolExecutor(initializer=step.get_thread_initializer()) as executor:
            for session in range(3):
                executor.submit(session_on_node, node_id, session)
            executor.shutdown(wait=True)


def test_multithreaded():
    with allure.step("Root") as step:
        with ThreadPoolExecutor(initializer=step.get_thread_initializer()) as executor:
            for node_id in range(3):
                executor.submit(parallel_task_for_specific_node, node_id)
            executor.shutdown(wait=True)