wehub-resource-sync
52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
"""Tests for opt-in CLI startup profiling."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import io
|
|
|
|
import pytest
|
|
|
|
from omnigent._startup_profile import StartupProfiler
|
|
|
|
|
|
def test_startup_profiler_env_opt_in(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
"""
|
|
Truthy env vars enable startup timing output.
|
|
|
|
:param monkeypatch: Pytest monkeypatch fixture.
|
|
:returns: None.
|
|
"""
|
|
monkeypatch.setenv("OMNIGENT_TEST_PROFILE", "yes")
|
|
stream = io.StringIO()
|
|
clock_values = iter([10.0, 10.25])
|
|
|
|
profiler = StartupProfiler.from_env(
|
|
name="test command",
|
|
env_var="OMNIGENT_TEST_PROFILE",
|
|
clock=lambda: next(clock_values),
|
|
stream=stream,
|
|
)
|
|
profiler.mark("first stage")
|
|
|
|
assert stream.getvalue() == "[test command startup +0.250s delta=0.250s] first stage\n"
|
|
|
|
|
|
def test_startup_profiler_disabled_without_env(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
"""
|
|
Missing env vars leave startup profiling silent by default.
|
|
|
|
:param monkeypatch: Pytest monkeypatch fixture.
|
|
:returns: None.
|
|
"""
|
|
monkeypatch.delenv("OMNIGENT_TEST_PROFILE", raising=False)
|
|
stream = io.StringIO()
|
|
|
|
profiler = StartupProfiler.from_env(
|
|
name="test command",
|
|
env_var="OMNIGENT_TEST_PROFILE",
|
|
stream=stream,
|
|
)
|
|
profiler.mark("hidden")
|
|
|
|
assert stream.getvalue() == ""
|