From a0b5d10c810dc3c5632a56e4fb88e112295c945f Mon Sep 17 00:00:00 2001 From: Choongwoo Han Date: Fri, 13 Nov 2020 03:22:53 +0900 Subject: [PATCH] Add target_workers to TaskUnitConfig (#305) --- src/api-service/__app__/onefuzzlib/tasks/config.py | 3 +++ src/pytypes/onefuzztypes/models.py | 1 + 2 files changed, 4 insertions(+) diff --git a/src/api-service/__app__/onefuzzlib/tasks/config.py b/src/api-service/__app__/onefuzzlib/tasks/config.py index 7c613af9d..3978096f4 100644 --- a/src/api-service/__app__/onefuzzlib/tasks/config.py +++ b/src/api-service/__app__/onefuzzlib/tasks/config.py @@ -267,6 +267,9 @@ def build_task_config( if TaskFeature.target_options_merge in definition.features: config.target_options_merge = task_config.task.target_options_merge or False + if TaskFeature.target_workers in definition.features: + config.target_workers = task_config.task.target_workers + if TaskFeature.rename_output in definition.features: config.rename_output = task_config.task.rename_output or False diff --git a/src/pytypes/onefuzztypes/models.py b/src/pytypes/onefuzztypes/models.py index cb8458d40..94e5bd7e0 100644 --- a/src/pytypes/onefuzztypes/models.py +++ b/src/pytypes/onefuzztypes/models.py @@ -297,6 +297,7 @@ class TaskUnitConfig(BaseModel): target_options: Optional[List[str]] target_timeout: Optional[int] target_options_merge: Optional[bool] + target_workers: Optional[int] check_asan_log: Optional[bool] check_debugger: Optional[bool] check_retry_count: Optional[int]