chrome/browser/recent_tabs/internal/android/java/src/org/chromium/chrome/browser/recent_tabs/RestoreTabsCoordinatorUnitTest.java

// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package org.chromium.chrome.browser.recent_tabs;

import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;

import static org.chromium.chrome.browser.recent_tabs.RestoreTabsProperties.CURRENT_SCREEN;
import static org.chromium.chrome.browser.recent_tabs.RestoreTabsProperties.ScreenType.HOME_SCREEN;

import android.app.Activity;
import android.widget.ViewFlipper;

import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.robolectric.Robolectric;
import org.robolectric.annotation.Config;

import org.chromium.base.test.BaseRobolectricTestRunner;
import org.chromium.chrome.browser.profiles.Profile;
import org.chromium.chrome.browser.tabmodel.TabCreatorManager;
import org.chromium.components.browser_ui.bottomsheet.BottomSheetController;
import org.chromium.ui.modelutil.PropertyModel;

/**
 * Unit tests for RestoreTabsCoordinator.
 */
@RunWith(BaseRobolectricTestRunner.class)
@Config(manifest = Config.NONE)
public class RestoreTabsCoordinatorUnitTest {
    @Mock
    private RestoreTabsMediator mMediator;
    @Mock
    private Profile mProfile;
    @Mock
    private RestoreTabsControllerFactory.ControllerListener mListener;
    @Mock
    private TabCreatorManager mTabCreatorManager;
    @Mock
    private BottomSheetController mBottomSheetController;

    private RestoreTabsCoordinator mCoordinator;
    private Activity mActivity;
    private PropertyModel mModel;
    private ViewFlipper mViewFlipperView;

    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);
        mActivity = Robolectric.buildActivity(Activity.class).setup().get();
        mCoordinator = new RestoreTabsCoordinator(mActivity, mProfile, mMediator, mListener,
                mTabCreatorManager, mBottomSheetController);
        mModel = mCoordinator.getPropertyModelForTesting();
        mViewFlipperView = mCoordinator.getViewFlipperForTesting();
    }

    @After
    public void tearDown() {
        mCoordinator.destroy();
    }

    @Test
    public void testRestoreTabsCoordinator_showHomeScreen() {
        mCoordinator.showHomeScreen();
        verify(mMediator, times(1)).showHomeScreen();
    }

    @Test
    public void testRestoreTabsCoordinator_viewFlipperScreenChange() {
        mModel.set(CURRENT_SCREEN, HOME_SCREEN);
        Assert.assertEquals(0, mViewFlipperView.getDisplayedChild());
    }
}