"""Test neo4j connector creation."""

import pytest

from unittest.mock import MagicMock
from unittest.mock import patch

from connectors import neo4j


@patch.object(neo4j, 'GraphDatabase')
def test_get_neo4j_driver_success(mock_db):
    """Test create neo4j driver."""
    mock_driver = MagicMock()
    mock_db.driver.return_value = mock_driver
    assert neo4j.get_driver() == mock_driver


@patch.object(neo4j, 'GraphDatabase')
def test_neo4j_driver_failed(mock_db):
    """Test error creating neo4j driver."""
    mock_db.driver.side_effect = ValueError
    with pytest.raises(ValueError):
        neo4j.get_driver()