OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/common/content_settings.h" | 5 #include "chrome/common/content_settings.h" |
6 #include "chrome/common/render_messages.h" | 6 #include "chrome/common/render_messages.h" |
7 #include "chrome/renderer/content_settings_observer.h" | 7 #include "chrome/renderer/content_settings_observer.h" |
8 #include "chrome/test/base/chrome_render_view_test.h" | 8 #include "chrome/test/base/chrome_render_view_test.h" |
9 #include "content/public/renderer/render_view.h" | 9 #include "content/public/renderer/render_view.h" |
10 #include "ipc/ipc_message_macros.h" | 10 #include "ipc/ipc_message_macros.h" |
(...skipping 327 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
338 ContentSettingsForOneType& image_setting_rules = | 338 ContentSettingsForOneType& image_setting_rules = |
339 content_setting_rules.image_rules; | 339 content_setting_rules.image_rules; |
340 image_setting_rules.push_back( | 340 image_setting_rules.push_back( |
341 ContentSettingPatternSource( | 341 ContentSettingPatternSource( |
342 ContentSettingsPattern::Wildcard(), | 342 ContentSettingsPattern::Wildcard(), |
343 ContentSettingsPattern::Wildcard(), | 343 ContentSettingsPattern::Wildcard(), |
344 CONTENT_SETTING_BLOCK, "", false)); | 344 CONTENT_SETTING_BLOCK, "", false)); |
345 | 345 |
346 ContentSettingsObserver* observer = ContentSettingsObserver::Get(view_); | 346 ContentSettingsObserver* observer = ContentSettingsObserver::Get(view_); |
347 observer->SetContentSettingRules(&content_setting_rules); | 347 observer->SetContentSettingRules(&content_setting_rules); |
348 observer->SetAsInterstitial(); | 348 observer->OnSetAsInterstitial(); |
349 | 349 |
350 // Load a page which contains a script. | 350 // Load a page which contains a script. |
351 std::string html = "<html>" | 351 std::string html = "<html>" |
352 "<head>" | 352 "<head>" |
353 "<script src='data:foo'></script>" | 353 "<script src='data:foo'></script>" |
354 "</head>" | 354 "</head>" |
355 "<body>" | 355 "<body>" |
356 "</body>" | 356 "</body>" |
357 "</html>"; | 357 "</html>"; |
358 LoadHTML(html.c_str()); | 358 LoadHTML(html.c_str()); |
359 | 359 |
360 // Verify that the script was allowed. | 360 // Verify that the script was allowed. |
361 bool was_blocked = false; | 361 bool was_blocked = false; |
362 for (size_t i = 0; i < render_thread_->sink().message_count(); ++i) { | 362 for (size_t i = 0; i < render_thread_->sink().message_count(); ++i) { |
363 const IPC::Message* msg = render_thread_->sink().GetMessageAt(i); | 363 const IPC::Message* msg = render_thread_->sink().GetMessageAt(i); |
364 if (msg->type() == ChromeViewHostMsg_ContentBlocked::ID) | 364 if (msg->type() == ChromeViewHostMsg_ContentBlocked::ID) |
365 was_blocked = true; | 365 was_blocked = true; |
366 } | 366 } |
367 EXPECT_FALSE(was_blocked); | 367 EXPECT_FALSE(was_blocked); |
368 | 368 |
369 // Verify that images are allowed. | 369 // Verify that images are allowed. |
370 EXPECT_CALL( | 370 EXPECT_CALL( |
371 mock_observer, | 371 mock_observer, |
372 OnContentBlocked(CONTENT_SETTINGS_TYPE_IMAGES, std::string())).Times(0); | 372 OnContentBlocked(CONTENT_SETTINGS_TYPE_IMAGES, std::string())).Times(0); |
373 EXPECT_TRUE(observer->AllowImage(GetMainFrame(), true, | 373 EXPECT_TRUE(observer->AllowImage(GetMainFrame(), true, |
374 mock_observer.image_url_)); | 374 mock_observer.image_url_)); |
375 ::testing::Mock::VerifyAndClearExpectations(&observer); | 375 ::testing::Mock::VerifyAndClearExpectations(&observer); |
376 } | 376 } |
OLD | NEW |