"""Shared fixtures."""

from unittest.mock import MagicMock

import pytest
import splitio

# mock this before all imports.
splitio.get_factory = MagicMock()

from src.model import neo4j_graph  # noqa
from src.app import sentry_sdk  # noqa


@pytest.fixture
def neo4j_driver_mock(mocker):
    """Neo4j driver mock."""
    return mocker.patch.object(neo4j_graph, 'neo4j_driver')


@pytest.fixture
def neo4j_session_mock(neo4j_driver_mock):
    """Neo4j session mock."""
    with neo4j_driver_mock.session() as session:
        yield session


@pytest.fixture
def neo4j_session_transaction_mock(neo4j_session_mock):
    """Neo4j session transaction mock."""
    return neo4j_session_mock.begin_transaction.return_value.__enter__.return_value


@pytest.fixture
def sentry_sdk_capture_exception_mock(mocker):
    """Sentry SDK capture_exception mock."""
    return mocker.patch.object(sentry_sdk, 'capture_exception')