| OLD | NEW |
|---|
| (Empty) |
| 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 "chrome/browser/tab_contents/render_view_context_menu.h" | |
| 6 | |
| 7 #include "base/prefs/pref_service.h" | |
| 8 #include "chrome/app/chrome_command_ids.h" | |
| 9 #include "chrome/browser/custom_handlers/protocol_handler_registry.h" | |
| 10 #include "chrome/browser/prefs/incognito_mode_prefs.h" | |
| 11 #include "chrome/browser/tab_contents/render_view_context_menu_test_util.h" | |
| 12 #include "chrome/test/base/chrome_render_view_host_test_harness.h" | |
| 13 #include "chrome/test/base/testing_profile.h" | |
| 14 #include "content/public/browser/web_contents.h" | |
| 15 #include "extensions/browser/extension_prefs.h" | |
| 16 #include "extensions/common/url_pattern.h" | |
| 17 #include "testing/gtest/include/gtest/gtest.h" | |
| 18 #include "third_party/WebKit/public/web/WebContextMenuData.h" | |
| 19 #include "url/gurl.h" | |
| 20 | |
| 21 using extensions::MenuItem; | |
| 22 using extensions::URLPatternSet; | |
| 23 | |
| 24 class RenderViewContextMenuTest : public testing::Test { | |
| 25 protected: | |
| 26 // Proxy defined here to minimize friend classes in RenderViewContextMenu | |
| 27 static bool ExtensionContextAndPatternMatch( | |
| 28 const content::ContextMenuParams& params, | |
| 29 MenuItem::ContextList contexts, | |
| 30 const URLPatternSet& patterns) { | |
| 31 return RenderViewContextMenu::ExtensionContextAndPatternMatch(params, | |
| 32 contexts, patterns); | |
| 33 } | |
| 34 }; | |
| 35 | |
| 36 // Generates a ContextMenuParams that matches the specified contexts. | |
| 37 static content::ContextMenuParams CreateParams(int contexts) { | |
| 38 content::ContextMenuParams rv; | |
| 39 rv.is_editable = false; | |
| 40 rv.media_type = blink::WebContextMenuData::MediaTypeNone; | |
| 41 rv.page_url = GURL("http://test.page/"); | |
| 42 | |
| 43 static const base::char16 selected_text[] = { 's', 'e', 'l', 0 }; | |
| 44 if (contexts & MenuItem::SELECTION) | |
| 45 rv.selection_text = selected_text; | |
| 46 | |
| 47 if (contexts & MenuItem::LINK) | |
| 48 rv.link_url = GURL("http://test.link/"); | |
| 49 | |
| 50 if (contexts & MenuItem::EDITABLE) | |
| 51 rv.is_editable = true; | |
| 52 | |
| 53 if (contexts & MenuItem::IMAGE) { | |
| 54 rv.src_url = GURL("http://test.image/"); | |
| 55 rv.media_type = blink::WebContextMenuData::MediaTypeImage; | |
| 56 } | |
| 57 | |
| 58 if (contexts & MenuItem::VIDEO) { | |
| 59 rv.src_url = GURL("http://test.video/"); | |
| 60 rv.media_type = blink::WebContextMenuData::MediaTypeVideo; | |
| 61 } | |
| 62 | |
| 63 if (contexts & MenuItem::AUDIO) { | |
| 64 rv.src_url = GURL("http://test.audio/"); | |
| 65 rv.media_type = blink::WebContextMenuData::MediaTypeAudio; | |
| 66 } | |
| 67 | |
| 68 if (contexts & MenuItem::FRAME) | |
| 69 rv.frame_url = GURL("http://test.frame/"); | |
| 70 | |
| 71 return rv; | |
| 72 } | |
| 73 | |
| 74 // Generates a URLPatternSet with a single pattern | |
| 75 static URLPatternSet CreatePatternSet(const std::string& pattern) { | |
| 76 URLPattern target(URLPattern::SCHEME_HTTP); | |
| 77 target.Parse(pattern); | |
| 78 | |
| 79 URLPatternSet rv; | |
| 80 rv.AddPattern(target); | |
| 81 | |
| 82 return rv; | |
| 83 } | |
| 84 | |
| 85 TEST_F(RenderViewContextMenuTest, TargetIgnoredForPage) { | |
| 86 content::ContextMenuParams params = CreateParams(0); | |
| 87 | |
| 88 MenuItem::ContextList contexts; | |
| 89 contexts.Add(MenuItem::PAGE); | |
| 90 | |
| 91 URLPatternSet patterns = CreatePatternSet("*://test.none/*"); | |
| 92 | |
| 93 EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 94 } | |
| 95 | |
| 96 TEST_F(RenderViewContextMenuTest, TargetCheckedForLink) { | |
| 97 content::ContextMenuParams params = CreateParams(MenuItem::LINK); | |
| 98 | |
| 99 MenuItem::ContextList contexts; | |
| 100 contexts.Add(MenuItem::PAGE); | |
| 101 contexts.Add(MenuItem::LINK); | |
| 102 | |
| 103 URLPatternSet patterns = CreatePatternSet("*://test.none/*"); | |
| 104 | |
| 105 EXPECT_FALSE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 106 } | |
| 107 | |
| 108 TEST_F(RenderViewContextMenuTest, TargetCheckedForImage) { | |
| 109 content::ContextMenuParams params = CreateParams(MenuItem::IMAGE); | |
| 110 | |
| 111 MenuItem::ContextList contexts; | |
| 112 contexts.Add(MenuItem::PAGE); | |
| 113 contexts.Add(MenuItem::IMAGE); | |
| 114 | |
| 115 URLPatternSet patterns = CreatePatternSet("*://test.none/*"); | |
| 116 | |
| 117 EXPECT_FALSE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 118 } | |
| 119 | |
| 120 TEST_F(RenderViewContextMenuTest, TargetCheckedForVideo) { | |
| 121 content::ContextMenuParams params = CreateParams(MenuItem::VIDEO); | |
| 122 | |
| 123 MenuItem::ContextList contexts; | |
| 124 contexts.Add(MenuItem::PAGE); | |
| 125 contexts.Add(MenuItem::VIDEO); | |
| 126 | |
| 127 URLPatternSet patterns = CreatePatternSet("*://test.none/*"); | |
| 128 | |
| 129 EXPECT_FALSE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 130 } | |
| 131 | |
| 132 TEST_F(RenderViewContextMenuTest, TargetCheckedForAudio) { | |
| 133 content::ContextMenuParams params = CreateParams(MenuItem::AUDIO); | |
| 134 | |
| 135 MenuItem::ContextList contexts; | |
| 136 contexts.Add(MenuItem::PAGE); | |
| 137 contexts.Add(MenuItem::AUDIO); | |
| 138 | |
| 139 URLPatternSet patterns = CreatePatternSet("*://test.none/*"); | |
| 140 | |
| 141 EXPECT_FALSE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 142 } | |
| 143 | |
| 144 TEST_F(RenderViewContextMenuTest, MatchWhenLinkedImageMatchesTarget) { | |
| 145 content::ContextMenuParams params = CreateParams(MenuItem::IMAGE | | |
| 146 MenuItem::LINK); | |
| 147 | |
| 148 MenuItem::ContextList contexts; | |
| 149 contexts.Add(MenuItem::LINK); | |
| 150 contexts.Add(MenuItem::IMAGE); | |
| 151 | |
| 152 URLPatternSet patterns = CreatePatternSet("*://test.link/*"); | |
| 153 | |
| 154 EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 155 } | |
| 156 | |
| 157 TEST_F(RenderViewContextMenuTest, MatchWhenLinkedImageMatchesSource) { | |
| 158 content::ContextMenuParams params = CreateParams(MenuItem::IMAGE | | |
| 159 MenuItem::LINK); | |
| 160 | |
| 161 MenuItem::ContextList contexts; | |
| 162 contexts.Add(MenuItem::LINK); | |
| 163 contexts.Add(MenuItem::IMAGE); | |
| 164 | |
| 165 URLPatternSet patterns = CreatePatternSet("*://test.image/*"); | |
| 166 | |
| 167 EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 168 } | |
| 169 | |
| 170 TEST_F(RenderViewContextMenuTest, NoMatchWhenLinkedImageMatchesNeither) { | |
| 171 content::ContextMenuParams params = CreateParams(MenuItem::IMAGE | | |
| 172 MenuItem::LINK); | |
| 173 | |
| 174 MenuItem::ContextList contexts; | |
| 175 contexts.Add(MenuItem::LINK); | |
| 176 contexts.Add(MenuItem::IMAGE); | |
| 177 | |
| 178 URLPatternSet patterns = CreatePatternSet("*://test.none/*"); | |
| 179 | |
| 180 EXPECT_FALSE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 181 } | |
| 182 | |
| 183 TEST_F(RenderViewContextMenuTest, TargetIgnoredForFrame) { | |
| 184 content::ContextMenuParams params = CreateParams(MenuItem::FRAME); | |
| 185 | |
| 186 MenuItem::ContextList contexts; | |
| 187 contexts.Add(MenuItem::FRAME); | |
| 188 | |
| 189 URLPatternSet patterns = CreatePatternSet("*://test.none/*"); | |
| 190 | |
| 191 EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 192 } | |
| 193 | |
| 194 TEST_F(RenderViewContextMenuTest, TargetIgnoredForEditable) { | |
| 195 content::ContextMenuParams params = CreateParams(MenuItem::EDITABLE); | |
| 196 | |
| 197 MenuItem::ContextList contexts; | |
| 198 contexts.Add(MenuItem::EDITABLE); | |
| 199 | |
| 200 URLPatternSet patterns = CreatePatternSet("*://test.none/*"); | |
| 201 | |
| 202 EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 203 } | |
| 204 | |
| 205 TEST_F(RenderViewContextMenuTest, TargetIgnoredForSelection) { | |
| 206 content::ContextMenuParams params = | |
| 207 CreateParams(MenuItem::SELECTION); | |
| 208 | |
| 209 MenuItem::ContextList contexts; | |
| 210 contexts.Add(MenuItem::SELECTION); | |
| 211 | |
| 212 URLPatternSet patterns = CreatePatternSet("*://test.none/*"); | |
| 213 | |
| 214 EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 215 } | |
| 216 | |
| 217 TEST_F(RenderViewContextMenuTest, TargetIgnoredForSelectionOnLink) { | |
| 218 content::ContextMenuParams params = CreateParams( | |
| 219 MenuItem::SELECTION | MenuItem::LINK); | |
| 220 | |
| 221 MenuItem::ContextList contexts; | |
| 222 contexts.Add(MenuItem::SELECTION); | |
| 223 contexts.Add(MenuItem::LINK); | |
| 224 | |
| 225 URLPatternSet patterns = CreatePatternSet("*://test.none/*"); | |
| 226 | |
| 227 EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 228 } | |
| 229 | |
| 230 TEST_F(RenderViewContextMenuTest, TargetIgnoredForSelectionOnImage) { | |
| 231 content::ContextMenuParams params = CreateParams( | |
| 232 MenuItem::SELECTION | MenuItem::IMAGE); | |
| 233 | |
| 234 MenuItem::ContextList contexts; | |
| 235 contexts.Add(MenuItem::SELECTION); | |
| 236 contexts.Add(MenuItem::IMAGE); | |
| 237 | |
| 238 URLPatternSet patterns = CreatePatternSet("*://test.none/*"); | |
| 239 | |
| 240 EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns)); | |
| 241 } | |
| 242 | |
| 243 class RenderViewContextMenuPrefsTest : public ChromeRenderViewHostTestHarness { | |
| 244 public: | |
| 245 virtual void SetUp() OVERRIDE { | |
| 246 ChromeRenderViewHostTestHarness::SetUp(); | |
| 247 registry_.reset(new ProtocolHandlerRegistry(profile(), NULL)); | |
| 248 } | |
| 249 | |
| 250 virtual void TearDown() OVERRIDE { | |
| 251 registry_.reset(); | |
| 252 ChromeRenderViewHostTestHarness::TearDown(); | |
| 253 } | |
| 254 | |
| 255 TestRenderViewContextMenu* CreateContextMenu() { | |
| 256 content::ContextMenuParams params = CreateParams(MenuItem::LINK); | |
| 257 params.unfiltered_link_url = params.link_url; | |
| 258 content::WebContents* wc = web_contents(); | |
| 259 TestRenderViewContextMenu* menu = new TestRenderViewContextMenu( | |
| 260 wc->GetMainFrame(), params); | |
| 261 // TestingProfile (returned by profile()) does not provide a protocol | |
| 262 // registry. | |
| 263 menu->protocol_handler_registry_ = registry_.get(); | |
| 264 menu->Init(); | |
| 265 return menu; | |
| 266 } | |
| 267 | |
| 268 private: | |
| 269 scoped_ptr<ProtocolHandlerRegistry> registry_; | |
| 270 }; | |
| 271 | |
| 272 // Verifies when Incognito Mode is not available (disabled by policy), | |
| 273 // Open Link in Incognito Window link in the context menu is disabled. | |
| 274 TEST_F(RenderViewContextMenuPrefsTest, | |
| 275 DisableOpenInIncognitoWindowWhenIncognitoIsDisabled) { | |
| 276 scoped_ptr<TestRenderViewContextMenu> menu(CreateContextMenu()); | |
| 277 | |
| 278 // Initially the Incognito mode is be enabled. So is the Open Link in | |
| 279 // Incognito Window link. | |
| 280 ASSERT_TRUE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKOFFTHERECORD)); | |
| 281 EXPECT_TRUE( | |
| 282 menu->IsCommandIdEnabled(IDC_CONTENT_CONTEXT_OPENLINKOFFTHERECORD)); | |
| 283 | |
| 284 // Disable Incognito mode. | |
| 285 IncognitoModePrefs::SetAvailability(profile()->GetPrefs(), | |
| 286 IncognitoModePrefs::DISABLED); | |
| 287 menu.reset(CreateContextMenu()); | |
| 288 ASSERT_TRUE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKOFFTHERECORD)); | |
| 289 EXPECT_FALSE( | |
| 290 menu->IsCommandIdEnabled(IDC_CONTENT_CONTEXT_OPENLINKOFFTHERECORD)); | |
| 291 } | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 178193005:
Move render_view_context_menu.* related files out of tab_contents. (Closed)
Base URL: https://chromium.googlesource.com/chromium/src.git@master