def test_runtime_metrics_api():
RuntimeMetrics.enable()
assert RuntimeWorker._instance is not None
assert RuntimeWorker._instance.status == ServiceStatus.RUNNING
RuntimeMetrics.disable()
assert RuntimeWorker._instance is None
def test_manually_start_runtime_metrics():
"""
When importing and manually starting runtime metrics
Runtime metrics worker starts and there are no errors
"""
from ddtrace.runtime import RuntimeMetrics
RuntimeMetrics.enable()
assert RuntimeMetrics._enabled
RuntimeMetrics.disable()
assert not RuntimeMetrics._enabled
def test_runtime_metrics_enable_environ(monkeypatch, environ):
try:
for (k, v) in environ.items():
monkeypatch.setenv(k, v)
RuntimeMetrics.enable()
assert RuntimeWorker._instance is not None
assert RuntimeWorker._instance.status == ServiceStatus.RUNNING
assert (RuntimeWorker._instance.interval == float(
environ["DD_RUNTIME_METRICS_INTERVAL"])
if "DD_RUNTIME_METRICS_INTERVAL" in environ else
RuntimeWorker._instance.interval == 10.0)
finally:
RuntimeMetrics.disable()
def test_runtime_metrics_api_idempotency():
RuntimeMetrics.enable()
instance = RuntimeWorker._instance
assert instance is not None
RuntimeMetrics.enable()
assert RuntimeWorker._instance is instance
RuntimeMetrics.disable()
assert RuntimeWorker._instance is None
RuntimeMetrics.disable()
def test_runtime_metrics_enable(enable_kwargs):
try:
RuntimeMetrics.enable(**enable_kwargs)
assert RuntimeWorker._instance is not None
assert RuntimeWorker._instance.status == ServiceStatus.RUNNING
assert (RuntimeWorker._instance.tracer == enable_kwargs["tracer"]
if "tracer" in enable_kwargs else
RuntimeWorker._instance.tracer is not None)
assert (RuntimeWorker._instance.dogstatsd_url
== enable_kwargs["dogstatsd_url"]
if "dogstatsd_url" in enable_kwargs else
RuntimeWorker._instance.dogstatsd_url is None)
assert (RuntimeWorker._instance.interval
== enable_kwargs["flush_interval"] if "flush_interval"
in enable_kwargs else RuntimeWorker._instance.interval == 10.0)
finally:
RuntimeMetrics.disable()
def test_runtime_metrics_via_env_var_manual_start(
monkeypatch, ddtrace_run_python_code_in_subprocess):
"""
When running with ddtrace-run and DD_RUNTIME_METRICS_ENABLED is set and trying to start RuntimeMetrics manually
Runtime metrics worker starts and there are no errors
"""
monkeypatch.setenv("DD_RUNTIME_METRICS_ENABLED", "true")
_, _, status, _ = ddtrace_run_python_code_in_subprocess(
"""
from ddtrace.runtime import RuntimeMetrics
assert RuntimeMetrics._enabled
RuntimeMetrics.enable()
assert RuntimeMetrics._enabled
""", )
assert status == 0
def test_manually_start_runtime_metrics(run_python_code_in_subprocess):
"""
When importing and manually starting runtime metrics
Runtime metrics worker starts and there are no errors
"""
out, err, status, pid = run_python_code_in_subprocess(
"""
from ddtrace.runtime import RuntimeMetrics
RuntimeMetrics.enable()
assert RuntimeMetrics._enabled
RuntimeMetrics.disable()
assert not RuntimeMetrics._enabled
""",
)
assert status == 0
def test_manually_start_runtime_metrics(run_python_code_in_subprocess):
"""
When importing and manually starting runtime metrics
Runtime metrics worker starts and there are no errors
"""
out, err, status, pid = run_python_code_in_subprocess(
"""
from ddtrace.runtime import RuntimeMetrics
RuntimeMetrics.enable()
from ddtrace.internal.runtime.runtime_metrics import RuntimeWorker
assert RuntimeWorker._instance is not None
RuntimeMetrics.disable()
assert RuntimeWorker._instance is None
""", )
assert status == 0