"""Shared fixtures."""

from model import base

import pytest


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


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