Merge "Fix testing fail for BatteryEntryTest from patch ag/14008232" into sc-dev

This commit is contained in:
YUKAI HUNG
2021-04-01 07:10:22 +00:00
committed by Android (Google) Code Review

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");