test_cli.py 954 B

123456789101112131415161718192021222324252627282930313233
  1. import pytest
  2. from pydantic import ValidationError
  3. from content_agent.cli import build_parser, request_from_args
  4. def test_cli_builds_request_with_demand_content_id():
  5. parser = build_parser()
  6. args = parser.parse_args(["--platform-mode", "real", "--demand-content-id", "130"])
  7. request = request_from_args(args)
  8. assert request.platform_mode == "real"
  9. assert request.demand_content_id == 130
  10. assert request.source is None
  11. assert request.run_label is None
  12. def test_cli_builds_request_with_run_label():
  13. parser = build_parser()
  14. args = parser.parse_args(["--run-label", "cfa_batch_001"])
  15. request = request_from_args(args)
  16. assert request.run_label == "cfa_batch_001"
  17. def test_cli_reuses_source_selector_exclusivity():
  18. parser = build_parser()
  19. args = parser.parse_args(["--source", "source.json", "--demand-content-id", "130"])
  20. with pytest.raises(ValidationError):
  21. request_from_args(args)