Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 696c2556 authored by Jacky Wang's avatar Jacky Wang
Browse files

[Catalyst] Refine MobileNetworkListScreenTest

Bug: 368355368
Flag: TEST_ONLY
Test: atest
Change-Id: I9e6f0200671bfc64c2c3b41c9b6f2c8669ddd497
parent eaaa0b34
Loading
Loading
Loading
Loading
+20 −3
Original line number Diff line number Diff line
@@ -15,14 +15,20 @@
 */
package com.android.settings.network

import androidx.test.ext.junit.runners.AndroidJUnit4
import android.content.Context
import android.platform.test.annotations.DisableFlags
import android.telephony.SubscriptionInfo
import android.telephony.SubscriptionManager
import androidx.test.core.app.ApplicationProvider
import com.android.settings.flags.Flags
import com.android.settingslib.preference.CatalystScreenTestCase
import com.google.common.truth.Truth.assertThat
import org.junit.Test
import org.junit.runner.RunWith
import org.mockito.kotlin.mock
import org.robolectric.Shadows.shadowOf
import org.robolectric.annotation.Config
import org.robolectric.shadows.ShadowSubscriptionManager

@RunWith(AndroidJUnit4::class)
class MobileNetworkListScreenTest : CatalystScreenTestCase() {
    override val preferenceScreenCreator = MobileNetworkListScreen()

@@ -33,4 +39,15 @@ class MobileNetworkListScreenTest : CatalystScreenTestCase() {
    fun key() {
        assertThat(preferenceScreenCreator.key).isEqualTo(MobileNetworkListScreen.KEY)
    }

    @DisableFlags(Flags.FLAG_IS_DUAL_SIM_ONBOARDING_ENABLED)
    @Config(shadows = [ShadowSubscriptionManager::class])
    override fun migration() {
        val context: Context = ApplicationProvider.getApplicationContext()
        val subscriptionManager =
            shadowOf(context.getSystemService(SubscriptionManager::class.java))
        val subscriptionInfo: SubscriptionInfo = mock()
        subscriptionManager.setAvailableSubscriptionInfos(subscriptionInfo)
        super.migration()
    }
}