Uninstall downloaded VM image after test is done
Bug: N/A
Test: N/A
Change-Id: I1b998d95f93bebad7263026a065775e3f5a0d6f3
diff --git a/libs/vm_launcher_lib/java/com/android/virtualization/vmlauncher/InstallUtils.java b/libs/vm_launcher_lib/java/com/android/virtualization/vmlauncher/InstallUtils.java
index 1febe27..03416c6 100644
--- a/libs/vm_launcher_lib/java/com/android/virtualization/vmlauncher/InstallUtils.java
+++ b/libs/vm_launcher_lib/java/com/android/virtualization/vmlauncher/InstallUtils.java
@@ -17,6 +17,7 @@
import android.content.Context;
import android.os.Environment;
+import android.os.FileUtils;
import android.util.Log;
import org.apache.commons.compress.archivers.ArchiveEntry;
@@ -59,6 +60,10 @@
}
}
+ public static void deleteInstallation(Context context) {
+ FileUtils.deleteContentsAndDir(getInternalStorageDir(context));
+ }
+
private static Path getPayloadPath() {
File payloadDir = Environment.getExternalStoragePublicDirectory(PAYLOAD_DIR);
if (payloadDir == null) {
diff --git a/tests/Terminal/src/com/android/virtualization/terminal/TerminalAppTest.java b/tests/Terminal/src/com/android/virtualization/terminal/TerminalAppTest.java
index 46fbc35..64ef79f 100644
--- a/tests/Terminal/src/com/android/virtualization/terminal/TerminalAppTest.java
+++ b/tests/Terminal/src/com/android/virtualization/terminal/TerminalAppTest.java
@@ -25,6 +25,9 @@
import androidx.test.InstrumentationRegistry;
import androidx.test.runner.AndroidJUnit4;
+import com.android.virtualization.vmlauncher.InstallUtils;
+
+import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
@@ -66,4 +69,9 @@
assertTrue("Failed to boot in 30s", activity.waitForBootCompleted(BOOT_TIMEOUT_MILLIS));
}
}
+
+ @After
+ public void tearDown() {
+ InstallUtils.deleteInstallation(mTargetContext);
+ }
}