from pytest import mark

from heliotrope.sanic import Heliotrope

url = "/api/proxy/"


@mark.flaky(reruns=3, reruns_delay=5)
def test_proxy_hitomi(app: Heliotrope, image_url: str):
    _, response = app.test_client.get(url + image_url)
    assert response.status == 200


def test_proxy_pixiv(app: Heliotrope):
    image_url = "https://i.pximg.net/img-master/img/2020/01/28/01/07/34/79136250_p0_master1200.jpg"
    _, response = app.test_client.get(url + image_url)
    assert response.status == 200


def test_proxy_invalid(app: Heliotrope):
    _, response = app.test_client.get(url + "test")
    assert response.status == 400


def test_proxy_not_found(app: Heliotrope):
    image_url = "https://i.pximg.net/img-master/img/1234/56/78/91/07/34/7312310_p0_master1200.jpg"
    _, response = app.test_client.get(url + image_url)
    assert response.status == 404