from unittest.mock import patch

from db import do_create
from db import do_match

from neo4j.graph import Graph
from neo4j.graph import Node


@patch('db.get_session')
def test_mock_session(mock_get_session):
    mock_session = mock_get_session.return_value

    assert not mock_get_session.called
    do_create('Neo', 0)
    assert mock_get_session.called
    assert mock_session.run.called_once_with(
        'CREATE (n:Person {name: $name})',
        name='Neo'
    )


@patch('db.get_session')
def test_mock_return_node(mock_get_session):
    mock_session = mock_get_session.return_value
    mock_node = Node(
        Graph(),
        '1',
        n_labels=['Person'],
        properties={'name': 'Neo'}
    )
    mock_session.run.return_value = [
        {'n': mock_node}
    ]

    results = do_match()

    assert results == ['Neo']