"""Lambda test module."""

from unittest.mock import MagicMock


def test_handler(mock_neo4j_session, query_result_with_write_node):
    """Test handler function."""
    # stub test for lambda handler skeleton
    import index
    session_mock = MagicMock()
    mock_neo4j_session.__enter__.return_value = session_mock
    session_mock.run.return_value.data.return_value = [
        query_result_with_write_node]
    result = index.handler(None, None)
    assert result == {'status': 'OK'}