From aa117aac847d05ef36a781a63d369edd2d0f5931 Mon Sep 17 00:00:00 2001
From: Kweku Adams <kwekua@google.com>
Date: Wed, 22 Mar 2023 18:32:22 +0000
Subject: [PATCH] Exit Doze for tests.

Exit device idle state so that job tests aren't affected by the state.

Bug: 274150342
Test: atest --rerun-until-failure 50 com.android.server.job.BiasSchedulingTest
Change-Id: I6935dafd154c72e7fa74b60f9c669d22e0f360b8
---
 .../src/com/android/server/job/BiasSchedulingTest.java          | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/services/tests/servicestests/src/com/android/server/job/BiasSchedulingTest.java b/services/tests/servicestests/src/com/android/server/job/BiasSchedulingTest.java
index 56d01b0e3a2ab..5e7dc33a1e520 100644
--- a/services/tests/servicestests/src/com/android/server/job/BiasSchedulingTest.java
+++ b/services/tests/servicestests/src/com/android/server/job/BiasSchedulingTest.java
@@ -21,6 +21,7 @@ import android.app.job.JobInfo;
 import android.app.job.JobScheduler;
 import android.content.ComponentName;
 import android.content.Context;
+import android.os.DeviceIdleManager;
 import android.test.AndroidTestCase;
 
 import com.android.server.job.MockBiasJobService.TestEnvironment;
@@ -48,6 +49,7 @@ public class BiasSchedulingTest extends AndroidTestCase {
         sJobServiceComponent = new ComponentName(getContext(), MockBiasJobService.class);
         mJobScheduler = (JobScheduler) getContext().getSystemService(Context.JOB_SCHEDULER_SERVICE);
         mJobScheduler.cancelAll();
+        getContext().getSystemService(DeviceIdleManager.class).endIdle("BiasSchedulingTest");
     }
 
     @Override