1 /*
2  * Copyright (C) 2020 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.datausage;
18 
19 import static com.google.common.truth.Truth.assertThat;
20 
21 import android.content.Context;
22 import android.telephony.SubscriptionInfo;
23 
24 import androidx.test.core.app.ApplicationProvider;
25 import androidx.test.ext.junit.runners.AndroidJUnit4;
26 
27 import org.junit.Before;
28 import org.junit.Test;
29 import org.junit.runner.RunWith;
30 import org.mockito.Mock;
31 import org.mockito.MockitoAnnotations;
32 
33 @RunWith(AndroidJUnit4.class)
34 public class CellDataPreferenceTest {
35 
36     @Mock
37     private SubscriptionInfo mSubInfo;
38 
39     private Context mContext;
40     private CellDataPreference mPreference;
41 
42     @Before
setUp()43     public void setUp() {
44         MockitoAnnotations.initMocks(this);
45 
46         mContext = ApplicationProvider.getApplicationContext();
47         mPreference = new CellDataPreference(mContext, null) {
48             @Override
49             SubscriptionInfo getActiveSubscriptionInfo(int subId) {
50                 return mSubInfo;
51             }
52         };
53     }
54 
55     @Test
updateEnabled_noActiveSub_shouldDisable()56     public void updateEnabled_noActiveSub_shouldDisable() {
57         mSubInfo = null;
58 
59         mPreference.mOnSubscriptionsChangeListener.onChanged();
60 
61         assertThat(mPreference.isEnabled()).isFalse();
62     }
63 
64     @Test
updateEnabled_hasActiveSub_shouldEnable()65     public void updateEnabled_hasActiveSub_shouldEnable() {
66         mPreference.mOnSubscriptionsChangeListener.onChanged();
67 
68         assertThat(mPreference.isEnabled()).isTrue();
69     }
70 }
71