Add infobar for PPAPI broker usage.

chrome/test/ppapi/ppapi_browsertest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/test/ppapi/ppapi_test.h"
 
 #include "base/test/test_timeouts.h"
 #include "build/build_config.h"
+#include "chrome/browser/content_settings/host_content_settings_map.h"
+#include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_navigator.h"
 #include "chrome/browser/ui/browser_tabstrip.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/url_constants.h"
 #include "content/public/test/test_renderer_host.h"
 
 using content::RenderViewHost;
+using ::testing::Return;
+using ::testing::StrictMock;
+using ::testing::_;
 
 // This macro finesses macro expansion to do what we want.
 #define STRIP_PREFIXES(test_name) StripPrefixes(#test_name)
 
 // Use these macros to run the tests for a specific interface.
 // Most interfaces should be tested with both macros.
 #define TEST_PPAPI_IN_PROCESS(test_name) \
     IN_PROC_BROWSER_TEST_F(PPAPITest, test_name) { \
       RunTest(STRIP_PREFIXES(test_name)); \
     }
@@ skipping 100 matching lines @@
 
 // Disable tests under ASAN.  http://crbug.com/104832.
 // This is a bit heavy handed, but the majority of these tests fail under ASAN.
 // See bug for history.
 #if !defined(ADDRESS_SANITIZER)
 
 TEST_PPAPI_IN_PROCESS(Broker)
 // Flaky, http://crbug.com/111355
 TEST_PPAPI_OUT_OF_PROCESS(DISABLED_Broker)
 
+IN_PROC_BROWSER_TEST_F(PPAPIBrokerInfoBarTest, Accept) {
+  // Accepting the infobar should grant permission to access the PPAPI broker.
+  StrictMock<InfoBarObserver> observer;
+  EXPECT_CALL(observer, ShouldAcceptInfoBar(_)).WillOnce(Return(true));
+
+  GURL url = GetTestFileUrl("Broker_ConnectPermissionGranted");
+  RunTestURL(url);
+
+  // It should also set a content settings exception for the site.
+  HostContentSettingsMap* content_settings =
+      browser()->profile()->GetHostContentSettingsMap();
+  EXPECT_EQ(CONTENT_SETTING_ALLOW,
+            content_settings->GetContentSetting(
+                url, url, CONTENT_SETTINGS_TYPE_PPAPI_BROKER, std::string()));
+}
+
+IN_PROC_BROWSER_TEST_F(PPAPIBrokerInfoBarTest, Deny) {
+  // Canceling the infobar should deny permission to access the PPAPI broker.
+  StrictMock<InfoBarObserver> observer;
+  EXPECT_CALL(observer, ShouldAcceptInfoBar(_)).WillOnce(Return(false));
+
+  GURL url = GetTestFileUrl("Broker_ConnectPermissionDenied");
+  RunTestURL(url);
+
+  // It should *not* set a content settings exception for the site.
+  HostContentSettingsMap* content_settings =
+      browser()->profile()->GetHostContentSettingsMap();
+  EXPECT_EQ(CONTENT_SETTING_ASK,
+            content_settings->GetContentSetting(
+                url, url, CONTENT_SETTINGS_TYPE_PPAPI_BROKER, std::string()));
+}
+
+IN_PROC_BROWSER_TEST_F(PPAPIBrokerInfoBarTest, Blocked) {
+  // Block access to the PPAPI broker.
+  browser()->profile()->GetHostContentSettingsMap()->SetDefaultContentSetting(
+      CONTENT_SETTINGS_TYPE_PPAPI_BROKER, CONTENT_SETTING_BLOCK);
+
+  // We shouldn't see an infobar.
+  StrictMock<InfoBarObserver> observer;
+
+  RunTest("Broker_ConnectPermissionDenied");
+}
+
+IN_PROC_BROWSER_TEST_F(PPAPIBrokerInfoBarTest, Allowed) {
+  // Always allow access to the PPAPI broker.
+  browser()->profile()->GetHostContentSettingsMap()->SetDefaultContentSetting(
+      CONTENT_SETTINGS_TYPE_PPAPI_BROKER, CONTENT_SETTING_ALLOW);
+
+  // We shouldn't see an infobar.
+  StrictMock<InfoBarObserver> observer;
+
+  RunTest("Broker_ConnectPermissionGranted");
+}
+
 TEST_PPAPI_IN_PROCESS(Core)
 TEST_PPAPI_OUT_OF_PROCESS(Core)
 TEST_PPAPI_NACL_VIA_HTTP(Core)
 
 // Times out on Linux. http://crbug.com/108859
 #if defined(OS_LINUX)
 #define MAYBE_InputEvent DISABLED_InputEvent
 #elif defined(OS_MACOSX)
 // Flaky on Mac. http://crbug.com/109258
 #define MAYBE_InputEvent DISABLED_InputEvent
@@ skipping 654 matching lines @@
 TEST_PPAPI_OUT_OF_PROCESS(FlashMessageLoop_RunWithoutQuit)
 
 TEST_PPAPI_IN_PROCESS(MouseCursor)
 TEST_PPAPI_OUT_OF_PROCESS(MouseCursor)
 TEST_PPAPI_NACL_VIA_HTTP(MouseCursor)
 
 // Only enabled in out-of-process mode.
 TEST_PPAPI_OUT_OF_PROCESS(FlashFile_CreateTemporaryFile)
 
 #endif // ADDRESS_SANITIZER