from unittest import mock import pytest from smelog.factory import SmeBoundLogger from dapd_scheduler_lambda.repository.batch import Batch from dapd_scheduler_lambda.repository.postgres import WorkflowDB from dapd_scheduler_lambda.repository.s3 import S3 from dapd_scheduler_lambda.service import Service @pytest.fixture(scope='function') def service(): logger = mock.create_autospec(SmeBoundLogger) params = mock.Mock() workflow_db_repo = mock.MagicMock(WorkflowDB) s3_repo = mock.MagicMock(S3) batch_repo = mock.MagicMock(Batch) yield Service(logger, params, workflow_db_repo, s3_repo, batch_repo)