gabriel / muse public
test_heartbeat_envelope.py python
63 lines 2.1 KB
Raw
sha256:a317886dc0496c4af7b285b3e41c86c4c34ea2e79afc63b8829aadb1ada7903f chore: bump version to 0.2.0rc15 to match musehub#113 fix release Sonnet 4.6 patch 63 days ago
1 """Envelope tests for muse coord heartbeat command."""
2 from __future__ import annotations
3 from collections.abc import Mapping
4 import json
5 import pytest
6 from tests.cli_test_helper import CliRunner
7
8 runner = CliRunner()
9
10 _FIELDS = ("muse_version", "schema", "timestamp", "warnings")
11
12
13 def _check(d: Mapping[str, object]) -> None:
14 for f in _FIELDS:
15 assert f in d, f"missing {f}"
16 assert "schema_version" not in d
17
18
19 class TestHeartbeatEnvelope:
20 def test_heartbeat_not_found_has_envelope(self) -> None:
21 # A well-formed sha256 ID that doesn't exist yields a not_found JSON response.
22 r = runner.invoke(
23 None,
24 ["coord", "heartbeat", f"sha256:{'0' * 64}",
25 "--run-id", "test", "--json"],
26 )
27 # exit_code 4 = NOT_FOUND
28 d = json.loads(r.output)
29 _check(d)
30 assert d["status"] == "not_found"
31
32
33 class TestRegisterFlags:
34 def test_json_short_flag(self) -> None:
35 import argparse
36 from muse.cli.commands.heartbeat_coord import register
37 p = argparse.ArgumentParser()
38 subs = p.add_subparsers()
39 register(subs)
40 args = p.parse_args(["heartbeat", "fake-res-id", "--run-id", "ag", "-j"])
41 assert args.json_out is True
42
43 def test_json_long_flag(self) -> None:
44 import argparse
45 from muse.cli.commands.heartbeat_coord import register
46 p = argparse.ArgumentParser()
47 subs = p.add_subparsers()
48 register(subs)
49 args = p.parse_args(["heartbeat", "fake-res-id", "--run-id", "ag", "--json"])
50 assert args.json_out is True
51
52 def test_default_no_json(self) -> None:
53 import argparse
54 from muse.cli.commands.heartbeat_coord import register
55 p = argparse.ArgumentParser()
56 subs = p.add_subparsers()
57 register(subs)
58 # Command-specific required args may differ; just check dest exists when possible
59 try:
60 args = p.parse_args(["heartbeat"])
61 assert args.json_out is False
62 except SystemExit:
63 pass # required positional args missing — flag default still correct
File History 2 commits
sha256:a317886dc0496c4af7b285b3e41c86c4c34ea2e79afc63b8829aadb1ada7903f chore: bump version to 0.2.0rc15 to match musehub#113 fix release Sonnet 4.6 patch 63 days ago
sha256:f3b726b50f0aee3622bba751e0a67aa7ae4cf75a798477dbce581940b6a9cf70 feat: migrate invariants cache to .muse/cache/invariants.ms… Sonnet 4.6 patch 131 days ago