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();
}
}