Merge "Mock WifiManager.class for HotspotConditionControllerTest"

This commit is contained in:
TreeHugger Robot
2022-01-22 08:35:50 +00:00
committed by Android (Google) Code Review

View File

@@ -18,28 +18,31 @@ package com.android.settings.homepage.contextualcards.conditional;
import static com.google.common.truth.Truth.assertThat; import static com.google.common.truth.Truth.assertThat;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.when;
import android.content.Context; import android.content.Context;
import android.net.wifi.SoftApConfiguration; import android.net.wifi.SoftApConfiguration;
import android.net.wifi.WifiManager; import android.net.wifi.WifiManager;
import androidx.test.core.app.ApplicationProvider;
import com.android.settings.homepage.contextualcards.ContextualCard; import com.android.settings.homepage.contextualcards.ContextualCard;
import com.android.settings.testutils.shadow.ShadowWifiManager;
import org.junit.Before; import org.junit.Before;
import org.junit.Ignore;
import org.junit.Test; import org.junit.Test;
import org.junit.runner.RunWith; import org.junit.runner.RunWith;
import org.mockito.Mock; import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.robolectric.RobolectricTestRunner; import org.robolectric.RobolectricTestRunner;
import org.robolectric.RuntimeEnvironment;
import org.robolectric.annotation.Config;
@RunWith(RobolectricTestRunner.class) @RunWith(RobolectricTestRunner.class)
@Config(shadows = {ShadowWifiManager.class})
public class HotspotConditionControllerTest { public class HotspotConditionControllerTest {
private static final String WIFI_AP_SSID = "Test Hotspot"; private static final String WIFI_AP_SSID = "Test Hotspot";
@Mock
private WifiManager mWifiManager;
@Mock @Mock
private ConditionManager mConditionManager; private ConditionManager mConditionManager;
@@ -48,12 +51,14 @@ public class HotspotConditionControllerTest {
@Before @Before
public void setUp() { public void setUp() {
mContext = RuntimeEnvironment.application; MockitoAnnotations.initMocks(this);
mContext = spy(ApplicationProvider.getApplicationContext());
when(mContext.getSystemService(WifiManager.class)).thenReturn(mWifiManager);
mController = new HotspotConditionController(mContext, mConditionManager); mController = new HotspotConditionController(mContext, mConditionManager);
} }
@Test @Test
@Ignore
public void buildContextualCard_hasWifiAp_shouldHaveWifiApSsid() { public void buildContextualCard_hasWifiAp_shouldHaveWifiApSsid() {
setupSoftApConfiguration(); setupSoftApConfiguration();
@@ -63,7 +68,6 @@ public class HotspotConditionControllerTest {
} }
@Test @Test
@Ignore
public void buildContextualCard_noWifiAp_shouldHaveEmptySsid() { public void buildContextualCard_noWifiAp_shouldHaveEmptySsid() {
final ContextualCard card = mController.buildContextualCard(); final ContextualCard card = mController.buildContextualCard();
@@ -73,6 +77,6 @@ public class HotspotConditionControllerTest {
private void setupSoftApConfiguration() { private void setupSoftApConfiguration() {
final SoftApConfiguration wifiApConfig = new SoftApConfiguration.Builder() final SoftApConfiguration wifiApConfig = new SoftApConfiguration.Builder()
.setSsid(WIFI_AP_SSID).build(); .setSsid(WIFI_AP_SSID).build();
mContext.getSystemService(WifiManager.class).setSoftApConfiguration(wifiApConfig); when(mWifiManager.getSoftApConfiguration()).thenReturn(wifiApConfig);
} }
} }