2 * Copyright (C) 2017 The Android Open Source Project
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
8 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 package com.android.settings.enterprise;
19 import android.content.Context;
20 import android.support.v7.preference.Preference;
22 import com.android.settings.R;
23 import com.android.settings.applications.ApplicationFeatureProvider;
24 import com.android.settings.testutils.FakeFeatureFactory;
26 import org.junit.Before;
27 import org.junit.Test;
28 import org.mockito.Answers;
29 import org.mockito.Mock;
30 import org.mockito.MockitoAnnotations;
31 import org.mockito.invocation.InvocationOnMock;
32 import org.mockito.stubbing.Answer;
34 import static com.google.common.truth.Truth.assertThat;
35 import static org.mockito.Matchers.anyObject;
36 import static org.mockito.Matchers.eq;
37 import static org.mockito.Mockito.doAnswer;
38 import static org.mockito.Mockito.when;
41 * Common base for testing subclasses of {@link AdminGrantedPermissionsPreferenceControllerBase}.
43 public abstract class AdminGrantedPermissionsPreferenceControllerTestBase {
44 protected final String mKey;
45 protected final String[] mPermissions;
46 protected final String mPermissionGroup;
48 @Mock(answer = Answers.RETURNS_DEEP_STUBS)
49 protected Context mContext;
50 private FakeFeatureFactory mFeatureFactory;
52 protected AdminGrantedPermissionsPreferenceControllerBase mController;
54 public AdminGrantedPermissionsPreferenceControllerTestBase(String key, String[] permissions,
55 String permissionGroup) {
57 mPermissions = permissions;
58 mPermissionGroup = permissionGroup;
63 MockitoAnnotations.initMocks(this);
64 FakeFeatureFactory.setupForTest(mContext);
65 mFeatureFactory = (FakeFeatureFactory) FakeFeatureFactory.getFactory(mContext);
66 mController = createController(true /* async */);
69 private void setNumberOfPackagesWithAdminGrantedPermissions(int number, boolean async) {
70 doAnswer(new Answer() {
71 public Object answer(InvocationOnMock invocation) {
72 ((ApplicationFeatureProvider.NumberOfAppsCallback)
73 invocation.getArguments()[2]).onNumberOfAppsResult(number);
75 }}).when(mFeatureFactory.applicationFeatureProvider)
76 .calculateNumberOfAppsWithAdminGrantedPermissions(eq(mPermissions),
77 eq(async), anyObject());
81 public void testUpdateState() {
82 final Preference preference = new Preference(mContext, null, 0, 0);
83 preference.setVisible(true);
85 setNumberOfPackagesWithAdminGrantedPermissions(0, true /* async */);
86 mController.updateState(preference);
87 assertThat(preference.isVisible()).isFalse();
89 setNumberOfPackagesWithAdminGrantedPermissions(20, true /* async */);
90 when(mContext.getResources().getQuantityString(
91 R.plurals.enterprise_privacy_number_packages_lower_bound, 20, 20))
92 .thenReturn("minimum 20 apps");
93 mController.updateState(preference);
94 assertThat(preference.getSummary()).isEqualTo("minimum 20 apps");
95 assertThat(preference.isVisible()).isTrue();
99 public void testIsAvailableSync() {
100 final AdminGrantedPermissionsPreferenceControllerBase controller
101 = createController(false /* async */);
103 setNumberOfPackagesWithAdminGrantedPermissions(0, false /* async */);
104 assertThat(controller.isAvailable()).isFalse();
106 setNumberOfPackagesWithAdminGrantedPermissions(20, false /* async */);
107 assertThat(controller.isAvailable()).isTrue();
111 public void testIsAvailableAsync() {
112 setNumberOfPackagesWithAdminGrantedPermissions(0, true /* async */);
113 assertThat(mController.isAvailable()).isTrue();
115 setNumberOfPackagesWithAdminGrantedPermissions(20, true /* async */);
116 assertThat(mController.isAvailable()).isTrue();
120 public void testHandlePreferenceTreeClick() {
121 assertThat(mController.handlePreferenceTreeClick(new Preference(mContext, null, 0, 0)))
126 public void testGetPreferenceKey() {
127 assertThat(mController.getPreferenceKey()).isEqualTo(mKey);
130 protected abstract AdminGrantedPermissionsPreferenceControllerBase createController(