1// Copyright (c) 2012 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/prefs/pref_service.h" 6#include "base/prefs/scoped_user_pref_update.h" 7#include "chrome/browser/extensions/extension_apitest.h" 8#include "chrome/browser/extensions/extension_service.h" 9#include "chrome/browser/extensions/extension_web_ui.h" 10#include "chrome/browser/profiles/profile.h" 11#include "chrome/browser/ui/browser.h" 12#include "chrome/browser/ui/tabs/tab_strip_model.h" 13#include "chrome/common/extensions/manifest_url_handler.h" 14#include "chrome/common/url_constants.h" 15#include "chrome/test/base/ui_test_utils.h" 16#include "content/public/browser/navigation_entry.h" 17#include "content/public/browser/web_contents.h" 18#include "extensions/common/constants.h" 19#include "extensions/test/result_catcher.h" 20 21using content::WebContents; 22 23class ExtensionOverrideTest : public ExtensionApiTest { 24 protected: 25 bool CheckHistoryOverridesContainsNoDupes() { 26 // There should be no duplicate entries in the preferences. 27 const base::DictionaryValue* overrides = 28 browser()->profile()->GetPrefs()->GetDictionary( 29 ExtensionWebUI::kExtensionURLOverrides); 30 31 const base::ListValue* values = NULL; 32 if (!overrides->GetList("history", &values)) 33 return false; 34 35 std::set<std::string> seen_overrides; 36 for (size_t i = 0; i < values->GetSize(); ++i) { 37 std::string value; 38 if (!values->GetString(i, &value)) 39 return false; 40 41 if (seen_overrides.find(value) != seen_overrides.end()) 42 return false; 43 44 seen_overrides.insert(value); 45 } 46 47 return true; 48 } 49}; 50 51IN_PROC_BROWSER_TEST_F(ExtensionOverrideTest, OverrideNewtab) { 52 ASSERT_TRUE(RunExtensionTest("override/newtab")) << message_; 53 { 54 extensions::ResultCatcher catcher; 55 // Navigate to the new tab page. The overridden new tab page 56 // will call chrome.test.notifyPass() . 57 ui_test_utils::NavigateToURL(browser(), GURL("chrome://newtab/")); 58 WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents(); 59 ASSERT_TRUE(tab->GetController().GetVisibleEntry()); 60 EXPECT_TRUE(tab->GetController().GetVisibleEntry()->GetURL(). 61 SchemeIs(extensions::kExtensionScheme)); 62 63 ASSERT_TRUE(catcher.GetNextResult()); 64 } 65 66 // TODO(erikkay) Load a second extension with the same override. 67 // Verify behavior, then unload the first and verify behavior, etc. 68} 69 70#if defined(OS_MACOSX) 71// Hangy: http://crbug.com/70511 72#define MAYBE_OverrideNewtabIncognito DISABLED_OverrideNewtabIncognito 73#else 74#define MAYBE_OverrideNewtabIncognito OverrideNewtabIncognito 75#endif 76IN_PROC_BROWSER_TEST_F(ExtensionOverrideTest, MAYBE_OverrideNewtabIncognito) { 77 ASSERT_TRUE(RunExtensionTest("override/newtab")) << message_; 78 79 // Navigate an incognito tab to the new tab page. We should get the actual 80 // new tab page because we can't load chrome-extension URLs in incognito. 81 Browser* otr_browser = ui_test_utils::OpenURLOffTheRecord( 82 browser()->profile(), GURL("chrome://newtab/")); 83 WebContents* tab = otr_browser->tab_strip_model()->GetActiveWebContents(); 84 ASSERT_TRUE(tab->GetController().GetVisibleEntry()); 85 EXPECT_FALSE(tab->GetController().GetVisibleEntry()->GetURL(). 86 SchemeIs(extensions::kExtensionScheme)); 87} 88 89// Times out consistently on Win, http://crbug.com/45173. 90#if defined(OS_WIN) 91#define MAYBE_OverrideHistory DISABLED_OverrideHistory 92#else 93#define MAYBE_OverrideHistory OverrideHistory 94#endif // defined(OS_WIN) 95 96IN_PROC_BROWSER_TEST_F(ExtensionOverrideTest, MAYBE_OverrideHistory) { 97 ASSERT_TRUE(RunExtensionTest("override/history")) << message_; 98 { 99 extensions::ResultCatcher catcher; 100 // Navigate to the history page. The overridden history page 101 // will call chrome.test.notifyPass() . 102 ui_test_utils::NavigateToURL(browser(), GURL("chrome://history/")); 103 ASSERT_TRUE(catcher.GetNextResult()); 104 } 105} 106 107// Regression test for http://crbug.com/41442. 108IN_PROC_BROWSER_TEST_F(ExtensionOverrideTest, ShouldNotCreateDuplicateEntries) { 109 const extensions::Extension* extension = 110 LoadExtension(test_data_dir_.AppendASCII("override/history")); 111 ASSERT_TRUE(extension); 112 113 // Simulate several LoadExtension() calls happening over the lifetime of 114 // a preferences file without corresponding UnloadExtension() calls. 115 for (size_t i = 0; i < 3; ++i) { 116 ExtensionWebUI::RegisterChromeURLOverrides( 117 browser()->profile(), 118 extensions::URLOverrides::GetChromeURLOverrides(extension)); 119 } 120 121 ASSERT_TRUE(CheckHistoryOverridesContainsNoDupes()); 122} 123 124IN_PROC_BROWSER_TEST_F(ExtensionOverrideTest, ShouldCleanUpDuplicateEntries) { 125 // Simulate several LoadExtension() calls happening over the lifetime of 126 // a preferences file without corresponding UnloadExtension() calls. This is 127 // the same as the above test, except for that it is testing the case where 128 // the file already contains dupes when an extension is loaded. 129 base::ListValue* list = new base::ListValue(); 130 for (size_t i = 0; i < 3; ++i) 131 list->Append(new base::StringValue("http://www.google.com/")); 132 133 { 134 DictionaryPrefUpdate update(browser()->profile()->GetPrefs(), 135 ExtensionWebUI::kExtensionURLOverrides); 136 update.Get()->Set("history", list); 137 } 138 139 ASSERT_FALSE(CheckHistoryOverridesContainsNoDupes()); 140 141 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("override/history"))); 142 143 ASSERT_TRUE(CheckHistoryOverridesContainsNoDupes()); 144} 145