Merge "Settings performance testing"

tests/perftests/src/com/android/settings/tests/perf/LaunchSettingsTest.java

@@ -15,8 +15,17 @@
  */
 package com.android.settings.tests.perf;
 
+import static androidx.test.platform.app.InstrumentationRegistry.getInstrumentation;
+
+import static junit.framework.TestCase.fail;
+
 import android.app.Instrumentation;
 import android.os.Bundle;
+import android.util.Log;
+import android.support.test.uiautomator.By;
+import android.support.test.uiautomator.UiDevice;
+import android.support.test.uiautomator.UiSelector;
+import android.support.test.uiautomator.Until;
 
 import androidx.test.InstrumentationRegistry;
 import androidx.test.runner.AndroidJUnit4;
@@ -26,23 +35,101 @@ import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 
+import java.util.ArrayList;
+import java.util.LinkedHashMap;
+import java.util.Map;
+import java.util.Collections;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
 @RunWith(AndroidJUnit4.class)
 public class LaunchSettingsTest {
 
+    private static final int TIME_OUT = 5000;
+    private static final int TEST_TIME = 10;
+    private static final Pattern PATTERN = Pattern.compile("TotalTime:\\s[0-9]*");
+    private static final String[] PAGES =
+            {"Settings", "Wi-Fi", "BlueTooth", "Application", "Battery"};
+    private Bundle mBundle;
+    private UiDevice mDevice;
+    private Instrumentation mInstrumentation;
+    private Map<String, ArrayList<Integer>> mResult;
+
     @Before
     public void setUp() throws Exception {
+        mBundle = new Bundle();
+        mDevice = UiDevice.getInstance(getInstrumentation());
+        mInstrumentation = InstrumentationRegistry.getInstrumentation();
+        mResult = new LinkedHashMap<>();
+        mDevice.pressHome();
+        mDevice.waitForIdle(TIME_OUT);
+
+        for (String string : PAGES) {
+            mResult.put(string, new ArrayList<Integer>());
+        }
     }
 
     @After
     public void tearDown() throws Exception {
+        sendResult();
+        mInstrumentation.sendStatus(0, mBundle);
     }
 
     @Test
-    public void testReportMetrics() throws Exception {
+    public void settingsPerformanceTest() throws Exception {
-        Instrumentation instrumentation = InstrumentationRegistry.getInstrumentation();
+        for (int i = 0; i < TEST_TIME; i++) {
-        final Bundle result = new Bundle();
+            executePreformanceTest("android.settings.SETTINGS", "Search settings", 0);
-        result.putString("LaunchSettingsTest_metric_key1", "1000");
+            executePreformanceTest("android.settings.WIFI_SETTINGS", "Use Wi‑Fi", 1);
-        result.putString("LaunchSettingsTest_metric_key2", "5000");
+            executePreformanceTest("android.settings.BLUETOOTH_SETTINGS", "Connected devices", 2);
-        instrumentation.sendStatus(0, result);
+            executePreformanceTest("android.settings.APPLICATION_SETTINGS", "App info", 3);
+            executePreformanceTest("android.intent.action.POWER_USAGE_SUMMARY", "Battery", 4);
+        }
+
     }
-}
+
+    private void executePreformanceTest(String activity, String text, int page) throws Exception {
+        final String mString = mDevice.executeShellCommand("am start -W -a" + activity);
+        mDevice.wait(Until.findObject(By.text(text)), TIME_OUT);
+        handleLaunchResult(page, mString);
+        closeApp();
+        mDevice.waitForIdle(TIME_OUT);
+    }
+
+    private void handleLaunchResult(int page, String s) {
+        Matcher mMatcher = PATTERN.matcher(s);
+        if (mMatcher.find()) {
+            mResult.get(PAGES[page]).add(Integer.valueOf(mMatcher.group().split("\\s")[1]));
+        } else {
+            fail("Some pages can't be found");
+        }
+    }
+
+    private void closeApp() throws Exception {
+        mDevice.pressRecentApps();
+        mDevice.findObject(new UiSelector().resourceId("com.android.launcher3:id/snapshot"))
+                .swipeUp(10);
+    }
+
+    private void sendResult() {
+        for (String string : mResult.keySet()) {
+            mBundle.putString(String.format("LaunchSettingsTest_%s_%s", string, "max"),
+                    getMax(mResult.get(string)));
+            mBundle.putString(String.format("LaunchSettingsTest_%s_%s", string, "min"),
+                    getMin(mResult.get(string)));
+            mBundle.putString(String.format("LaunchSettingsTest_%s_%s", string, "avg"),
+                    getAvg(mResult.get(string)));
+        }
+    }
+
+    private String getMax(ArrayList<Integer> launchResult) {
+        return String.format("%s", launchResult.isEmpty() ? "null" : Collections.max(launchResult));
+    }
+
+    private String getMin(ArrayList<Integer> launchResult) {
+        return String.format("%s", launchResult.isEmpty() ? "null" : Collections.min(launchResult));
+    }
+
+    private String getAvg(ArrayList<Integer> launchResult) {
+        return String.valueOf((int) launchResult.stream().mapToInt(i -> i).average().orElse(0));
+    }
+}