Merge "Add test for content inflation during row inflation" into rvc-dev am:...

import static android.app.NotificationManager.IMPORTANCE_DEFAULT;

import static com.android.systemui.statusbar.notification.row.NotificationRowContentBinder.FLAG_CONTENT_VIEW_HEADS_UP;

import static junit.framework.Assert.assertNotNull;

import static org.mockito.ArgumentMatchers.any;

    private NotificationRowBinderImpl mRowBinder;

    private Handler mHandler;

    private FakeExecutor mBgExecutor;

    private RowContentBindStage mRowContentBindStage;

    @Before

    public void setUp() {

        mHandler = Handler.createAsync(TestableLooper.get(this).getLooper());

        // Add an action so heads up content views are made

        Notification.Action action = new Notification.Action.Builder(null, null, null).build();

        Notification notification = new Notification.Builder(mContext)

                .setSmallIcon(R.drawable.ic_person)

                .setContentTitle(TEST_TITLE)

                .setContentText(TEST_TEXT)

                .setActions(action)

                .build();

        mSbn = new SbnBuilder()

                .setNotification(notification)

                () -> mock(SmartReplyController.class),

                mock(ConversationNotificationProcessor.class),

                mBgExecutor);

        RowContentBindStage stage = new RowContentBindStage(

        mRowContentBindStage = new RowContentBindStage(

                binder,

                mock(NotifInflationErrorManager.class),

                mock(RowContentBindStageLogger.class));

        pipeline.setStage(stage);

        pipeline.setStage(mRowContentBindStage);

        ArgumentCaptor<ExpandableNotificationRow> viewCaptor =

                ArgumentCaptor.forClass(ExpandableNotificationRow.class);

                                "FOOBAR", "FOOBAR",

                                mKeyguardBypassController,

                                mGroupManager,

                                stage,

                                mRowContentBindStage,

                                mock(NotificationLogger.class),

                                mHeadsUpManager,

                                mPresenter,

                mRemoteInputManager,

                mLockscreenUserManager,

                pipeline,

                stage,

                mRowContentBindStage,

                mNotificationInterruptionStateProvider,

                RowInflaterTask::new,

                mExpandableNotificationRowComponentBuilder,

        verify(mPresenter).updateNotificationViews();

    }

    @Test

    public void testContentViewInflationDuringRowInflationInflatesCorrectViews() {

        // GIVEN a notification is added and the row is inflating

        mEntryManager.addNotification(mSbn, mRankingMap);

        ArgumentCaptor<NotificationEntry> entryCaptor = ArgumentCaptor.forClass(

                NotificationEntry.class);

        verify(mEntryListener).onPendingEntryAdded(entryCaptor.capture());

        NotificationEntry entry = entryCaptor.getValue();

        // WHEN we try to bind a content view

        mRowContentBindStage.getStageParams(entry).requireContentViews(FLAG_CONTENT_VIEW_HEADS_UP);

        mRowContentBindStage.requestRebind(entry, null);

        waitForInflation();

        // THEN the notification has its row and all relevant content views inflated

        assertNotNull(entry.getRow());

        assertNotNull(entry.getRow().getPrivateLayout().getContractedChild());

        assertNotNull(entry.getRow().getPrivateLayout().getHeadsUpChild());

    }

    /**

     * Wait for inflation to finish.

     *