mirror of
https://github.com/ThePhaseless/Byparr.git
synced 2025-03-16 02:00:21 +08:00
64 lines
1.5 KiB
Python
64 lines
1.5 KiB
Python
from http import HTTPStatus
|
|
|
|
import httpx
|
|
import pytest
|
|
from starlette.testclient import TestClient
|
|
|
|
from main import app
|
|
from src.models.requests import LinkRequest
|
|
|
|
client = TestClient(app)
|
|
|
|
test_websites = [
|
|
"https://btmet.com/",
|
|
"https://ext.to/",
|
|
]
|
|
|
|
# Not testable on github actions
|
|
github_restricted = [
|
|
"https://www.ygg.re/",
|
|
"https://extratorrent.st/",
|
|
"https://idope.se/",
|
|
"https://speed.cd/login",
|
|
]
|
|
|
|
# if os.getenv("GITHUB_ACTIONS") != "true":
|
|
test_websites.extend(github_restricted)
|
|
|
|
|
|
@pytest.mark.parametrize("website", test_websites)
|
|
def test_bypass(website: str):
|
|
"""
|
|
Tests if the service can bypass cloudflare/DDOS-GUARD on given websites.
|
|
|
|
This test is skipped if the website is not reachable or does not have cloudflare/DDOS-GUARD.
|
|
"""
|
|
test_request = httpx.get(
|
|
website,
|
|
)
|
|
if (
|
|
test_request.status_code != HTTPStatus.OK
|
|
and "Just a moment..." not in test_request.text
|
|
):
|
|
pytest.skip(f"Skipping {website} due to {test_request.status_code}")
|
|
|
|
response = client.post(
|
|
"/v1",
|
|
json=LinkRequest.model_construct(
|
|
url=website, max_timeout=30, cmd="request.get"
|
|
).model_dump(),
|
|
)
|
|
|
|
assert response.status_code == HTTPStatus.OK
|
|
|
|
|
|
def test_health_check():
|
|
"""
|
|
Tests the health check endpoint.
|
|
|
|
This test ensures that the health check
|
|
endpoint returns HTTPStatus.OK.
|
|
"""
|
|
response = client.get("/health")
|
|
assert response.status_code == HTTPStatus.OK
|