Android/frameworks_base
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge "Keep a baseline heap before a test." into tm-qpr-dev

Browse Source
This commit is contained in:
David Saff
2023-01-11 16:14:14 +00:00
committed by Android (Google) Code Review
parent dbefd37340 ced9626ec7
commit 68d61e4f51
1 changed files with 31 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
packages/SystemUI/tests/utils/src/com/android/systemui/MemoryTrackingTestCase.java
View File

@@ -40,24 +40,49 @@ import java.io.IOException;
public class MemoryTrackingTestCase extends SysuiTestCase {
private static File sFilesDir = null;
private static String sLatestTestClassName = null;
private static int sHeapCount = 0;
private static File sLatestBaselineHeapFile = null;
@Before public void grabFilesDir() {
// Ideally, we would do this in @BeforeClass just once, but we need mContext to get the files
// dir, and that does not exist until @Before on each test method.
@Before
public void grabFilesDir() throws IOException {
// This should happen only once per suite
if (sFilesDir == null) {
sFilesDir = mContext.getFilesDir();
}
sLatestTestClassName = getClass().getName();
// This will happen before the first test method in each class
if (sLatestTestClassName == null) {
sLatestTestClassName = getClass().getName();
sLatestBaselineHeapFile = dump("baseline" + (++sHeapCount), "before-test");
}
}
@AfterClass
public static void dumpHeap() throws IOException {
File afterTestHeap = dump(sLatestTestClassName, "after-test");
if (sLatestBaselineHeapFile != null && afterTestHeap != null) {
Log.w("MEMORY", "To compare heap to baseline (use go/ahat):");
Log.w("MEMORY", " adb pull " + sLatestBaselineHeapFile);
Log.w("MEMORY", " adb pull " + afterTestHeap);
Log.w("MEMORY",
" java -jar ahat.jar --baseline " + sLatestBaselineHeapFile.getName() + " "
+ afterTestHeap.getName());
}
sLatestTestClassName = null;
}
private static File dump(String basename, String heapKind) throws IOException {
if (sFilesDir == null) {
Log.e("MEMORY", "Somehow no test cases??");
return;
return null;
}
mockitoTearDown();
Log.w("MEMORY", "about to dump heap");
File path = new File(sFilesDir, sLatestTestClassName + ".ahprof");
Log.w("MEMORY", "about to dump " + heapKind + " heap");
File path = new File(sFilesDir, basename + ".ahprof");
Debug.dumpHprofData(path.getPath());
Log.w("MEMORY", "did it! Location: " + path);
Log.w("MEMORY", "Success! Location: " + path);
return path;
}
}