chrome/browser/share/android/javatests/src/org/chromium/chrome/browser/share/send_tab_to_self/SendTabToSelfBottomSheetRenderTest.java

// Copyright 2022 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.share.send_tab_to_self;

import static org.mockito.Mockito.when;

import android.graphics.Bitmap;
import android.graphics.Canvas;
import android.graphics.drawable.Drawable;
import android.view.View;

import androidx.appcompat.content.res.AppCompatResources;
import androidx.test.filters.MediumTest;

import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;

import org.chromium.base.test.util.Feature;
import org.chromium.chrome.browser.profiles.Profile;
import org.chromium.chrome.browser.signin.services.IdentityServicesProvider;
import org.chromium.chrome.test.ChromeJUnit4ClassRunner;
import org.chromium.components.browser_ui.bottomsheet.BottomSheetController;
import org.chromium.components.signin.base.AccountCapabilities;
import org.chromium.components.signin.base.AccountInfo;
import org.chromium.components.signin.base.CoreAccountId;
import org.chromium.components.signin.identitymanager.ConsentLevel;
import org.chromium.components.signin.identitymanager.IdentityManager;
import org.chromium.components.signin.test.util.R;
import org.chromium.components.sync_device_info.FormFactor;
import org.chromium.content_public.browser.test.util.TestThreadUtils;
import org.chromium.ui.test.util.BlankUiTestActivityTestCase;
import org.chromium.ui.test.util.RenderTestRule;
import org.chromium.url.JUnitTestGURLs;

import java.util.Arrays;
import java.util.Calendar;
import java.util.HashMap;
import java.util.List;
import java.util.concurrent.TimeUnit;

/** Render tests for the send-tab-to-self bottom sheets. */
@RunWith(ChromeJUnit4ClassRunner.class)
public class SendTabToSelfBottomSheetRenderTest extends BlankUiTestActivityTestCase {
    @Rule
    public final RenderTestRule mRenderTestRule =
            RenderTestRule.Builder.withPublicCorpus()
                    .setBugComponent(RenderTestRule.Component.UI_BROWSER_SHARING)
                    .setRevision(4)
                    .build();
    @Rule
    public final MockitoRule mMockitoRule = MockitoJUnit.rule();

    @Mock
    private Profile mProfile;
    @Mock
    private IdentityServicesProvider mIdentityServicesProvider;
    @Mock
    private IdentityManager mIdentityManager;
    @Mock
    private BottomSheetController mBottomSheetController;

    @Before
    public void setUp() {
        // Set up account data to be shown by the UI.
        AccountInfo account = createFakeAccount();
        when(mIdentityManager.getPrimaryAccountInfo(ConsentLevel.SIGNIN)).thenReturn(account);
        when(mIdentityManager.findExtendedAccountInfoByEmailAddress(account.getEmail()))
                .thenReturn(account);
        when(mIdentityServicesProvider.getIdentityManager(mProfile)).thenReturn(mIdentityManager);
        IdentityServicesProvider.setInstanceForTests(mIdentityServicesProvider);
        Profile.setLastUsedProfileForTesting(mProfile);
    }

    @Test
    @MediumTest
    @Feature("RenderTest")
    public void testDevicePickerBottomSheet() throws Throwable {
        long todayTimestamp = Calendar.getInstance().getTimeInMillis();
        List<TargetDeviceInfo> devices = Arrays.asList(
                new TargetDeviceInfo("My Phone", "guid1", FormFactor.PHONE, todayTimestamp),
                new TargetDeviceInfo("My Computer", "guid2", FormFactor.DESKTOP,
                        todayTimestamp - TimeUnit.DAYS.toMillis(1)));
        View view = TestThreadUtils.runOnUiThreadBlockingNoException(() -> {
            DevicePickerBottomSheetContent sheetContent =
                    new DevicePickerBottomSheetContent(getActivity(), JUnitTestGURLs.HTTP_URL,
                            "Title", mBottomSheetController, devices, mProfile);
            getActivity().setContentView(sheetContent.getContentView());
            return sheetContent.getContentView();
        });
        mRenderTestRule.render(view, "device_picker");
    }

    @Test
    @MediumTest
    @Feature("RenderTest")
    public void testNoTargetDeviceBottomSheetWithPromoFeatureDisabled() throws Throwable {
        View view = TestThreadUtils.runOnUiThreadBlockingNoException(() -> {
            NoTargetDeviceBottomSheetContent sheetContent = new NoTargetDeviceBottomSheetContent(
                    getActivity(), /*isPromoFeatureEnabled=*/false);
            getActivity().setContentView(sheetContent.getContentView());
            return sheetContent.getContentView();
        });
        mRenderTestRule.render(view, "no_target_device");
    }

    @Test
    @MediumTest
    @Feature("RenderTest")
    public void testNoTargetDeviceBottomSheetWithPromoFeatureEnabled() throws Throwable {
        View view = TestThreadUtils.runOnUiThreadBlockingNoException(() -> {
            NoTargetDeviceBottomSheetContent sheetContent = new NoTargetDeviceBottomSheetContent(
                    getActivity(), /*isPromoFeatureEnabled=*/true);
            getActivity().setContentView(sheetContent.getContentView());
            return sheetContent.getContentView();
        });
        mRenderTestRule.render(view, "no_target_device_with_account");
    }

    // TODO(crbug.com/1219434): This duplicates the account in AccountManagerTestRule, so tests can
    // later adopt the rule without failing the golden diffs. That's not done now because it
    // requires changing the device picker to depend on ProfileDataCache instead of IdentityManager.
    private AccountInfo createFakeAccount() {
        Drawable drawable =
                AppCompatResources.getDrawable(getActivity(), R.drawable.test_profile_picture);
        Bitmap bitmap = Bitmap.createBitmap(drawable.getIntrinsicWidth(),
                drawable.getIntrinsicHeight(), Bitmap.Config.ARGB_8888);
        Canvas canvas = new Canvas(bitmap);
        drawable.setBounds(0, 0, canvas.getWidth(), canvas.getHeight());
        drawable.draw(canvas);

        return new AccountInfo(new CoreAccountId("id"), "test@gmail.com", "gaiaId", "John Doe",
                "John", bitmap, new AccountCapabilities(new HashMap<>()));
    }
}