| /* |
| * Copyright (C) 2019 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package com.android.settingslib.wifi; |
| |
| import static com.google.common.truth.Truth.assertThat; |
| |
| import static org.mockito.Mockito.when; |
| |
| import android.content.Context; |
| import android.graphics.drawable.Drawable; |
| |
| import com.android.wifitrackerlib.WifiEntry; |
| |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.mockito.Mock; |
| import org.mockito.MockitoAnnotations; |
| import org.robolectric.RobolectricTestRunner; |
| import org.robolectric.RuntimeEnvironment; |
| |
| import java.util.ArrayList; |
| import java.util.List; |
| |
| @RunWith(RobolectricTestRunner.class) |
| public class WifiEntryPreferenceTest { |
| |
| private Context mContext; |
| |
| @Mock |
| private WifiEntry mMockWifiEntry; |
| @Mock |
| private WifiEntryPreference.IconInjector mMockIconInjector; |
| |
| @Mock |
| private Drawable mMockDrawable0; |
| @Mock |
| private Drawable mMockDrawable1; |
| @Mock |
| private Drawable mMockDrawable2; |
| @Mock |
| private Drawable mMockDrawable3; |
| @Mock |
| private Drawable mMockDrawable4; |
| |
| private static final String MOCK_TITLE = "title"; |
| private static final String MOCK_SUMMARY = "summary"; |
| |
| |
| @Before |
| public void setUp() { |
| mContext = RuntimeEnvironment.application; |
| |
| MockitoAnnotations.initMocks(this); |
| |
| when(mMockWifiEntry.getTitle()).thenReturn(MOCK_TITLE); |
| when(mMockWifiEntry.getSummary(false /* concise */)).thenReturn(MOCK_SUMMARY); |
| |
| when(mMockIconInjector.getIcon(0)).thenReturn(mMockDrawable0); |
| when(mMockIconInjector.getIcon(1)).thenReturn(mMockDrawable1); |
| when(mMockIconInjector.getIcon(2)).thenReturn(mMockDrawable2); |
| when(mMockIconInjector.getIcon(3)).thenReturn(mMockDrawable3); |
| when(mMockIconInjector.getIcon(4)).thenReturn(mMockDrawable4); |
| } |
| |
| @Test |
| public void constructor_shouldSetWifiEntryTitleAndSummary() { |
| final WifiEntryPreference pref = |
| new WifiEntryPreference(mContext, mMockWifiEntry, mMockIconInjector); |
| |
| assertThat(pref.getTitle()).isEqualTo(MOCK_TITLE); |
| assertThat(pref.getSummary()).isEqualTo(MOCK_SUMMARY); |
| } |
| |
| @Test |
| public void constructor_shouldSetIcon() { |
| when(mMockWifiEntry.getLevel()).thenReturn(0); |
| |
| final WifiEntryPreference pref = |
| new WifiEntryPreference(mContext, mMockWifiEntry, mMockIconInjector); |
| |
| assertThat(pref.getIcon()).isEqualTo(mMockDrawable0); |
| } |
| |
| @Test |
| public void titleChanged_refresh_shouldUpdateTitle() { |
| final WifiEntryPreference pref = |
| new WifiEntryPreference(mContext, mMockWifiEntry, mMockIconInjector); |
| final String updatedTitle = "updated title"; |
| when(mMockWifiEntry.getTitle()).thenReturn(updatedTitle); |
| |
| pref.refresh(); |
| |
| assertThat(pref.getTitle()).isEqualTo(updatedTitle); |
| } |
| |
| @Test |
| public void summaryChanged_refresh_shouldUpdateSummary() { |
| final WifiEntryPreference pref = |
| new WifiEntryPreference(mContext, mMockWifiEntry, mMockIconInjector); |
| final String updatedSummary = "updated summary"; |
| when(mMockWifiEntry.getSummary(false /* concise */)).thenReturn(updatedSummary); |
| |
| pref.refresh(); |
| |
| assertThat(pref.getSummary()).isEqualTo(updatedSummary); |
| } |
| |
| @Test |
| public void levelChanged_refresh_shouldUpdateLevelIcon() { |
| final List<Drawable> iconList = new ArrayList<>(); |
| final WifiEntryPreference pref = |
| new WifiEntryPreference(mContext, mMockWifiEntry, mMockIconInjector); |
| |
| when(mMockWifiEntry.getLevel()).thenReturn(0); |
| pref.refresh(); |
| iconList.add(pref.getIcon()); |
| when(mMockWifiEntry.getLevel()).thenReturn(1); |
| pref.refresh(); |
| iconList.add(pref.getIcon()); |
| when(mMockWifiEntry.getLevel()).thenReturn(2); |
| pref.refresh(); |
| iconList.add(pref.getIcon()); |
| when(mMockWifiEntry.getLevel()).thenReturn(3); |
| pref.refresh(); |
| iconList.add(pref.getIcon()); |
| when(mMockWifiEntry.getLevel()).thenReturn(4); |
| pref.refresh(); |
| iconList.add(pref.getIcon()); |
| when(mMockWifiEntry.getLevel()).thenReturn(-1); |
| pref.refresh(); |
| iconList.add(pref.getIcon()); |
| |
| assertThat(iconList).containsExactly(mMockDrawable0, mMockDrawable1, |
| mMockDrawable2, mMockDrawable3, mMockDrawable4, null); |
| } |
| } |