service/tests/wifitests/src/com/android/server/wifi/AvailableNetworkNotifierTest.java

/*
 * Copyright (C) 2021 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.server.wifi;

import static com.android.server.wifi.ConnectToNetworkNotificationBuilder.ACTION_CONNECT_TO_NETWORK;

import static org.mockito.Mockito.*;

import android.content.BroadcastReceiver;
import android.content.Intent;
import android.net.wifi.ScanResult;
import android.net.wifi.ScanResult.InformationElement;
import android.net.wifi.WifiSsid;
import android.os.Looper;

import androidx.test.filters.SmallTest;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.mockito.ArgumentCaptor;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

import java.nio.charset.StandardCharsets;

/**
 * Unit tests for {@link com.android.server.wifi.EapFailureNotifier}.
 */
@SmallTest
public class AvailableNetworkNotifierTest extends WifiBaseTest {
    private AvailableNetworkNotifier mAvailableNetworkNotifier;
    @Mock WifiContext mContext;
    @Mock Looper mLooper;
    @Mock FrameworkFacade mFrameworkFacade;
    @Mock Clock mClock;
    @Mock WifiMetrics mWifiMetrics;
    @Mock WifiConfigManager mWifiConfigManager;
    @Mock WifiConfigStore mWifiConfigStore;
    @Mock ConnectHelper mConnectHelper;
    @Mock ConnectToNetworkNotificationBuilder mConnectToNetworkNotificationBuilder;
    @Mock MakeBeforeBreakManager mMakeBeforeBreakManager;
    @Mock WifiNotificationManager mWifiNotificationManager;

    BroadcastReceiver mBroadcastReceiver;

    /**
     * Sets up for unit test
     */
    @Before
    public void setUp() throws Exception {
        MockitoAnnotations.initMocks(this);

        mAvailableNetworkNotifier = new AvailableNetworkNotifier(
                "AvailableNetworkNotifierTest",
                "storeDataIdentifier",
                "toggleSettingsName",
                1, // notificationIdentifier
                1, // nominatorId
                mContext,
                mLooper,
                mFrameworkFacade,
                mClock,
                mWifiMetrics,
                mWifiConfigManager,
                mWifiConfigStore,
                mConnectHelper,
                mConnectToNetworkNotificationBuilder,
                mMakeBeforeBreakManager,
                mWifiNotificationManager);

        ArgumentCaptor<BroadcastReceiver> captor = ArgumentCaptor.forClass(BroadcastReceiver.class);
        verify(mContext).registerReceiver(captor.capture(), any(), any(), any());
        mBroadcastReceiver = captor.getValue();
    }

    @After
    public void cleanUp() throws Exception {
        validateMockitoUsage();
    }

    /** Verify that connecting to an unknown akm network works normally.
     */
    @Test
    public void testConnectToUnknownAkmNetwork() throws Exception {
        mAvailableNetworkNotifier.mState =
                AvailableNetworkNotifier.STATE_SHOWING_RECOMMENDATION_NOTIFICATION;
        final String ssid = "UnknownAkm-Network";
        final String caps = "[RSN-?-TKIP+CCMP][ESS][WPS]";
        ScanResult result = new ScanResult(WifiSsid.createFromAsciiEncoded(ssid), ssid,
                "ab:cd:01:ef:45:89", 1245, 0, caps, -78, 2450, 1025, 22, 33, 20, 0,
                0, true);
        InformationElement ie = new InformationElement();
        ie.id = InformationElement.EID_SSID;
        ie.bytes = ssid.getBytes(StandardCharsets.UTF_8);
        result.informationElements = new InformationElement[] { ie };
        mAvailableNetworkNotifier.mRecommendedNetwork = result;

        Intent intent = new Intent();
        intent.setAction(ACTION_CONNECT_TO_NETWORK);
        mBroadcastReceiver.onReceive(mContext, intent);
        verify(mWifiConfigManager, never()).addOrUpdateNetwork(any(), anyInt());
    }
}