1 /* 2 * Copyright (C) 2017 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.wifi; 18 19 import android.content.BroadcastReceiver; 20 import android.content.Context; 21 import android.content.Intent; 22 import android.content.IntentFilter; 23 import android.net.wifi.WifiManager; 24 25 import com.android.settings.R; 26 import com.android.settings.SettingsRobolectricTestRunner; 27 import com.android.settings.TestConfig; 28 import com.android.settings.widget.SummaryUpdater.OnSummaryChangeListener; 29 import com.android.settingslib.wifi.WifiStatusTracker; 30 31 import org.junit.Before; 32 import org.junit.Test; 33 import org.junit.runner.RunWith; 34 import org.mockito.Answers; 35 import org.mockito.Mock; 36 import org.mockito.MockitoAnnotations; 37 import org.robolectric.RuntimeEnvironment; 38 import org.robolectric.annotation.Config; 39 40 import static com.google.common.truth.Truth.assertThat; 41 import static org.mockito.Matchers.any; 42 import static org.mockito.Matchers.anyString; 43 import static org.mockito.Mockito.spy; 44 import static org.mockito.Mockito.verify; 45 46 @RunWith(SettingsRobolectricTestRunner.class) 47 @Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION) 48 public class WifiSummaryUpdaterTest { 49 50 @Mock(answer = Answers.RETURNS_DEEP_STUBS) 51 private WifiManager mWifiManager; 52 @Mock 53 private SummaryListener mListener; 54 55 private Context mContext; 56 private WifiSummaryUpdater mSummaryUpdater; 57 private WifiStatusTracker mWifiTracker; 58 59 @Before setUp()60 public void setUp() { 61 MockitoAnnotations.initMocks(this); 62 mWifiTracker = new WifiStatusTracker(mWifiManager); 63 64 mContext = spy(RuntimeEnvironment.application.getApplicationContext()); 65 mSummaryUpdater = new WifiSummaryUpdater(mContext, mListener, mWifiTracker); 66 } 67 68 @Test register_true_shouldRegisterListener()69 public void register_true_shouldRegisterListener() { 70 mSummaryUpdater.register(true); 71 72 verify(mContext).registerReceiver(any(BroadcastReceiver.class), any(IntentFilter.class)); 73 } 74 75 @Test register_false_shouldUnregisterListener()76 public void register_false_shouldUnregisterListener() { 77 mSummaryUpdater.register(true); 78 mSummaryUpdater.register(false); 79 80 verify(mContext).unregisterReceiver(any(BroadcastReceiver.class)); 81 } 82 83 @Test onReceive_networkStateChanged_shouldSendSummaryChange()84 public void onReceive_networkStateChanged_shouldSendSummaryChange() { 85 mSummaryUpdater.register(true); 86 mContext.sendBroadcast(new Intent(WifiManager.NETWORK_STATE_CHANGED_ACTION)); 87 88 89 verify(mListener).onSummaryChanged(anyString()); 90 } 91 92 @Test onReceive_rssiChanged_shouldSendSummaryChange()93 public void onReceive_rssiChanged_shouldSendSummaryChange() { 94 mSummaryUpdater.register(true); 95 mContext.sendBroadcast(new Intent(WifiManager.RSSI_CHANGED_ACTION)); 96 97 98 verify(mListener).onSummaryChanged(anyString()); 99 } 100 101 @Test getSummary_wifiDisabled_shouldReturnDisabled()102 public void getSummary_wifiDisabled_shouldReturnDisabled() { 103 mWifiTracker.enabled = false; 104 105 assertThat(mSummaryUpdater.getSummary()).isEqualTo( 106 mContext.getString(R.string.switch_off_text)); 107 } 108 109 @Test getSummary_wifiDisconnected_shouldReturnDisconnected()110 public void getSummary_wifiDisconnected_shouldReturnDisconnected() { 111 mWifiTracker.enabled = true; 112 mWifiTracker.connected = false; 113 114 assertThat(mSummaryUpdater.getSummary()).isEqualTo( 115 mContext.getString(R.string.disconnected)); 116 } 117 118 @Test getSummary_wifiConnected_shouldReturnSsid()119 public void getSummary_wifiConnected_shouldReturnSsid() { 120 mWifiTracker.enabled = true; 121 mWifiTracker.connected = true; 122 mWifiTracker.ssid = "Test Ssid"; 123 124 assertThat(mSummaryUpdater.getSummary()).isEqualTo("Test Ssid"); 125 } 126 127 private class SummaryListener implements OnSummaryChangeListener { 128 String summary; 129 130 @Override onSummaryChanged(String summary)131 public void onSummaryChanged(String summary) { 132 this.summary = summary; 133 } 134 } 135 136 } 137