mirror of
https://github.com/n8n-io/n8n
synced 2026-05-10 13:32:23 +02:00
109 lines
2.8 KiB
Python
109 lines
2.8 KiB
Python
import pytest_asyncio
|
|
from src.message_types.broker import Items
|
|
from src.message_serde import NODE_MODE_MAP
|
|
|
|
from tests.fixtures.local_task_broker import LocalTaskBroker
|
|
from tests.fixtures.task_runner_manager import TaskRunnerManager
|
|
from tests.fixtures.test_constants import (
|
|
TASK_RESPONSE_WAIT,
|
|
)
|
|
|
|
NODE_MODE_TO_BROKER_STYLE = {v: k for k, v in NODE_MODE_MAP.items()}
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def broker():
|
|
broker = LocalTaskBroker()
|
|
await broker.start()
|
|
yield broker
|
|
await broker.stop()
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def manager(broker):
|
|
manager = TaskRunnerManager(task_broker_url=broker.get_url())
|
|
await manager.start()
|
|
yield manager
|
|
await manager.stop()
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def manager_with_stdlib_wildcard(broker):
|
|
manager = TaskRunnerManager(
|
|
task_broker_url=broker.get_url(),
|
|
custom_env={
|
|
"N8N_RUNNERS_STDLIB_ALLOW": "*",
|
|
},
|
|
)
|
|
await manager.start()
|
|
yield manager
|
|
await manager.stop()
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def manager_with_env_access_blocked(broker):
|
|
manager = TaskRunnerManager(
|
|
task_broker_url=broker.get_url(),
|
|
custom_env={
|
|
"N8N_RUNNERS_STDLIB_ALLOW": "os",
|
|
"N8N_BLOCK_RUNNER_ENV_ACCESS": "true",
|
|
},
|
|
)
|
|
await manager.start()
|
|
yield manager
|
|
await manager.stop()
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def manager_with_env_access_allowed(broker):
|
|
manager = TaskRunnerManager(
|
|
task_broker_url=broker.get_url(),
|
|
custom_env={
|
|
"N8N_RUNNERS_STDLIB_ALLOW": "os",
|
|
"N8N_BLOCK_RUNNER_ENV_ACCESS": "false",
|
|
},
|
|
)
|
|
await manager.start()
|
|
yield manager
|
|
await manager.stop()
|
|
|
|
|
|
def create_task_settings(
|
|
code: str,
|
|
node_mode: str,
|
|
items: Items | None = None,
|
|
continue_on_fail: bool = False,
|
|
):
|
|
return {
|
|
"code": code,
|
|
"nodeMode": NODE_MODE_TO_BROKER_STYLE[node_mode],
|
|
"items": items if items is not None else [],
|
|
"continueOnFail": continue_on_fail,
|
|
}
|
|
|
|
|
|
async def wait_for_task_done(broker, task_id: str, timeout: float = TASK_RESPONSE_WAIT):
|
|
return await broker.wait_for_msg(
|
|
"runner:taskdone",
|
|
timeout=timeout,
|
|
predicate=lambda msg: msg.get("taskId") == task_id,
|
|
)
|
|
|
|
|
|
async def wait_for_task_error(
|
|
broker, task_id: str, timeout: float = TASK_RESPONSE_WAIT
|
|
):
|
|
return await broker.wait_for_msg(
|
|
"runner:taskerror",
|
|
timeout=timeout,
|
|
predicate=lambda msg: msg.get("taskId") == task_id,
|
|
)
|
|
|
|
|
|
def get_browser_console_msgs(broker: LocalTaskBroker, task_id: str) -> list[list[str]]:
|
|
console_msgs = []
|
|
for msg in broker.get_task_rpc_messages(task_id):
|
|
if msg.get("method") == "logNodeOutput":
|
|
console_msgs.append(msg.get("params", []))
|
|
return console_msgs
|