import pytest
from db_schema.schemas import slz

from slz_uow_reset.service import DBService


@pytest.mark.integration
@pytest.mark.parametrize(
    'uow_code, repr_id', [
        ('apple-20191124-theorchard-amEvent-v1_2', ''),
        ('apple-20191124-theorchard-amEvent-v1_2', '123'),
        ('apple-20191124-theorchard-amEvent-v1_5', '555'),
    ]
)
def test_get_uow_by_uow_id(uow_code, repr_id, db, logger_test, now, unit_of_work_stubs):
    uow_1 = slz.UnitOfWork(**unit_of_work_stubs[1])
    uow_1.unit_of_work_code = uow_code
    uow_1.reprocess_id = repr_id
    db.session.add(uow_1)
    db.session.commit()
    db_service = DBService(
        logger=logger_test,
        db_conn=db,
    )
    result = db_service.get_uow_by_uow_id(uow_1.readable)
    assert bool(result) is True
    assert result == (uow_1.unit_of_work_id, )


@pytest.mark.integration
def test_reset_uow(db, logger_test, now, unit_of_work_stubs):
    uow_1 = slz.UnitOfWork(**unit_of_work_stubs[1])
    db.session.add(uow_1)
    db.session.commit()
    db_service = DBService(
        logger=logger_test,
        db_conn=db,
    )
    assert uow_1.last_updated_at != now
    result = db_service.update_uow_activity_status(uow_1.unit_of_work_id)
    assert result is True
    assert uow_1.last_updated_at == now
    assert uow_1.activity_status == slz.ActivityStatusEnum.NOT_IN_PROGRESS