42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import pytest
|
|
|
|
from batchmortal.browser import (
|
|
build_review_url,
|
|
normalize_review_language,
|
|
normalize_review_ui,
|
|
)
|
|
|
|
|
|
def test_normalize_review_language():
|
|
assert normalize_review_language(None) == "zh-CN"
|
|
assert normalize_review_language("zh-cn") == "zh-CN"
|
|
assert normalize_review_language("zh_CN") == "zh-CN"
|
|
assert normalize_review_language("en") == "en"
|
|
assert normalize_review_language("jp") == "ja"
|
|
assert normalize_review_language("ko-KR") == "ko"
|
|
|
|
|
|
def test_build_review_url():
|
|
assert build_review_url("zh-CN") == "https://mjai.ekyu.moe/zh-cn.html"
|
|
assert build_review_url("en") == "https://mjai.ekyu.moe/"
|
|
assert build_review_url("ja") == "https://mjai.ekyu.moe/ja.html"
|
|
assert build_review_url("ko") == "https://mjai.ekyu.moe/ko.html"
|
|
|
|
|
|
def test_normalize_review_ui():
|
|
assert normalize_review_ui(None) == "classic"
|
|
assert normalize_review_ui("classic") == "classic"
|
|
assert normalize_review_ui("killerducky") == "killerducky"
|
|
assert normalize_review_ui("killer-ducky") == "killerducky"
|
|
assert normalize_review_ui("kd") == "killerducky"
|
|
|
|
|
|
def test_normalize_review_ui_rejects_unknown_values():
|
|
with pytest.raises(ValueError, match="Unsupported review UI"):
|
|
normalize_review_ui("foo")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_normalize_review_language()
|
|
test_build_review_url()
|