/*
 * Copyright (C) 2023 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.android.car.carlauncher.recents.view;

import static androidx.recyclerview.widget.RecyclerView.SCROLL_STATE_IDLE;

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

import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import android.content.Context;
import android.content.pm.ApplicationInfo;
import android.graphics.Rect;
import android.testing.TestableContext;
import android.util.LayoutDirection;
import android.view.View;
import android.view.WindowMetrics;
import android.widget.Button;
import android.widget.ImageView;

import androidx.recyclerview.widget.GridLayoutManager;
import androidx.recyclerview.widget.RecyclerView;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import androidx.test.platform.app.InstrumentationRegistry;

import com.android.car.carlauncher.R;
import com.android.car.carlauncher.recents.RecentTasksViewModel;

import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;
import org.mockito.Captor;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

@RunWith(AndroidJUnit4.class)
public class RecentsRecyclerViewTest {
    private static final int WINDOW_WIDTH = 1000;
    private static final int FIRST_TASK_WIDTH = 800;
    private static final int TASK_WIDTH = 300;
    private static final int COL_SPACING_BETWEEN_TASKS = 50;
    private static final int CURRENT_ADAPTER_POSITION = 2;
    private static final int PREVIOUS_ADAPTER_POSITION = 1;
    private static final int NEXT_ADAPTER_POSITION = 3;
    private RecentsRecyclerView mRecentsRecyclerView;

    @Mock
    private RecentTasksViewModel mRecentTasksViewModel;
    @Mock
    private WindowMetrics mWindowMetrics;
    @Mock
    private GridLayoutManager mGridLayoutManager;
    @Mock
    private View mCurrentTaskView;
    @Mock
    private ImageView mCurrentThumbnailView;
    @Mock
    private View mCurrentDismissView;
    @Mock
    private View mNextTaskView;
    @Mock
    private ImageView mNextThumbnailView;
    @Mock
    private View mNextDismissView;
    @Mock
    private View mPreviousTaskView;
    @Mock
    private ImageView mPreviousThumbnailView;
    @Mock
    private View mPreviousDismissView;
    @Mock
    private View mClearAllView;
    @Mock
    private Button mClearAllButtonView;
    @Mock
    private View.OnClickListener mClearAllButtonClickListener;

    @Captor
    ArgumentCaptor<RecyclerView.OnScrollListener> mOnScrollListenerCaptor;

    private BaseTaskViewHolder mCurrentTaskViewHolder;
    private BaseTaskViewHolder mNextTaskViewHolder;
    private BaseTaskViewHolder mPreviousTaskViewHolder;
    private ClearAllViewHolder mClearAllTaskViewHolder;

    @Rule
    public final TestableContext mContext = new TestableContext(
            InstrumentationRegistry.getInstrumentation().getTargetContext()) {
        @Override
        public Context createApplicationContext(ApplicationInfo application, int flags) {
            return this;
        }
    };

    @Before
    public void setup() {
        MockitoAnnotations.initMocks(this);
        initViews();
        Rect windowBounds = new Rect(/* left= */ 0, /* top= */ 0,
                /* right= */ WINDOW_WIDTH, /* bottom= */ 0);
        when(mWindowMetrics.getBounds()).thenReturn(windowBounds);
        mContext.getOrCreateTestableResources().addOverride(R.dimen.recent_task_width_first,
                /* value= */ FIRST_TASK_WIDTH);
        mContext.getOrCreateTestableResources().addOverride(R.dimen.recent_task_width,
                /* value= */ TASK_WIDTH);
        mContext.getOrCreateTestableResources().addOverride(R.dimen.recent_task_col_space,
                /* value= */ COL_SPACING_BETWEEN_TASKS);

        mRecentsRecyclerView = new RecentsRecyclerView(mContext, mRecentTasksViewModel,
                mWindowMetrics);
        mRecentsRecyclerView = spy(mRecentsRecyclerView);
        mRecentsRecyclerView.setLayoutManager(mGridLayoutManager);
    }

    @Test
    public void resetPadding_setsStartPadding_toZero_whenNoRecentTasksPresent() {
        when(mRecentTasksViewModel.getRecentTasksSize()).thenReturn(0);

        mRecentsRecyclerView.resetPadding();

        assertThat(mRecentsRecyclerView.getPaddingStart()).isEqualTo(0);
    }


    @Test
    public void resetPadding_setsEndPadding_toZero_whenNoRecentTasksPresent() {
        when(mRecentTasksViewModel.getRecentTasksSize()).thenReturn(0);

        mRecentsRecyclerView.resetPadding();

        assertThat(mRecentsRecyclerView.getPaddingEnd()).isEqualTo(0);
    }

    @Test
    public void resetPadding_setsStartPadding_toStart_noReverseLayout_noRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mRecentTasksViewModel.getRecentTasksSize()).thenReturn(5);

        int startPadding = mRecentsRecyclerView.calculateFirstItemPadding(WINDOW_WIDTH);
        mRecentsRecyclerView.resetPadding();

        assertThat(mRecentsRecyclerView.getPaddingStart()).isEqualTo(startPadding);
    }

    @Test
    public void resetPadding_setsEndPadding_toEnd_noReverseLayout_noRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mRecentTasksViewModel.getRecentTasksSize()).thenReturn(5);

        int endPadding = mRecentsRecyclerView.calculateLastItemPadding();
        mRecentsRecyclerView.resetPadding();

        assertThat(mRecentsRecyclerView.getPaddingEnd()).isEqualTo(endPadding);
    }

    @Test
    public void resetPadding_setsStartPadding_toEnd_noReverseLayout_RTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.RTL);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mRecentTasksViewModel.getRecentTasksSize()).thenReturn(5);

        int startPadding = mRecentsRecyclerView.calculateFirstItemPadding(WINDOW_WIDTH);
        mRecentsRecyclerView.resetPadding();

        assertThat(mRecentsRecyclerView.getPaddingEnd()).isEqualTo(startPadding);
    }

    @Test
    public void resetPadding_setsEndPadding_toStart_noReverseLayout_RTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.RTL);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mRecentTasksViewModel.getRecentTasksSize()).thenReturn(5);

        int endPadding = mRecentsRecyclerView.calculateLastItemPadding();
        mRecentsRecyclerView.resetPadding();

        assertThat(mRecentsRecyclerView.getPaddingStart()).isEqualTo(endPadding);
    }

    @Test
    public void resetPadding_setsStartPadding_toEnd_reverseLayout_noRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(true);
        when(mRecentTasksViewModel.getRecentTasksSize()).thenReturn(5);

        int startPadding = mRecentsRecyclerView.calculateFirstItemPadding(WINDOW_WIDTH);
        mRecentsRecyclerView.resetPadding();

        assertThat(mRecentsRecyclerView.getPaddingEnd()).isEqualTo(startPadding);
    }

    @Test
    public void resetPadding_setsEndPadding_toStart_reverseLayout_noRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(true);
        when(mRecentTasksViewModel.getRecentTasksSize()).thenReturn(5);

        int endPadding = mRecentsRecyclerView.calculateLastItemPadding();
        mRecentsRecyclerView.resetPadding();

        assertThat(mRecentsRecyclerView.getPaddingStart()).isEqualTo(endPadding);
    }

    @Test
    public void resetPadding_setsStartPadding_toStart_reverseLayout_RTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mRecentTasksViewModel.getRecentTasksSize()).thenReturn(5);

        int startPadding = mRecentsRecyclerView.calculateFirstItemPadding(WINDOW_WIDTH);
        mRecentsRecyclerView.resetPadding();

        assertThat(mRecentsRecyclerView.getPaddingStart()).isEqualTo(startPadding);
    }

    @Test
    public void resetPadding_setsEndPadding_toEnd_reverseLayout_RTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mRecentTasksViewModel.getRecentTasksSize()).thenReturn(5);

        int endPadding = mRecentsRecyclerView.calculateLastItemPadding();
        mRecentsRecyclerView.resetPadding();

        assertThat(mRecentsRecyclerView.getPaddingEnd()).isEqualTo(endPadding);
    }

    @Test
    public void focusSearch_dismissFocused_forward_sameViewThumbnailFocus_noReverseLayout_NoRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mCurrentDismissView.isFocused()).thenReturn(true);
        doReturn(mCurrentTaskViewHolder).when(mRecentsRecyclerView).findContainingViewHolder(
                mCurrentDismissView);

        View nextFocusedView = mRecentsRecyclerView.focusSearch(mCurrentDismissView,
                View.FOCUS_FORWARD);

        assertThat(nextFocusedView).isEqualTo(mCurrentThumbnailView);
    }

    @Test
    public void focusSearch_thumbnailFocused_forward_nextViewDismissFocus_noReverseLayout_NoRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mCurrentThumbnailView.isFocused()).thenReturn(true);
        doReturn(mCurrentTaskViewHolder).when(mRecentsRecyclerView).findContainingViewHolder(
                mCurrentThumbnailView);
        doReturn(mNextTaskViewHolder).when(mRecentsRecyclerView).findViewHolderForAdapterPosition(
                NEXT_ADAPTER_POSITION);

        View nextFocusedView = mRecentsRecyclerView.focusSearch(mCurrentThumbnailView,
                View.FOCUS_FORWARD);

        assertThat(nextFocusedView).isEqualTo(mNextDismissView);
    }

    @Test
    public void focusSearch_dismissFocused_backward_sameViewThumbnailFocus_reverseLayout_NoRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(true);
        when(mCurrentDismissView.isFocused()).thenReturn(true);
        doReturn(mCurrentTaskViewHolder).when(mRecentsRecyclerView).findContainingViewHolder(
                mCurrentDismissView);

        View nextFocusedView = mRecentsRecyclerView.focusSearch(mCurrentDismissView,
                View.FOCUS_BACKWARD);

        assertThat(nextFocusedView).isEqualTo(mCurrentThumbnailView);
    }

    @Test
    public void focusSearch_thumbnailFocused_backward_nextViewDismissFocus_reverseLayout_NoRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(true);
        when(mCurrentThumbnailView.isFocused()).thenReturn(true);
        doReturn(mCurrentTaskViewHolder).when(mRecentsRecyclerView).findContainingViewHolder(
                mCurrentThumbnailView);
        doReturn(mNextTaskViewHolder).when(mRecentsRecyclerView).findViewHolderForAdapterPosition(
                NEXT_ADAPTER_POSITION);

        View nextFocusedView = mRecentsRecyclerView.focusSearch(mCurrentThumbnailView,
                View.FOCUS_BACKWARD);

        assertThat(nextFocusedView).isEqualTo(mNextDismissView);
    }

    @Test
    public void focusSearch_thumbnailFocused_backward_sameViewDismissFocus_noReverseLayout_NoRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mCurrentThumbnailView.isFocused()).thenReturn(true);
        doReturn(mCurrentTaskViewHolder).when(mRecentsRecyclerView).findContainingViewHolder(
                mCurrentThumbnailView);

        View nextFocusedView = mRecentsRecyclerView.focusSearch(mCurrentThumbnailView,
                View.FOCUS_BACKWARD);

        assertThat(nextFocusedView).isEqualTo(mCurrentDismissView);
    }

    @Test
    public void focusSearch_dismissFocused_backward_prevViewThumbnailFocus_noReverseLayout_NoRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mCurrentDismissView.isFocused()).thenReturn(true);
        doReturn(mCurrentTaskViewHolder).when(mRecentsRecyclerView).findContainingViewHolder(
                mCurrentDismissView);
        doReturn(mPreviousTaskViewHolder).when(
                mRecentsRecyclerView).findViewHolderForAdapterPosition(
                PREVIOUS_ADAPTER_POSITION);

        View nextFocusedView = mRecentsRecyclerView.focusSearch(mCurrentDismissView,
                View.FOCUS_BACKWARD);

        assertThat(nextFocusedView).isEqualTo(mPreviousThumbnailView);
    }

    @Test
    public void focusSearch_thumbnailFocused_forward_sameViewDismissFocus_reverseLayout_NoRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(true);
        when(mCurrentThumbnailView.isFocused()).thenReturn(true);
        doReturn(mCurrentTaskViewHolder).when(mRecentsRecyclerView).findContainingViewHolder(
                mCurrentThumbnailView);

        View nextFocusedView = mRecentsRecyclerView.focusSearch(mCurrentThumbnailView,
                View.FOCUS_FORWARD);

        assertThat(nextFocusedView).isEqualTo(mCurrentDismissView);
    }

    @Test
    public void focusSearch_dismissFocused_forward_prevViewThumbnailFocus_reverseLayout_NoRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(true);
        when(mCurrentDismissView.isFocused()).thenReturn(true);
        doReturn(mCurrentTaskViewHolder).when(mRecentsRecyclerView).findContainingViewHolder(
                mCurrentDismissView);
        doReturn(mPreviousTaskViewHolder).when(
                mRecentsRecyclerView).findViewHolderForAdapterPosition(
                PREVIOUS_ADAPTER_POSITION);

        View nextFocusedView = mRecentsRecyclerView.focusSearch(mCurrentDismissView,
                View.FOCUS_FORWARD);

        assertThat(nextFocusedView).isEqualTo(mPreviousThumbnailView);
    }

    @Test
    public void focusSearch_nextViewNotFound_scrollOnce_nextViewFound_focusRequested() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mCurrentThumbnailView.isFocused()).thenReturn(true);
        doReturn(mCurrentTaskViewHolder).when(mRecentsRecyclerView).findContainingViewHolder(
                mCurrentThumbnailView);
        doReturn(null).when(mRecentsRecyclerView).findViewHolderForAdapterPosition(
                NEXT_ADAPTER_POSITION);

        View nextFocusedView = mRecentsRecyclerView.focusSearch(mCurrentThumbnailView,
                View.FOCUS_FORWARD);

        // since view is not loaded, no focused view is returned
        assertThat(nextFocusedView).isEqualTo(null);
        verify(mRecentsRecyclerView, times(1)).smoothScrollBy(anyInt(), eq(0));
        doReturn(mNextTaskViewHolder).when(mRecentsRecyclerView)
                .findViewHolderForAdapterPosition(NEXT_ADAPTER_POSITION);
        verify(mRecentsRecyclerView).addOnScrollListener(mOnScrollListenerCaptor.capture());
        mOnScrollListenerCaptor.getValue().onScrollStateChanged(mRecentsRecyclerView,
                SCROLL_STATE_IDLE);
        verify(mNextDismissView).requestFocus();
    }

    @Test
    public void focusSearch_nextViewNotFound_scrollOnce_nextViewNotFound_noMoreScroll() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mCurrentThumbnailView.isFocused()).thenReturn(true);
        doReturn(mCurrentTaskViewHolder).when(mRecentsRecyclerView).findContainingViewHolder(
                mCurrentThumbnailView);
        doReturn(null).when(mRecentsRecyclerView).findViewHolderForAdapterPosition(
                NEXT_ADAPTER_POSITION);

        View nextFocusedView = mRecentsRecyclerView.focusSearch(mCurrentThumbnailView,
                View.FOCUS_FORWARD);

        // since view is not loaded, no focused view is returned
        assertThat(nextFocusedView).isEqualTo(null);
        verify(mRecentsRecyclerView, times(1)).smoothScrollBy(anyInt(), eq(0));
        // view holder still null at NEXT_ADAPTER_POSITION
        verify(mRecentsRecyclerView).addOnScrollListener(mOnScrollListenerCaptor.capture());
        mOnScrollListenerCaptor.getValue().onScrollStateChanged(mRecentsRecyclerView,
                SCROLL_STATE_IDLE);
        verify(mRecentsRecyclerView, times(1)).smoothScrollBy(anyInt(), eq(0));
    }

    @Test
    public void focusSearch_thumbnailFocused_forward_clearAllViewFocus_noReverseLayout_NoRTL() {
        mRecentsRecyclerView.setLayoutDirection(LayoutDirection.LTR);
        when(mGridLayoutManager.getReverseLayout()).thenReturn(false);
        when(mCurrentThumbnailView.isFocused()).thenReturn(true);
        doReturn(mCurrentTaskViewHolder).when(mRecentsRecyclerView).findContainingViewHolder(
                mCurrentThumbnailView);
        doReturn(mClearAllTaskViewHolder).when(mRecentsRecyclerView)
                .findViewHolderForAdapterPosition(NEXT_ADAPTER_POSITION);

        View nextFocusedView = mRecentsRecyclerView.focusSearch(mCurrentThumbnailView,
                View.FOCUS_FORWARD);

        assertThat(nextFocusedView).isEqualTo(mClearAllButtonView);
    }

    private void initViews() {
        when(mCurrentTaskView.findViewById(R.id.task_thumbnail)).thenReturn(mCurrentThumbnailView);
        when(mCurrentTaskView.findViewById(R.id.task_dismiss_button)).thenReturn(
                mCurrentDismissView);
        when(mCurrentThumbnailView.getId()).thenReturn(R.id.task_thumbnail);
        when(mCurrentDismissView.getId()).thenReturn(R.id.task_dismiss_button);
        when(mNextTaskView.findViewById(R.id.task_thumbnail)).thenReturn(mNextThumbnailView);
        when(mNextTaskView.findViewById(R.id.task_dismiss_button)).thenReturn(mNextDismissView);
        when(mNextThumbnailView.getId()).thenReturn(R.id.task_thumbnail);
        when(mNextDismissView.getId()).thenReturn(R.id.task_dismiss_button);
        when(mPreviousTaskView.findViewById(R.id.task_thumbnail)).thenReturn(
                mPreviousThumbnailView);
        when(mPreviousTaskView.findViewById(R.id.task_dismiss_button)).thenReturn(
                mPreviousDismissView);
        when(mPreviousThumbnailView.getId()).thenReturn(R.id.task_thumbnail);
        when(mPreviousDismissView.getId()).thenReturn(R.id.task_dismiss_button);
        when(mClearAllView.findViewById(R.id.recents_clear_all_button)).thenReturn(
                mClearAllButtonView);

        mCurrentTaskViewHolder = spy(new BaseTaskViewHolder(mCurrentTaskView));
        doReturn(CURRENT_ADAPTER_POSITION).when(
                mCurrentTaskViewHolder).getAbsoluteAdapterPosition();
        mNextTaskViewHolder = spy(new BaseTaskViewHolder(mNextTaskView));
        doReturn(NEXT_ADAPTER_POSITION).when(mNextTaskViewHolder).getAbsoluteAdapterPosition();
        mPreviousTaskViewHolder = spy(new BaseTaskViewHolder(mPreviousTaskView));
        doReturn(PREVIOUS_ADAPTER_POSITION).when(
                mPreviousTaskViewHolder).getAbsoluteAdapterPosition();
        mClearAllTaskViewHolder = spy(
                new ClearAllViewHolder(mClearAllView, mClearAllButtonClickListener));
        doReturn(NEXT_ADAPTER_POSITION).when(
                mClearAllTaskViewHolder).getAbsoluteAdapterPosition();
    }
}