Merge "Fix b/270656372 unit test failure." into udc-dev

import org.mockito.MockitoAnnotations;

import org.robolectric.RobolectricTestRunner;

import org.robolectric.RuntimeEnvironment;

import org.robolectric.Shadows;

import java.util.ArrayList;

import java.util.List;

    private Context mContext;

    @Mock

    private PackageManager mPackageManager;

    @Mock private PackageManager mPackageManager;

    @Mock private UserManager mUserManager;

    @Mock private ContentResolver mMockContentResolver;

    @Mock private ContentResolver mMockContentResolver2;

    @Mock private BatteryUsageStats mBatteryUsageStats;

        doReturn("com.fake.package").when(mContext).getPackageName();

        doReturn(mMockContext).when(mContext).createPackageContextAsUser(

                "com.fake.package", /*flags=*/ 0, UserHandle.OWNER);

        BatteryTestUtils.setWorkProfile(mContext);

        doReturn(UserHandle.CURRENT).when(mContext).getUser();

        doReturn(mUserManager).when(mContext).getSystemService(UserManager.class);

        doReturn(true).when(mUserManager).isManagedProfile();

        doReturn(UserHandle.SYSTEM).when(mUserManager).getProfileParent(UserHandle.CURRENT);

        DatabaseUtils.sFakeBatteryStateSupplier = () -> getMatrixCursor();

        final Map<Long, Map<String, BatteryHistEntry>> batteryHistMap =