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