1 /*
2  * Copyright (C) 2017 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 
17 package com.android.settings.security;
18 
19 import static com.google.common.truth.Truth.assertThat;
20 
21 import android.content.Context;
22 import android.os.UserManager;
23 
24 import androidx.lifecycle.LifecycleOwner;
25 
26 import com.android.settings.testutils.shadow.ShadowUserManager;
27 import com.android.settingslib.core.lifecycle.Lifecycle;
28 
29 import org.junit.Before;
30 import org.junit.Test;
31 import org.junit.runner.RunWith;
32 import org.robolectric.RobolectricTestRunner;
33 import org.robolectric.RuntimeEnvironment;
34 import org.robolectric.annotation.Config;
35 
36 @RunWith(RobolectricTestRunner.class)
37 @Config(shadows = ShadowUserManager.class)
38 public class RestrictedEncryptionPreferenceControllerTest {
39 
40     private Context mContext;
41     private ShadowUserManager mUserManager;
42     private CredentialStoragePreferenceController mCredentialStoragePreferenceController;
43     private InstallCertificatePreferenceController mInstallCertificatePreferenceController;
44     private ResetCredentialsPreferenceController mResetCredentialsPreferenceController;
45     private UserCredentialsPreferenceController mUserCredentialsPreferenceController;
46     private InstallCaCertificatePreferenceController mInstallCaCertificatePreferenceController;
47     private InstallUserCertificatePreferenceController mInstallUserCertificatePreferenceController;
48     private InstallWifiCertificatePreferenceController mInstallWifiCertificatePreferenceController;
49     private Lifecycle mLifecycle;
50     private LifecycleOwner mLifecycleOwner;
51 
52     @Before
setUp()53     public void setUp() {
54         mContext = RuntimeEnvironment.application;
55         mLifecycleOwner = () -> mLifecycle;
56         mLifecycle = new Lifecycle(mLifecycleOwner);
57         mCredentialStoragePreferenceController =
58                 new CredentialStoragePreferenceController(mContext);
59         mInstallCertificatePreferenceController =
60                 new InstallCertificatePreferenceController(mContext);
61         mResetCredentialsPreferenceController =
62                 new ResetCredentialsPreferenceController(mContext, mLifecycle);
63         mUserCredentialsPreferenceController =
64                 new UserCredentialsPreferenceController(mContext);
65         mInstallCaCertificatePreferenceController =
66                 new InstallCaCertificatePreferenceController(mContext);
67         mInstallUserCertificatePreferenceController =
68                 new InstallUserCertificatePreferenceController(mContext);
69         mInstallWifiCertificatePreferenceController =
70                 new InstallWifiCertificatePreferenceController(mContext);
71         mUserManager = ShadowUserManager.getShadow();
72     }
73 
74     @Test
isAvailable_noRestriction_shouldReturnTrue()75     public void isAvailable_noRestriction_shouldReturnTrue() {
76         assertThat(mCredentialStoragePreferenceController.isAvailable()).isTrue();
77         assertThat(mInstallCertificatePreferenceController.isAvailable()).isTrue();
78         assertThat(mResetCredentialsPreferenceController.isAvailable()).isTrue();
79         assertThat(mUserCredentialsPreferenceController.isAvailable()).isTrue();
80         assertThat(mInstallCaCertificatePreferenceController.isAvailable()).isTrue();
81         assertThat(mInstallUserCertificatePreferenceController.isAvailable()).isTrue();
82         assertThat(mInstallWifiCertificatePreferenceController.isAvailable()).isTrue();
83     }
84 
85     @Test
isAvailable_hasRestriction_shouldReturnFalse()86     public void isAvailable_hasRestriction_shouldReturnFalse() {
87         mUserManager.addBaseUserRestriction(UserManager.DISALLOW_CONFIG_CREDENTIALS);
88 
89         assertThat(mCredentialStoragePreferenceController.isAvailable()).isFalse();
90         assertThat(mInstallCertificatePreferenceController.isAvailable()).isFalse();
91         assertThat(mResetCredentialsPreferenceController.isAvailable()).isFalse();
92         assertThat(mUserCredentialsPreferenceController.isAvailable()).isFalse();
93     }
94 }