def test_ext_without_prometheus_flask_metrics(mock_prometheus_metrics): PROMETHEUS_ENABLE_EXPORTER_FLASK = False app = Flask("testapp") app.config.update( {"PROMETHEUS_ENABLE_EXPORTER_FLASK": PROMETHEUS_ENABLE_EXPORTER_FLASK}) InspireLogger(app) mock_prometheus_metrics.assert_not_called()
def test_ext_with_prometheus_flask_metrics(mock_prometheus_metrics, monkeypatch): PROMETHEUS_ENABLE_EXPORTER_FLASK = True monkeypatch.setenv("prometheus_multiproc_dir", "/tmp") app = Flask("testapp") app.config.update( {"PROMETHEUS_ENABLE_EXPORTER_FLASK": PROMETHEUS_ENABLE_EXPORTER_FLASK}) InspireLogger(app) mock_prometheus_metrics.assert_called_once()
def test_ext_without_dsn(mock_sentry_sdk): SENTRY_DSN = None SENTRY_SEND_DEFAULT_PII = True app = Flask("testapp") app.config.update({ "SENTRY_DSN": SENTRY_DSN, "SENTRY_SEND_DEFAULT_PII": SENTRY_SEND_DEFAULT_PII }) InspireLogger(app) mock_sentry_sdk.assert_not_called()
def test_ext_with_dsn(mock_sentry_sdk): SENTRY_DSN = "TEST_DSN_URL_FOR_SENTRY" SENTRY_SEND_DEFAULT_PII = True app = Flask("testapp") app.config.update({ "SENTRY_DSN": SENTRY_DSN, "SENTRY_SEND_DEFAULT_PII": SENTRY_SEND_DEFAULT_PII }) InspireLogger(app) mock_sentry_sdk.assert_called_once()
def test_ext(): """Test extension initialization.""" app = Flask("testapp") InspireLogger(app) assert "inspirehep-logger" in app.extensions