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 package com.android.managedprovisioning.common;
17 
18 import static com.android.managedprovisioning.common.ManagedProvisioningSharedPreferences.SHARED_PREFERENCE;
19 
20 import static com.google.common.truth.Truth.assertThat;
21 
22 import static org.junit.Assert.assertEquals;
23 import static org.mockito.Mockito.eq;
24 import static org.mockito.Mockito.when;
25 
26 import android.content.Context;
27 import android.content.SharedPreferences;
28 
29 import androidx.test.InstrumentationRegistry;
30 import androidx.test.filters.SmallTest;
31 
32 import org.junit.After;
33 import org.junit.Before;
34 import org.junit.Test;
35 import org.mockito.Mock;
36 import org.mockito.MockitoAnnotations;
37 
38 @SmallTest
39 public class ManagedProvisioningSharedPreferencesTest {
40 
41     private static final String KEY_TEST_SHARED_PREFERENCE =
42             "ManagedProvisioningSharedPreferencesTest";
43 
44     @Mock
45     private Context mContext;
46     private SharedPreferences mSharedPreferences;
47     private ManagedProvisioningSharedPreferences mManagedProvisioningSharedPreferences;
48 
49     @Before
setUp()50     public void setUp() {
51         MockitoAnnotations.initMocks(this);
52         Context targetContext = InstrumentationRegistry.getTargetContext();
53         mSharedPreferences = targetContext.getSharedPreferences(KEY_TEST_SHARED_PREFERENCE,
54                 Context.MODE_PRIVATE);
55         cleanUp();
56 
57         when(mContext.getSharedPreferences(eq(SHARED_PREFERENCE), eq(Context.MODE_PRIVATE)))
58                 .thenReturn(mSharedPreferences);
59         mManagedProvisioningSharedPreferences = new ManagedProvisioningSharedPreferences(mContext);
60     }
61 
62     @After
tearDown()63     public void tearDown() {
64         cleanUp();
65     }
66 
cleanUp()67     private void cleanUp() {
68         clearSharedPreferences();
69     }
70 
clearSharedPreferences()71     private boolean clearSharedPreferences() {
72         return mSharedPreferences.edit().clear().commit();
73     }
74 
75     @Test
testGetAndIncrementProvisioningId()76     public void testGetAndIncrementProvisioningId() {
77         assertEquals(mManagedProvisioningSharedPreferences.incrementAndGetProvisioningId(), 1L);
78         assertEquals(mManagedProvisioningSharedPreferences.getProvisioningId(), 1L);
79 
80         assertEquals(mManagedProvisioningSharedPreferences.incrementAndGetProvisioningId(), 2L);
81         assertEquals(mManagedProvisioningSharedPreferences.getProvisioningId(), 2L);
82     }
83 
84     @Test
testProvisioningStartedTimestamp()85     public void testProvisioningStartedTimestamp() {
86         mManagedProvisioningSharedPreferences.writeProvisioningStartedTimestamp(1234);
87 
88         assertThat(mManagedProvisioningSharedPreferences.getProvisioningStartedTimestamp())
89                 .isEqualTo(1234);
90     }
91 
92     @Test
setIsProvisioningFlowDelegatedToRoleHolder_setToTrue_works()93     public void setIsProvisioningFlowDelegatedToRoleHolder_setToTrue_works() {
94         mManagedProvisioningSharedPreferences.setIsProvisioningFlowDelegatedToRoleHolder(true);
95 
96         assertThat(mManagedProvisioningSharedPreferences.isProvisioningFlowDelegatedToRoleHolder())
97                 .isTrue();
98     }
99 
100     @Test
setIsProvisioningFlowDelegatedToRoleHolder_setToFalse_works()101     public void setIsProvisioningFlowDelegatedToRoleHolder_setToFalse_works() {
102         mManagedProvisioningSharedPreferences.setIsProvisioningFlowDelegatedToRoleHolder(false);
103 
104         assertThat(mManagedProvisioningSharedPreferences.isProvisioningFlowDelegatedToRoleHolder())
105                 .isFalse();
106     }
107 
108     @Test
setIsProvisioningFlowDelegatedToRoleHolder_sharedPreferencesCleared_valueIsFalse()109     public void setIsProvisioningFlowDelegatedToRoleHolder_sharedPreferencesCleared_valueIsFalse() {
110         clearSharedPreferences();
111 
112         assertThat(mManagedProvisioningSharedPreferences.isProvisioningFlowDelegatedToRoleHolder())
113                 .isFalse();
114     }
115 }
116