solpbc.org
personal memory agent
1# SPDX-License-Identifier: AGPL-3.0-only
2# Copyright (c) 2026 sol pbc
3
4"""Guard package-import defaults for telemetry-off environment variables."""
5
6import os
7import subprocess
8import sys
9
10TELEMETRY_ENV_KEYS = {
11 "HF_HUB_DISABLE_TELEMETRY",
12 "DO_NOT_TRACK",
13 "OTEL_SDK_DISABLED",
14}
15
16
17def test_telemetry_env_defaults_set_at_import():
18 env = {
19 key: value for key, value in os.environ.items() if key not in TELEMETRY_ENV_KEYS
20 }
21
22 result = subprocess.run(
23 [
24 sys.executable,
25 "-c",
26 (
27 "import os; "
28 "import solstone; "
29 "assert os.environ['HF_HUB_DISABLE_TELEMETRY'] == '1'; "
30 "assert os.environ['DO_NOT_TRACK'] == '1'; "
31 "assert os.environ['OTEL_SDK_DISABLED'] == 'true'"
32 ),
33 ],
34 env=env,
35 check=False,
36 text=True,
37 capture_output=True,
38 )
39
40 assert result.returncode == 0, result.stderr
41
42
43def test_otel_env_default_preserves_explicit_value():
44 env = {**os.environ, "OTEL_SDK_DISABLED": "false"}
45
46 result = subprocess.run(
47 [
48 sys.executable,
49 "-c",
50 (
51 "import os; "
52 "import solstone; "
53 "assert os.environ['OTEL_SDK_DISABLED'] == 'false'"
54 ),
55 ],
56 env=env,
57 check=False,
58 text=True,
59 capture_output=True,
60 )
61
62 assert result.returncode == 0, result.stderr