test_forecast_envelope.py
file-level
1
files
1
commits
0
hotspots
0
🧊 dead
0
💥 blast risk
| 1 | """Envelope tests for code forecast command.""" |
| 2 | from __future__ import annotations |
| 3 | from collections.abc import Mapping |
| 4 | import argparse |
| 5 | import json |
| 6 | import pytest |
| 7 | from tests.cli_test_helper import CliRunner |
| 8 | |
| 9 | runner = CliRunner() |
| 10 | |
| 11 | _FIELDS = ("muse_version", "schema", "timestamp", "warnings") |
| 12 | |
| 13 | |
| 14 | def _check(d: Mapping[str, object]) -> None: |
| 15 | for f in _FIELDS: |
| 16 | assert f in d, f"missing {f}" |
| 17 | assert "schema_version" not in d |
| 18 | |
| 19 | |
| 20 | class TestForecastEnvelope: |
| 21 | def test_forecast_has_envelope(self) -> None: |
| 22 | r = runner.invoke(None, ["coord", "forecast", "--json"]) |
| 23 | assert r.exit_code == 0, r.output |
| 24 | _check(json.loads(r.output)) |
| 25 | |
| 26 | |
| 27 | # --------------------------------------------------------------------------- |
| 28 | # TestRegisterFlags — argparse-level verification |
| 29 | # --------------------------------------------------------------------------- |
| 30 | |
| 31 | |
| 32 | class TestRegisterFlags: |
| 33 | """Verify that register() wires --json / -j correctly.""" |
| 34 | |
| 35 | def _make_parser(self) -> "argparse.ArgumentParser": |
| 36 | import argparse |
| 37 | from muse.cli.commands.forecast import register |
| 38 | ap = argparse.ArgumentParser() |
| 39 | subs = ap.add_subparsers() |
| 40 | register(subs) |
| 41 | return ap |
| 42 | |
| 43 | def test_json_flag_long(self) -> None: |
| 44 | ns = self._make_parser().parse_args(["forecast", "--json"]) |
| 45 | assert ns.json_out is True |
| 46 | |
| 47 | def test_j_alias(self) -> None: |
| 48 | ns = self._make_parser().parse_args(["forecast", "-j"]) |
| 49 | assert ns.json_out is True |
| 50 | |
| 51 | def test_default_is_text(self) -> None: |
| 52 | ns = self._make_parser().parse_args(["forecast"]) |
| 53 | assert ns.json_out is False |
| 54 | |
| 55 | def test_dest_is_json_out(self) -> None: |
| 56 | ns = self._make_parser().parse_args(["forecast", "-j"]) |
| 57 | assert hasattr(ns, "json_out") |
| 58 | assert not hasattr(ns, "fmt") |