mirror of
https://github.com/rennf93/roboco.git
synced 2026-08-03 07:23:24 +02:00
28 lines
811 B
Python
28 lines
811 B
Python
|
"""Tests for roboco.mcp.utils._get_agent_headers UNSIGNED-token guard."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import TYPE_CHECKING
|
||
|
|
|
||
|
|
if TYPE_CHECKING:
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_agent_headers_omits_unsigned_token(
|
||
|
|
monkeypatch: pytest.MonkeyPatch,
|
||
|
|
) -> None:
|
||
|
|
from roboco.mcp import utils as mcp_utils
|
||
|
|
|
||
|
|
monkeypatch.setenv("ROBOCO_AGENT_TOKEN", "UNSIGNED")
|
||
|
|
h = mcp_utils._get_agent_headers("be-dev-1")
|
||
|
|
assert "X-Agent-Token" not in h
|
||
|
|
assert h["X-Agent-ID"] == "be-dev-1"
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_agent_headers_sends_real_token(monkeypatch: pytest.MonkeyPatch) -> None:
|
||
|
|
from roboco.mcp import utils as mcp_utils
|
||
|
|
|
||
|
|
monkeypatch.setenv("ROBOCO_AGENT_TOKEN", "signed-token-abc")
|
||
|
|
h = mcp_utils._get_agent_headers("be-dev-1")
|
||
|
|
assert h["X-Agent-Token"] == "signed-token-abc"
|