diff options
author | Google Team <no-reply@google.com> | 2024-02-05 16:40:24 -0800 |
---|---|---|
committer | Copybara-Service <copybara-worker@google.com> | 2024-02-05 16:40:55 -0800 |
commit | 401295a3f5c3460108b757d9c76fb41489a6d1fc (patch) | |
tree | d47d542c8b0449edfbaacd4e75f60649add97935 | |
parent | 266db13d7227d1f24b773a7a544f1c680946fab2 (diff) | |
download | tradefed_cluster-401295a3f5c3460108b757d9c76fb41489a6d1fc.tar.gz |
No public description
PiperOrigin-RevId: 604470607
Change-Id: Id5cffe56445181b916aaf84e30860a7ea447d66b
-rw-r--r-- | tradefed_cluster/plugins/google_cloud.py | 2 | ||||
-rw-r--r-- | tradefed_cluster/plugins/google_cloud_test.py | 8 |
2 files changed, 5 insertions, 5 deletions
diff --git a/tradefed_cluster/plugins/google_cloud.py b/tradefed_cluster/plugins/google_cloud.py index f4a3d51..4cf74f4 100644 --- a/tradefed_cluster/plugins/google_cloud.py +++ b/tradefed_cluster/plugins/google_cloud.py @@ -75,7 +75,7 @@ class TaskScheduler(base.TaskScheduler): task['schedule_time'] = timestamp if target: task['app_engine_http_request']['app_engine_routing']['service'] = target - task = self._GetClient().create_task(parent, task, + task = self._GetClient().create_task(parent=parent, task=task, retry=DEFAULT_RETRY_OPTION) return base.Task(name=task.name.split('/')[-1], payload=payload, eta=eta) diff --git a/tradefed_cluster/plugins/google_cloud_test.py b/tradefed_cluster/plugins/google_cloud_test.py index ea00c42..f67ec5f 100644 --- a/tradefed_cluster/plugins/google_cloud_test.py +++ b/tradefed_cluster/plugins/google_cloud_test.py @@ -61,8 +61,8 @@ class TaskSchedulerTest(unittest.TestCase): self.mock_client.task_path.assert_called_once_with( 'project', 'location', 'queue_name', 'task_name') self.mock_client.create_task.assert_called_once_with( - 'queue_path', - { + parent='queue_path', + task={ 'name': 'task_path', 'schedule_time': timestamp, 'app_engine_http_request': { @@ -96,8 +96,8 @@ class TaskSchedulerTest(unittest.TestCase): self.mock_client.queue_path.assert_called_once_with( 'project', 'location', 'queue_name') self.mock_client.create_task.assert_called_once_with( - 'queue_path', - { + parent='queue_path', + task={ 'app_engine_http_request': { 'http_method': 'POST', 'app_engine_routing': {}, |