Settings performance testing

 */

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;

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 {

        Instrumentation instrumentation = InstrumentationRegistry.getInstrumentation();

        final Bundle result = new Bundle();

        result.putString("LaunchSettingsTest_metric_key1", "1000");

        result.putString("LaunchSettingsTest_metric_key2", "5000");

        instrumentation.sendStatus(0, result);

    public void settingsPerformanceTest() throws Exception {

        for (int i = 0; i < TEST_TIME; i++) {

            executePreformanceTest("android.settings.SETTINGS", "Search settings", 0);

            executePreformanceTest("android.settings.WIFI_SETTINGS", "Use Wi‑Fi", 1);

            executePreformanceTest("android.settings.BLUETOOTH_SETTINGS", "Connected devices", 2);

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

    }

}

 No newline at end of file