MCPX MCP Gateway

# type: ignore # This is since `behave` is really mistyped and all dynamic. # Might be handled later. import os import asyncio from behave import when from behave.api.async_step import async_run_until_complete from typing import Any from utils.consts import * from utils.flows import ( write_flow_file, write_resource_file, FlowRepresentation, ResourceQuotaRepresentation, GatewayConfigRequests, ) @when("flow file is saved") @async_run_until_complete async def step_impl(context: Any): flow_yaml = context.flow.build_yaml() print(f"flow yaml:\n{flow_yaml}") await write_flow_file(flow_filename="flow.yaml", flow_yaml=flow_yaml) @when("resource file is saved") @async_run_until_complete async def step_impl(context: Any): resource_yaml = context.resource.build_yaml() print(f"resource yaml:\n{resource_yaml}") await write_resource_file( filename="resource_quota.yaml", resource_yaml=resource_yaml, directory_path=QUOTAS_DIRECTORY, ) @when("flow file is saved with name {file_name}") @async_run_until_complete async def step_impl(context: Any, file_name: str): flow_yaml = context.flow.build_yaml() print(f"flow yaml:\n{flow_yaml}") await write_flow_file(flow_filename=file_name, flow_yaml=flow_yaml) @when("resource file is saved with name {file_name}") @async_run_until_complete async def step_impl(context: Any, file_name: str): resource_yaml = context.resource.build_yaml() print(f"resource yaml:\n{resource_yaml}") await write_resource_file( filename=file_name, resource_yaml=resource_yaml, directory_path=QUOTAS_DIRECTORY ) @when("flow file is saved on {container_name}") @async_run_until_complete async def step_impl(context: Any, container_name: str): flow_yaml = context.flow.build_yaml() print(f"flow yaml:\n{flow_yaml}") await write_flow_file( flow_filename="flow.yaml", flow_yaml=flow_yaml, container_name=container_name ) @when("resource file is saved on {container_name}") @async_run_until_complete async def step_impl(context: Any, container_name: str): resource_yaml = context.resource.build_yaml() print(f"resource yaml:\n{resource_yaml}") await write_resource_file( filename="resource_quota.yaml", resource_yaml=resource_yaml, container_name=container_name, directory_path=QUOTAS_DIRECTORY, ) @when("Flow configuration is used") @async_run_until_complete async def step_impl(context: Any): context.user_flow = FlowRepresentation() context.gateway_config = GatewayConfigRequests() context.quota = ResourceQuotaRepresentation() @when("Flow configuration is saved") @async_run_until_complete async def step_impl(context: Any): assert await context.gateway_config.build_yaml() @when("load_flows command is run") @async_run_until_complete async def step_impl(_: Any): assert os.system("docker exec lunar-proxy load_flows") == 0 @when("load_flows command is run on {container_name}") @async_run_until_complete async def step_impl(_: Any, container_name: str): for _ in range(10): if os.system(f"docker exec {container_name} load_flows") == 0: return True await asyncio.sleep(1)