Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit ced9626e authored by David Saff's avatar David Saff
Browse files

Keep a baseline heap before a test. 

This makes it even clearer which allocations are the fault of the
investigated test.

Bug: 262272836
Test: presubmit
Change-Id: I984a188713322db49c2358f9c1b2391652e03e19
parent 679ef9c1

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

+31 −6
Original line number Diff line number Diff line
@@ -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();

        }



        // 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;

    }

}