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

Commit 876897fb authored by Treehugger Robot's avatar Treehugger Robot Committed by Automerger Merge Worker
Browse files

Merge "Finish mocking in finally clause" am: 6385740b am: 9620fa37 am: 800ccac4

Change-Id: I5383d3f736ed45bd1d4b73ce53b43ec4626288c5
parents 61c81b2a 800ccac4
Loading
Loading
Loading
Loading
+10 −8
Original line number Original line Diff line number Diff line
@@ -95,15 +95,17 @@ class ModuleNetworkStackClientTest {
        // Force NetworkStack.getService() to return null: this cannot be done with
        // Force NetworkStack.getService() to return null: this cannot be done with
        // setServiceForTest, as passing null just restores default behavior.
        // setServiceForTest, as passing null just restores default behavior.
        val session = mockitoSession().spyStatic(NetworkStack::class.java).startMocking()
        val session = mockitoSession().spyStatic(NetworkStack::class.java).startMocking()
        try {
            doReturn(null).`when` { NetworkStack.getService() }
            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)

        } finally {
            // Restore behavior of NetworkStack to return what was set in setServiceForTest
            // Restore behavior of NetworkStack to return what was set in setServiceForTest
            session.finishMocking()
            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)