amira_test.py 文件源码

def _patch_and_run_amira(
            self, region_name, queue_name, contents, created_objects):
        """Patches all the external dependencies and runs AMIRA."""
        self._results_uploader_mock = MagicMock()

        with nested(
            patch.object(
                S3Handler, '__init__', autospec=True, return_value=None),
            patch.object(
                S3Handler, 'get_contents_as_string', autospec=True,
                side_effect=contents),
            patch.object(
                SqsHandler, '__init__', autospec=True, return_value=None),
            patch.object(
                SqsHandler, 'get_created_objects', autospec=True,
                side_effect=created_objects),
        ) as (
            __,
            self._patched_get_contents_as_string,
            __,
            self._patched_get_created_objects,
        ):
            amira = AMIRA(region_name, queue_name)
            amira.register_results_uploader(self._results_uploader_mock)
            amira.run()