Merge "De-flake scheduler tests." into udc-mainline-prod
diff --git a/tests/unittests/src/com/android/server/healthconnect/HealthConnectThreadSchedulerTest.java b/tests/unittests/src/com/android/server/healthconnect/HealthConnectThreadSchedulerTest.java
index fca5767..afcb889 100644
--- a/tests/unittests/src/com/android/server/healthconnect/HealthConnectThreadSchedulerTest.java
+++ b/tests/unittests/src/com/android/server/healthconnect/HealthConnectThreadSchedulerTest.java
@@ -57,6 +57,8 @@
     public void setUp() {
         MockitoAnnotations.initMocks(this);
 
+        HealthConnectThreadScheduler.resetThreadPools();
+
         mInternalTaskScheduler = HealthConnectThreadScheduler.sInternalBackgroundExecutor;
         mInternalTaskSchedulerCompletedJobs = mInternalTaskScheduler.getCompletedTaskCount();
         mControllerTaskScheduler = HealthConnectThreadScheduler.sControllerExecutor;
@@ -130,8 +132,6 @@
 
     @Test
     public void testHealthConnectSchedulerClear() {
-        HealthConnectThreadScheduler.resetThreadPools();
-        setUp();
         Truth.assertThat(mInternalTaskSchedulerCompletedJobs).isEqualTo(0);
         Truth.assertThat(mControllerTaskSchedulerCompletedJobs).isEqualTo(0);
         Truth.assertThat(mForegroundTaskSchedulerCompletedJobs).isEqualTo(0);
diff --git a/tests/unittests/src/com/android/server/healthconnect/storage/TestUtils.java b/tests/unittests/src/com/android/server/healthconnect/storage/TestUtils.java
index cee8798..21e86f2 100644
--- a/tests/unittests/src/com/android/server/healthconnect/storage/TestUtils.java
+++ b/tests/unittests/src/com/android/server/healthconnect/storage/TestUtils.java
@@ -30,7 +30,7 @@
             } catch (Exception e) {
                 // ignore
             } finally {
-                if (ChronoUnit.SECONDS.between(Instant.now(), startTime) > 1) {
+                if (ChronoUnit.SECONDS.between(startTime, Instant.now()) > 3) {
                     throw new TimeoutException();
                 }
             }