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

Commit ecde596d authored by Etan Cohen's avatar Etan Cohen Committed by Android (Google) Code Review
Browse files

Merge "[Provider Model][SysUI] Mock global provider flag"

parents 3b9dd183 5487e9e8

packages/SystemUI/tests/src/com/android/systemui/statusbar/policy/NetworkControllerBaseTest.java

+13 −2
Original line number Diff line number Diff line
@@ -62,6 +62,7 @@ import android.util.Log; 


import androidx.test.InstrumentationRegistry;



import com.android.dx.mockito.inline.extended.ExtendedMockito;

import com.android.settingslib.R;

import com.android.settingslib.graph.SignalDrawable;

import com.android.settingslib.mobile.MobileMappings.Config;
@@ -82,6 +83,8 @@ import org.junit.rules.TestWatcher; 
import org.junit.runner.Description;

import org.mockito.ArgumentCaptor;

import org.mockito.Mockito;

import org.mockito.MockitoSession;

import org.mockito.quality.Strictness;



import java.io.PrintWriter;

import java.io.StringWriter;
@@ -126,6 +129,8 @@ public class NetworkControllerBaseTest extends SysuiTestCase { 
    private ConnectivityManager.NetworkCallback mDefaultCallbackInNetworkController;

    private ConnectivityManager.NetworkCallback mNetworkCallback;



    MockitoSession mMockingSession = null;



    @Rule

    public TestWatcher failWatcher = new TestWatcher() {

        @Override
@@ -145,7 +150,11 @@ public class NetworkControllerBaseTest extends SysuiTestCase { 


    @Before

    public void setUp() throws Exception {

        FeatureFlagUtils.setEnabled(mContext, FeatureFlagUtils.SETTINGS_PROVIDER_MODEL, true);

        mMockingSession = ExtendedMockito.mockitoSession().strictness(Strictness.LENIENT)

                .mockStatic(FeatureFlagUtils.class).startMocking();

        ExtendedMockito.doReturn(true).when(() -> FeatureFlagUtils.isEnabled(mContext,

                FeatureFlagUtils.SETTINGS_PROVIDER_MODEL));



        mInstrumentation = InstrumentationRegistry.getInstrumentation();

        Settings.Global.putInt(mContext.getContentResolver(), Global.AIRPLANE_MODE_ON, 0);

        TestableResources res = mContext.getOrCreateTestableResources();
@@ -227,7 +236,9 @@ public class NetworkControllerBaseTest extends SysuiTestCase { 


    @After

    public void tearDown() throws Exception {

        FeatureFlagUtils.setEnabled(mContext, FeatureFlagUtils.SETTINGS_PROVIDER_MODEL, false);

        if (mMockingSession != null) {

            mMockingSession.finishMocking();

        }

    }



    protected void setupNetworkController() {