Fix testIpClientServiceAvailableAfterPolling am: eb138e5d am: 6bcd214f am: 3c928c5d

import android.os.IBinder

import android.os.IBinder

import androidx.test.ext.junit.runners.AndroidJUnit4

import androidx.test.ext.junit.runners.AndroidJUnit4

import androidx.test.filters.SmallTest

import androidx.test.filters.SmallTest

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

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

import com.android.testutils.DevSdkIgnoreRule

import com.android.testutils.DevSdkIgnoreRule

import org.junit.After

import org.junit.After

import org.junit.Before

import org.junit.Before

import org.junit.runner.RunWith

import org.junit.runner.RunWith

import org.mockito.ArgumentMatchers.any

import org.mockito.ArgumentMatchers.any

import org.mockito.Mock

import org.mockito.Mock

import org.mockito.Mockito.doReturn

import org.mockito.Mockito.never

import org.mockito.Mockito.never

import org.mockito.Mockito.timeout

import org.mockito.Mockito.timeout

import org.mockito.Mockito.verify

import org.mockito.Mockito.verify

    @Test

    @Test

    fun testIpClientServiceAvailableAfterPolling() {

    fun testIpClientServiceAvailableAfterPolling() {

        // Force NetworkStack.getService() to return null: this cannot be done with

        // setServiceForTest, as passing null just restores default behavior.

        val session = mockitoSession().spyStatic(NetworkStack::class.java).startMocking()

        doReturn(null).`when` { NetworkStack.getService() }

        ModuleNetworkStackClient.getInstance(mContext).makeIpClient(TEST_IFNAME, mIpClientCb)

        ModuleNetworkStackClient.getInstance(mContext).makeIpClient(TEST_IFNAME, mIpClientCb)

        Thread.sleep(TEST_TIMEOUT_MS)

        Thread.sleep(TEST_TIMEOUT_MS)

        verify(mConnector, never()).makeIpClient(any(), any())

        verify(mConnector, never()).makeIpClient(any(), any())

        NetworkStack.setServiceForTest(mConnectorBinder)

        NetworkStack.setServiceForTest(mConnectorBinder)

        // Restore behavior of NetworkStack to return what was set in setServiceForTest

        session.finishMocking()

        // Use a longer timeout as polling can cause larger delays

        // Use a longer timeout as polling can cause larger delays

        verify(mConnector, timeout(TEST_TIMEOUT_MS * 4)).makeIpClient(TEST_IFNAME, mIpClientCb)

        verify(mConnector, timeout(TEST_TIMEOUT_MS * 4)).makeIpClient(TEST_IFNAME, mIpClientCb)

    }

    }