AutoRpa/rpa_cart/tests/test_main.py

24 lines
632 B
Python

from main import parse_args
def test_parse_args_defaults():
args = parse_args([])
assert args.days == 0
assert args.dry_run is False
assert args.xls is None
assert args.command == "pay"
def test_parse_args_xls_dry_run():
args = parse_args(["--xls", "/tmp/orders.xls", "--dry-run"])
assert args.dry_run is True
assert str(args.xls).endswith("orders.xls")
assert args.command == "pay"
def test_parse_args_cart_subcommand():
args = parse_args(["cart", "--limit", "5", "--platform", "yaoex"])
assert args.command == "cart"
assert args.limit == 5
assert args.platform == "yaoex"