gabriel / muse public
test_heartbeat_envelope.py python
63 lines 2.1 KB
Raw
sha256:be3641f35bdbcc094677776a77b9aa6a5dab891f8fab201dc162d03c2bab5aea fix(read): strip position:null from structured_delta ops in… Sonnet 4.6 patch 23 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 3 commits
sha256:36c3cb3e76619d4c30a6d9bf81b5ec4ff148e30dcfed913e3114ca7b43b81c7e fix: rename objects→blobs in push client and all stale test… Sonnet 4.6 patch 22 days ago
sha256:c06a9b9b9fee26c68ea725b44d54b2c0a171301ce9de746d5b656617b4463a9a fix: repair four test failures from post-migration audit Sonnet 4.6 patch 28 days ago
sha256:1900655993c83c4107067375548a7be823e471d2515830842f1a12cba4bd3cdf fix: unified object store migration — idempotent writes, JS… Sonnet 4.6 minor 29 days ago