From 134221b92b6c3ae68ed59b428ab82a840f632d3a Mon Sep 17 00:00:00 2001
From: Sam Dubey <dubeyshubham@google.com>
Date: Tue, 2 May 2023 17:45:56 +0000
Subject: [PATCH] Add libraries to record screen when needed

Iw tests are recorded in staged testing which helps in debugging.

Test: go/abtd/run/L82800000960365904
Bug: 280563885
Change-Id: Idecfc69c37626d5991f09a12d0a43573c898ff00
---
 core/tests/coretests/Android.bp      | 1 +
 core/tests/coretests/AndroidTest.xml | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/core/tests/coretests/Android.bp b/core/tests/coretests/Android.bp
index ab500f730fc8c..286697cbd9403 100644
--- a/core/tests/coretests/Android.bp
+++ b/core/tests/coretests/Android.bp
@@ -36,6 +36,7 @@ android_test {
         "-c fa",
     ],
     static_libs: [
+        "collector-device-lib-platform",
         "frameworks-base-testutils",
         "core-test-rules", // for libcore.dalvik.system.CloseGuardSupport
         "core-tests-support",
diff --git a/core/tests/coretests/AndroidTest.xml b/core/tests/coretests/AndroidTest.xml
index e2cdbf36b5303..3e4c47b36ed90 100644
--- a/core/tests/coretests/AndroidTest.xml
+++ b/core/tests/coretests/AndroidTest.xml
@@ -32,6 +32,13 @@
 
     <target_preparer class="com.android.compatibility.common.tradefed.targetprep.DeviceInteractionHelperInstaller" />
 
+    <metrics_collector class="com.android.tradefed.device.metric.FilePullerLogCollector">
+        <option name="directory-keys" value="/data/user/0/com.android.frameworks.coretests/files" />
+        <option name="pull-pattern-keys"
+                value="android.device.collectors.ScreenRecordCollector.*\.mp4"/>
+        <option name="collect-on-run-ended-only" value="true" />
+    </metrics_collector>
+
     <option name="test-tag" value="FrameworksCoreTests" />
     <test class="com.android.tradefed.testtype.AndroidJUnitTest" >
         <option name="package" value="com.android.frameworks.coretests" />