diff options
author | Mikhail Shchatko <mikhail.shchatko@mongodb.com> | 2021-11-02 11:40:35 +0300 |
---|---|---|
committer | Evergreen Agent <no-reply@evergreen.mongodb.com> | 2021-11-02 14:04:03 +0000 |
commit | dae19caf147fb767b64ec983fbc8de46222f2fa4 (patch) | |
tree | aaeb6c8a72d72ef2bfa0b2bbe6ae564bd2cf37a9 | |
parent | 18e4ac84db47e93dcbbfeffec798e9f746b15f9d (diff) | |
download | mongo-dae19caf147fb767b64ec983fbc8de46222f2fa4.tar.gz |
SERVER-61169 Inject ResmokeProxyService in MultiversionGenTaskDecorator
3 files changed, 6 insertions, 4 deletions
diff --git a/buildscripts/task_generation/task_types/fuzzer_tasks.py b/buildscripts/task_generation/task_types/fuzzer_tasks.py index 9292108f040..869835315b3 100644 --- a/buildscripts/task_generation/task_types/fuzzer_tasks.py +++ b/buildscripts/task_generation/task_types/fuzzer_tasks.py @@ -83,7 +83,7 @@ class FuzzerGenTaskService: def __init__(self): """Initialize the service.""" - self.multiversion_decorator = MultiversionGenTaskDecorator() + self.multiversion_decorator = MultiversionGenTaskDecorator() # pylint: disable=no-value-for-parameter def generate_tasks(self, params: FuzzerGenTaskParams) -> FuzzerTask: """ diff --git a/buildscripts/task_generation/task_types/multiversion_decorator.py b/buildscripts/task_generation/task_types/multiversion_decorator.py index b0203919c64..36d8bcfd395 100644 --- a/buildscripts/task_generation/task_types/multiversion_decorator.py +++ b/buildscripts/task_generation/task_types/multiversion_decorator.py @@ -2,6 +2,7 @@ import copy from typing import List, Set, Union, Optional +import inject import structlog from shrub.v2 import Task, FunctionCall from shrub.v2.command import ShrubCommand @@ -26,9 +27,10 @@ class MultiversionGenTaskDecorator: """Multiversion decorator for basic generated tasks.""" # pylint: disable=no-self-use - def __init__(self): + @inject.autoparams() + def __init__(self, resmoke_proxy: ResmokeProxyService): """Initialize multiversion decorator.""" - self.resmoke_proxy = ResmokeProxyService() + self.resmoke_proxy = resmoke_proxy self.old_versions = self._init_old_versions() def decorate_tasks(self, sub_tasks: Set[Task], params) -> Set[Task]: diff --git a/buildscripts/tests/task_generation/task_types/test_multiversion_decorator.py b/buildscripts/tests/task_generation/task_types/test_multiversion_decorator.py index 3008d7957bc..6ada07a195b 100644 --- a/buildscripts/tests/task_generation/task_types/test_multiversion_decorator.py +++ b/buildscripts/tests/task_generation/task_types/test_multiversion_decorator.py @@ -7,7 +7,7 @@ from buildscripts.task_generation.task_types import multiversion_decorator as un from buildscripts.task_generation.task_types.fuzzer_tasks import FuzzerGenTaskParams -# pylint: disable=missing-docstring,invalid-name,unused-argument,no-self-use,protected-access +# pylint: disable=missing-docstring,invalid-name,unused-argument,no-self-use,protected-access,no-value-for-parameter def build_mock_fuzzer_params(): return FuzzerGenTaskParams( task_name="task_name", |