Merge "Add Simulate maitanance mode in app startup test cases."

 */

package com.android.tests.applaunch;

import java.io.OutputStreamWriter;

import android.accounts.Account;

import android.accounts.AccountManager;

import android.app.ActivityManager;

import android.app.ActivityManager.ProcessErrorStateInfo;

import android.app.IActivityManager;

import android.app.UiAutomation;

import android.content.Context;

import android.content.Intent;

import android.content.pm.PackageManager;

import android.content.pm.PackageManager.NameNotFoundException;

import android.content.pm.ResolveInfo;

import android.os.Bundle;

import android.os.ParcelFileDescriptor;

import android.os.RemoteException;

import android.os.UserHandle;

import android.app.UiAutomation;

import android.app.IActivityManager;

import android.support.test.rule.logging.AtraceLogger;

import android.test.InstrumentationTestCase;

import android.test.InstrumentationTestRunner;

import android.util.Log;

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import java.io.OutputStreamWriter;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.HashSet;

import java.util.LinkedHashMap;

import java.util.List;

import java.util.ArrayList;

import java.util.Map;

import java.util.Set;

import android.os.ParcelFileDescriptor;

import java.io.FileInputStream;

import java.io.FileOutputStream;

import java.io.InputStream;

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.InputStreamReader;

/**

 * This test is intended to measure the time it takes for the apps to start.

 * Names of the applications are passed in command line, and the

    private static final String KEY_LAUNCH_ITERATIONS = "launch_iterations";

    private static final String KEY_LAUNCH_ORDER = "launch_order";

    private static final String KEY_DROP_CACHE = "drop_cache";

    private static final String KEY_SIMULATE_MAINTANANCE = "simulate_maintanance";

    private static final String KEY_SIMPLEPPERF_CMD = "simpleperf_cmd";

    private static final String KEY_TRACE_ITERATIONS = "trace_iterations";

    private static final String KEY_LAUNCH_DIRECTORY = "launch_directory";

    private static final String DROP_CACHE_SCRIPT = "/data/local/tmp/dropCache.sh";

    private static final String APP_LAUNCH_CMD = "am start -W -n";

    private static final String SUCCESS_MESSAGE = "Status: ok";

    private static final String PROFILE_COMPILE_SUCCESS = "Success";

    private static final String THIS_TIME = "ThisTime:";

    private static final String LAUNCH_ITERATION = "LAUNCH_ITERATION - %d";

    private static final String TRACE_ITERATION = "TRACE_ITERATION - %d";

    private static final String TRACE_ITERATION_PREFIX = "TRACE_ITERATION";

    private static final String LAUNCH_ORDER_CYCLIC = "cyclic";

    private static final String LAUNCH_ORDER_SEQUENTIAL = "sequential";

    private static final String SPEED_PROFILE_CMD = "cmd package compile -f -m speed-profile %s";

    private Map<String, Intent> mNameToIntent;

    private File mFile = null;

    private FileOutputStream mOutputStream = null;

    private BufferedWriter mBufferedWriter = null;

    private boolean mSimulateMaintanance = false;

    @Override

        mDropCache = Boolean.parseBoolean(args.getString(KEY_DROP_CACHE));

        mSimplePerfCmd = args.getString(KEY_SIMPLEPPERF_CMD);

        mLaunchOrder = args.getString(KEY_LAUNCH_ORDER, LAUNCH_ORDER_CYCLIC);

        mSimulateMaintanance =  Boolean.parseBoolean(args.getString(KEY_SIMULATE_MAINTANANCE));

        createMappings();

        parseArgs(args);

        checkAccountSignIn();

                    sleep(INITIAL_LAUNCH_IDLE_TIMEOUT);

                    closeApp(launch.getApp(), true);

                    dropCache();

                    if (mSimulateMaintanance) {

                        String appPkgName = mNameToIntent.get(launch.getApp())

                                .getComponent().getPackageName();

                        assertTrue(String.format("Not able to speed profile the app : %s",

                                appPkgName), profileCompileApp(appPkgName));

                    }

                    sleep(BETWEEN_LAUNCH_SLEEP_TIMEOUT);

                }

        instrumentation.sendStatus(0, mResult);

    }

    /**

     * Compile the app package using speed compile command and return true or false

     * based on status of the compilation command.

     */

    private boolean profileCompileApp(String appPkgName) throws IOException {

        Log.i(TAG, "Starting to speed profile " + appPkgName);

        try (ParcelFileDescriptor result = getInstrumentation().getUiAutomation().

                executeShellCommand(String.format(SPEED_PROFILE_CMD, appPkgName));

                BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(

                        new FileInputStream(result.getFileDescriptor())))) {

            String line;

            while ((line = bufferedReader.readLine()) != null) {

                if (line.contains(PROFILE_COMPILE_SUCCESS)) {

                    return true;

                }

            }

            return false;

        }

    }

    /**

     * If launch order is "cyclic" then apps will be launched one after the

     * other for each iteration count.