mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
Paul Hernandez
816ee85fb9
Signed-off-by: phernandez <paul@basicmachines.co> Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
"""Telemetry coverage for CLI command boundaries."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
from typing import Any, cast
|
|
|
|
import logfire
|
|
|
|
from basic_memory.cli import app as cli_app
|
|
|
|
|
|
class FakeContext:
|
|
"""Small Typer-like context for callback testing."""
|
|
|
|
def __init__(self, invoked_subcommand: str | None) -> None:
|
|
self.invoked_subcommand = invoked_subcommand
|
|
self.resources: list[object] = []
|
|
self.close_callbacks: list[object] = []
|
|
|
|
def with_resource(self, resource: object) -> None:
|
|
self.resources.append(resource)
|
|
|
|
def call_on_close(self, callback) -> None:
|
|
self.close_callbacks.append(callback)
|
|
|
|
|
|
def test_app_callback_registers_command_operation(monkeypatch) -> None:
|
|
operations: list[tuple[str, dict]] = []
|
|
resource = object()
|
|
|
|
monkeypatch.setattr(cli_app, "init_cli_logging", lambda: None)
|
|
# Container needs a `config` attribute because app_callback passes it to the
|
|
# uvloop policy installer; the helper itself is stubbed below for this
|
|
# telemetry-focused test.
|
|
monkeypatch.setattr(
|
|
cli_app.CliContainer, "create", staticmethod(lambda: SimpleNamespace(config=object()))
|
|
)
|
|
monkeypatch.setattr(cli_app, "set_container", lambda container: None)
|
|
# app_callback installs the uvloop policy for the Postgres backend; stub the
|
|
# helper so this telemetry test does not depend on event-loop policy state.
|
|
monkeypatch.setattr("basic_memory.db.maybe_install_uvloop", lambda config: False)
|
|
monkeypatch.setattr(cli_app, "maybe_show_init_line", lambda command_name: None)
|
|
monkeypatch.setattr(cli_app, "maybe_show_cloud_promo", lambda command_name: None)
|
|
monkeypatch.setattr(cli_app, "maybe_run_periodic_auto_update", lambda command_name: None)
|
|
|
|
def fake_span(name: str, **attrs):
|
|
operations.append((name, attrs))
|
|
return resource
|
|
|
|
monkeypatch.setattr(logfire, "span", fake_span)
|
|
|
|
ctx = FakeContext(invoked_subcommand="status")
|
|
cli_app.app_callback(cast(Any, ctx), version=None)
|
|
|
|
assert ctx.resources == [resource]
|
|
assert operations == [
|
|
(
|
|
"cli.command.status",
|
|
{"entrypoint": "cli", "command_name": "status"},
|
|
)
|
|
]
|