import unittest import uuid from datetime import date, datetime from unittest.mock import Mock from slz_clean_up.services.sf_execution import JobService class JobServiceTestCase(unittest.TestCase): def setUp(self) -> None: self.__sf_client_mock = Mock() self.__arn = str(uuid.uuid4()) self.__step_function_arn = str(uuid.uuid1()) self.__job_stoper = JobService( logger=Mock(), sf_client=self.__sf_client_mock, step_function_arn=self.__step_function_arn, step_function_execution_arn=self.__arn ) def test_sf_should_not_stop_itself(self): job_ids = [self.__arn] self.__job_stoper.stop_uow_executions(job_ids) self.__sf_client_mock.stop_execution.assert_not_called()