[Swarming] Adds Response proto & discards tasks with missing dimensions (#5289)

## Overview
It was request for me to add a log to easily debug when swarming
scheduled a task, but i noticed we didnt had the protos for parsing the
response, hence i added them.

Also this PR handles an edge case in which swarming will not return a
task_id if the tasks wasn't scheduled due to swarming related issues(not
request, nor pc contract).

#### Background
This PR comes from a discussion & agreement to a required change from
this [PR](#5277)

## Changes
- Adds the missing swarming_task_response proto
  - Ran `protos/generate.sh` which updated a ton of files
- The swarming service now marks tasks as not scheduled when no task_id
is found, task_id cannot be found if swarming didn't queue the task,
because of wrong dimensions or other swarming internal reasons
- As the `count_tasks()` api method, now the `push_task()` will also
only raise `SwarmingAPIErrors`

---------

Co-authored-by: Titouan Rigoudy <titouan@chromium.org>

Author: IvanBM18

src/clusterfuzz/_internal/protos/swarming.proto

@@ -406,6 +406,38 @@ message NewTaskRequest {
   string realm = 18;
 }
 
+// Description of a task request as registered by the server.
+// This message is used when retrieving information about an existing task.
+// See NewTaskRequest for more details.
+message TaskRequestResponse {
+  string task_id = 1;
+  int32 expiration_secs = 2;
+  string name = 3;
+  string parent_task_id = 4;
+  int32 priority = 5;
+  // For some amount of time, the properties will be copied into the
+  // task_slices and vice-versa, to give time to the clients to update.
+  // Eventually, only task_slices will be supported.
+  TaskProperties properties = 6;
+  repeated string tags = 7;
+  google.protobuf.Timestamp created_ts = 8;
+  string user = 9;
+  // User name of whoever posted this task, extracted from the credentials.
+  string authenticated = 10;
+  repeated TaskSlice task_slices = 11;
+  // Indicates what OAuth2 credentials the task uses when calling other services.
+  string service_account = 12;
+  string realm = 13;
+
+  // Configuration of Swarming:ResultDB integration.
+  ResultDBCfg resultdb = 14;
+
+  string pubsub_topic = 15;
+  string pubsub_userdata = 16;
+  int32 bot_ping_tolerance_secs = 17;
+  string rbe_instance = 18;
+}
+
 message TasksCountRequest {
   google.protobuf.Timestamp start = 1;
   google.protobuf.Timestamp end = 2;

src/clusterfuzz/_internal/protos/swarming_pb2.py

@@ -29,15 +29,15 @@
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+clusterfuzz/_internal/protos/swarming.proto\x12\x0bswarming.v2\x1a\x1fgoogle/protobuf/timestamp.proto\"(\n\nStringPair\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\",\n\x0eStringListPair\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x03(\t\"*\n\x06\x44igest\x12\x0c\n\x04hash\x18\x01 \x01(\t\x12\x12\n\nsize_bytes\x18\x02 \x01(\x03\"I\n\x0c\x43\x41SReference\x12\x14\n\x0c\x63\x61s_instance\x18\x01 \x01(\t\x12#\n\x06\x64igest\x18\x02 \x01(\x0b\x32\x13.swarming.v2.Digest\"B\n\x0b\x43ipdPackage\x12\x14\n\x0cpackage_name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\x12\x0c\n\x04path\x18\x03 \x01(\t\"y\n\tCipdInput\x12\x0e\n\x06server\x18\x01 \x01(\t\x12\x30\n\x0e\x63lient_package\x18\x02 \x01(\x0b\x32\x18.swarming.v2.CipdPackage\x12*\n\x08packages\x18\x03 \x03(\x0b\x32\x18.swarming.v2.CipdPackage\"(\n\nCacheEntry\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\t\"\xf2\x01\n\x0b\x43ontainment\x12\x16\n\x0elower_priority\x18\x01 \x01(\x08\x12\x42\n\x10\x63ontainment_type\x18\x02 \x01(\x0e\x32(.swarming.v2.Containment.ContainmentType\x12\x17\n\x0flimit_processes\x18\x03 \x01(\x03\x12$\n\x1climit_total_committed_memory\x18\x04 \x01(\x03\"H\n\x0f\x43ontainmentType\x12\x11\n\rNOT_SPECIFIED\x10\x00\x12\x08\n\x04NONE\x10\x01\x12\x08\n\x04\x41UTO\x10\x02\x12\x0e\n\nJOB_OBJECT\x10\x03\"\x83\x04\n\x0eTaskProperties\x12\'\n\x06\x63\x61\x63hes\x18\x01 \x03(\x0b\x32\x17.swarming.v2.CacheEntry\x12*\n\ncipd_input\x18\x02 \x01(\x0b\x32\x16.swarming.v2.CipdInput\x12\x0f\n\x07\x63ommand\x18\x03 \x03(\t\x12\x14\n\x0crelative_cwd\x18\x04 \x01(\t\x12+\n\ndimensions\x18\x05 \x03(\x0b\x32\x17.swarming.v2.StringPair\x12$\n\x03\x65nv\x18\x06 \x03(\x0b\x32\x17.swarming.v2.StringPair\x12\x31\n\x0c\x65nv_prefixes\x18\x07 \x03(\x0b\x32\x1b.swarming.v2.StringListPair\x12\x1e\n\x16\x65xecution_timeout_secs\x18\x08 \x01(\x05\x12\x19\n\x11grace_period_secs\x18\t \x01(\x05\x12\x12\n\nidempotent\x18\n \x01(\x08\x12\x31\n\x0e\x63\x61s_input_root\x18\x0b \x01(\x0b\x32\x19.swarming.v2.CASReference\x12\x17\n\x0fio_timeout_secs\x18\x0c \x01(\x05\x12\x0f\n\x07outputs\x18\r \x03(\t\x12\x14\n\x0csecret_bytes\x18\x0e \x01(\x0c\x12-\n\x0b\x63ontainment\x18\x0f \x01(\x0b\x32\x18.swarming.v2.Containment\"p\n\tTaskSlice\x12/\n\nproperties\x18\x01 \x01(\x0b\x32\x1b.swarming.v2.TaskProperties\x12\x17\n\x0f\x65xpiration_secs\x18\x02 \x01(\x05\x12\x19\n\x11wait_for_capacity\x18\x03 \x01(\x08\"\x1d\n\x0bResultDBCfg\x12\x0e\n\x06\x65nable\x18\x01 \x01(\x08\"\xe8\x04\n\x0eNewTaskRequest\x12\x17\n\x0f\x65xpiration_secs\x18\x01 \x01(\x05\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x16\n\x0eparent_task_id\x18\x03 \x01(\t\x12\x10\n\x08priority\x18\x04 \x01(\x05\x12/\n\nproperties\x18\x05 \x01(\x0b\x32\x1b.swarming.v2.TaskProperties\x12+\n\x0btask_slices\x18\x06 \x03(\x0b\x32\x16.swarming.v2.TaskSlice\x12\x0c\n\x04tags\x18\x07 \x03(\t\x12\x0c\n\x04user\x18\x08 \x01(\t\x12\x17\n\x0fservice_account\x18\t \x01(\t\x12\x14\n\x0cpubsub_topic\x18\n \x01(\t\x12\x19\n\x11pubsub_auth_token\x18\x0b \x01(\t\x12\x17\n\x0fpubsub_userdata\x18\x0c \x01(\t\x12\x15\n\revaluate_only\x18\r \x01(\x08\x12M\n\x12pool_task_template\x18\x0e \x01(\x0e\x32\x31.swarming.v2.NewTaskRequest.PoolTaskTemplateField\x12\x1f\n\x17\x62ot_ping_tolerance_secs\x18\x0f \x01(\x05\x12\x14\n\x0crequest_uuid\x18\x10 \x01(\t\x12*\n\x08resultdb\x18\x11 \x01(\x0b\x32\x18.swarming.v2.ResultDBCfg\x12\r\n\x05realm\x18\x12 \x01(\t\"P\n\x15PoolTaskTemplateField\x12\x08\n\x04\x41UTO\x10\x00\x12\x11\n\rCANARY_PREFER\x10\x01\x12\x10\n\x0c\x43\x41NARY_NEVER\x10\x02\x12\x08\n\x04SKIP\x10\x03\"\x9d\x01\n\x11TasksCountRequest\x12)\n\x05start\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\'\n\x03\x65nd\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12&\n\x05state\x18\x03 \x01(\x0e\x32\x17.swarming.v2.StateQuery\x12\x0c\n\x04tags\x18\x04 \x03(\t\"D\n\nTasksCount\x12\r\n\x05\x63ount\x18\x01 \x01(\x05\x12\'\n\x03now\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp*\xcf\x02\n\nStateQuery\x12\x11\n\rQUERY_PENDING\x10\x00\x12\x11\n\rQUERY_RUNNING\x10\x01\x12\x19\n\x15QUERY_PENDING_RUNNING\x10\x02\x12\x13\n\x0fQUERY_COMPLETED\x10\x03\x12\x1b\n\x17QUERY_COMPLETED_SUCCESS\x10\x04\x12\x1b\n\x17QUERY_COMPLETED_FAILURE\x10\x05\x12\x11\n\rQUERY_EXPIRED\x10\x06\x12\x13\n\x0fQUERY_TIMED_OUT\x10\x07\x12\x12\n\x0eQUERY_BOT_DIED\x10\x08\x12\x12\n\x0eQUERY_CANCELED\x10\t\x12\r\n\tQUERY_ALL\x10\n\x12\x11\n\rQUERY_DEDUPED\x10\x0b\x12\x10\n\x0cQUERY_KILLED\x10\x0c\x12\x15\n\x11QUERY_NO_RESOURCE\x10\r\x12\x16\n\x12QUERY_CLIENT_ERROR\x10\x0e\x62\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+clusterfuzz/_internal/protos/swarming.proto\x12\x0bswarming.v2\x1a\x1fgoogle/protobuf/timestamp.proto\"(\n\nStringPair\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t\",\n\x0eStringListPair\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x03(\t\"*\n\x06\x44igest\x12\x0c\n\x04hash\x18\x01 \x01(\t\x12\x12\n\nsize_bytes\x18\x02 \x01(\x03\"I\n\x0c\x43\x41SReference\x12\x14\n\x0c\x63\x61s_instance\x18\x01 \x01(\t\x12#\n\x06\x64igest\x18\x02 \x01(\x0b\x32\x13.swarming.v2.Digest\"B\n\x0b\x43ipdPackage\x12\x14\n\x0cpackage_name\x18\x01 \x01(\t\x12\x0f\n\x07version\x18\x02 \x01(\t\x12\x0c\n\x04path\x18\x03 \x01(\t\"y\n\tCipdInput\x12\x0e\n\x06server\x18\x01 \x01(\t\x12\x30\n\x0e\x63lient_package\x18\x02 \x01(\x0b\x32\x18.swarming.v2.CipdPackage\x12*\n\x08packages\x18\x03 \x03(\x0b\x32\x18.swarming.v2.CipdPackage\"(\n\nCacheEntry\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\t\"\xf2\x01\n\x0b\x43ontainment\x12\x16\n\x0elower_priority\x18\x01 \x01(\x08\x12\x42\n\x10\x63ontainment_type\x18\x02 \x01(\x0e\x32(.swarming.v2.Containment.ContainmentType\x12\x17\n\x0flimit_processes\x18\x03 \x01(\x03\x12$\n\x1climit_total_committed_memory\x18\x04 \x01(\x03\"H\n\x0f\x43ontainmentType\x12\x11\n\rNOT_SPECIFIED\x10\x00\x12\x08\n\x04NONE\x10\x01\x12\x08\n\x04\x41UTO\x10\x02\x12\x0e\n\nJOB_OBJECT\x10\x03\"\x83\x04\n\x0eTaskProperties\x12\'\n\x06\x63\x61\x63hes\x18\x01 \x03(\x0b\x32\x17.swarming.v2.CacheEntry\x12*\n\ncipd_input\x18\x02 \x01(\x0b\x32\x16.swarming.v2.CipdInput\x12\x0f\n\x07\x63ommand\x18\x03 \x03(\t\x12\x14\n\x0crelative_cwd\x18\x04 \x01(\t\x12+\n\ndimensions\x18\x05 \x03(\x0b\x32\x17.swarming.v2.StringPair\x12$\n\x03\x65nv\x18\x06 \x03(\x0b\x32\x17.swarming.v2.StringPair\x12\x31\n\x0c\x65nv_prefixes\x18\x07 \x03(\x0b\x32\x1b.swarming.v2.StringListPair\x12\x1e\n\x16\x65xecution_timeout_secs\x18\x08 \x01(\x05\x12\x19\n\x11grace_period_secs\x18\t \x01(\x05\x12\x12\n\nidempotent\x18\n \x01(\x08\x12\x31\n\x0e\x63\x61s_input_root\x18\x0b \x01(\x0b\x32\x19.swarming.v2.CASReference\x12\x17\n\x0fio_timeout_secs\x18\x0c \x01(\x05\x12\x0f\n\x07outputs\x18\r \x03(\t\x12\x14\n\x0csecret_bytes\x18\x0e \x01(\x0c\x12-\n\x0b\x63ontainment\x18\x0f \x01(\x0b\x32\x18.swarming.v2.Containment\"p\n\tTaskSlice\x12/\n\nproperties\x18\x01 \x01(\x0b\x32\x1b.swarming.v2.TaskProperties\x12\x17\n\x0f\x65xpiration_secs\x18\x02 \x01(\x05\x12\x19\n\x11wait_for_capacity\x18\x03 \x01(\x08\"\x1d\n\x0bResultDBCfg\x12\x0e\n\x06\x65nable\x18\x01 \x01(\x08\"\xe8\x04\n\x0eNewTaskRequest\x12\x17\n\x0f\x65xpiration_secs\x18\x01 \x01(\x05\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x16\n\x0eparent_task_id\x18\x03 \x01(\t\x12\x10\n\x08priority\x18\x04 \x01(\x05\x12/\n\nproperties\x18\x05 \x01(\x0b\x32\x1b.swarming.v2.TaskProperties\x12+\n\x0btask_slices\x18\x06 \x03(\x0b\x32\x16.swarming.v2.TaskSlice\x12\x0c\n\x04tags\x18\x07 \x03(\t\x12\x0c\n\x04user\x18\x08 \x01(\t\x12\x17\n\x0fservice_account\x18\t \x01(\t\x12\x14\n\x0cpubsub_topic\x18\n \x01(\t\x12\x19\n\x11pubsub_auth_token\x18\x0b \x01(\t\x12\x17\n\x0fpubsub_userdata\x18\x0c \x01(\t\x12\x15\n\revaluate_only\x18\r \x01(\x08\x12M\n\x12pool_task_template\x18\x0e \x01(\x0e\x32\x31.swarming.v2.NewTaskRequest.PoolTaskTemplateField\x12\x1f\n\x17\x62ot_ping_tolerance_secs\x18\x0f \x01(\x05\x12\x14\n\x0crequest_uuid\x18\x10 \x01(\t\x12*\n\x08resultdb\x18\x11 \x01(\x0b\x32\x18.swarming.v2.ResultDBCfg\x12\r\n\x05realm\x18\x12 \x01(\t\"P\n\x15PoolTaskTemplateField\x12\x08\n\x04\x41UTO\x10\x00\x12\x11\n\rCANARY_PREFER\x10\x01\x12\x10\n\x0c\x43\x41NARY_NEVER\x10\x02\x12\x08\n\x04SKIP\x10\x03\"\xf2\x03\n\x13TaskRequestResponse\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x17\n\x0f\x65xpiration_secs\x18\x02 \x01(\x05\x12\x0c\n\x04name\x18\x03 \x01(\t\x12\x16\n\x0eparent_task_id\x18\x04 \x01(\t\x12\x10\n\x08priority\x18\x05 \x01(\x05\x12/\n\nproperties\x18\x06 \x01(\x0b\x32\x1b.swarming.v2.TaskProperties\x12\x0c\n\x04tags\x18\x07 \x03(\t\x12.\n\ncreated_ts\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\x0c\n\x04user\x18\t \x01(\t\x12\x15\n\rauthenticated\x18\n \x01(\t\x12+\n\x0btask_slices\x18\x0b \x03(\x0b\x32\x16.swarming.v2.TaskSlice\x12\x17\n\x0fservice_account\x18\x0c \x01(\t\x12\r\n\x05realm\x18\r \x01(\t\x12*\n\x08resultdb\x18\x0e \x01(\x0b\x32\x18.swarming.v2.ResultDBCfg\x12\x14\n\x0cpubsub_topic\x18\x0f \x01(\t\x12\x17\n\x0fpubsub_userdata\x18\x10 \x01(\t\x12\x1f\n\x17\x62ot_ping_tolerance_secs\x18\x11 \x01(\x05\x12\x14\n\x0crbe_instance\x18\x12 \x01(\t\"\x9d\x01\n\x11TasksCountRequest\x12)\n\x05start\x18\x01 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12\'\n\x03\x65nd\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp\x12&\n\x05state\x18\x03 \x01(\x0e\x32\x17.swarming.v2.StateQuery\x12\x0c\n\x04tags\x18\x04 \x03(\t\"D\n\nTasksCount\x12\r\n\x05\x63ount\x18\x01 \x01(\x05\x12\'\n\x03now\x18\x02 \x01(\x0b\x32\x1a.google.protobuf.Timestamp*\xcf\x02\n\nStateQuery\x12\x11\n\rQUERY_PENDING\x10\x00\x12\x11\n\rQUERY_RUNNING\x10\x01\x12\x19\n\x15QUERY_PENDING_RUNNING\x10\x02\x12\x13\n\x0fQUERY_COMPLETED\x10\x03\x12\x1b\n\x17QUERY_COMPLETED_SUCCESS\x10\x04\x12\x1b\n\x17QUERY_COMPLETED_FAILURE\x10\x05\x12\x11\n\rQUERY_EXPIRED\x10\x06\x12\x13\n\x0fQUERY_TIMED_OUT\x10\x07\x12\x12\n\x0eQUERY_BOT_DIED\x10\x08\x12\x12\n\x0eQUERY_CANCELED\x10\t\x12\r\n\tQUERY_ALL\x10\n\x12\x11\n\rQUERY_DEDUPED\x10\x0b\x12\x10\n\x0cQUERY_KILLED\x10\x0c\x12\x15\n\x11QUERY_NO_RESOURCE\x10\r\x12\x16\n\x12QUERY_CLIENT_ERROR\x10\x0e\x62\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'clusterfuzz._internal.protos.swarming_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
   DESCRIPTOR._options = None
-  _globals['_STATEQUERY']._serialized_start=2291
-  _globals['_STATEQUERY']._serialized_end=2626
+  _globals['_STATEQUERY']._serialized_start=2792
+  _globals['_STATEQUERY']._serialized_end=3127
   _globals['_STRINGPAIR']._serialized_start=93
   _globals['_STRINGPAIR']._serialized_end=133
   _globals['_STRINGLISTPAIR']._serialized_start=135
@@ -66,8 +66,10 @@
   _globals['_NEWTASKREQUEST']._serialized_end=2058
   _globals['_NEWTASKREQUEST_POOLTASKTEMPLATEFIELD']._serialized_start=1978
   _globals['_NEWTASKREQUEST_POOLTASKTEMPLATEFIELD']._serialized_end=2058
-  _globals['_TASKSCOUNTREQUEST']._serialized_start=2061
-  _globals['_TASKSCOUNTREQUEST']._serialized_end=2218
-  _globals['_TASKSCOUNT']._serialized_start=2220
-  _globals['_TASKSCOUNT']._serialized_end=2288
+  _globals['_TASKREQUESTRESPONSE']._serialized_start=2061
+  _globals['_TASKREQUESTRESPONSE']._serialized_end=2559
+  _globals['_TASKSCOUNTREQUEST']._serialized_start=2562
+  _globals['_TASKSCOUNTREQUEST']._serialized_end=2719
+  _globals['_TASKSCOUNT']._serialized_start=2721
+  _globals['_TASKSCOUNT']._serialized_end=2789
 # @@protoc_insertion_point(module_scope)

src/clusterfuzz/_internal/protos/swarming_pb2.pyi

@@ -797,6 +797,90 @@ class NewTaskRequest(google.protobuf.message.Message):
 
 global___NewTaskRequest = NewTaskRequest
 
+@typing_extensions.final
+class TaskRequestResponse(google.protobuf.message.Message):
+    """Description of a task request as registered by the server.
+    This message is used when retrieving information about an existing task.
+    See NewTaskRequest for more details.
+    """
+
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_ID_FIELD_NUMBER: builtins.int
+    EXPIRATION_SECS_FIELD_NUMBER: builtins.int
+    NAME_FIELD_NUMBER: builtins.int
+    PARENT_TASK_ID_FIELD_NUMBER: builtins.int
+    PRIORITY_FIELD_NUMBER: builtins.int
+    PROPERTIES_FIELD_NUMBER: builtins.int
+    TAGS_FIELD_NUMBER: builtins.int
+    CREATED_TS_FIELD_NUMBER: builtins.int
+    USER_FIELD_NUMBER: builtins.int
+    AUTHENTICATED_FIELD_NUMBER: builtins.int
+    TASK_SLICES_FIELD_NUMBER: builtins.int
+    SERVICE_ACCOUNT_FIELD_NUMBER: builtins.int
+    REALM_FIELD_NUMBER: builtins.int
+    RESULTDB_FIELD_NUMBER: builtins.int
+    PUBSUB_TOPIC_FIELD_NUMBER: builtins.int
+    PUBSUB_USERDATA_FIELD_NUMBER: builtins.int
+    BOT_PING_TOLERANCE_SECS_FIELD_NUMBER: builtins.int
+    RBE_INSTANCE_FIELD_NUMBER: builtins.int
+    task_id: builtins.str
+    expiration_secs: builtins.int
+    name: builtins.str
+    parent_task_id: builtins.str
+    priority: builtins.int
+    @property
+    def properties(self) -> global___TaskProperties:
+        """For some amount of time, the properties will be copied into the
+        task_slices and vice-versa, to give time to the clients to update.
+        Eventually, only task_slices will be supported.
+        """
+    @property
+    def tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
+    @property
+    def created_ts(self) -> google.protobuf.timestamp_pb2.Timestamp: ...
+    user: builtins.str
+    authenticated: builtins.str
+    """User name of whoever posted this task, extracted from the credentials."""
+    @property
+    def task_slices(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___TaskSlice]: ...
+    service_account: builtins.str
+    """Indicates what OAuth2 credentials the task uses when calling other services."""
+    realm: builtins.str
+    @property
+    def resultdb(self) -> global___ResultDBCfg:
+        """Configuration of Swarming:ResultDB integration."""
+    pubsub_topic: builtins.str
+    pubsub_userdata: builtins.str
+    bot_ping_tolerance_secs: builtins.int
+    rbe_instance: builtins.str
+    def __init__(
+        self,
+        *,
+        task_id: builtins.str = ...,
+        expiration_secs: builtins.int = ...,
+        name: builtins.str = ...,
+        parent_task_id: builtins.str = ...,
+        priority: builtins.int = ...,
+        properties: global___TaskProperties | None = ...,
+        tags: collections.abc.Iterable[builtins.str] | None = ...,
+        created_ts: google.protobuf.timestamp_pb2.Timestamp | None = ...,
+        user: builtins.str = ...,
+        authenticated: builtins.str = ...,
+        task_slices: collections.abc.Iterable[global___TaskSlice] | None = ...,
+        service_account: builtins.str = ...,
+        realm: builtins.str = ...,
+        resultdb: global___ResultDBCfg | None = ...,
+        pubsub_topic: builtins.str = ...,
+        pubsub_userdata: builtins.str = ...,
+        bot_ping_tolerance_secs: builtins.int = ...,
+        rbe_instance: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["created_ts", b"created_ts", "properties", b"properties", "resultdb", b"resultdb"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["authenticated", b"authenticated", "bot_ping_tolerance_secs", b"bot_ping_tolerance_secs", "created_ts", b"created_ts", "expiration_secs", b"expiration_secs", "name", b"name", "parent_task_id", b"parent_task_id", "priority", b"priority", "properties", b"properties", "pubsub_topic", b"pubsub_topic", "pubsub_userdata", b"pubsub_userdata", "rbe_instance", b"rbe_instance", "realm", b"realm", "resultdb", b"resultdb", "service_account", b"service_account", "tags", b"tags", "task_id", b"task_id", "task_slices", b"task_slices", "user", b"user"]) -> None: ...
+
+global___TaskRequestResponse = TaskRequestResponse
+
 @typing_extensions.final
 class TasksCountRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor

src/clusterfuzz/_internal/swarming/api.py

@@ -134,24 +134,40 @@ def _make_request(self, endpoint: str, body: str) -> str | None:
 
     return response
 
-  def push_task(self, task_request: swarming_pb2.NewTaskRequest) -> str | None:  # pylint: disable=no-member
+  def push_task(
+      self,
+      task_request: swarming_pb2.NewTaskRequest  # pylint: disable=no-member
+  ) -> swarming_pb2.TaskRequestResponse:  # pylint: disable=no-member
     """Schedules a task on swarming.
     
     Args:
       task_request: The NewTaskRequest proto message.
       
     Returns:
-      The raw JSON response string from the server, or None if the response is
-      empty.
+      The TaskRequestResponse parsed proto message from the server.
 
     Raises:
-      requests.exceptions.HTTPError: If the request fails with a 4xx or 5xx
-        status code.
+      SwarmingApiError: If the pRPC request fails or response parsing fails.
     """
     message_body = json_format.MessageToJson(task_request)
 
-    response = self._make_request(_NEW_TASK_ENDPOINT, message_body)
-    return response
+    try:
+      raw_response = self._make_request(_NEW_TASK_ENDPOINT, message_body)
+    except HTTPError as e:
+      raise SwarmingApiError(f'HTTP error calling push_task: {e}') from e
+
+    if not raw_response:
+      raise SwarmingApiError(
+          'RPC Contract failure, got an empty response from Swarming API.')
+
+    try:
+      return json_format.Parse(
+          raw_response,
+          swarming_pb2.TaskRequestResponse()  # pylint: disable=no-member
+      )
+    except (json_format.ParseError, AttributeError) as e:
+      raise SwarmingApiError(
+          f'RPC Contract failure, failed to parse response: {e}') from e
 
   def count_tasks(
       self,