Fix testing fail for BatteryEntryTest from patch ag/14008232

Bug: 184138623
Test: make RunSettingsRoboTests
Test: make RunSettingsGoogleRoboTests
Change-Id: Ic770b09cc303a9f213e31174f279940c2aa1f87d
This commit is contained in:
ykhung
2021-04-01 13:34:48 +08:00
parent e3e3fc915a
commit 1b8a100c7a

View File

@@ -20,6 +20,7 @@ import static com.google.common.truth.Truth.assertThat;
import static org.mockito.ArgumentMatchers.any; import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyString; import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.ArgumentMatchers.eq; import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock; import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when; import static org.mockito.Mockito.when;
@@ -235,6 +236,7 @@ public class BatteryEntryTest {
@Test @Test
public void getKey_UserBatteryConsumer_returnUserId() { public void getKey_UserBatteryConsumer_returnUserId() {
doReturn(mockUserManager).when(mMockContext).getSystemService(UserManager.class);
final BatteryEntry entry = createUserBatteryConsumer(2); final BatteryEntry entry = createUserBatteryConsumer(2);
final String key = entry.getKey(); final String key = entry.getKey();
assertThat(key).isEqualTo("U|2"); assertThat(key).isEqualTo("U|2");