Internal change

PiperOrigin-RevId: 935921835

Author: OmniLab Team

src/devtools/mobileharness/platform/android/instrumentation/result/proto/test_suite_result.proto

@@ -47,6 +47,12 @@ message TestSuiteMetaData {
 
   // Number of test cases scheduled to be run.
   int32 scheduled_test_case_count = 2;
+
+  // Start time of the test suite execution.
+  google.protobuf.Timestamp start_time = 3;
+
+  // End time of the test suite execution.
+  google.protobuf.Timestamp end_time = 4;
 }
 
 // Test status of a test suite or test case

src/java/com/google/devtools/mobileharness/platform/android/instrumentation/parser/AmInstrumentationParser.kt

@@ -42,10 +42,15 @@ import java.time.Instant
  * INSTRUMENTATION_CODE: -1
  * ```
  */
-class AmInstrumentationParser(
+class AmInstrumentationParser
+@JvmOverloads
+constructor(
   private val listeners: Set<AmInstrumentationListener> = emptySet(),
   private val testTimeTrackerFactory: () -> TestTimeTracker = { TestTimeTracker() },
+  private val systemClock: () -> Instant = { Instant.now() },
 ) {
+  private val startTime = systemClock()
+
   /** The result of the Instrumentation, available after parse completion. */
   var result: InstrumentationResult? = null
     private set
@@ -262,7 +267,14 @@ class AmInstrumentationParser(
       reportInstrumentationFailed("Test run failed to complete. $error")
     }
     val result =
-      InstrumentationResult(code, bundle.toMutableMap().also { knownStatus.forEach(it::remove) })
+      InstrumentationResult(
+          code = code,
+          bundle = bundle.toMutableMap().also { knownStatus.forEach(it::remove) },
+        )
+        .apply {
+          startTime = this@AmInstrumentationParser.startTime
+          endTime = systemClock()
+        }
     this.result = result
     reportInstrumentationEnded(result)
   }

src/java/com/google/devtools/mobileharness/platform/android/instrumentation/parser/AmInstrumentationResultBuilder.kt

@@ -61,5 +61,8 @@ class AmInstrumentationResultBuilder(private val testSuiteResultBuilder: TestSui
       } else {
         ProtoTestStatus.FAILED
       }
+    val metaDataBuilder = testSuiteResultBuilder.testSuiteMetaDataBuilder
+    instrumentationResult.startTime?.let { metaDataBuilder.startTime = it.toProtoTimestamp() }
+    instrumentationResult.endTime?.let { metaDataBuilder.endTime = it.toProtoTimestamp() }
   }
 }

src/java/com/google/devtools/mobileharness/platform/android/instrumentation/parser/InstrumentationResult.kt

@@ -1,5 +1,7 @@
 package com.google.devtools.mobileharness.platform.android.instrumentation.parser
 
+import java.time.Instant
+
 /**
  * Results reported at the end of an `am instrument` command.
  *
@@ -9,6 +11,9 @@ package com.google.devtools.mobileharness.platform.android.instrumentation.parse
  * * other: Failure
  */
 data class InstrumentationResult(val code: Int? = null, val bundle: Map<String, String> = mapOf()) {
+  var startTime: Instant? = null
+  var endTime: Instant? = null
+
   val success
     get() = code == -1
 }