1be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka/* 2be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * Copyright (C) 2014 The Android Open Source Project 3be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * 4be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * Licensed under the Apache License, Version 2.0 (the "License"); 5be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * you may not use this file except in compliance with the License. 6be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * You may obtain a copy of the License at 7be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * 8be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * http://www.apache.org/licenses/LICENSE-2.0 9be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * 10be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * Unless required by applicable law or agreed to in writing, software 11be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * distributed under the License is distributed on an "AS IS" BASIS, 12be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * See the License for the specific language governing permissions and 14be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka * limitations under the License. 15be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka */ 16be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 17be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaokapackage com.android.inputmethod.latin.utils; 18be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 19604158669b407a40cd0f23538fad4dce5d738f24Mohammadinamul Sheikimport static com.android.inputmethod.latin.utils.ImportantNoticeUtils.KEY_TIMESTAMP_OF_CONTACTS_NOTICE; 208ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyenimport static org.mockito.Mockito.when; 21be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 22be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaokaimport android.content.Context; 23be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaokaimport android.content.SharedPreferences; 24be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaokaimport android.test.AndroidTestCase; 2590058c3d1f09e37a74bf6f44d9ebd314be3e680bTadashi G. Takaokaimport android.test.suitebuilder.annotation.MediumTest; 26be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaokaimport android.text.TextUtils; 27be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 288ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyenimport com.android.inputmethod.latin.settings.SettingsValues; 298ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen 308ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyenimport org.mockito.Mock; 318ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyenimport org.mockito.MockitoAnnotations; 328ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen 33be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaokaimport java.util.concurrent.TimeUnit; 34be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 3590058c3d1f09e37a74bf6f44d9ebd314be3e680bTadashi G. Takaoka@MediumTest 36be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaokapublic class ImportantNoticeUtilsTests extends AndroidTestCase { 37be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 38be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka private ImportantNoticePreferences mImportantNoticePreferences; 39be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 408ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen @Mock private SettingsValues mMockSettingsValues; 418ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen 42be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka private static class ImportantNoticePreferences { 43be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka private final SharedPreferences mPref; 44be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 45be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka private Integer mVersion; 46be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka private Long mLastTime; 47be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 48be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka public ImportantNoticePreferences(final Context context) { 49be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka mPref = ImportantNoticeUtils.getImportantNoticePreferences(context); 50be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 51be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 52be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka private Integer getInt(final String key) { 53be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka if (mPref.contains(key)) { 54be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka return mPref.getInt(key, 0); 55be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 56be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka return null; 57be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 58be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 59be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka public Long getLong(final String key) { 60be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka if (mPref.contains(key)) { 61be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka return mPref.getLong(key, 0); 62be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 63be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka return null; 64be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 65be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 66be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka private void putInt(final String key, final Integer value) { 67be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka if (value == null) { 68be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka removePreference(key); 69be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } else { 70be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka mPref.edit().putInt(key, value).apply(); 71be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 72be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 73be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 74be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka private void putLong(final String key, final Long value) { 75be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka if (value == null) { 76be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka removePreference(key); 77be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } else { 78be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka mPref.edit().putLong(key, value).apply(); 79be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 80be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 81be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 82be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka private void removePreference(final String key) { 83be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka mPref.edit().remove(key).apply(); 84be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 85be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 86be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka public void save() { 87604158669b407a40cd0f23538fad4dce5d738f24Mohammadinamul Sheik mLastTime = getLong(KEY_TIMESTAMP_OF_CONTACTS_NOTICE); 88be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 89be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 90be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka public void restore() { 91604158669b407a40cd0f23538fad4dce5d738f24Mohammadinamul Sheik putLong(KEY_TIMESTAMP_OF_CONTACTS_NOTICE, mLastTime); 92be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 93be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 94be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka public void clear() { 95604158669b407a40cd0f23538fad4dce5d738f24Mohammadinamul Sheik removePreference(KEY_TIMESTAMP_OF_CONTACTS_NOTICE); 96be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 97be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 98be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 99be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka @Override 100be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka protected void setUp() throws Exception { 101be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka super.setUp(); 1028ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen MockitoAnnotations.initMocks(this); 103be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka mImportantNoticePreferences = new ImportantNoticePreferences(getContext()); 104be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka mImportantNoticePreferences.save(); 1058ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen when(mMockSettingsValues.isPersonalizationEnabled()).thenReturn(true); 106be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 107be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 108be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka @Override 109be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka protected void tearDown() throws Exception { 110be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka super.tearDown(); 111be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka mImportantNoticePreferences.restore(); 112be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka } 113be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka 1148ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen public void testPersonalizationSetting() { 1158ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen mImportantNoticePreferences.clear(); 1168ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen 1178ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen // Personalization enabled. 1188ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen when(mMockSettingsValues.isPersonalizationEnabled()).thenReturn(true); 1198ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen assertEquals("Current boolean with personalization enabled", true, 1208ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen ImportantNoticeUtils.shouldShowImportantNotice(getContext(), mMockSettingsValues)); 1218ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen 1228ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen // Personalization disabled. 1238ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen when(mMockSettingsValues.isPersonalizationEnabled()).thenReturn(false); 1248ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen assertEquals("Current boolean with personalization disabled", false, 1258ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen ImportantNoticeUtils.shouldShowImportantNotice(getContext(), mMockSettingsValues)); 1268ec36026c93d13c1118eda936ac12325e0085561Chieu Nguyen } 127be4e0d0e3921af3d575f745cb356704974362bafTadashi G. Takaoka} 128