1/*
2 * Copyright (C) 2016 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.android.managedprovisioning;
18
19import android.app.Activity;
20import android.app.Application;
21import android.app.Application.ActivityLifecycleCallbacks;
22import android.content.Context;
23import android.content.Intent;
24import android.os.Bundle;
25import android.support.test.runner.AndroidJUnitRunner;
26import android.util.ArrayMap;
27
28import android.view.WindowManager;
29import java.util.Map;
30
31public class TestInstrumentationRunner extends AndroidJUnitRunner {
32    // hard-coded package name as context.getPackageName() provides ManagedProvisioning app name
33    // instead of test package name
34    public static final String TEST_PACKAGE_NAME = "com.android.managedprovisioning.tests";
35
36    private static final String TAG = "TestInstrumentationRunner";
37    private static final Map<String, OnActivityCreatedCallback> sReplacedActivityMap =
38            new ArrayMap();
39
40    public static void registerReplacedActivity(Class<?> oldActivity,
41            OnActivityCreatedCallback onActivityCreatedCallback) {
42        sReplacedActivityMap.put(oldActivity.getCanonicalName(), onActivityCreatedCallback);
43    }
44
45    public static void registerReplacedActivity(Class<?> oldActivity,
46            Class<? extends Activity> newActivity) {
47        registerReplacedActivity(oldActivity,
48                (classLoader, className, intent) -> newActivity.newInstance());
49    }
50
51    public static void unregisterReplacedActivity(Class<?> oldActivity) {
52        sReplacedActivityMap.remove(oldActivity.getCanonicalName());
53    }
54
55    @Override
56    public Activity newActivity(ClassLoader cl, String className, Intent intent)
57            throws InstantiationException, IllegalAccessException, ClassNotFoundException {
58        OnActivityCreatedCallback callback = sReplacedActivityMap.get(className);
59        if (callback != null) {
60            return callback.createActivity(cl, className, intent);
61        } else {
62            return super.newActivity(cl, className, intent);
63        }
64    }
65
66    @Override
67    public Application newApplication(ClassLoader cl, String className, Context context)
68            throws InstantiationException, IllegalAccessException, ClassNotFoundException {
69        Application app = super.newApplication(cl, className, context);
70        app.registerActivityLifecycleCallbacks(new ActivityLifecycleCallbacks() {
71            @Override
72            public void onActivityCreated(Activity activity, Bundle bundle) {
73                // Show activity on top of keyguard
74                activity.getWindow().addFlags(WindowManager.LayoutParams.FLAG_DISMISS_KEYGUARD);
75                // Turn on screen to prevent activity being paused by system. See b/31262906
76                activity.getWindow().addFlags(WindowManager.LayoutParams.FLAG_TURN_SCREEN_ON);
77                activity.getWindow().addFlags(WindowManager.LayoutParams.FLAG_KEEP_SCREEN_ON);
78            }
79
80            @Override
81            public void onActivityStarted(Activity activity) {}
82
83            @Override
84            public void onActivityResumed(Activity activity) {}
85
86            @Override
87            public void onActivityPaused(Activity activity) {}
88
89            @Override
90            public void onActivityStopped(Activity activity) {}
91
92            @Override
93            public void onActivitySaveInstanceState(Activity activity, Bundle bundle) {}
94
95            @Override
96            public void onActivityDestroyed(Activity activity) {}
97        });
98        return app;
99    }
100
101    public interface OnActivityCreatedCallback {
102        Activity createActivity(ClassLoader cl, String className, Intent intent)
103                throws IllegalAccessException, InstantiationException;
104    }
105}
106