Merge "Remove NetworkShimImpl mock in TcpSocketTrackerTest"

import static androidx.test.platform.app.InstrumentationRegistry.getInstrumentation;

import static androidx.test.platform.app.InstrumentationRegistry.getInstrumentation;

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

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

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

import static com.android.net.module.util.netlink.NetlinkConstants.SOCKDIAG_MSG_HEADER_SIZE;

import static com.android.net.module.util.netlink.NetlinkConstants.SOCKDIAG_MSG_HEADER_SIZE;

import static junit.framework.Assert.assertEquals;

import static junit.framework.Assert.assertEquals;

import com.android.net.module.util.netlink.StructNlMsgHdr;

import com.android.net.module.util.netlink.StructNlMsgHdr;

import com.android.networkstack.apishim.ConstantsShim;

import com.android.networkstack.apishim.ConstantsShim;

import com.android.networkstack.apishim.NetworkShimImpl;

import com.android.testutils.DevSdkIgnoreRule;

import com.android.testutils.DevSdkIgnoreRule;

import com.android.testutils.DevSdkIgnoreRule.IgnoreAfter;

import com.android.testutils.DevSdkIgnoreRule.IgnoreAfter;

import com.android.testutils.DevSdkIgnoreRule.IgnoreUpTo;

import com.android.testutils.DevSdkIgnoreRule.IgnoreUpTo;

import org.junit.runner.RunWith;

import org.junit.runner.RunWith;

import org.mockito.Mock;

import org.mockito.Mock;

import org.mockito.MockitoAnnotations;

import org.mockito.MockitoAnnotations;

import org.mockito.MockitoSession;

import org.mockito.quality.Strictness;

import java.io.FileDescriptor;

import java.io.FileDescriptor;

import java.nio.ByteBuffer;

import java.nio.ByteBuffer;

    @Mock private INetd mNetd;

    @Mock private INetd mNetd;

    private final Network mNetwork = new Network(TEST_NETID1);

    private final Network mNetwork = new Network(TEST_NETID1);

    private final Network mOtherNetwork = new Network(TEST_NETID2);

    private final Network mOtherNetwork = new Network(TEST_NETID2);

    private MockitoSession mSession;

    private TerribleFailureHandler mOldWtfHandler;

    private TerribleFailureHandler mOldWtfHandler;

    @Rule

    @Rule

                eq(NAMESPACE_CONNECTIVITY),

                eq(NAMESPACE_CONNECTIVITY),

                eq(CONFIG_TCP_PACKETS_FAIL_PERCENTAGE),

                eq(CONFIG_TCP_PACKETS_FAIL_PERCENTAGE),

                anyInt())).thenReturn(DEFAULT_TCP_PACKETS_FAIL_PERCENTAGE);

                anyInt())).thenReturn(DEFAULT_TCP_PACKETS_FAIL_PERCENTAGE);

        mSession = mockitoSession()

                .spyStatic(NetworkShimImpl.class)

                .strictness(Strictness.WARN)

                .startMocking();

        when(mNetd.getFwmarkForNetwork(eq(TEST_NETID1)))

        when(mNetd.getFwmarkForNetwork(eq(TEST_NETID1)))

                .thenReturn(makeMarkMaskParcel(NETID_MASK, TEST_NETID1_FWMARK));

                .thenReturn(makeMarkMaskParcel(NETID_MASK, TEST_NETID1_FWMARK));

    @After

    @After

    public void tearDown() {

    public void tearDown() {

        mSession.finishMocking();

        Log.setWtfHandler(mOldWtfHandler);

        Log.setWtfHandler(mOldWtfHandler);

    }

    }