import logging
import unittest.mock

from db_schema.schemas import slz

from slz_meta_status_reset.handler import handler


class HandlerTestCase(unittest.TestCase):
    def setUp(self):
        self._logger = logging.getLogger('tests')

    def test_handler(self):
        payload = {
            'uow_id': 'test-id',
            'content_name': 'test-content-name',
            'context': 'US',
        }
        pg_storage = unittest.mock.Mock()
        handler(payload, pg_storage)

        pg_storage.update_content_status_metadata_by_uow_id.assert_called_once_with(
            uow_id='test-id',
            context='US',
            process_status=slz.ContentMetadataStatusEnum.FAILED,
        )