Skip to content

Commit d45af03

Browse files
committed
Fixes broken UT
1 parent 0ede2ed commit d45af03

2 files changed

Lines changed: 7 additions & 1 deletion

File tree

src/clusterfuzz/_internal/base/tasks/__init__.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,11 @@ def get_task_duration(command):
101101
UTASK_MAIN_QUEUE = 'utask_main'
102102
PREPROCESS_QUEUE = 'preprocess'
103103

104+
SWARMING_QUEUES = {
105+
PREPROCESS_QUEUE: 'preprocess_swarming',
106+
UTASK_MAIN_QUEUE: 'utask_main_swarming',
107+
}
108+
104109
# See https://github.com/google/clusterfuzz/issues/3347 for usage
105110
SUBQUEUE_IDENTIFIER = ':'
106111

src/clusterfuzz/_internal/swarming/api.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,8 @@ class SwarmingAPI:
3737

3838
def __init__(self):
3939
self._config = get_swarming_config()
40-
self._base_url = f"https://{self._config.get('swarming_server')}/prpc/"
40+
if self._config:
41+
self._base_url = f"https://{self._config.get('swarming_server')}/prpc/"
4142

4243
def _get_headers(self) -> dict[str, str]:
4344
"""Checks config and returns headers for pRPC request.

0 commit comments

Comments
 (0)