mirror of
https://github.com/warp-tech/warpgate.git
synced 2024-11-15 12:31:55 +08:00
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
|
import requests
|
||
|
|
|
||
|
|
|
||
|
|
from .util import wait_port
|
||
|
|
|
||
|
|
|
||
|
|
class TestHTTPUserAuthPassword:
|
||
|
|
def test_auth_password_success(
|
||
|
|
self,
|
||
|
|
http_common_wg_port,
|
||
|
|
):
|
||
|
|
wait_port(http_common_wg_port, recv=False)
|
||
|
|
session = requests.Session()
|
||
|
|
session.verify = False
|
||
|
|
url = f'https://localhost:{http_common_wg_port}'
|
||
|
|
|
||
|
|
response = session.get(f'{url}/?warpgate-target=echo', allow_redirects=False)
|
||
|
|
assert response.status_code // 100 != 2
|
||
|
|
|
||
|
|
response = session.post(
|
||
|
|
f'{url}/@warpgate/api/auth/login',
|
||
|
|
json={
|
||
|
|
'username': 'user',
|
||
|
|
'password': '123',
|
||
|
|
},
|
||
|
|
)
|
||
|
|
assert response.status_code // 100 == 2
|
||
|
|
|
||
|
|
response = session.get(
|
||
|
|
f'{url}/some/path?a=b&warpgate-target=echo&c=d', allow_redirects=False
|
||
|
|
)
|
||
|
|
assert response.status_code // 100 == 2
|
||
|
|
assert response.json()['path'] == '/some/path'
|
||
|
|
|
||
|
|
def test_auth_password_fail(
|
||
|
|
self,
|
||
|
|
http_common_wg_port,
|
||
|
|
):
|
||
|
|
wait_port(http_common_wg_port, recv=False)
|
||
|
|
session = requests.Session()
|
||
|
|
session.verify = False
|
||
|
|
url = f'https://localhost:{http_common_wg_port}'
|
||
|
|
|
||
|
|
response = session.post(
|
||
|
|
f'{url}/@warpgate/api/auth/login',
|
||
|
|
json={
|
||
|
|
'username': 'user',
|
||
|
|
'password': '321321',
|
||
|
|
},
|
||
|
|
)
|
||
|
|
assert response.status_code // 100 != 2
|
||
|
|
|
||
|
|
response = session.get(
|
||
|
|
f'{url}/some/path?a=b&warpgate-target=echo&c=d', allow_redirects=False
|
||
|
|
)
|
||
|
|
assert response.status_code // 100 != 2
|