Merge "Fix AppInfoWithHeaderTest" into main

import static com.google.common.truth.Truth.assertThat;

import static org.mockito.ArgumentMatchers.any;

import static org.mockito.ArgumentMatchers.eq;

import static org.mockito.Mockito.mock;

import static org.mockito.Mockito.verify;

import static org.mockito.Mockito.when;

import org.junit.After;

import org.junit.Before;

import org.junit.Ignore;

import org.junit.Rule;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Answers;

import org.mockito.Mock;

import org.mockito.MockitoAnnotations;

import org.mockito.junit.MockitoJUnit;

import org.mockito.junit.MockitoRule;

import org.robolectric.RobolectricTestRunner;

import org.robolectric.RuntimeEnvironment;

import org.robolectric.annotation.Config;

@RunWith(RobolectricTestRunner.class)

@Config(shadows = {ShadowEntityHeaderController.class, ShadowSettingsLibUtils.class})

public class AppInfoWithHeaderTest {

    @Rule

    public final MockitoRule mMockitoRule = MockitoJUnit.rule();

    @Mock(answer = Answers.RETURNS_DEEP_STUBS)

    private EntityHeaderController mHeaderController;

    @Before

    public void setUp() {

        MockitoAnnotations.initMocks(this);

        mFactory = FakeFeatureFactory.setupForTest();

        when(mFactory.metricsFeatureProvider.getMetricsCategory(any(Object.class)))

                .thenReturn(MetricsProto.MetricsEvent.SETTINGS_APP_NOTIF_CATEGORY);

        assertThat(mAppInfoWithHeader.mPackageRemovedCalled).isTrue();

    }

    @Ignore("b/315135755")

    @Test

    public void noExtraUserHandleInIntent_retrieveAppEntryWithMyUserId()

            throws PackageManager.NameNotFoundException {

        when(mAppInfoWithHeader.mState.getEntry(packageName,

                UserHandle.myUserId())).thenReturn(entry);

        when(mAppInfoWithHeader.mPm.getPackageInfoAsUser(entry.info.packageName,

                PackageManager.MATCH_DISABLED_COMPONENTS |

                        PackageManager.GET_SIGNING_CERTIFICATES |

                        PackageManager.GET_PERMISSIONS, UserHandle.myUserId())).thenReturn(

        when(mAppInfoWithHeader.mPm.getPackageInfoAsUser(eq(entry.info.packageName),

                any(), eq(UserHandle.myUserId()))).thenReturn(

                mAppInfoWithHeader.mPackageInfo);

        mAppInfoWithHeader.retrieveAppEntry();

        assertThat(mAppInfoWithHeader.mAppEntry).isNotNull();

    }

    @Ignore("b/315135755")

    @Test

    public void extraUserHandleInIntent_retrieveAppEntryWithMyUserId()

            throws PackageManager.NameNotFoundException {

        entry.info.packageName = packageName;

        when(mAppInfoWithHeader.mState.getEntry(packageName, USER_ID)).thenReturn(entry);

        when(mAppInfoWithHeader.mPm.getPackageInfoAsUser(entry.info.packageName,

                PackageManager.MATCH_DISABLED_COMPONENTS |

                        PackageManager.GET_SIGNING_CERTIFICATES |

                        PackageManager.GET_PERMISSIONS, USER_ID)).thenReturn(

        when(mAppInfoWithHeader.mPm.getPackageInfoAsUser(eq(entry.info.packageName),

                any(), eq(USER_ID))).thenReturn(

                mAppInfoWithHeader.mPackageInfo);

        mAppInfoWithHeader.retrieveAppEntry();

        }

        @Override

        protected Intent getIntent() { return mIntent; }

        protected Intent getIntent() {

            return mIntent;

        }

    }

}