Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2015 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.systemui.statusbar.policy; |
| 17 | |
| 18 | import android.os.HandlerThread; |
Geoffrey Pitsch | 2c403db | 2016-08-26 09:09:39 -0400 | [diff] [blame] | 19 | import android.support.test.runner.AndroidJUnit4; |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 20 | import android.telephony.SubscriptionInfo; |
Sundeep Ghuman | 9d10a3c | 2017-06-16 00:05:16 +0000 | [diff] [blame] | 21 | import android.test.AndroidTestCase; |
Jason Monk | e427cb8 | 2015-07-28 09:18:08 -0400 | [diff] [blame] | 22 | import android.test.suitebuilder.annotation.SmallTest; |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 23 | import com.android.systemui.R; |
Jason Monk | 25a52b6 | 2017-05-23 10:42:59 -0400 | [diff] [blame] | 24 | import com.android.systemui.SysuiTestCase; |
Jason Monk | 46dbfb4 | 2016-02-25 14:59:20 -0500 | [diff] [blame] | 25 | import com.android.systemui.statusbar.policy.NetworkController.EmergencyListener; |
Jason Monk | 33f8ae7 | 2015-05-08 10:45:15 -0400 | [diff] [blame] | 26 | import com.android.systemui.statusbar.policy.NetworkController.IconState; |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 27 | import com.android.systemui.statusbar.policy.NetworkController.SignalCallback; |
Geoffrey Pitsch | 2c403db | 2016-08-26 09:09:39 -0400 | [diff] [blame] | 28 | import java.util.ArrayList; |
| 29 | import java.util.List; |
| 30 | import org.junit.Before; |
| 31 | import org.junit.runner.RunWith; |
| 32 | import org.junit.Test; |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 33 | import org.mockito.ArgumentCaptor; |
| 34 | import org.mockito.Mock; |
| 35 | import org.mockito.Mockito; |
| 36 | import org.mockito.MockitoAnnotations; |
| 37 | |
Geoffrey Pitsch | 2c403db | 2016-08-26 09:09:39 -0400 | [diff] [blame] | 38 | import static junit.framework.Assert.assertTrue; |
| 39 | import static junit.framework.Assert.assertEquals; |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 40 | |
Jason Monk | 0f0de13 | 2016-12-19 15:36:13 -0500 | [diff] [blame] | 41 | import static org.mockito.Matchers.eq; |
| 42 | |
Jason Monk | e427cb8 | 2015-07-28 09:18:08 -0400 | [diff] [blame] | 43 | @SmallTest |
Geoffrey Pitsch | 2c403db | 2016-08-26 09:09:39 -0400 | [diff] [blame] | 44 | @RunWith(AndroidJUnit4.class) |
Jason Monk | 25a52b6 | 2017-05-23 10:42:59 -0400 | [diff] [blame] | 45 | public class CallbackHandlerTest extends SysuiTestCase { |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 46 | |
| 47 | private CallbackHandler mHandler; |
| 48 | private HandlerThread mHandlerThread; |
| 49 | |
| 50 | @Mock |
| 51 | private EmergencyListener mEmengencyListener; |
| 52 | @Mock |
| 53 | private SignalCallback mSignalCallback; |
| 54 | |
Geoffrey Pitsch | 2c403db | 2016-08-26 09:09:39 -0400 | [diff] [blame] | 55 | @Before |
| 56 | public void setUp() throws Exception { |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 57 | mHandlerThread = new HandlerThread("TestThread"); |
| 58 | mHandlerThread.start(); |
| 59 | mHandler = new CallbackHandler(mHandlerThread.getLooper()); |
| 60 | |
| 61 | MockitoAnnotations.initMocks(this); |
| 62 | mHandler.setListening(mEmengencyListener, true); |
| 63 | mHandler.setListening(mSignalCallback, true); |
| 64 | } |
| 65 | |
Geoffrey Pitsch | 2c403db | 2016-08-26 09:09:39 -0400 | [diff] [blame] | 66 | @Test |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 67 | public void testEmergencyListener() { |
| 68 | mHandler.setEmergencyCallsOnly(true); |
| 69 | waitForCallbacks(); |
| 70 | |
| 71 | ArgumentCaptor<Boolean> captor = ArgumentCaptor.forClass(Boolean.class); |
| 72 | Mockito.verify(mEmengencyListener).setEmergencyCallsOnly(captor.capture()); |
| 73 | assertTrue(captor.getValue()); |
| 74 | } |
| 75 | |
Geoffrey Pitsch | 2c403db | 2016-08-26 09:09:39 -0400 | [diff] [blame] | 76 | @Test |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 77 | public void testSignalCallback_setWifiIndicators() { |
| 78 | boolean enabled = true; |
| 79 | IconState status = new IconState(true, 0, ""); |
| 80 | IconState qs = new IconState(true, 1, ""); |
| 81 | boolean in = true; |
| 82 | boolean out = true; |
| 83 | String description = "Test"; |
Jason Monk | 110e5f6 | 2017-03-31 13:45:12 -0400 | [diff] [blame] | 84 | mHandler.setWifiIndicators(enabled, status, qs, in, out, description, true); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 85 | waitForCallbacks(); |
| 86 | |
| 87 | ArgumentCaptor<Boolean> enableArg = ArgumentCaptor.forClass(Boolean.class); |
| 88 | ArgumentCaptor<IconState> statusArg = ArgumentCaptor.forClass(IconState.class); |
| 89 | ArgumentCaptor<IconState> qsArg = ArgumentCaptor.forClass(IconState.class); |
| 90 | ArgumentCaptor<Boolean> inArg = ArgumentCaptor.forClass(Boolean.class); |
| 91 | ArgumentCaptor<Boolean> outArg = ArgumentCaptor.forClass(Boolean.class); |
| 92 | ArgumentCaptor<String> descArg = ArgumentCaptor.forClass(String.class); |
Jason Monk | 110e5f6 | 2017-03-31 13:45:12 -0400 | [diff] [blame] | 93 | ArgumentCaptor<Boolean> isTransient = ArgumentCaptor.forClass(Boolean.class); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 94 | Mockito.verify(mSignalCallback).setWifiIndicators(enableArg.capture(), |
| 95 | statusArg.capture(), qsArg.capture(), inArg.capture(), outArg.capture(), |
Jason Monk | 110e5f6 | 2017-03-31 13:45:12 -0400 | [diff] [blame] | 96 | descArg.capture(), isTransient.capture()); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 97 | assertEquals(enabled, (boolean) enableArg.getValue()); |
| 98 | assertEquals(status, statusArg.getValue()); |
| 99 | assertEquals(qs, qsArg.getValue()); |
| 100 | assertEquals(in, (boolean) inArg.getValue()); |
| 101 | assertEquals(out, (boolean) outArg.getValue()); |
| 102 | assertEquals(description, descArg.getValue()); |
Jason Monk | 110e5f6 | 2017-03-31 13:45:12 -0400 | [diff] [blame] | 103 | assertTrue(isTransient.getValue()); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 104 | } |
| 105 | |
Geoffrey Pitsch | 2c403db | 2016-08-26 09:09:39 -0400 | [diff] [blame] | 106 | @Test |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 107 | public void testSignalCallback_setMobileDataIndicators() { |
| 108 | IconState status = new IconState(true, 0, ""); |
| 109 | IconState qs = new IconState(true, 1, ""); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 110 | boolean in = true; |
| 111 | boolean out = true; |
| 112 | String typeDescription = "Test 1"; |
| 113 | String description = "Test 2"; |
| 114 | int type = R.drawable.stat_sys_data_fully_connected_1x; |
Jason Monk | 01df36f | 2017-06-07 13:02:47 -0400 | [diff] [blame] | 115 | int qsType = R.drawable.ic_qs_signal_1x; |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 116 | boolean wide = true; |
| 117 | int subId = 5; |
Jason Monk | 0f0de13 | 2016-12-19 15:36:13 -0500 | [diff] [blame] | 118 | boolean roaming = true; |
Jason Monk | 7e6c83c | 2017-04-26 14:35:24 -0400 | [diff] [blame] | 119 | mHandler.setMobileDataIndicators(status, qs, type, qsType, in, out, typeDescription, |
| 120 | description, wide, subId, roaming); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 121 | waitForCallbacks(); |
| 122 | |
| 123 | ArgumentCaptor<IconState> statusArg = ArgumentCaptor.forClass(IconState.class); |
Jason Monk | 7e6c83c | 2017-04-26 14:35:24 -0400 | [diff] [blame] | 124 | ArgumentCaptor<IconState> qsArg = ArgumentCaptor.forClass(IconState.class); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 125 | ArgumentCaptor<Integer> typeIconArg = ArgumentCaptor.forClass(Integer.class); |
Jason Monk | 7e6c83c | 2017-04-26 14:35:24 -0400 | [diff] [blame] | 126 | ArgumentCaptor<Integer> qsTypeIconArg = ArgumentCaptor.forClass(Integer.class); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 127 | ArgumentCaptor<Boolean> inArg = ArgumentCaptor.forClass(Boolean.class); |
| 128 | ArgumentCaptor<Boolean> outArg = ArgumentCaptor.forClass(Boolean.class); |
| 129 | ArgumentCaptor<String> typeContentArg = ArgumentCaptor.forClass(String.class); |
Jason Monk | 7e6c83c | 2017-04-26 14:35:24 -0400 | [diff] [blame] | 130 | ArgumentCaptor<String> descArg = ArgumentCaptor.forClass(String.class); |
| 131 | ArgumentCaptor<Boolean> wideArg = ArgumentCaptor.forClass(Boolean.class); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 132 | ArgumentCaptor<Integer> subIdArg = ArgumentCaptor.forClass(Integer.class); |
Andrew Flynn | 2fdbe12 | 2015-06-01 16:34:21 -0400 | [diff] [blame] | 133 | Mockito.verify(mSignalCallback).setMobileDataIndicators(statusArg.capture(), |
Jason Monk | 7e6c83c | 2017-04-26 14:35:24 -0400 | [diff] [blame] | 134 | qsArg.capture(), typeIconArg.capture(), qsTypeIconArg.capture(), inArg.capture(), |
| 135 | outArg.capture(), typeContentArg.capture(), descArg.capture(), wideArg.capture(), |
| 136 | subIdArg.capture(), eq(roaming)); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 137 | assertEquals(status, statusArg.getValue()); |
Jason Monk | 7e6c83c | 2017-04-26 14:35:24 -0400 | [diff] [blame] | 138 | assertEquals(qs, qsArg.getValue()); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 139 | assertEquals(type, (int) typeIconArg.getValue()); |
Jason Monk | 7e6c83c | 2017-04-26 14:35:24 -0400 | [diff] [blame] | 140 | assertEquals(qsType, (int) qsTypeIconArg.getValue()); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 141 | assertEquals(in, (boolean) inArg.getValue()); |
| 142 | assertEquals(out, (boolean) outArg.getValue()); |
| 143 | assertEquals(typeDescription, typeContentArg.getValue()); |
Jason Monk | 7e6c83c | 2017-04-26 14:35:24 -0400 | [diff] [blame] | 144 | assertEquals(description, descArg.getValue()); |
| 145 | assertEquals(wide, (boolean) wideArg.getValue()); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 146 | assertEquals(subId, (int) subIdArg.getValue()); |
| 147 | } |
| 148 | |
| 149 | @SuppressWarnings("unchecked") |
Geoffrey Pitsch | 2c403db | 2016-08-26 09:09:39 -0400 | [diff] [blame] | 150 | @Test |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 151 | public void testSignalCallback_setSubs() { |
| 152 | List<SubscriptionInfo> subs = new ArrayList<>(); |
| 153 | mHandler.setSubs(subs); |
| 154 | waitForCallbacks(); |
| 155 | |
| 156 | ArgumentCaptor<ArrayList> subsArg = ArgumentCaptor.forClass(ArrayList.class); |
| 157 | Mockito.verify(mSignalCallback).setSubs(subsArg.capture()); |
| 158 | assertTrue(subs == subsArg.getValue()); |
| 159 | } |
| 160 | |
Geoffrey Pitsch | 2c403db | 2016-08-26 09:09:39 -0400 | [diff] [blame] | 161 | @Test |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 162 | public void testSignalCallback_setNoSims() { |
| 163 | boolean noSims = true; |
Jason Monk | 1ff7766 | 2017-09-29 11:17:05 -0400 | [diff] [blame] | 164 | boolean simDetected = false; |
| 165 | mHandler.setNoSims(noSims, simDetected); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 166 | waitForCallbacks(); |
| 167 | |
Jason Monk | 1ff7766 | 2017-09-29 11:17:05 -0400 | [diff] [blame] | 168 | Mockito.verify(mSignalCallback).setNoSims(eq(noSims), eq(simDetected)); |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 169 | } |
| 170 | |
Geoffrey Pitsch | 2c403db | 2016-08-26 09:09:39 -0400 | [diff] [blame] | 171 | @Test |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 172 | public void testSignalCallback_setEthernetIndicators() { |
| 173 | IconState state = new IconState(true, R.drawable.stat_sys_ethernet, "Test Description"); |
| 174 | mHandler.setEthernetIndicators(state); |
| 175 | waitForCallbacks(); |
| 176 | |
| 177 | ArgumentCaptor<IconState> iconArg = ArgumentCaptor.forClass(IconState.class); |
| 178 | Mockito.verify(mSignalCallback).setEthernetIndicators(iconArg.capture()); |
| 179 | assertEquals(state, iconArg.getValue()); |
| 180 | } |
| 181 | |
Geoffrey Pitsch | 2c403db | 2016-08-26 09:09:39 -0400 | [diff] [blame] | 182 | @Test |
Jason Monk | 07b75fe | 2015-05-14 16:47:03 -0400 | [diff] [blame] | 183 | public void testSignalCallback_setIsAirplaneMode() { |
| 184 | IconState state = new IconState(true, R.drawable.stat_sys_airplane_mode, "Test Description"); |
| 185 | mHandler.setIsAirplaneMode(state); |
| 186 | waitForCallbacks(); |
| 187 | |
| 188 | ArgumentCaptor<IconState> iconArg = ArgumentCaptor.forClass(IconState.class); |
| 189 | Mockito.verify(mSignalCallback).setIsAirplaneMode(iconArg.capture()); |
| 190 | assertEquals(state, iconArg.getValue()); |
| 191 | } |
| 192 | |
| 193 | private void waitForCallbacks() { |
| 194 | mHandlerThread.quitSafely(); |
| 195 | try { |
| 196 | mHandlerThread.join(); |
| 197 | } catch (InterruptedException e) { |
| 198 | } |
| 199 | } |
| 200 | |
| 201 | } |