"""SQL connector tests."""
from sqlalchemy import orm

from availability.connectors import sql


def test_db_session_wrap_creates_new_session():
    """Test that decorator creates new session if no existing is passed."""
    func = sql.db_session_wrap(lambda session: session)

    result_session = func()
    assert isinstance(result_session, orm.Session)


def test_db_session_wrap_passes_existing_session():
    """Test that decorator passes existing session if any."""
    func = sql.db_session_wrap(lambda session: session)

    with sql.session_scope() as session:
        result_session = func(session=session)
        assert result_session is session