1 /* 2 * Copyright (C) 2024 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.Context 20 import androidx.test.core.app.ApplicationProvider 21 import com.android.settings.core.BasePreferenceController.AVAILABLE 22 import com.android.settings.core.BasePreferenceController.CONDITIONALLY_UNAVAILABLE 23 import com.android.wifitrackerlib.WifiEntry 24 import com.google.common.truth.Truth.assertThat 25 import org.junit.Rule 26 import org.junit.Test 27 import org.mockito.Mock 28 import org.mockito.Spy 29 import org.mockito.junit.MockitoJUnit 30 import org.mockito.junit.MockitoRule 31 import org.mockito.Mockito.`when` as whenever 32 33 class WepLessSecureWarningControllerTest { 34 @get:Rule 35 val mockito: MockitoRule = MockitoJUnit.rule() 36 @Spy 37 private val context: Context = ApplicationProvider.getApplicationContext() 38 @Mock 39 private lateinit var mockWifiEntry: WifiEntry 40 41 private val controller = WepLessSecureWarningController(context, TEST_KEY) 42 43 @Test getAvailabilityStatus_default_conditionallyUnavailablenull44 fun getAvailabilityStatus_default_conditionallyUnavailable() { 45 assertThat(controller.availabilityStatus).isEqualTo(CONDITIONALLY_UNAVAILABLE) 46 } 47 48 @Test getAvailabilityStatus_noWepSecurityType_conditionallyUnavailablenull49 fun getAvailabilityStatus_noWepSecurityType_conditionallyUnavailable() { 50 whenever(mockWifiEntry.securityTypes).thenReturn(listOf(WifiEntry.SECURITY_PSK)) 51 52 controller.setWifiEntry(mockWifiEntry) 53 54 assertThat(controller.availabilityStatus).isEqualTo(CONDITIONALLY_UNAVAILABLE) 55 } 56 57 @Test getAvailabilityStatus_containsWepSecurityType_availablenull58 fun getAvailabilityStatus_containsWepSecurityType_available() { 59 whenever(mockWifiEntry.securityTypes).thenReturn(listOf(WifiEntry.SECURITY_WEP)) 60 61 controller.setWifiEntry(mockWifiEntry) 62 63 assertThat(controller.availabilityStatus).isEqualTo(AVAILABLE) 64 } 65 66 private companion object { 67 const val TEST_KEY = "test_key" 68 } 69 } 70