Whitelist chrome.screenlockPrivate only for component apps on ChromeOS.

chrome/browser/extensions/api/screenlock_private/screenlock_private_apitest.cc

 // Copyright 2013 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/browser/chrome_notification_types.h"
 #include "chrome/browser/extensions/api/screenlock_private/screenlock_private_api.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/browser/signin/signin_manager_factory.h"
 #include "components/signin/core/browser/signin_manager.h"
 #include "components/signin/core/common/profile_management_switches.h"
 #include "content/public/browser/notification_service.h"
 #include "extensions/browser/api/test/test_api.h"
+#include "extensions/common/switches.h"
 
 namespace extensions {
 
 namespace {
 
+const char kAttemptClickAuthMessage[] = "attemptClickAuth";
+const char kTestExtensionId[] = "lkegkdgachcnekllcdfkijonogckdnjo";
 const char kTestUser[] = "testuser@gmail.com";
-const char kAttemptClickAuthMessage[] = "attemptClickAuth";
 
 }  // namespace
 
 class ScreenlockPrivateApiTest : public ExtensionApiTest,
                                  public content::NotificationObserver {
  public:
   ScreenlockPrivateApiTest() {}
 
   virtual ~ScreenlockPrivateApiTest() {}
 
   // ExtensionApiTest
   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     ExtensionApiTest::SetUpCommandLine(command_line);
+    command_line->AppendSwitchASCII(
+        extensions::switches::kWhitelistedExtensionID, kTestExtensionId);
 
 #if !defined(OS_CHROMEOS)
     // New profile management needs to be on for non-ChromeOS lock.
-    switches::EnableNewProfileManagementForTesting(command_line);
+    ::switches::EnableNewProfileManagementForTesting(command_line);
 #endif
   }
 
   virtual void SetUpOnMainThread() OVERRIDE {
     SigninManagerFactory::GetForProfile(profile())
         ->SetAuthenticatedUsername(kTestUser);
     ExtensionApiTest::SetUpOnMainThread();
   }
 
  protected:
@@ skipping 13 matching lines @@
     const std::string& content = *content::Details<std::string>(details).ptr();
     if (content == kAttemptClickAuthMessage) {
       extensions::ScreenlockPrivateEventRouter* router =
           extensions::ScreenlockPrivateEventRouter::GetFactoryInstance()->Get(
               profile());
       router->OnAuthAttempted(
           ScreenlockBridge::Get()->lock_handler()->GetAuthType(kTestUser), "");
     }
   }
 
+  // Loads |extension_name| as appropriate for the platform and waits for a
+  // pass / fail notification.
+  void RunTest(const std::string& extension_name) {
+#if defined(OS_CHROMEOS)
+    ASSERT_TRUE(RunComponentExtensionTest(extension_name)) << message_;
+#else
+    ASSERT_TRUE(RunExtensionTest(extension_name)) << message_;
+#endif
+  }
+
  private:
   content::NotificationRegistrar registrar_;
 
   DISALLOW_COPY_AND_ASSIGN(ScreenlockPrivateApiTest);
 };
 
 IN_PROC_BROWSER_TEST_F(ScreenlockPrivateApiTest, LockUnlock) {
-  ASSERT_TRUE(RunExtensionTest("screenlock_private/lock_unlock")) << message_;
+  RunTest("screenlock_private/lock_unlock");
 }
 
 IN_PROC_BROWSER_TEST_F(ScreenlockPrivateApiTest, AuthType) {
-  ASSERT_TRUE(RunExtensionTest("screenlock_private/auth_type")) << message_;
+  RunTest("screenlock_private/auth_type");
 }
 
 }  // namespace extensions