mirror of
https://github.com/warp-tech/warpgate.git
synced 2024-11-14 11:57:36 +08:00
96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
from urllib.parse import unquote
|
|
from uuid import uuid4
|
|
import requests
|
|
|
|
from tests.conftest import WarpgateProcess
|
|
|
|
from .api_client import (
|
|
api_admin_session,
|
|
api_create_target,
|
|
api_create_user,
|
|
api_create_role,
|
|
api_add_role_to_user,
|
|
api_add_role_to_target,
|
|
)
|
|
from .test_http_common import * # noqa
|
|
|
|
|
|
class Test:
|
|
def test_basic(
|
|
self,
|
|
echo_server_port,
|
|
shared_wg: WarpgateProcess,
|
|
):
|
|
url = f"https://localhost:{shared_wg.http_port}"
|
|
|
|
with api_admin_session(url) as session:
|
|
role = api_create_role(url, session, {"name": f"role-{uuid4()}"})
|
|
user = api_create_user(
|
|
url,
|
|
session,
|
|
{
|
|
"username": "user",
|
|
"credentials": [
|
|
{
|
|
"kind": "Password",
|
|
"hash": "123",
|
|
}
|
|
],
|
|
},
|
|
)
|
|
api_add_role_to_user(url, session, user["id"], role["id"])
|
|
echo_target = api_create_target(
|
|
url,
|
|
session,
|
|
{
|
|
"name": f"echo-{uuid4()}",
|
|
"options": {
|
|
"kind": "Http",
|
|
"url": f"http://localhost:{echo_server_port}",
|
|
"tls": {
|
|
"mode": "Disabled",
|
|
"verify": False,
|
|
},
|
|
},
|
|
},
|
|
)
|
|
api_add_role_to_target(url, session, echo_target["id"], role["id"])
|
|
|
|
session = requests.Session()
|
|
session.verify = False
|
|
|
|
response = session.get(
|
|
f"{url}/?warpgate-target={echo_target['name']}", allow_redirects=False
|
|
)
|
|
assert response.status_code == 307
|
|
redirect = response.headers["location"]
|
|
print(unquote(redirect))
|
|
assert (
|
|
unquote(redirect)
|
|
== f"/@warpgate#/login?next=/?warpgate-target={echo_target['name']}"
|
|
)
|
|
|
|
response = session.get(f"{url}/@warpgate/api/info").json()
|
|
assert response["username"] is None
|
|
|
|
response = session.post(
|
|
f"{url}/@warpgate/api/auth/login",
|
|
json={
|
|
"username": user["username"],
|
|
"password": "123",
|
|
},
|
|
)
|
|
assert response.status_code == 201
|
|
|
|
response = session.get(f"{url}/@warpgate/api/info").json()
|
|
assert response["username"] == user["username"]
|
|
|
|
response = session.get(
|
|
f"{url}/some/path?a=b&warpgate-target={echo_target['name']}&c=d",
|
|
allow_redirects=False,
|
|
)
|
|
assert response.status_code == 200
|
|
assert response.json()["method"] == "GET"
|
|
assert response.json()["path"] == "/some/path"
|
|
assert response.json()["args"]["a"] == "b"
|
|
assert response.json()["args"]["c"] == "d"
|