OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/chrome_notification_types.h" | 5 #include "chrome/browser/chrome_notification_types.h" |
6 #include "chrome/browser/chromeos/login/auth/user_context.h" | 6 #include "chrome/browser/extensions/api/screenlock_private/screenlock_private_ap
i.h" |
7 #include "chrome/browser/chromeos/login/lock/screen_locker.h" | |
8 #include "chrome/browser/chromeos/login/users/user.h" | |
9 #include "chrome/browser/extensions/extension_apitest.h" | 7 #include "chrome/browser/extensions/extension_apitest.h" |
10 #include "chrome/browser/signin/signin_manager_factory.h" | 8 #include "chrome/browser/signin/signin_manager_factory.h" |
11 #include "components/signin/core/browser/signin_manager.h" | 9 #include "components/signin/core/browser/signin_manager.h" |
| 10 #include "components/signin/core/common/signin_switches.h" |
12 #include "content/public/browser/notification_service.h" | 11 #include "content/public/browser/notification_service.h" |
13 #include "extensions/browser/api/test/test_api.h" | 12 #include "extensions/browser/api/test/test_api.h" |
14 | 13 |
15 namespace extensions { | 14 namespace extensions { |
16 | 15 |
17 namespace { | 16 namespace { |
18 | 17 |
| 18 const char kTestUser[] = "testuser@gmail.com"; |
19 const char kAttemptClickAuthMessage[] = "attemptClickAuth"; | 19 const char kAttemptClickAuthMessage[] = "attemptClickAuth"; |
20 | 20 |
21 } // namespace | 21 } // namespace |
22 | 22 |
23 class ScreenlockPrivateApiTest : public ExtensionApiTest, | 23 class ScreenlockPrivateApiTest : public ExtensionApiTest, |
24 public content::NotificationObserver { | 24 public content::NotificationObserver { |
25 public: | 25 public: |
26 ScreenlockPrivateApiTest() {} | 26 ScreenlockPrivateApiTest() {} |
27 | 27 |
28 virtual ~ScreenlockPrivateApiTest() {} | 28 virtual ~ScreenlockPrivateApiTest() {} |
29 | 29 |
30 // ExtensionApiTest | 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 command_line->AppendSwitch(switches::kNewProfileManagement); |
| 37 #endif |
| 38 } |
| 39 |
31 virtual void SetUpOnMainThread() OVERRIDE { | 40 virtual void SetUpOnMainThread() OVERRIDE { |
32 SigninManagerFactory::GetForProfile(profile()) | 41 SigninManagerFactory::GetForProfile(profile()) |
33 ->SetAuthenticatedUsername("testuser@gmail.com"); | 42 ->SetAuthenticatedUsername(kTestUser); |
34 ExtensionApiTest::SetUpOnMainThread(); | 43 ExtensionApiTest::SetUpOnMainThread(); |
35 } | 44 } |
36 | 45 |
37 protected: | 46 protected: |
38 chromeos::ScreenLocker* GetScreenLocker() { | |
39 chromeos::ScreenLocker* locker = | |
40 chromeos::ScreenLocker::default_screen_locker(); | |
41 EXPECT_TRUE(locker); | |
42 return locker; | |
43 } | |
44 | |
45 // ExtensionApiTest override: | 47 // ExtensionApiTest override: |
46 virtual void RunTestOnMainThreadLoop() OVERRIDE { | 48 virtual void RunTestOnMainThreadLoop() OVERRIDE { |
47 registrar_.Add(this, | 49 registrar_.Add(this, |
48 chrome::NOTIFICATION_EXTENSION_TEST_MESSAGE, | 50 chrome::NOTIFICATION_EXTENSION_TEST_MESSAGE, |
49 content::NotificationService::AllSources()); | 51 content::NotificationService::AllSources()); |
50 ExtensionApiTest::RunTestOnMainThreadLoop(); | 52 ExtensionApiTest::RunTestOnMainThreadLoop(); |
51 registrar_.RemoveAll(); | 53 registrar_.RemoveAll(); |
52 } | 54 } |
53 | 55 |
54 // content::NotificationObserver override: | 56 // content::NotificationObserver override: |
55 virtual void Observe(int type, | 57 virtual void Observe(int type, |
56 const content::NotificationSource& source, | 58 const content::NotificationSource& source, |
57 const content::NotificationDetails& details) OVERRIDE { | 59 const content::NotificationDetails& details) OVERRIDE { |
58 const std::string& content = *content::Details<std::string>(details).ptr(); | 60 const std::string& content = *content::Details<std::string>(details).ptr(); |
59 if (content == kAttemptClickAuthMessage) { | 61 if (content == kAttemptClickAuthMessage) { |
60 chromeos::ScreenLocker* locker = GetScreenLocker(); | 62 extensions::ScreenlockPrivateEventRouter* router = |
61 const chromeos::UserList& users = locker->users(); | 63 extensions::ScreenlockPrivateEventRouter::GetFactoryInstance()->Get( |
62 EXPECT_GE(1u, users.size()); | 64 profile()); |
63 GetScreenLocker()->Authenticate(chromeos::UserContext(users[0]->email())); | 65 router->OnAuthAttempted( |
| 66 ScreenlockBridge::Get()->lock_handler()->GetAuthType(kTestUser), ""); |
64 } | 67 } |
65 } | 68 } |
66 | 69 |
| 70 private: |
67 content::NotificationRegistrar registrar_; | 71 content::NotificationRegistrar registrar_; |
| 72 |
| 73 DISALLOW_COPY_AND_ASSIGN(ScreenlockPrivateApiTest); |
68 }; | 74 }; |
69 | 75 |
70 IN_PROC_BROWSER_TEST_F(ScreenlockPrivateApiTest, LockUnlock) { | 76 IN_PROC_BROWSER_TEST_F(ScreenlockPrivateApiTest, LockUnlock) { |
71 ASSERT_TRUE(RunExtensionTest("screenlock_private/lock_unlock")) << message_; | 77 ASSERT_TRUE(RunExtensionTest("screenlock_private/lock_unlock")) << message_; |
72 } | 78 } |
73 | 79 |
74 IN_PROC_BROWSER_TEST_F(ScreenlockPrivateApiTest, AuthType) { | 80 IN_PROC_BROWSER_TEST_F(ScreenlockPrivateApiTest, AuthType) { |
75 ASSERT_TRUE(RunExtensionTest("screenlock_private/auth_type")) << message_; | 81 ASSERT_TRUE(RunExtensionTest("screenlock_private/auth_type")) << message_; |
76 } | 82 } |
77 | 83 |
78 } // namespace extensions | 84 } // namespace extensions |
OLD | NEW |