1 /* 2 * Copyright (C) 2014 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 package com.android.cts.managedprofile; 17 18 import android.app.admin.DeviceAdminReceiver; 19 import android.app.admin.DevicePolicyManager; 20 import android.content.ComponentName; 21 import android.content.Context; 22 import android.support.test.uiautomator.UiDevice; 23 import android.test.InstrumentationTestCase; 24 25 /** 26 * Base class for profile-owner based tests. 27 * 28 * This class handles making sure that the test is the profile owner and that it has an active admin 29 * registered, so that all tests may assume these are done. 30 */ 31 public class BaseManagedProfileTest extends InstrumentationTestCase { 32 33 public static class BasicAdminReceiver extends DeviceAdminReceiver { 34 } 35 36 public static final ComponentName ADMIN_RECEIVER_COMPONENT = new ComponentName( 37 BasicAdminReceiver.class.getPackage().getName(), BasicAdminReceiver.class.getName()); 38 39 protected DevicePolicyManager mDevicePolicyManager; 40 protected DevicePolicyManager mParentDevicePolicyManager; 41 protected Context mContext; 42 43 @Override setUp()44 protected void setUp() throws Exception { 45 super.setUp(); 46 mContext = getInstrumentation().getContext(); 47 48 mDevicePolicyManager = (DevicePolicyManager) 49 mContext.getSystemService(Context.DEVICE_POLICY_SERVICE); 50 mParentDevicePolicyManager = 51 mDevicePolicyManager.getParentProfileInstance(ADMIN_RECEIVER_COMPONENT); 52 assertNotNull(mDevicePolicyManager); 53 54 // TODO: Only check the below if we are running as the profile user. If running under the 55 // user owner, can we check that there is a profile and that the below holds for it? If we 56 // don't want to do these checks every time we could get rid of this class altogether and 57 // just have a single test case running under the profile user that do them. 58 assertTrue(mDevicePolicyManager.isAdminActive(ADMIN_RECEIVER_COMPONENT)); 59 assertTrue(mDevicePolicyManager.isProfileOwnerApp( 60 ADMIN_RECEIVER_COMPONENT.getPackageName())); 61 } 62 getDevicePolicyManager(boolean isParent)63 protected DevicePolicyManager getDevicePolicyManager(boolean isParent) { 64 return isParent ? mParentDevicePolicyManager : mDevicePolicyManager; 65 } 66 } 67