mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-04-28 03:30:23 +00:00
51 lines
1.2 KiB
Python
51 lines
1.2 KiB
Python
import sys
|
|
from pathlib import Path
|
|
|
|
PROJECT_ROOT = Path(__file__).resolve().parents[1]
|
|
if str(PROJECT_ROOT) not in sys.path:
|
|
sys.path.insert(0, str(PROJECT_ROOT))
|
|
|
|
from helpers.ws import validate_ws_origin
|
|
|
|
|
|
def test_validate_ws_origin_allows_same_origin_with_explicit_port():
|
|
ok, reason = validate_ws_origin(
|
|
{
|
|
"HTTP_ORIGIN": "http://localhost:5000",
|
|
"HTTP_HOST": "localhost:5000",
|
|
}
|
|
)
|
|
assert ok is True
|
|
assert reason is None
|
|
|
|
|
|
def test_validate_ws_origin_allows_default_https_port_without_explicit_port():
|
|
ok, reason = validate_ws_origin(
|
|
{
|
|
"HTTP_ORIGIN": "https://example.com",
|
|
"HTTP_HOST": "example.com",
|
|
}
|
|
)
|
|
assert ok is True
|
|
assert reason is None
|
|
|
|
|
|
def test_validate_ws_origin_rejects_missing_origin():
|
|
ok, reason = validate_ws_origin(
|
|
{
|
|
"HTTP_HOST": "localhost:5000",
|
|
}
|
|
)
|
|
assert ok is False
|
|
assert reason == "missing_origin"
|
|
|
|
|
|
def test_validate_ws_origin_rejects_cross_origin():
|
|
ok, reason = validate_ws_origin(
|
|
{
|
|
"HTTP_ORIGIN": "http://evil.test",
|
|
"HTTP_HOST": "localhost:5000",
|
|
}
|
|
)
|
|
assert ok is False
|
|
assert reason == "origin_host_mismatch"
|