Spaces:
Sleeping
Sleeping
File size: 4,862 Bytes
1ab5126 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 |
#!/usr/bin/env python3
"""
Test script to verify ENABLE_TRACING toggle works correctly.
Usage:
# Test with tracing enabled (default)
python test_tracing_toggle.py
# Test with tracing disabled
ENABLE_TRACING=false python test_tracing_toggle.py
"""
import os
import sys
# Add src to path
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))
from utils.tracing import is_tracing_enabled, init_tracer
def test_is_tracing_enabled():
"""Test the is_tracing_enabled function with various env values."""
print("Testing is_tracing_enabled()...")
test_cases = [
("true", True),
("True", True),
("TRUE", True),
("1", True),
("yes", True),
("YES", True),
("false", False),
("False", False),
("FALSE", False),
("0", False),
("no", False),
("NO", False),
("", True), # Default is enabled
(None, True), # Default is enabled
]
for value, expected in test_cases:
# Set or unset environment variable
if value is None:
os.environ.pop("ENABLE_TRACING", None)
else:
os.environ["ENABLE_TRACING"] = value
result = is_tracing_enabled()
status = "✓" if result == expected else "✗"
print(f" {status} ENABLE_TRACING={value!r} → {result} (expected {expected})")
if result != expected:
return False
return True
def test_init_tracer():
"""Test that init_tracer respects the ENABLE_TRACING flag."""
print("\nTesting init_tracer()...")
# Test with tracing disabled
os.environ["ENABLE_TRACING"] = "false"
os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = "http://localhost:4318"
print(" Testing with ENABLE_TRACING=false...")
tracer = init_tracer("test-service")
# Should return a tracer (no-op) but not initialize OpenTelemetry
if tracer is None:
print(" ✗ init_tracer returned None (should return no-op tracer)")
return False
print(" ✓ init_tracer returned tracer (no-op)")
# Test with tracing enabled
os.environ["ENABLE_TRACING"] = "true"
print(" Testing with ENABLE_TRACING=true...")
# Note: This will actually try to connect to Jaeger if available
# but won't fail if Jaeger is down
try:
tracer = init_tracer("test-service-enabled")
print(" ✓ init_tracer returned tracer (full initialization)")
except Exception as e:
print(f" ⚠ init_tracer failed: {e}")
print(" (This is expected if Jaeger is not running)")
return True
def test_span_creation():
"""Test that spans are created/skipped based on toggle."""
print("\nTesting span creation...")
from opentelemetry import trace
# Test with tracing disabled
os.environ["ENABLE_TRACING"] = "false"
if is_tracing_enabled():
print(" ✗ is_tracing_enabled() returned True when ENABLE_TRACING=false")
return False
# This should be a no-op
tracer = trace.get_tracer(__name__)
span = tracer.start_span("test-span")
span.end()
print(" ✓ No-op span created successfully when disabled")
# Test with tracing enabled
os.environ["ENABLE_TRACING"] = "true"
if not is_tracing_enabled():
print(" ✗ is_tracing_enabled() returned False when ENABLE_TRACING=true")
return False
print(" ✓ Tracing enabled check passed")
return True
def main():
"""Run all tests."""
print("=" * 60)
print("ENABLE_TRACING Toggle Test Suite")
print("=" * 60)
current_value = os.environ.get("ENABLE_TRACING", "<not set>")
print(f"\nCurrent ENABLE_TRACING: {current_value}")
print()
tests = [
("is_tracing_enabled()", test_is_tracing_enabled),
("init_tracer()", test_init_tracer),
("span creation", test_span_creation),
]
results = []
for name, test_func in tests:
try:
success = test_func()
results.append((name, success))
except Exception as e:
print(f"\n✗ Test '{name}' raised exception: {e}")
results.append((name, False))
# Print summary
print("\n" + "=" * 60)
print("Test Summary")
print("=" * 60)
passed = sum(1 for _, success in results if success)
total = len(results)
for name, success in results:
status = "✓ PASS" if success else "✗ FAIL"
print(f"{status}: {name}")
print(f"\n{passed}/{total} tests passed")
if passed == total:
print("\n✓ All tests passed!")
return 0
else:
print(f"\n✗ {total - passed} test(s) failed")
return 1
if __name__ == "__main__":
sys.exit(main())
|