def disable_oob_telemetry():
oob_telemetry_service = TelemetryService.get_instance()
original_state = oob_telemetry_service.enabled
oob_telemetry_service.disable()
yield None
if original_state:
oob_telemetry_service.enable()
def test_telemetry_oob_close_empty(telemetry_setup):
"""Tests closing the Telemetry Service when the queue is empty."""
telemetry = TelemetryService.get_instance()
assert telemetry.size() == 0
telemetry.close()
assert telemetry.size() == 0
def telemetry_setup(request):
"""
Sets up the telemetry service by enabling it and flushing any entries
"""
telemetry = TelemetryService.get_instance()
telemetry.update_context(DEV_CONFIG)
telemetry.enable()
telemetry.flush()
def test_telemetry_oob_close(telemetry_setup):
"""Tests closing the Telemetry Service when there are still messages in the queue."""
telemetry = TelemetryService.get_instance()
telemetry.log_ocsp_exception(event_type, telemetry_data, exception=exception, stack_trace=stack_trace)
assert telemetry.size() == 1
telemetry.close()
assert telemetry.size() == 0
def test_telemetry_oob_simple_flush(telemetry_setup):
"""Tests capturing and sending a simple OCSP Exception message."""
telemetry = TelemetryService.get_instance()
telemetry.log_ocsp_exception(event_type, telemetry_data, exception=exception, stack_trace=stack_trace)
assert telemetry.size() == 1
telemetry.flush()
assert telemetry.size() == 0
def test_telemetry_oob_http_log_urgent(telemetry_setup):
"""Tests sending an urgent HTTP request telemetry event."""
telemetry = TelemetryService.get_instance()
assert telemetry.size() == 0
telemetry.log_http_request_error(event_name, url, method, SQLSTATE_CONNECTION_WAS_NOT_ESTABLISHED, ER_FAILED_TO_REQUEST,
exception=exception, stack_trace=stack_trace, urgent=True)
assert telemetry.size() == 0
def test_telemetry_oob_log_when_disabled(telemetry_setup):
"""Tests trying to log to the telemetry service when it is disabled."""
telemetry = TelemetryService.get_instance()
assert telemetry.size() == 0
telemetry.disable()
telemetry.log_ocsp_exception(event_type, telemetry_data, exception=exception, stack_trace=stack_trace)
assert telemetry.size() == 0
telemetry.enable()
def test_telemetry_oob_urgent(telemetry_setup):
"""Tests sending an urgent OCSP Exception message."""
telemetry = TelemetryService.get_instance()
telemetry.log_ocsp_exception(event_type,
telemetry_data,
exception=exception,
stack_trace=stack_trace,
urgent=True)
assert telemetry.size() == 0
def test_telemetry_oob_http_log(telemetry_setup):
"""
Tests sending a simple HTTP request telemetry event
"""
telemetry = TelemetryService.get_instance()
telemetry.log_http_request(event_name,
url,
method,
SQLSTATE_CONNECTION_WAS_NOT_ESTABLISHED,
ER_FAILED_TO_REQUEST,
exception=exception,
stack_trace=stack_trace)
assert telemetry.size() == 1
telemetry.flush()
assert telemetry.size() == 0
