diff --git a/tests/unit/java/com/android/server/connectivity/AutomaticOnOffKeepaliveTrackerTest.java b/tests/unit/java/com/android/server/connectivity/AutomaticOnOffKeepaliveTrackerTest.java index 1b71fb93bf..986c389b60 100644 --- a/tests/unit/java/com/android/server/connectivity/AutomaticOnOffKeepaliveTrackerTest.java +++ b/tests/unit/java/com/android/server/connectivity/AutomaticOnOffKeepaliveTrackerTest.java @@ -367,6 +367,10 @@ public class AutomaticOnOffKeepaliveTrackerTest { @After public void teardown() throws Exception { TestKeepaliveInfo.closeAllSockets(); + if (mHandlerThread != null) { + mHandlerThread.quitSafely(); + mHandlerThread.join(); + } } private final class AOOTestHandler extends Handler { diff --git a/tests/unit/java/com/android/server/connectivity/KeepaliveStatsTrackerTest.java b/tests/unit/java/com/android/server/connectivity/KeepaliveStatsTrackerTest.java index fa703eb93a..90a0eddc0a 100644 --- a/tests/unit/java/com/android/server/connectivity/KeepaliveStatsTrackerTest.java +++ b/tests/unit/java/com/android/server/connectivity/KeepaliveStatsTrackerTest.java @@ -67,6 +67,7 @@ import com.android.testutils.DevSdkIgnoreRule; import com.android.testutils.DevSdkIgnoreRunner; import com.android.testutils.HandlerUtils; +import org.junit.After; import org.junit.Before; import org.junit.Rule; import org.junit.Test; @@ -240,6 +241,14 @@ public class KeepaliveStatsTrackerTest { HandlerUtils.waitForIdle(mTestHandler, TIMEOUT_MS); } + @After + public void tearDown() throws Exception { + if (mHandlerThread != null) { + mHandlerThread.quitSafely(); + mHandlerThread.join(); + } + } + private void setElapsedRealtime(long time) { doReturn(time).when(mDependencies).getElapsedRealtime(); }