OLD | NEW |
---|---|
(Empty) | |
1 // Copyright (c) 2010 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 "base/message_loop.h" | |
6 #include "chrome/browser/browser_thread.h" | |
7 #include "chrome/browser/chromeos/cros/mock_cryptohome_library.h" | |
8 #include "chrome/browser/chromeos/cros/mock_login_library.h" | |
9 #include "chrome/browser/chromeos/cros/mock_network_library.h" | |
10 #include "chrome/browser/chromeos/login/login_screen.h" | |
11 #include "chrome/browser/chromeos/login/mock_authenticator.h" | |
12 #include "chrome/browser/chromeos/login/mock_screen_observer.h" | |
13 #include "chrome/browser/chromeos/login/wizard_controller.h" | |
14 #include "chrome/browser/chromeos/login/wizard_in_process_browser_test.h" | |
15 #include "chrome/browser/chromeos/login/wizard_screen.h" | |
16 #include "grit/generated_resources.h" | |
17 #include "testing/gmock/include/gmock/gmock.h" | |
18 #include "testing/gtest/include/gtest/gtest.h" | |
19 | |
20 namespace chromeos { | |
21 | |
22 using ::testing::AnyNumber; | |
23 using ::testing::InvokeWithoutArgs; | |
24 using ::testing::Return; | |
25 | |
26 const char kUsername[] = "test_user@gmail.com"; | |
27 const char kPassword[] = "test_password"; | |
28 | |
29 class LoginScreenTest : public WizardInProcessBrowserTest { | |
30 public: | |
31 LoginScreenTest(): WizardInProcessBrowserTest("login"), | |
32 mock_cryptohome_library_(NULL), | |
33 mock_login_library_(NULL), | |
34 mock_network_library_(NULL) { | |
35 } | |
36 | |
37 protected: | |
38 MockCryptohomeLibrary *mock_cryptohome_library_; | |
39 MockLoginLibrary *mock_login_library_; | |
40 MockNetworkLibrary *mock_network_library_; | |
41 | |
42 virtual void SetUpInProcessBrowserTestFixture() { | |
43 WizardInProcessBrowserTest::SetUpInProcessBrowserTestFixture(); | |
44 cros_mock_->InitStatusAreaMocks(); | |
45 cros_mock_->SetStatusAreaMocksExpectations(); | |
46 | |
47 mock_network_library_ = cros_mock_->mock_network_library(); | |
48 | |
49 mock_login_library_ = new MockLoginLibrary(); | |
50 EXPECT_CALL(*mock_login_library_, EmitLoginPromptReady()) | |
51 .Times(1); | |
52 cros_mock_->test_api()->SetLoginLibrary(mock_login_library_, true); | |
53 | |
54 cros_mock_->InitMockCryptohomeLibrary(); | |
55 mock_cryptohome_library_ = cros_mock_->mock_cryptohome_library(); | |
56 EXPECT_CALL(*mock_cryptohome_library_, IsMounted()) | |
57 .Times(AnyNumber()) | |
58 .WillRepeatedly((Return(true))); | |
59 LoginUtils::Set(new MockLoginUtils(kUsername, kPassword)); | |
60 } | |
61 | |
62 virtual void TearDownInProcessBrowserTestFixture() { | |
63 WizardInProcessBrowserTest::TearDownInProcessBrowserTestFixture(); | |
64 cros_mock_->test_api()->SetLoginLibrary(NULL, false); | |
65 } | |
66 | |
67 private: | |
68 DISALLOW_COPY_AND_ASSIGN(LoginScreenTest); | |
69 }; | |
70 | |
71 static void Quit() { | |
72 VLOG(1) << "Posting a QuitTask to UI thread"; | |
73 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, | |
74 new MessageLoop::QuitTask); | |
75 } | |
76 IN_PROC_BROWSER_TEST_F(LoginScreenTest, TestBasic) { | |
Dmitry Polukhin
2010/12/14 14:06:18
It would be good to add in-process tests for new l
| |
77 ASSERT_TRUE(controller() != NULL); | |
78 ASSERT_EQ(controller()->current_screen(), controller()->GetLoginScreen()); | |
79 | |
80 scoped_ptr<MockScreenObserver> mock_screen_observer( | |
81 new MockScreenObserver()); | |
82 EXPECT_CALL(*mock_screen_observer, | |
83 OnExit(ScreenObserver::LOGIN_SIGN_IN_SELECTED)) | |
84 .WillOnce(InvokeWithoutArgs(Quit)); | |
85 | |
86 controller()->set_observer(mock_screen_observer.get()); | |
87 NewUserView* login = controller()->GetLoginScreen()->view(); | |
88 login->SetUsername(kUsername); | |
89 login->SetPassword(kPassword); | |
90 | |
91 bool old_state = MessageLoop::current()->NestableTasksAllowed(); | |
92 MessageLoop::current()->SetNestableTasksAllowed(true); | |
93 login->Login(); | |
94 MessageLoop::current()->Run(); | |
95 MessageLoop::current()->SetNestableTasksAllowed(old_state); | |
96 controller()->set_observer(NULL); | |
97 } | |
98 | |
99 IN_PROC_BROWSER_TEST_F(LoginScreenTest, AuthenticationFailed) { | |
100 ASSERT_TRUE(controller() != NULL); | |
101 ASSERT_EQ(controller()->current_screen(), controller()->GetLoginScreen()); | |
102 | |
103 scoped_ptr<MockScreenObserver> mock_screen_observer( | |
104 new MockScreenObserver()); | |
105 controller()->set_observer(mock_screen_observer.get()); | |
106 | |
107 EXPECT_CALL(*mock_network_library_, Connected()) | |
108 .Times(AnyNumber()) | |
109 .WillRepeatedly((Return(true))); | |
110 | |
111 NewUserView* login = controller()->GetLoginScreen()->view(); | |
112 login->SetUsername(kUsername); | |
113 login->SetPassword("wrong password"); | |
114 | |
115 bool old_state = MessageLoop::current()->NestableTasksAllowed(); | |
116 MessageLoop::current()->SetNestableTasksAllowed(true); | |
117 login->Login(); | |
118 MessageLoop::current()->Run(); | |
119 MessageLoop::current()->SetNestableTasksAllowed(old_state); | |
120 | |
121 ASSERT_EQ(controller()->current_screen(), controller()->GetLoginScreen()); | |
122 EXPECT_TRUE(controller()->GetLoginScreen()->IsErrorShown()); | |
123 controller()->set_observer(NULL); | |
124 } | |
125 | |
126 IN_PROC_BROWSER_TEST_F(LoginScreenTest, IncognitoLogin) { | |
127 ASSERT_TRUE(controller() != NULL); | |
128 ASSERT_EQ(controller()->current_screen(), controller()->GetLoginScreen()); | |
129 | |
130 scoped_ptr<MockScreenObserver> mock_screen_observer( | |
131 new MockScreenObserver()); | |
132 controller()->set_observer(mock_screen_observer.get()); | |
133 NewUserView* login = controller()->GetLoginScreen()->view(); | |
134 | |
135 login->LinkActivated(login->guest_link_, 0); | |
136 controller()->set_observer(NULL); | |
137 Quit(); | |
138 } | |
139 | |
140 } // namespace chromeos | |
OLD | NEW |