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