| // Copyright 2012 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <memory> |
| |
| #include "base/ios/device_util.h" |
| #include "base/run_loop.h" |
| #include "base/strings/sys_string_conversions.h" |
| #include "components/prefs/pref_service.h" |
| #include "ios/chrome/browser/browser_state/test_chrome_browser_state.h" |
| #include "ios/chrome/browser/pref_names.h" |
| #import "ios/chrome/browser/prerender/preload_controller.h" |
| #include "ios/web/public/test/web_task_environment.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/platform_test.h" |
| |
| #if !defined(__has_feature) || !__has_feature(objc_arc) |
| #error "This file requires ARC support." |
| #endif |
| |
| namespace { |
| |
| // Override NetworkChangeNotifier to simulate connection type changes for tests. |
| class TestNetworkChangeNotifier : public net::NetworkChangeNotifier { |
| public: |
| TestNetworkChangeNotifier() |
| : net::NetworkChangeNotifier(), |
| connection_type_to_return_( |
| net::NetworkChangeNotifier::CONNECTION_UNKNOWN) {} |
| |
| // Simulates a change of the connection type to |type|. This will notify any |
| // objects that are NetworkChangeNotifiers. |
| void SimulateNetworkConnectionChange( |
| net::NetworkChangeNotifier::ConnectionType type) { |
| connection_type_to_return_ = type; |
| net::NetworkChangeNotifier::NotifyObserversOfConnectionTypeChange(); |
| base::RunLoop().RunUntilIdle(); |
| } |
| |
| private: |
| ConnectionType GetCurrentConnectionType() const override { |
| return connection_type_to_return_; |
| } |
| |
| // The currently simulated network connection type. If this is set to |
| // CONNECTION_NONE, then NetworkChangeNotifier::IsOffline will return true. |
| net::NetworkChangeNotifier::ConnectionType connection_type_to_return_; |
| |
| DISALLOW_COPY_AND_ASSIGN(TestNetworkChangeNotifier); |
| }; |
| |
| class PreloadControllerTest : public PlatformTest { |
| protected: |
| void SetUp() override { |
| TestChromeBrowserState::Builder test_cbs_builder; |
| chrome_browser_state_ = test_cbs_builder.Build(); |
| // Set up a NetworkChangeNotifier so that the test can simulate Wi-Fi vs. |
| // cellular connection. |
| network_change_notifier_.reset(new TestNetworkChangeNotifier); |
| |
| controller_ = [[PreloadController alloc] |
| initWithBrowserState:chrome_browser_state_.get()]; |
| } |
| |
| // Set the "Preload webpages" setting to "Always". |
| void PreloadWebpagesAlways() { |
| chrome_browser_state_->GetPrefs()->SetBoolean( |
| prefs::kNetworkPredictionEnabled, YES); |
| chrome_browser_state_->GetPrefs()->SetBoolean( |
| prefs::kNetworkPredictionWifiOnly, NO); |
| } |
| |
| // Set the "Preload webpages" setting to "Only on Wi-Fi". |
| void PreloadWebpagesWiFiOnly() { |
| chrome_browser_state_->GetPrefs()->SetBoolean( |
| prefs::kNetworkPredictionEnabled, YES); |
| chrome_browser_state_->GetPrefs()->SetBoolean( |
| prefs::kNetworkPredictionWifiOnly, YES); |
| } |
| |
| // Set the "Preload webpages" setting to "Never". |
| void PreloadWebpagesNever() { |
| chrome_browser_state_->GetPrefs()->SetBoolean( |
| prefs::kNetworkPredictionEnabled, NO); |
| } |
| |
| void SimulateWiFiConnection() { |
| network_change_notifier_->SimulateNetworkConnectionChange( |
| net::NetworkChangeNotifier::CONNECTION_WIFI); |
| } |
| |
| void SimulateCellularConnection() { |
| network_change_notifier_->SimulateNetworkConnectionChange( |
| net::NetworkChangeNotifier::CONNECTION_3G); |
| } |
| |
| web::WebTaskEnvironment task_environment_; |
| std::unique_ptr<TestChromeBrowserState> chrome_browser_state_; |
| std::unique_ptr<TestNetworkChangeNotifier> network_change_notifier_; |
| PreloadController* controller_; |
| }; |
| |
| // Tests that the preload controller does not try to preload non-web urls. |
| TEST_F(PreloadControllerTest, DontPreloadNonWebURLs) { |
| const web::Referrer kReferrer; |
| const ui::PageTransition kTransition = ui::PAGE_TRANSITION_LINK; |
| |
| // Attempt to prerender an empty URL and verify that no WebState was created |
| // to preload. |
| [controller_ prerenderURL:GURL() |
| referrer:kReferrer |
| transition:kTransition |
| immediately:YES]; |
| EXPECT_FALSE([controller_ releasePrerenderContents]); |
| |
| // Attempt to prerender the NTP and verify that no WebState was created |
| // to preload. |
| [controller_ prerenderURL:GURL("chrome://newtab") |
| referrer:kReferrer |
| transition:kTransition |
| immediately:YES]; |
| EXPECT_FALSE([controller_ releasePrerenderContents]); |
| |
| // Attempt to prerender the flags UI and verify that no WebState was created |
| // to preload. |
| [controller_ prerenderURL:GURL("about:flags") |
| referrer:kReferrer |
| transition:kTransition |
| immediately:YES]; |
| EXPECT_FALSE([controller_ releasePrerenderContents]); |
| } |
| |
| TEST_F(PreloadControllerTest, TestIsPrerenderingEnabled_preloadAlways) { |
| // With the "Preload Webpages" setting set to "Always", prerendering is |
| // enabled regardless of network type. |
| PreloadWebpagesAlways(); |
| |
| SimulateWiFiConnection(); |
| EXPECT_TRUE(controller_.enabled || ios::device_util::IsSingleCoreDevice() || |
| !ios::device_util::RamIsAtLeast512Mb()); |
| |
| SimulateCellularConnection(); |
| EXPECT_TRUE(controller_.enabled || ios::device_util::IsSingleCoreDevice() || |
| !ios::device_util::RamIsAtLeast512Mb()); |
| } |
| |
| TEST_F(PreloadControllerTest, TestIsPrerenderingEnabled_preloadWiFiOnly) { |
| // With the Chrome "Preload Webpages" setting set to "Only on Wi-Fi", |
| // prerendering is enabled only on WiFi. |
| PreloadWebpagesWiFiOnly(); |
| |
| SimulateWiFiConnection(); |
| EXPECT_TRUE(controller_.enabled || ios::device_util::IsSingleCoreDevice() || |
| !ios::device_util::RamIsAtLeast512Mb()); |
| |
| SimulateCellularConnection(); |
| EXPECT_FALSE(controller_.enabled); |
| } |
| |
| TEST_F(PreloadControllerTest, TestIsPrerenderingEnabled_preloadNever) { |
| // With the Chrome "Preload Webpages" setting set to "Never", prerendering |
| // is never enabled, regardless of the network type. |
| PreloadWebpagesNever(); |
| |
| SimulateWiFiConnection(); |
| EXPECT_FALSE(controller_.enabled); |
| |
| SimulateCellularConnection(); |
| EXPECT_FALSE(controller_.enabled); |
| } |
| |
| } // anonymous namespace |