From 35c3c76d577e04f24a72a2b0d5a76430f6940dbc Mon Sep 17 00:00:00 2001
From: vadimt <vadimt@google.com>
Date: Fri, 2 Jul 2021 17:43:32 -0700
Subject: [PATCH] Deleting screen records for succeeded tests

Test: local runs
Bug: 187761685
Change-Id: I3b1a75b775d1d6677b331869a063fc6132217947
---
 .../com/android/launcher3/util/rule/ScreenRecordRule.java | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/tests/src/com/android/launcher3/util/rule/ScreenRecordRule.java b/tests/src/com/android/launcher3/util/rule/ScreenRecordRule.java
index 00b1cdd35e..7a5cf2c2cd 100644
--- a/tests/src/com/android/launcher3/util/rule/ScreenRecordRule.java
+++ b/tests/src/com/android/launcher3/util/rule/ScreenRecordRule.java
@@ -62,12 +62,17 @@ public class ScreenRecordRule implements TestRule {
                 ParcelFileDescriptor output =
                         automation.executeShellCommand("screenrecord " + outputFile);
                 String screenRecordPid = device.executeShellCommand("pidof screenrecord");
+                boolean success = false;
                 try {
                     base.evaluate();
+                    success = true;
                 } finally {
                     device.executeShellCommand("kill -INT " + screenRecordPid);
                     Log.e(TAG, "Screenrecord captured at: " + outputFile);
                     output.close();
+                    if (success) {
+                        automation.executeShellCommand("rm " + outputFile);
+                    }
                 }
             }
         };
@@ -78,5 +83,6 @@ public class ScreenRecordRule implements TestRule {
      */
     @Retention(RetentionPolicy.RUNTIME)
     @Target(ElementType.METHOD)
-    public @interface ScreenRecord { }
+    public @interface ScreenRecord {
+    }
 }