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:
35
tests/unit_tests/test_internal_auth.py
Normal file
35
tests/unit_tests/test_internal_auth.py
Normal 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
|
||||
Reference in New Issue
Block a user