from unittest import mock

import pytest

from slz_uow_reset.const import STATUS_OK
from slz_uow_reset.handler import DBService, app_entry
from slz_uow_reset.service import ServiceException


@mock.patch.object(DBService, 'update_uow_activity_status', return_value=True)
@mock.patch('slz_uow_reset.handler.Connection')
def test_success(conn_mock, service_mock, test_config):
    result = app_entry(None, mock.Mock(), config=test_config, payload={'uow_id': 'apple-9999'})

    assert {'meta': {'status': STATUS_OK}} == result


@mock.patch.object(DBService, 'update_uow_activity_status', return_value=False)
@mock.patch('slz_uow_reset.handler.Connection')
def test_fail(conn_mock, service_mock, test_config):
    with pytest.raises(ServiceException):
        app_entry(None, mock.Mock(), config=test_config, payload={'uow_id': 'apple-9999'})