From 1b8a100c7a9bbda40fa78c3f3dfcff13562e9500 Mon Sep 17 00:00:00 2001
From: ykhung <ykhung@google.com>
Date: Thu, 1 Apr 2021 13:34:48 +0800
Subject: [PATCH] Fix testing fail for BatteryEntryTest from patch ag/14008232

Bug: 184138623
Test: make RunSettingsRoboTests
Test: make RunSettingsGoogleRoboTests
Change-Id: Ic770b09cc303a9f213e31174f279940c2aa1f87d
---
 .../src/com/android/settings/fuelgauge/BatteryEntryTest.java    | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/robotests/src/com/android/settings/fuelgauge/BatteryEntryTest.java b/tests/robotests/src/com/android/settings/fuelgauge/BatteryEntryTest.java
index 6858579479b..c1bafca6489 100644
--- a/tests/robotests/src/com/android/settings/fuelgauge/BatteryEntryTest.java
+++ b/tests/robotests/src/com/android/settings/fuelgauge/BatteryEntryTest.java
@@ -20,6 +20,7 @@ import static com.google.common.truth.Truth.assertThat;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.anyString;
 import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.doReturn;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
@@ -235,6 +236,7 @@ public class BatteryEntryTest {
 
     @Test
     public void getKey_UserBatteryConsumer_returnUserId() {
+        doReturn(mockUserManager).when(mMockContext).getSystemService(UserManager.class);
         final BatteryEntry entry = createUserBatteryConsumer(2);
         final String key = entry.getKey();
         assertThat(key).isEqualTo("U|2");