57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
"""Tests for the CLI argument parsing in main.py."""
|
|
|
|
import pytest
|
|
|
|
import main
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# selected_sources: comma-separated --source values
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_all_expands_to_every_source():
|
|
assert main.selected_sources("all") == ["rightmove", "onthemarket", "zoopla"]
|
|
|
|
|
|
def test_single_source():
|
|
assert main.selected_sources("rightmove") == ["rightmove"]
|
|
|
|
|
|
def test_comma_separated_sources():
|
|
assert main.selected_sources("rightmove,onthemarket") == [
|
|
"rightmove",
|
|
"onthemarket",
|
|
]
|
|
|
|
|
|
def test_whitespace_and_case_are_tolerated():
|
|
assert main.selected_sources(" Rightmove , ZOOPLA ") == ["rightmove", "zoopla"]
|
|
|
|
|
|
def test_order_is_canonical_and_deduplicated():
|
|
# Caller order ignored, duplicates collapsed: keeps SOURCES order so the
|
|
# downstream merge/dedup stays deterministic.
|
|
assert main.selected_sources("zoopla,rightmove,rightmove") == [
|
|
"rightmove",
|
|
"zoopla",
|
|
]
|
|
|
|
|
|
def test_all_anywhere_in_the_list_wins():
|
|
assert main.selected_sources("rightmove,all") == [
|
|
"rightmove",
|
|
"onthemarket",
|
|
"zoopla",
|
|
]
|
|
|
|
|
|
def test_empty_value_is_rejected():
|
|
with pytest.raises(SystemExit):
|
|
main.selected_sources(" , ")
|
|
|
|
|
|
def test_unknown_source_is_rejected():
|
|
with pytest.raises(SystemExit) as excinfo:
|
|
main.selected_sources("rightmove,foo")
|
|
assert "foo" in str(excinfo.value)
|