"""Test database utilities."""

import sqlalchemy

from asset_file_details.utils import db


def test_create_engine_queue_pool(mocker):
    """Test creating a DB engine using the QueuePool and associated params."""
    mocker.patch.multiple(
        db.config,
        POOL_CLASS=sqlalchemy.pool.QueuePool,
        POOL_MAX_OVERFLOW=1,
        POOL_PRE_PING=2,
        POOL_RECYCLE_MS=3,
        POOL_SIZE=4
    )
    mock_create_engine = mocker.patch.object(db.sqlalchemy, 'create_engine')

    db.create_engine('testurl')

    mock_create_engine.assert_called_with(
        'testurl',
        max_overflow=1,
        pool_pre_ping=2,
        pool_recycle=3,
        pool_size=4
    )


def test_create_engine_static_pool(mocker):
    """Test creating a DB engine using the StaticPool."""
    mocker.patch.object(db.config, 'POOL_CLASS', sqlalchemy.pool.StaticPool)
    mock_create_engine = mocker.patch.object(db.sqlalchemy, 'create_engine')

    db.create_engine('testurl')

    mock_create_engine.assert_called_with(
        'testurl',
        poolclass=sqlalchemy.pool.StaticPool
    )


def test_create_session(mocker):
    """Test creating a session."""
    session_mock = mocker.patch('sqlalchemy.orm.session.Session')
    session_maker_stub = mocker.stub()
    session_maker_stub.return_value = session_mock

    with db.create_session(session_maker_stub):
        pass

    session_maker_stub.assert_called_once()
    session_mock.commit.assert_called_once()
    session_mock.close.assert_called_once()
    session_mock.rollback.assert_not_called()


def test_create_session_with_error(mocker):
    """Test creating a session which encounters an error."""
    session_mock = mocker.patch('sqlalchemy.orm.session.Session')
    session_maker_stub = mocker.stub()
    session_maker_stub.return_value = session_mock

    try:
        with db.create_session(session_maker_stub):
            raise 'Fail'
    except Exception:
        pass

    session_maker_stub.assert_called_once()
    session_mock.rollback.assert_called_once()
    session_mock.close.assert_called_once()
    session_mock.commit.assert_not_called()