Handle timeouts when uploading ResultStore results

1 files changed, 11 insertions, 1 deletions

diff --git a/tools/run_tests/python_utils/upload_rbe_results.py b/tools/run_tests/python_utils/upload_rbe_results.py
index d302024883..5955b3792f 100644
--- a/tools/run_tests/python_utils/upload_rbe_results.py
+++ b/tools/run_tests/python_utils/upload_rbe_results.py
@@ -146,7 +146,15 @@ if __name__ == "__main__":
             test_cases = [{
                 'testCase': {
                     'caseName': str(action['id']['actionId']),
-                    'result': str(action['statusAttributes']['status'])
+                }
+            }]
+        # Test timeouts have a different dictionary structure compared to pass and
+        # fail results.
+        elif action['statusAttributes']['status'] == 'TIMED_OUT':
+            test_cases = [{
+                'testCase': {
+                    'caseName': str(action['id']['actionId']),
+                    'timedOut': True
                 }
             }]
         else:
@@ -155,6 +163,8 @@ if __name__ == "__main__":
         for test_case in test_cases:
             if 'errors' in test_case['testCase']:
                 result = 'FAILED'
+            elif 'timedOut' in test_case['testCase']:
+                result = 'TIMEOUT'
             else:
                 result = 'PASSED'
             bq_rows.append({