Byparr/tests/main_test.py
2024-10-21 13:34:07 +00:00

43 lines
987 B
Python

from http import HTTPStatus
from time import sleep
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://ext.to/",
"https://btmet.com/",
"https://extratorrent.st/", # github is blocking these
"https://idope.se/", # github is blocking these
"https://www.ygg.re/",
"https://speed.cd/browse/freeleech",
]
@pytest.mark.parametrize("website", test_websites)
def test_bypass(website: str):
sleep(3)
test_request = httpx.get(
website,
)
if test_request.status_code != HTTPStatus.OK:
pytest.skip()
response = client.post(
"/v1",
json=LinkRequest(url=website, maxTimeout=30, cmd="request.get").model_dump(),
)
assert response.status_code == HTTPStatus.OK
def test_health_check():
response = client.get("/health")
assert response.status_code == HTTPStatus.OK