mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-10-22 00:15:20 +08:00
fix(cloud): retain all execution records for activity counts
This commit is contained in:
@@ -0,0 +1,34 @@
|
||||
from types import SimpleNamespace
|
||||
|
||||
from langbot.pkg.telemetry.execution import ExecutionCounters
|
||||
from langbot.pkg.utils import constants
|
||||
|
||||
|
||||
def test_cloud_counts_are_not_sampled(monkeypatch):
|
||||
monkeypatch.setattr(constants, 'edition', 'cloud')
|
||||
counters = ExecutionCounters(SimpleNamespace(telemetry_config={'execution_trace': 'sampled'}))
|
||||
assert counters.trace_mode() == 'all'
|
||||
state = SimpleNamespace(failure_reason='', stages=[], synthetic=False, trace_id='00000001')
|
||||
assert counters._trace_emitted(state)
|
||||
|
||||
|
||||
def test_community_retains_sampling(monkeypatch):
|
||||
monkeypatch.setattr(constants, 'edition', 'community')
|
||||
counters = ExecutionCounters(SimpleNamespace(telemetry_config={'execution_trace': 'sampled'}))
|
||||
assert counters.trace_mode() == 'sampled'
|
||||
|
||||
|
||||
def test_dns_survives_saturated_executor():
|
||||
import socket
|
||||
import threading
|
||||
from langbot.pkg.utils.bounded_executor import BoundedThreadPoolExecutor
|
||||
|
||||
gate = threading.Event()
|
||||
executor = BoundedThreadPoolExecutor(max_workers=1, max_pending=0)
|
||||
try:
|
||||
blocked = executor.submit(gate.wait)
|
||||
assert executor.submit(socket.getaddrinfo, '127.0.0.1', 443).result(timeout=2)
|
||||
finally:
|
||||
gate.set()
|
||||
blocked.result(timeout=2)
|
||||
executor.shutdown()
|
||||
Reference in New Issue
Block a user