// Copyright 2021 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.tab;
import androidx.test.filters.SmallTest;
import org.hamcrest.Matchers;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentMatchers;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.MockitoAnnotations;
import org.chromium.base.ThreadUtils;
import org.chromium.base.test.util.Batch;
import org.chromium.base.test.util.CallbackHelper;
import org.chromium.base.test.util.CommandLineFlags;
import org.chromium.base.test.util.Criteria;
import org.chromium.base.test.util.CriteriaHelper;
import org.chromium.base.test.util.HistogramWatcher;
import org.chromium.base.test.util.Restriction;
import org.chromium.chrome.browser.browsing_data.BrowsingDataBridge;
import org.chromium.chrome.browser.browsing_data.BrowsingDataType;
import org.chromium.chrome.browser.browsing_data.TimePeriod;
import org.chromium.chrome.browser.feature_engagement.TrackerFactory;
import org.chromium.chrome.browser.flags.ChromeSwitches;
import org.chromium.chrome.browser.profiles.ProfileManager;
import org.chromium.chrome.browser.ui.appmenu.AppMenuObserver;
import org.chromium.chrome.browser.ui.appmenu.AppMenuTestSupport;
import org.chromium.chrome.test.ChromeJUnit4ClassRunner;
import org.chromium.chrome.test.ChromeTabbedActivityTestRule;
import org.chromium.chrome.test.R;
import org.chromium.components.browser_ui.site_settings.WebsitePreferenceBridge;
import org.chromium.components.content_settings.ContentSettingsType;
import org.chromium.components.feature_engagement.Tracker;
import org.chromium.ui.test.util.UiRestriction;
import java.util.concurrent.TimeoutException;
/** Test for user flows around {@link ContentSettingsType.REQUEST_DESKTOP_SITE}. */
@RunWith(ChromeJUnit4ClassRunner.class)
@Batch(Batch.PER_CLASS)
@Restriction(UiRestriction.RESTRICTION_TYPE_PHONE)
@CommandLineFlags.Add({ChromeSwitches.DISABLE_FIRST_RUN_EXPERIENCE})
public class RequestDesktopSiteTest {
private static final String URL_1 = "https://www.chromium.org/";
private static final String URL_2 = "https://www.example.com/";
private CallbackHelper mMenuObserver;
@Rule
public ChromeTabbedActivityTestRule mActivityTestRule = new ChromeTabbedActivityTestRule();
@Mock Tracker mMockTracker;
@Before
public void setup() {
MockitoAnnotations.initMocks(this);
// Disable IPH to avoid interruptions on the app menu.
TrackerFactory.setTrackerForTests(mMockTracker);
Mockito.doReturn(false)
.when(mMockTracker)
.shouldTriggerHelpUI(ArgumentMatchers.anyString());
mActivityTestRule.startMainActivityOnBlankPage();
mMenuObserver = new CallbackHelper();
mActivityTestRule
.getAppMenuCoordinator()
.getAppMenuHandler()
.addObserver(
new AppMenuObserver() {
@Override
public void onMenuVisibilityChanged(boolean isVisible) {
mMenuObserver.notifyCalled();
}
@Override
public void onMenuHighlightChanged(boolean highlighting) {}
});
}
@After
public void tearDown() throws TimeoutException {
// Clean up content settings.
CallbackHelper helper = new CallbackHelper();
ThreadUtils.runOnUiThreadBlocking(
() -> {
BrowsingDataBridge.getForProfile(ProfileManager.getLastUsedRegularProfile())
.clearBrowsingData(
helper::notifyCalled,
new int[] {BrowsingDataType.SITE_SETTINGS},
TimePeriod.ALL_TIME);
});
helper.waitForCallback(0);
}
@Test
@SmallTest
public void testGlobalSiteSettingsAndException() throws TimeoutException {
Tab tab = mActivityTestRule.loadUrlInNewTab(URL_1);
// Explicitly set the global setting to mobile to avoid flakiness.
updateGlobalSetting(tab, false);
assertUsingDesktopUserAgent(
tab, false, "User agent should be mobile according to global site settings.");
ThreadUtils.runOnUiThreadBlocking(
() -> {
AppMenuTestSupport.showAppMenu(
mActivityTestRule.getAppMenuCoordinator(), null, false);
});
mMenuObserver.waitForCallback(0);
HistogramWatcher histogramExpectation =
HistogramWatcher.newSingleRecordWatcher(
"Android.RequestDesktopSite.UserSwitchToDesktop", true);
toggleFromAppMenu(tab);
assertUsingDesktopUserAgent(
tab, true, "User agent should be desktop according to site settings.");
histogramExpectation.assertExpected();
mActivityTestRule.loadUrl(URL_2);
assertUsingDesktopUserAgent(
tab, false, "Site settings exceptions should not affect other URL.");
// Change site settings and reload.
updateGlobalSetting(tab, true);
assertUsingDesktopUserAgent(
tab, true, "User agent should be desktop according to global site settings.");
}
@Test
@SmallTest
public void testUnsetPerTabSettings() throws TimeoutException {
Tab tab = mActivityTestRule.loadUrlInNewTab(URL_1);
// Explicitly set the global setting to mobile to avoid flakiness.
updateGlobalSetting(tab, false);
assertUsingDesktopUserAgent(
tab,
false,
"Tab layout should be <Mobile>, while global settings is <Mobile> and tab level"
+ " settings is <Default>.");
ThreadUtils.runOnUiThreadBlocking(
() -> {
AppMenuTestSupport.showAppMenu(
mActivityTestRule.getAppMenuCoordinator(), null, false);
});
mMenuObserver.waitForCallback(0);
toggleFromAppMenu(tab);
assertUsingDesktopUserAgent(
tab,
true,
"Tab layout should be <Desktop>, while global settings is <Mobile> and tab level"
+ " settings is <Desktop>.");
updateGlobalSetting(tab, true);
assertUsingDesktopUserAgent(
tab,
true,
"Tab layout should be <Desktop>, while global settings is <Desktop> and tab level"
+ " settings is <Desktop>.");
updateGlobalSetting(tab, false);
assertUsingDesktopUserAgent(
tab,
true,
"Tab layout should be <Desktop>, while global settings is <Mobile> and tab level"
+ " settings is <Desktop>.");
toggleFromAppMenu(tab);
assertUsingDesktopUserAgent(
tab,
false,
"Tab layout should be <Mobile>, while global settings is <Mobile> and tab level"
+ " settings is <Default>.");
updateGlobalSetting(tab, true);
assertUsingDesktopUserAgent(
tab,
true,
"Tab layout should be <Desktop>, while global settings is <Desktop> and tab level"
+ " settings is <Default>.");
}
@Test
@SmallTest
public void testGlobalAndPerTabSettings() throws TimeoutException {
Tab tab = mActivityTestRule.loadUrlInNewTab(URL_1);
updateGlobalSetting(tab, true);
assertUsingDesktopUserAgent(
tab,
true,
"Tab layout should be <Desktop>, while global settings is <Desktop> and tab level"
+ " settings is <Default>.");
ThreadUtils.runOnUiThreadBlocking(
() -> {
AppMenuTestSupport.showAppMenu(
mActivityTestRule.getAppMenuCoordinator(), null, false);
});
mMenuObserver.waitForCallback(0);
toggleFromAppMenu(tab);
assertUsingDesktopUserAgent(
tab,
false,
"Tab layout should be <Mobile>, while global settings is <Desktop> and tab level"
+ " settings is <Mobile>.");
updateGlobalSetting(tab, false);
assertUsingDesktopUserAgent(
tab,
false,
"Tab layout should be <Mobile>, while global settings is <Mobile> and tab level"
+ " settings is <Mobile>.");
toggleFromAppMenu(tab);
assertUsingDesktopUserAgent(
tab,
true,
"Tab layout should be <Desktop>, while global settings is <Mobile> and tab level"
+ " settings is <Desktop>.");
}
private void assertUsingDesktopUserAgent(
Tab tab, boolean useDesktopUserAgent, String errorMessage) {
Assert.assertNotNull("Tab does not have a WebContent.", tab.getWebContents());
Assert.assertEquals(
errorMessage,
useDesktopUserAgent,
tab.getWebContents().getNavigationController().getUseDesktopUserAgent());
}
private void updateGlobalSetting(Tab tab, boolean setting) {
ThreadUtils.runOnUiThreadBlocking(
() -> {
WebsitePreferenceBridge.setContentSettingEnabled(
tab.getProfile(), ContentSettingsType.REQUEST_DESKTOP_SITE, setting);
tab.reload();
});
CriteriaHelper.pollUiThread(() -> Criteria.checkThat(tab.isLoading(), Matchers.is(false)));
}
private void toggleFromAppMenu(Tab tab) {
ThreadUtils.runOnUiThreadBlocking(
() -> {
AppMenuTestSupport.callOnItemClick(
mActivityTestRule.getAppMenuCoordinator(),
R.id.request_desktop_site_id);
});
CriteriaHelper.pollUiThread(() -> Criteria.checkThat(tab.isLoading(), Matchers.is(false)));
}
}
Codebase Browser
chromium