test_support.py

from __future__ import annotations

import socket
import threading
from pathlib import Path
from typing import Any

import pytest


def can_bind_loopback() -> bool:
    sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    try:
        sock.bind(('127.0.0.1', 0))
    except PermissionError:
        return False
    finally:
        sock.close()
    return True


def can_start_threads(count: int = 1) -> bool:
    threads: list[threading.Thread] = []
    release = threading.Event()
    try:
        for _ in range(count):
            thread = threading.Thread(target=release.wait)
            thread.start()
            threads.append(thread)
    except RuntimeError as exc:
        if "can't start new thread" in str(exc):
            return False
        raise
    finally:
        release.set()
        for thread in threads:
            thread.join(timeout=1.0)
    return True


def skip_if_socket_bind_is_blocked() -> None:
    """Skip tests that require a loopback TCP listener when the environment forbids it."""

    if not can_bind_loopback():
        pytest.skip('sandbox forbids socket.bind() on loopback')


def skip_if_thread_start_is_blocked(count: int = 1) -> None:
    """Skip tests that require spawning worker threads when the environment forbids it."""

    if not can_start_threads(count):
        pytest.skip('environment has thread resource limits')


def mcp_test_governance(root: str | Path, permission_mode: str = 'prompt') -> Any:
    """Build a governed MCP server context (approval policy + run log) under ``root``."""
    from teaagent.mcp_server import MCPGovernance
    from teaagent.policy import PermissionMode

    return MCPGovernance.for_workspace(
        root, permission_mode=PermissionMode(permission_mode), transport='test'
    )