1 // Copyright 2013 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "base/memory/scoped_ptr.h" 6 #include "base/prefs/pref_service.h" 7 #include "chrome/browser/content_settings/host_content_settings_map.h" 8 #include "chrome/browser/search/instant_service_factory.h" 9 #include "chrome/browser/search/search.h" 10 #include "chrome/browser/ui/search/instant_ntp.h" 11 #include "chrome/browser/ui/search/instant_ntp_prerenderer.h" 12 #include "chrome/common/content_settings.h" 13 #include "chrome/common/pref_names.h" 14 #include "chrome/test/base/testing_profile.h" 15 #include "content/public/test/test_browser_thread_bundle.h" 16 #include "testing/gtest/include/gtest/gtest.h" 17 18 class TestableInstantNTP : public InstantNTP { 19 public: 20 TestableInstantNTP(InstantNTPPrerenderer* ntp_prerenderer, 21 const std::string& instant_url, 22 Profile* profile) 23 : InstantNTP(ntp_prerenderer, "UNUSED", profile), 24 test_instant_url_(instant_url), 25 test_supports_instant_(true), 26 test_is_local_(false) { 27 } 28 29 // Overrides from InstantPage 30 virtual bool supports_instant() const OVERRIDE { 31 return test_supports_instant_; 32 } 33 34 virtual bool IsLocal() const OVERRIDE { 35 return test_is_local_; 36 }; 37 38 virtual const std::string& instant_url() const OVERRIDE { 39 return test_instant_url_; 40 } 41 42 void set_instant_url(const std::string& instant_url) { 43 test_instant_url_ = instant_url; 44 } 45 46 void set_supports_instant(bool supports_instant) { 47 test_supports_instant_ = supports_instant; 48 } 49 50 void set_is_local(bool is_local) { 51 test_is_local_ = is_local; 52 } 53 54 private: 55 std::string test_instant_url_; 56 bool test_supports_instant_; 57 bool test_is_local_; 58 }; 59 60 class TestableInstantNTPPrerenderer : public InstantNTPPrerenderer { 61 public: 62 explicit TestableInstantNTPPrerenderer(TestingProfile* profile, 63 InstantService* instant_service) 64 : InstantNTPPrerenderer(profile, instant_service, NULL), 65 test_instant_url_("http://test_url"), 66 override_javascript_enabled_(true), 67 test_javascript_enabled_(true), 68 test_in_startup_(false), 69 test_ntp_(NULL) { 70 } 71 72 // Overrides from InstantNTPPrerenderer 73 virtual std::string GetInstantURL() const OVERRIDE { 74 return test_instant_url_; 75 } 76 77 virtual std::string GetLocalInstantURL() const OVERRIDE { 78 return "http://local_instant_url"; 79 } 80 81 virtual InstantNTP* ntp() const OVERRIDE { 82 return test_ntp_; 83 } 84 85 virtual bool IsJavascriptEnabled() const OVERRIDE { 86 if (override_javascript_enabled_) 87 return test_javascript_enabled_; 88 else 89 return InstantNTPPrerenderer::IsJavascriptEnabled(); 90 } 91 92 virtual bool InStartup() const OVERRIDE { 93 return test_in_startup_; 94 } 95 96 void set_instant_url(const std::string& instant_url) { 97 test_instant_url_ = instant_url; 98 } 99 100 void set_ntp(InstantNTP* ntp) { 101 test_ntp_ = ntp; 102 } 103 104 void set_javascript_enabled(bool javascript_enabled) { 105 override_javascript_enabled_ = true; 106 test_javascript_enabled_ = javascript_enabled; 107 } 108 109 void set_override_javascript_enabled(bool override_javascript_enabled) { 110 override_javascript_enabled_ = override_javascript_enabled; 111 } 112 113 void set_in_startup(bool in_startup) { 114 test_in_startup_ = in_startup; 115 } 116 117 private: 118 std::string test_instant_url_; 119 bool override_javascript_enabled_; 120 bool test_javascript_enabled_; 121 bool test_in_startup_; 122 InstantNTP* test_ntp_; 123 }; 124 125 class InstantNTPPrerendererTest : public testing::Test { 126 public: 127 virtual void SetUp() OVERRIDE { 128 instant_service_ = InstantServiceFactory::GetForProfile(&profile_); 129 instant_ntp_prerenderer_.reset( 130 new TestableInstantNTPPrerenderer(&profile_, instant_service_)); 131 } 132 133 virtual void TearDown() OVERRIDE { 134 instant_ntp_prerenderer_.reset(); 135 } 136 137 TestableInstantNTPPrerenderer* instant_ntp_prerenderer() { 138 return instant_ntp_prerenderer_.get(); 139 } 140 141 Profile* profile() { 142 return instant_ntp_prerenderer()->profile(); 143 } 144 145 private: 146 content::TestBrowserThreadBundle thread_bundle_; 147 scoped_ptr<TestableInstantNTPPrerenderer> instant_ntp_prerenderer_; 148 InstantService* instant_service_; 149 mutable TestingProfile profile_; 150 }; 151 152 TEST_F(InstantNTPPrerendererTest, PrefersRemoteNTPOnStartup) { 153 std::string instant_url("http://instant_url"); 154 scoped_ptr<TestableInstantNTP> ntp(new TestableInstantNTP( 155 instant_ntp_prerenderer(), instant_url, profile())); 156 instant_ntp_prerenderer()->set_ntp(ntp.get()); 157 instant_ntp_prerenderer()->set_instant_url(instant_url); 158 instant_ntp_prerenderer()->set_in_startup(true); 159 EXPECT_FALSE(instant_ntp_prerenderer()->ShouldSwitchToLocalNTP()); 160 } 161 162 TEST_F(InstantNTPPrerendererTest, SwitchesToLocalNTPIfNoInstantSupport) { 163 std::string instant_url("http://instant_url"); 164 scoped_ptr<TestableInstantNTP> ntp(new TestableInstantNTP( 165 instant_ntp_prerenderer(), instant_url, profile())); 166 instant_ntp_prerenderer()->set_ntp(ntp.get()); 167 instant_ntp_prerenderer()->set_instant_url(instant_url); 168 ntp->set_supports_instant(false); 169 EXPECT_TRUE(instant_ntp_prerenderer()->ShouldSwitchToLocalNTP()); 170 } 171 172 TEST_F(InstantNTPPrerendererTest, SwitchesToLocalNTPIfPathBad) { 173 std::string instant_url("http://instant_url"); 174 scoped_ptr<TestableInstantNTP> ntp(new TestableInstantNTP( 175 instant_ntp_prerenderer(), instant_url, profile())); 176 instant_ntp_prerenderer()->set_ntp(ntp.get()); 177 instant_ntp_prerenderer()->set_instant_url("http://bogus_url"); 178 EXPECT_TRUE(instant_ntp_prerenderer()->ShouldSwitchToLocalNTP()); 179 } 180 181 TEST_F(InstantNTPPrerendererTest, DoesNotSwitchToLocalNTPIfOnCurrentNTP) { 182 std::string instant_url("http://instant_url"); 183 scoped_ptr<TestableInstantNTP> ntp(new TestableInstantNTP( 184 instant_ntp_prerenderer(), instant_url, profile())); 185 instant_ntp_prerenderer()->set_ntp(ntp.get()); 186 instant_ntp_prerenderer()->set_instant_url(instant_url); 187 EXPECT_FALSE(instant_ntp_prerenderer()->ShouldSwitchToLocalNTP()); 188 } 189 190 TEST_F(InstantNTPPrerendererTest, DoesNotSwitchToLocalNTPIfOnLocalNTP) { 191 std::string instant_url("http://instant_url"); 192 scoped_ptr<TestableInstantNTP> ntp(new TestableInstantNTP( 193 instant_ntp_prerenderer(), instant_url, profile())); 194 instant_ntp_prerenderer()->set_ntp(ntp.get()); 195 instant_ntp_prerenderer()->set_instant_url(instant_url); 196 ntp->set_instant_url("http://local_instant_url"); 197 EXPECT_FALSE(instant_ntp_prerenderer()->ShouldSwitchToLocalNTP()); 198 } 199 200 TEST_F(InstantNTPPrerendererTest, SwitchesToLocalNTPIfJSDisabled) { 201 std::string instant_url("http://instant_url"); 202 scoped_ptr<TestableInstantNTP> ntp(new TestableInstantNTP( 203 instant_ntp_prerenderer(), instant_url, profile())); 204 instant_ntp_prerenderer()->set_ntp(ntp.get()); 205 instant_ntp_prerenderer()->set_javascript_enabled(false); 206 instant_ntp_prerenderer()->set_instant_url(instant_url); 207 ntp->set_instant_url("http://local_instant_url"); 208 EXPECT_TRUE(instant_ntp_prerenderer()->ShouldSwitchToLocalNTP()); 209 } 210 211 TEST_F(InstantNTPPrerendererTest, SwitchesToLocalNTPIfNoNTPReady) { 212 EXPECT_TRUE(instant_ntp_prerenderer()->ShouldSwitchToLocalNTP()); 213 } 214 215 TEST_F(InstantNTPPrerendererTest, IsJavascriptEnabled) { 216 instant_ntp_prerenderer()->set_override_javascript_enabled(false); 217 EXPECT_TRUE(instant_ntp_prerenderer()->IsJavascriptEnabled()); 218 } 219 220 TEST_F(InstantNTPPrerendererTest, IsJavascriptEnabledChecksContentSettings) { 221 instant_ntp_prerenderer()->set_override_javascript_enabled(false); 222 instant_ntp_prerenderer()->profile()->GetHostContentSettingsMap() 223 ->SetDefaultContentSetting(CONTENT_SETTINGS_TYPE_JAVASCRIPT, 224 CONTENT_SETTING_DEFAULT); 225 EXPECT_TRUE(instant_ntp_prerenderer()->IsJavascriptEnabled()); 226 instant_ntp_prerenderer()->profile()->GetHostContentSettingsMap() 227 ->SetDefaultContentSetting(CONTENT_SETTINGS_TYPE_JAVASCRIPT, 228 CONTENT_SETTING_ALLOW); 229 EXPECT_TRUE(instant_ntp_prerenderer()->IsJavascriptEnabled()); 230 instant_ntp_prerenderer()->profile()->GetHostContentSettingsMap() 231 ->SetDefaultContentSetting(CONTENT_SETTINGS_TYPE_JAVASCRIPT, 232 CONTENT_SETTING_BLOCK); 233 EXPECT_FALSE(instant_ntp_prerenderer()->IsJavascriptEnabled()); 234 } 235 236 TEST_F(InstantNTPPrerendererTest, IsJavascriptEnabledChecksPrefs) { 237 instant_ntp_prerenderer()->set_override_javascript_enabled(false); 238 instant_ntp_prerenderer()->profile()->GetPrefs()->SetBoolean( 239 prefs::kWebKitJavascriptEnabled, true); 240 EXPECT_TRUE(instant_ntp_prerenderer()->IsJavascriptEnabled()); 241 instant_ntp_prerenderer()->profile()->GetPrefs()->SetBoolean( 242 prefs::kWebKitJavascriptEnabled, false); 243 EXPECT_FALSE(instant_ntp_prerenderer()->IsJavascriptEnabled()); 244 } 245
