| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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/browser/content_settings/host_content_settings_map.h" | 5 #include "chrome/browser/content_settings/host_content_settings_map.h" |
| 6 #include "chrome/browser/content_settings/tab_specific_content_settings.h" | 6 #include "chrome/browser/content_settings/tab_specific_content_settings.h" |
| 7 #include "chrome/browser/prerender/prerender_manager.h" | 7 #include "chrome/browser/prerender/prerender_manager.h" |
| 8 #include "chrome/browser/profiles/profile.h" | 8 #include "chrome/browser/profiles/profile.h" |
| 9 #include "chrome/browser/ui/content_settings/content_setting_image_model.h" | 9 #include "chrome/browser/ui/content_settings/content_setting_image_model.h" |
| 10 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h" | 10 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h" |
| 11 #include "chrome/browser/ui/tab_contents/test_tab_contents_wrapper.h" | 11 #include "chrome/browser/ui/tab_contents/test_tab_contents.h" |
| 12 #include "chrome/test/base/testing_profile.h" | 12 #include "chrome/test/base/testing_profile.h" |
| 13 #include "content/public/test/test_browser_thread.h" | 13 #include "content/public/test/test_browser_thread.h" |
| 14 #include "content/public/test/test_renderer_host.h" | 14 #include "content/public/test/test_renderer_host.h" |
| 15 #include "net/cookies/cookie_options.h" | 15 #include "net/cookies/cookie_options.h" |
| 16 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 17 | 17 |
| 18 using content::BrowserThread; | 18 using content::BrowserThread; |
| 19 | 19 |
| 20 class ContentSettingImageModelTest : public TabContentsWrapperTestHarness { | 20 class ContentSettingImageModelTest : public TabContentsTestHarness { |
| 21 public: | 21 public: |
| 22 ContentSettingImageModelTest() | 22 ContentSettingImageModelTest() |
| 23 : ui_thread_(BrowserThread::UI, &message_loop_) {} | 23 : ui_thread_(BrowserThread::UI, &message_loop_) {} |
| 24 | 24 |
| 25 private: | 25 private: |
| 26 content::TestBrowserThread ui_thread_; | 26 content::TestBrowserThread ui_thread_; |
| 27 | 27 |
| 28 DISALLOW_COPY_AND_ASSIGN(ContentSettingImageModelTest); | 28 DISALLOW_COPY_AND_ASSIGN(ContentSettingImageModelTest); |
| 29 }; | 29 }; |
| 30 | 30 |
| 31 TEST_F(ContentSettingImageModelTest, UpdateFromWebContents) { | 31 TEST_F(ContentSettingImageModelTest, UpdateFromWebContents) { |
| 32 TabSpecificContentSettings* content_settings = | 32 TabSpecificContentSettings* content_settings = |
| 33 contents_wrapper()->content_settings(); | 33 tab_contents()->content_settings(); |
| 34 scoped_ptr<ContentSettingImageModel> content_setting_image_model( | 34 scoped_ptr<ContentSettingImageModel> content_setting_image_model( |
| 35 ContentSettingImageModel::CreateContentSettingImageModel( | 35 ContentSettingImageModel::CreateContentSettingImageModel( |
| 36 CONTENT_SETTINGS_TYPE_IMAGES)); | 36 CONTENT_SETTINGS_TYPE_IMAGES)); |
| 37 EXPECT_FALSE(content_setting_image_model->is_visible()); | 37 EXPECT_FALSE(content_setting_image_model->is_visible()); |
| 38 EXPECT_EQ(0, content_setting_image_model->get_icon()); | 38 EXPECT_EQ(0, content_setting_image_model->get_icon()); |
| 39 EXPECT_TRUE(content_setting_image_model->get_tooltip().empty()); | 39 EXPECT_TRUE(content_setting_image_model->get_tooltip().empty()); |
| 40 | 40 |
| 41 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_IMAGES, | 41 content_settings->OnContentBlocked(CONTENT_SETTINGS_TYPE_IMAGES, |
| 42 std::string()); | 42 std::string()); |
| 43 content_setting_image_model->UpdateFromWebContents(contents()); | 43 content_setting_image_model->UpdateFromWebContents(contents()); |
| 44 | 44 |
| 45 EXPECT_TRUE(content_setting_image_model->is_visible()); | 45 EXPECT_TRUE(content_setting_image_model->is_visible()); |
| 46 EXPECT_NE(0, content_setting_image_model->get_icon()); | 46 EXPECT_NE(0, content_setting_image_model->get_icon()); |
| 47 EXPECT_FALSE(content_setting_image_model->get_tooltip().empty()); | 47 EXPECT_FALSE(content_setting_image_model->get_tooltip().empty()); |
| 48 } | 48 } |
| 49 | 49 |
| 50 TEST_F(ContentSettingImageModelTest, CookieAccessed) { | 50 TEST_F(ContentSettingImageModelTest, CookieAccessed) { |
| 51 TabSpecificContentSettings* content_settings = | 51 TabSpecificContentSettings* content_settings = |
| 52 contents_wrapper()->content_settings(); | 52 tab_contents()->content_settings(); |
| 53 profile()->GetHostContentSettingsMap()->SetDefaultContentSetting( | 53 profile()->GetHostContentSettingsMap()->SetDefaultContentSetting( |
| 54 CONTENT_SETTINGS_TYPE_COOKIES, CONTENT_SETTING_BLOCK); | 54 CONTENT_SETTINGS_TYPE_COOKIES, CONTENT_SETTING_BLOCK); |
| 55 scoped_ptr<ContentSettingImageModel> content_setting_image_model( | 55 scoped_ptr<ContentSettingImageModel> content_setting_image_model( |
| 56 ContentSettingImageModel::CreateContentSettingImageModel( | 56 ContentSettingImageModel::CreateContentSettingImageModel( |
| 57 CONTENT_SETTINGS_TYPE_COOKIES)); | 57 CONTENT_SETTINGS_TYPE_COOKIES)); |
| 58 EXPECT_FALSE(content_setting_image_model->is_visible()); | 58 EXPECT_FALSE(content_setting_image_model->is_visible()); |
| 59 EXPECT_EQ(0, content_setting_image_model->get_icon()); | 59 EXPECT_EQ(0, content_setting_image_model->get_icon()); |
| 60 EXPECT_TRUE(content_setting_image_model->get_tooltip().empty()); | 60 EXPECT_TRUE(content_setting_image_model->get_tooltip().empty()); |
| 61 | 61 |
| 62 net::CookieOptions options; | 62 net::CookieOptions options; |
| 63 content_settings->OnCookieChanged(GURL("http://google.com"), | 63 content_settings->OnCookieChanged(GURL("http://google.com"), |
| 64 GURL("http://google.com"), | 64 GURL("http://google.com"), |
| 65 "A=B", | 65 "A=B", |
| 66 options, | 66 options, |
| 67 false); | 67 false); |
| 68 content_setting_image_model->UpdateFromWebContents(contents()); | 68 content_setting_image_model->UpdateFromWebContents(contents()); |
| 69 EXPECT_TRUE(content_setting_image_model->is_visible()); | 69 EXPECT_TRUE(content_setting_image_model->is_visible()); |
| 70 EXPECT_NE(0, content_setting_image_model->get_icon()); | 70 EXPECT_NE(0, content_setting_image_model->get_icon()); |
| 71 EXPECT_FALSE(content_setting_image_model->get_tooltip().empty()); | 71 EXPECT_FALSE(content_setting_image_model->get_tooltip().empty()); |
| 72 } | 72 } |
| OLD | NEW |