import pytest
from db_schema.factories.apps import UnitOfWorkFactory
from db_schema.schemas.apps import FailureLog

from apps_etl_set_uow_status import update_unit_of_work


@pytest.mark.integration
def test_update_unit_of_work_with_error(session):
    uow = UnitOfWorkFactory(
        unit_of_work_code='apple-20191112-report',
        activity_status='IN_PROGRESS',
        completeness_status='FAILED',
    )
    session.commit()
    session.refresh(uow)

    update_unit_of_work(session, 'test-execution-id', uow.unit_of_work_id, {}, error='Timeout')  # pylint: disable=E1101

    session.refresh(uow)
    assert uow.failure_count == 1  # pylint: disable=E1101
    assert session.query(FailureLog).filter_by(unit_of_work_id=uow.unit_of_work_id).first()  # pylint: disable=E1101