Differential D8430 Diff 30391 swh/loader/core/tests/conftest.py

Changeset View

Standalone View

				# Copyright (C) 2022 The Software Heritage developers
				anlambertAuthorUnsubmitted Done Inline Actions Copied and pasted from `swh-indexer` conftest file. It could be moved to a pytest plugin in `swh-core` to avoid code duplication. anlambert: Copied and pasted from `swh-indexer` conftest file. It could be moved to a pytest plugin in…
				# See the AUTHORS file at the top-level directory of this distribution
				# License: GNU General Public License version 3, or any later version
				# See top-level LICENSE file for more information

				import pytest
				import sentry_sdk


				@pytest.fixture
				def sentry_init():
				# Inspired by
				# https://github.com/getsentry/sentry-python/blob/1.5.9/tests/conftest.py#L168-L184

				initialized = False

				def inner(a, *kw):
				nonlocal initialized
				assert not initialized, "already initialized"
				initialized = True
				hub = sentry_sdk.Hub.current
				client = sentry_sdk.Client(a, *kw)
				hub.bind_client(client)
				client.transport = TestTransport()

				class TestTransport:
				def __init__(self):
				self.events = []
				self.envelopes = []

				def capture_event(self, event):
				self.events.append(event)

				def capture_envelope(self, envelope):
				self.envelopes.append(envelope)

				with sentry_sdk.Hub(None):
				yield inner


				@pytest.fixture
				def sentry_events(monkeypatch, sentry_init):
				sentry_init()
				return sentry_sdk.Hub.current.client.transport.events