"""Unit tests for the Neo4j connector."""

from unittest.mock import MagicMock

from neo4j import Driver

from contributor.connectors.neo4j import neo4j_session


def test_neo4j_session_yields_session() -> None:
    mock_driver = MagicMock(spec=Driver)
    mock_session = MagicMock()
    mock_driver.session.return_value.__enter__ = lambda s: mock_session
    mock_driver.session.return_value.__exit__ = MagicMock(return_value=False)

    with neo4j_session(mock_driver) as session:
        assert session is mock_session

    mock_driver.session.assert_called_once_with(
        default_access_mode="READ",
    )