Index: chrome/browser/content_settings/permission_context_base_unittest.cc |
diff --git a/chrome/browser/content_settings/permission_context_base_unittest.cc b/chrome/browser/content_settings/permission_context_base_unittest.cc |
index 22ce437a905dbc93ef65417a5a624c4ee39fada5..aa29454f2ffb3a7630720bc9ff42b73a3051787f 100644 |
--- a/chrome/browser/content_settings/permission_context_base_unittest.cc |
+++ b/chrome/browser/content_settings/permission_context_base_unittest.cc |
@@ -9,6 +9,7 @@ |
#include "chrome/browser/content_settings/permission_queue_controller.h" |
#include "chrome/browser/content_settings/permission_request_id.h" |
#include "chrome/browser/infobars/infobar_service.h" |
+#include "chrome/browser/ui/website_settings/permission_bubble_manager.h" |
#include "chrome/common/content_settings.h" |
#include "chrome/test/base/chrome_render_view_host_test_harness.h" |
#include "chrome/test/base/testing_profile.h" |
@@ -23,11 +24,22 @@ class PermissionContextBaseTests : public ChromeRenderViewHostTestHarness { |
PermissionContextBaseTests() {} |
virtual ~PermissionContextBaseTests() {} |
+ // Accept or dismiss the permission bubble. |
+ void RespondPermissionBubble(bool accept) { |
+ PermissionBubbleManager* manager = |
+ PermissionBubbleManager::FromWebContents(web_contents()); |
+ if (accept) |
+ manager->Accept(); |
+ else |
+ manager->Closing(); |
+ } |
+ |
private: |
// ChromeRenderViewHostTestHarness: |
virtual void SetUp() OVERRIDE { |
ChromeRenderViewHostTestHarness::SetUp(); |
InfoBarService::CreateForWebContents(web_contents()); |
+ PermissionBubbleManager::CreateForWebContents(web_contents()); |
} |
DISALLOW_COPY_AND_ASSIGN(PermissionContextBaseTests); |
@@ -96,8 +108,12 @@ TEST_F(PermissionContextBaseTests, TestAskAndGrant) { |
base::Bind(&TestPermissionContext::TrackPermissionDecision, |
base::Unretained(&permission_context))); |
- permission_context.GetInfoBarController()->OnPermissionSet( |
- id, url, url, true, true); |
+ if (!PermissionBubbleManager::Enabled()) { |
+ permission_context.GetInfoBarController()->OnPermissionSet( |
+ id, url, url, true, true); |
+ } else { |
+ RespondPermissionBubble(true); |
+ } |
EXPECT_TRUE(permission_context.permission_set()); |
EXPECT_TRUE(permission_context.permission_granted()); |
EXPECT_TRUE(permission_context.tab_context_updated()); |
@@ -127,8 +143,12 @@ TEST_F(PermissionContextBaseTests, TestAskAndDismiss) { |
base::Bind(&TestPermissionContext::TrackPermissionDecision, |
base::Unretained(&permission_context))); |
- permission_context.GetInfoBarController()->OnPermissionSet( |
- id, url, url, false, false); |
+ if (!PermissionBubbleManager::Enabled()) { |
+ permission_context.GetInfoBarController()->OnPermissionSet( |
+ id, url, url, false, false); |
+ } else { |
+ RespondPermissionBubble(false); |
+ } |
EXPECT_TRUE(permission_context.permission_set()); |
EXPECT_FALSE(permission_context.permission_granted()); |
EXPECT_TRUE(permission_context.tab_context_updated()); |