Byparr/tests/main_test.py

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=120 * len(test_websites), cmd="request.get"
        ).model_dump(),
    )
    if response.status_code == HTTPStatus.TOO_MANY_REQUESTS:
        # if rate limited
        assert True
    assert response.status_code == HTTPStatus.OK