OpenHands/tests/unit/test_microagent_utils.py

import os

import pytest
from pytest import MonkeyPatch

import openhands.agenthub  # noqa: F401
from openhands.core.exceptions import (
    AgentNotRegisteredError,
    MicroAgentValidationError,
)
from openhands.utils.microagent import MicroAgent

CONTENT = (
    '# dummy header\n' 'dummy content\n' '## dummy subheader\n' 'dummy subcontent\n'
)


def test_micro_agent_load(tmp_path, monkeypatch: MonkeyPatch):
    with open(os.path.join(tmp_path, 'dummy.md'), 'w') as f:
        f.write(
            (
                '---\n'
                'name: dummy\n'
                'agent: CodeActAgent\n'
                'require_env_var:\n'
                '  SANDBOX_OPENHANDS_TEST_ENV_VAR: "Set this environment variable for testing purposes"\n'
                '---\n' + CONTENT
            )
        )

    # Patch the required environment variable
    monkeypatch.setenv('SANDBOX_OPENHANDS_TEST_ENV_VAR', 'dummy_value')

    micro_agent = MicroAgent(os.path.join(tmp_path, 'dummy.md'))
    assert micro_agent is not None
    assert micro_agent.content == CONTENT.strip()


def test_not_existing_agent(tmp_path, monkeypatch: MonkeyPatch):
    with open(os.path.join(tmp_path, 'dummy.md'), 'w') as f:
        f.write(
            (
                '---\n'
                'name: dummy\n'
                'agent: NotExistingAgent\n'
                'require_env_var:\n'
                '  SANDBOX_OPENHANDS_TEST_ENV_VAR: "Set this environment variable for testing purposes"\n'
                '---\n' + CONTENT
            )
        )
    monkeypatch.setenv('SANDBOX_OPENHANDS_TEST_ENV_VAR', 'dummy_value')

    with pytest.raises(AgentNotRegisteredError):
        MicroAgent(os.path.join(tmp_path, 'dummy.md'))


def test_not_existing_env_var(tmp_path):
    with open(os.path.join(tmp_path, 'dummy.md'), 'w') as f:
        f.write(
            (
                '---\n'
                'name: dummy\n'
                'agent: CodeActAgent\n'
                'require_env_var:\n'
                '  SANDBOX_OPENHANDS_TEST_ENV_VAR: "Set this environment variable for testing purposes"\n'
                '---\n' + CONTENT
            )
        )

    with pytest.raises(MicroAgentValidationError) as excinfo:
        MicroAgent(os.path.join(tmp_path, 'dummy.md'))

    assert 'Set this environment variable for testing purposes' in str(excinfo.value)