From e0500885f06f348f8875b22ec52624c1c4654c9a Mon Sep 17 00:00:00 2001
From: Jay Aliomer <aaliomer@google.com>
Date: Sun, 10 May 2020 20:45:25 -0400
Subject: [PATCH] Dark theme not turning on/off when in twilight mode

Fixes: 156254231
Test: UiModeManagerTest
Change-Id: I570319bfe0acabce418b071f21d75b760f5464a2
---
 .../core/java/com/android/server/UiModeManagerService.java | 7 ++++---
 .../src/com/android/server/UiModeManagerServiceTest.java   | 3 ++-
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/services/core/java/com/android/server/UiModeManagerService.java b/services/core/java/com/android/server/UiModeManagerService.java
index dd3de359a01e7..f42e32dad4d9e 100644
--- a/services/core/java/com/android/server/UiModeManagerService.java
+++ b/services/core/java/com/android/server/UiModeManagerService.java
@@ -165,9 +165,11 @@ final class UiModeManagerService extends SystemService {
     }
 
     @VisibleForTesting
-    protected UiModeManagerService(Context context, boolean setupWizardComplete) {
+    protected UiModeManagerService(Context context, boolean setupWizardComplete,
+            TwilightManager tm) {
         this(context);
         mSetupWizardComplete = setupWizardComplete;
+        mTwilightManager = tm;
     }
 
     private static Intent buildHomeIntent(String category) {
@@ -337,7 +339,6 @@ final class UiModeManagerService extends SystemService {
                 mAlarmManager = (AlarmManager) getContext().getSystemService(Context.ALARM_SERVICE);
                 mLocalPowerManager =
                         LocalServices.getService(PowerManagerInternal.class);
-                mTwilightManager = getLocalService(TwilightManager.class);
                 initPowerSave();
                 mCarModeEnabled = mDockState == Intent.EXTRA_DOCK_STATE_CAR;
                 registerVrStateListener();
@@ -380,7 +381,7 @@ final class UiModeManagerService extends SystemService {
                 com.android.internal.R.bool.config_enableCarDockHomeLaunch);
         mUiModeLocked = res.getBoolean(com.android.internal.R.bool.config_lockUiMode);
         mNightModeLocked = res.getBoolean(com.android.internal.R.bool.config_lockDayNightMode);
-
+        mTwilightManager = getLocalService(TwilightManager.class);
         final PackageManager pm = context.getPackageManager();
         mTelevision = pm.hasSystemFeature(PackageManager.FEATURE_TELEVISION)
                 || pm.hasSystemFeature(PackageManager.FEATURE_LEANBACK);
diff --git a/services/tests/uiservicestests/src/com/android/server/UiModeManagerServiceTest.java b/services/tests/uiservicestests/src/com/android/server/UiModeManagerServiceTest.java
index 69ef499749a9f..3062584aee203 100644
--- a/services/tests/uiservicestests/src/com/android/server/UiModeManagerServiceTest.java
+++ b/services/tests/uiservicestests/src/com/android/server/UiModeManagerServiceTest.java
@@ -146,7 +146,8 @@ public class UiModeManagerServiceTest extends UiServiceTestCase {
         addLocalService(PowerManagerInternal.class, mLocalPowerManager);
         addLocalService(TwilightManager.class, mTwilightManager);
         
-        mUiManagerService = new UiModeManagerService(mContext, true);
+        mUiManagerService = new UiModeManagerService(mContext, true,
+                mTwilightManager);
         try {
             mUiManagerService.onBootPhase(SystemService.PHASE_SYSTEM_SERVICES_READY);
         } catch (SecurityException e) {/* ignore for permission denial */}