import unittest
from unittest.mock import MagicMock, Mock

from slz_appreciationengine_scrapper.dsp import exceptions
from slz_appreciationengine_scrapper.dsp.appreciationengine.clients import ActivityFeedFileClient
from slz_appreciationengine_scrapper.entities import Job


class ActivityFeedFileTestCase(unittest.TestCase):
    def setUp(self) -> None:
        self.__content_status_repository_mock = Mock()
        self.__boto_client = Mock()
        self.__client = ActivityFeedFileClient(logger=Mock())
        self.__client.client = Mock()
        self.__client.client.head_object = MagicMock(return_value={'ContentLength': 1})
        self.job = Job(
            uow_id='appreciationengine-20220608-sme-activityfeed-v2',
            unit_of_work_id=0,
            dsp='appreciationengine',
            report_type='activityfeed',
            subtype='Summary',
            version='v2',
            report_date='2022-06-08',
            licensor='sme',
            extension='txt',
            config_bucket='delphi-configs',
            context=['US'],
            context_params=None,
            job_id="test_job_id",
        )

    def test_original_content_name(self):
        self.__client._get_file_name = MagicMock(return_value='US_20220608.json')
        self.assertEqual('US_20220608.json', self.__client.get_original_content_name(self.job))