wehub-resource-sync
555e282cc4
ci / changelog_check (push) Waiting to run
ci / check_changes (push) Waiting to run
ci / build_mem0 (3.10) (push) Blocked by required conditions
ci / build_mem0 (3.11) (push) Blocked by required conditions
ci / build_mem0 (3.12) (push) Blocked by required conditions
CLI Node CI / lint (push) Waiting to run
CLI Node CI / test (20) (push) Waiting to run
CLI Node CI / test (22) (push) Waiting to run
CLI Node CI / build (push) Waiting to run
CLI Python CI / lint (push) Waiting to run
CLI Python CI / test (3.10) (push) Waiting to run
CLI Python CI / test (3.11) (push) Waiting to run
CLI Python CI / test (3.12) (push) Waiting to run
CLI Python CI / build (push) Waiting to run
openclaw checks / lint (push) Waiting to run
openclaw checks / test (20) (push) Waiting to run
openclaw checks / test (22) (push) Waiting to run
openclaw checks / build (push) Waiting to run
opencode-plugin checks / build (push) Waiting to run
pi-agent-plugin checks / lint (push) Waiting to run
pi-agent-plugin checks / test (20) (push) Waiting to run
pi-agent-plugin checks / test (22) (push) Waiting to run
pi-agent-plugin checks / build (push) Waiting to run
TypeScript SDK CI / check_changes (push) Waiting to run
TypeScript SDK CI / changelog_check (push) Blocked by required conditions
TypeScript SDK CI / build_ts_sdk (20) (push) Blocked by required conditions
TypeScript SDK CI / build_ts_sdk (22) (push) Blocked by required conditions
TypeScript SDK CI / integration_ts_sdk (20) (push) Blocked by required conditions
TypeScript SDK CI / integration_ts_sdk (22) (push) Blocked by required conditions
81 lines
2.1 KiB
Python
81 lines
2.1 KiB
Python
"""Tests for telemetry subprocess secret handling."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import io
|
|
import json
|
|
import subprocess
|
|
import sys
|
|
|
|
from mem0_cli.config import Mem0Config, save_config
|
|
from mem0_cli.telemetry import capture_event
|
|
from mem0_cli.telemetry_sender import _load_context
|
|
|
|
|
|
class _CaptureStdin:
|
|
def __init__(self):
|
|
self.buffer = ""
|
|
self.closed = False
|
|
|
|
def write(self, value: str) -> None:
|
|
self.buffer += value
|
|
|
|
def close(self) -> None:
|
|
self.closed = True
|
|
|
|
|
|
class _DummyProcess:
|
|
def __init__(self):
|
|
self.stdin = _CaptureStdin()
|
|
|
|
|
|
def test_capture_event_writes_context_to_stdin_not_argv(isolate_config, monkeypatch):
|
|
config = Mem0Config()
|
|
config.platform.api_key = "m0-test-secret"
|
|
config.telemetry.anonymous_id = "cli-anon-test"
|
|
save_config(config)
|
|
|
|
captured: dict[str, object] = {}
|
|
proc = _DummyProcess()
|
|
|
|
def fake_popen(args, **kwargs):
|
|
captured["args"] = args
|
|
captured["kwargs"] = kwargs
|
|
return proc
|
|
|
|
monkeypatch.setattr("mem0_cli.telemetry.subprocess.Popen", fake_popen)
|
|
|
|
capture_event("unit_test_event", {"case": "stdin-secret"})
|
|
|
|
argv = captured["args"]
|
|
assert argv == [sys.executable, "-m", "mem0_cli.telemetry_sender"]
|
|
assert all("m0-test-secret" not in arg for arg in argv)
|
|
|
|
kwargs = captured["kwargs"]
|
|
assert kwargs["stdin"] == subprocess.PIPE
|
|
assert kwargs["text"] is True
|
|
|
|
ctx = json.loads(proc.stdin.buffer)
|
|
assert ctx["mem0_api_key"] == "m0-test-secret"
|
|
assert ctx["payload"]["event"] == "unit_test_event"
|
|
|
|
assert proc.stdin.closed
|
|
|
|
|
|
def test_load_context_reads_from_stdin(monkeypatch):
|
|
monkeypatch.setattr("sys.argv", ["telemetry_sender"])
|
|
monkeypatch.setattr("sys.stdin", io.StringIO('{"payload": {"event": "stdin"}}'))
|
|
|
|
ctx = _load_context()
|
|
|
|
assert ctx["payload"]["event"] == "stdin"
|
|
|
|
|
|
def test_load_context_falls_back_to_argv(monkeypatch):
|
|
monkeypatch.setattr("sys.argv", ["telemetry_sender", '{"payload": {"event": "argv"}}'])
|
|
monkeypatch.setattr("sys.stdin", io.StringIO(""))
|
|
|
|
ctx = _load_context()
|
|
|
|
assert ctx["payload"]["event"] == "argv"
|