feat: self healing skyvern api key (#3614)

Co-authored-by: Suchintan <suchintan@users.noreply.github.com>
Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>
This commit is contained in:
greg niemeyer
2025-10-13 07:55:59 -07:00
committed by GitHub
parent a8179ae61c
commit 2faf4e102f
16 changed files with 638 additions and 46 deletions

View File

@@ -0,0 +1,35 @@
from starlette.requests import Request
from skyvern.forge.sdk.routes.internal_auth import _is_local_request
def _make_request(host: str | None) -> Request:
scope = {
"type": "http",
"client": (host, 12345) if host else None,
"headers": [],
"method": "GET",
"path": "/",
"scheme": "http",
}
return Request(scope)
def test_is_local_request_returns_false_for_public_ip() -> None:
request = _make_request("8.8.8.8") # public IPv4 address
assert _is_local_request(request) is False
def test_is_local_request_accepts_loopback() -> None:
request = _make_request("127.0.0.1")
assert _is_local_request(request) is True
def test_is_local_request_accepts_private_networks() -> None:
request = _make_request("192.168.1.20")
assert _is_local_request(request) is True
def test_is_local_request_handles_missing_client() -> None:
request = _make_request(None)
assert _is_local_request(request) is False