27 lines
675 B
Python
27 lines
675 B
Python
from http import HTTPStatus
|
|
|
|
import pytest
|
|
from starlette.testclient import TestClient
|
|
|
|
from main import app
|
|
from src.models.requests import LinkRequest
|
|
|
|
client = TestClient(app)
|
|
|
|
test_websites = [
|
|
"https://ext.to/",
|
|
"https://btmet.com/",
|
|
# "https://extratorrent.st/", # github is blocking these
|
|
# "https://idope.se/", # github is blocking these
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize("website", test_websites)
|
|
def test_bypass(website: str):
|
|
response = client.post(
|
|
"/v1",
|
|
json=LinkRequest(
|
|
url=website, maxTimeout=60 * len(test_websites), cmd="request.get"
|
|
).model_dump(),
|
|
)
|
|
assert response.status_code == HTTPStatus.OK
|