tests/robotests/src/com/android/settings/deviceinfo/ManageStoragePreferenceControllerTest.java

   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.settings.deviceinfo;
  17
  18
  19 import android.content.Context;
  20 import android.support.v7.preference.Preference;
  21 import android.support.v7.preference.PreferenceScreen;
  22 import com.android.settings.SettingsRobolectricTestRunner;
  23 import com.android.settings.TestConfig;
  24 import java.util.ArrayList;
  25 import java.util.List;
  26 import org.junit.Before;
  27 import org.junit.Test;
  28 import org.junit.runner.RunWith;
  29 import org.mockito.Answers;
  30 import org.mockito.Mock;
  31 import org.mockito.MockitoAnnotations;
  32 import org.robolectric.annotation.Config;
  33
  34 import static com.google.common.truth.Truth.assertThat;
  35 import static org.mockito.Answers.RETURNS_DEEP_STUBS;
  36 import static org.mockito.Matchers.any;
  37 import static org.mockito.Mockito.never;
  38 import static org.mockito.Mockito.verify;
  39 import static org.mockito.Mockito.when;
  40
  41 @RunWith(SettingsRobolectricTestRunner.class)
  42 @Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION)
  43 public class ManageStoragePreferenceControllerTest {
  44
  45
  46     @Mock(answer = Answers.RETURNS_DEEP_STUBS)
  47     private Context mContext;
  48     @Mock(answer = RETURNS_DEEP_STUBS)
  49     private PreferenceScreen mScreen;
  50
  51     private ManageStoragePreferenceController mController;
  52
  53     @Before
  54     public void setUp() {
  55         MockitoAnnotations.initMocks(this);
  56         mController = new ManageStoragePreferenceController(mContext);
  57     }
  58
  59     @Test
  60     public void updateNonIndexableKey_prefUnavaiable_shouldUpdate() {
  61         final List<String> keys = new ArrayList<>();
  62         mController.updateNonIndexableKeys(keys);
  63
  64         assertThat(keys).isNotEmpty();
  65     }
  66
  67     @Test
  68     public void updateNonIndexableKey_prefAvaiable_shouldNotUpdate() {
  69         final List<String> keys = new ArrayList<>();
  70         when(mContext.getResources().getBoolean(
  71                 com.android.settings.R.bool.config_storage_manager_settings_enabled))
  72                 .thenReturn(true);
  73
  74         mController.updateNonIndexableKeys(keys);
  75
  76         assertThat(keys).isEmpty();
  77     }
  78
  79     @Test
  80     public void displayPref_prefAvaiable_shouldDisplay() {
  81         when(mContext.getResources().getBoolean(
  82                 com.android.settings.R.bool.config_storage_manager_settings_enabled))
  83                 .thenReturn(true);
  84
  85         mController.displayPreference(mScreen);
  86
  87         verify(mScreen, never()).removePreference(any(Preference.class));
  88     }
  89
  90     @Test
  91     public void displayPref_prefAvaiable_shouldNotDisplay() {
  92         mController.displayPreference(mScreen);
  93
  94         verify(mScreen).removePreference(any(Preference.class));
  95     }
  96 }