Ubuntu
63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
"""Unit tests for formatter (no network required)."""
|
|
|
|
from xhs_cli.formatter import coerce_int, extract_note_id, format_count, parse_note_reference
|
|
|
|
|
|
class TestFormatCount:
|
|
def test_small_number(self):
|
|
assert format_count(123) == "123"
|
|
|
|
def test_wan(self):
|
|
assert format_count(12345) == "1.2万"
|
|
|
|
def test_yi(self):
|
|
assert format_count(123456789) == "1.2亿"
|
|
|
|
def test_string_input(self):
|
|
assert format_count("5678") == "5678"
|
|
|
|
def test_string_large(self):
|
|
assert format_count("50000") == "5.0万"
|
|
|
|
|
|
class TestCoerceInt:
|
|
def test_int_input(self):
|
|
assert coerce_int(3) == 3
|
|
|
|
def test_string_input(self):
|
|
assert coerce_int("42") == 42
|
|
|
|
def test_invalid_string_falls_back(self):
|
|
assert coerce_int("10+") == 0
|
|
|
|
|
|
class TestExtractNoteId:
|
|
def test_plain_id(self):
|
|
assert extract_note_id("abc123def") == "abc123def"
|
|
|
|
def test_explore_url(self):
|
|
result = extract_note_id("https://www.xiaohongshu.com/explore/abc123def")
|
|
assert result == "abc123def"
|
|
|
|
def test_url_with_params(self):
|
|
result = extract_note_id("https://www.xiaohongshu.com/explore/abc123?xsec_token=xxx")
|
|
assert result == "abc123"
|
|
|
|
def test_discovery_url(self):
|
|
result = extract_note_id("https://www.xiaohongshu.com/discovery/item/abc123")
|
|
assert result == "abc123"
|
|
|
|
def test_trailing_slash(self):
|
|
result = extract_note_id("https://www.xiaohongshu.com/explore/abc123/")
|
|
assert result == "abc123"
|
|
|
|
|
|
class TestParseNoteReference:
|
|
def test_extracts_token_and_source(self):
|
|
note_id, token, source = parse_note_reference(
|
|
"https://www.xiaohongshu.com/explore/abc123?xsec_token=token-1&xsec_source=pc_search"
|
|
)
|
|
assert note_id == "abc123"
|
|
assert token == "token-1"
|
|
assert source == "pc_search"
|