20 lines
767 B
Python
20 lines
767 B
Python
from __future__ import annotations
|
|
|
|
from app.core.pximg_reverse_proxy import pick_pximg_mirror_host_for_request
|
|
|
|
|
|
def test_pick_pximg_mirror_host_for_request_prefers_re_for_cn() -> None:
|
|
host = pick_pximg_mirror_host_for_request(headers={"CF-IPCountry": "CN"}, fallback_host="i.pixiv.cat")
|
|
assert host == "i.pixiv.re"
|
|
|
|
|
|
def test_pick_pximg_mirror_host_for_request_uses_fallback_for_non_cn() -> None:
|
|
host = pick_pximg_mirror_host_for_request(headers={"CF-IPCountry": "US"}, fallback_host="i.pixiv.nl")
|
|
assert host == "i.pixiv.nl"
|
|
|
|
|
|
def test_pick_pximg_mirror_host_for_request_falls_back_to_default_on_invalid_fallback() -> None:
|
|
host = pick_pximg_mirror_host_for_request(headers=None, fallback_host="not-a-host")
|
|
assert host == "i.pixiv.cat"
|
|
|