Home | History | Annotate | Download | only in screenlock_private 
      1 // Copyright 2013 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #include "chrome/browser/chrome_notification_types.h"
      6 #include "chrome/browser/extensions/api/screenlock_private/screenlock_private_api.h"
      7 #include "chrome/browser/extensions/extension_apitest.h"
      8 #include "chrome/browser/signin/signin_manager_factory.h"
      9 #include "components/signin/core/browser/signin_manager.h"
     10 #include "components/signin/core/common/profile_management_switches.h"
     11 #include "content/public/browser/notification_service.h"
     12 #include "extensions/browser/api/test/test_api.h"
     13 
     14 namespace extensions {
     15 
     16 namespace {
     17 
     18 const char kTestUser[] = "testuser (at) gmail.com";
     19 const char kAttemptClickAuthMessage[] = "attemptClickAuth";
     20 
     21 }  // namespace
     22 
     23 class ScreenlockPrivateApiTest : public ExtensionApiTest,
     24                                  public content::NotificationObserver {
     25  public:
     26   ScreenlockPrivateApiTest() {}
     27 
     28   virtual ~ScreenlockPrivateApiTest() {}
     29 
     30   // ExtensionApiTest
     31   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     32     ExtensionApiTest::SetUpCommandLine(command_line);
     33 
     34 #if !defined(OS_CHROMEOS)
     35     // New profile management needs to be on for non-ChromeOS lock.
     36     switches::EnableNewProfileManagementForTesting(command_line);
     37 #endif
     38   }
     39 
     40   virtual void SetUpOnMainThread() OVERRIDE {
     41     SigninManagerFactory::GetForProfile(profile())
     42         ->SetAuthenticatedUsername(kTestUser);
     43     ExtensionApiTest::SetUpOnMainThread();
     44   }
     45 
     46  protected:
     47   // ExtensionApiTest override:
     48   virtual void RunTestOnMainThreadLoop() OVERRIDE {
     49     registrar_.Add(this,
     50                    chrome::NOTIFICATION_EXTENSION_TEST_MESSAGE,
     51                    content::NotificationService::AllSources());
     52     ExtensionApiTest::RunTestOnMainThreadLoop();
     53     registrar_.RemoveAll();
     54   }
     55 
     56   // content::NotificationObserver override:
     57   virtual void Observe(int type,
     58                        const content::NotificationSource& source,
     59                        const content::NotificationDetails& details) OVERRIDE {
     60     const std::string& content = *content::Details<std::string>(details).ptr();
     61     if (content == kAttemptClickAuthMessage) {
     62       extensions::ScreenlockPrivateEventRouter* router =
     63           extensions::ScreenlockPrivateEventRouter::GetFactoryInstance()->Get(
     64               profile());
     65       router->OnAuthAttempted(
     66           ScreenlockBridge::Get()->lock_handler()->GetAuthType(kTestUser), "");
     67     }
     68   }
     69 
     70  private:
     71   content::NotificationRegistrar registrar_;
     72 
     73   DISALLOW_COPY_AND_ASSIGN(ScreenlockPrivateApiTest);
     74 };
     75 
     76 IN_PROC_BROWSER_TEST_F(ScreenlockPrivateApiTest, LockUnlock) {
     77   ASSERT_TRUE(RunExtensionTest("screenlock_private/lock_unlock")) << message_;
     78 }
     79 
     80 IN_PROC_BROWSER_TEST_F(ScreenlockPrivateApiTest, AuthType) {
     81   ASSERT_TRUE(RunExtensionTest("screenlock_private/auth_type")) << message_;
     82 }
     83 
     84 }  // namespace extensions
     85