1 // Copyright (c) 2011 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/command_line.h"
6 #include "base/time.h"
7 #include "chrome/browser/chromeos/cros/cros_in_process_browser_test.h"
8 #include "chrome/browser/chromeos/cros/mock_cryptohome_library.h"
9 #include "chrome/browser/chromeos/cros/mock_input_method_library.h"
10 #include "chrome/browser/chromeos/cros/mock_library_loader.h"
11 #include "chrome/browser/chromeos/cros/mock_network_library.h"
12 #include "chrome/browser/chromeos/cros/mock_power_library.h"
13 #include "chrome/browser/chromeos/cros/mock_screen_lock_library.h"
14 #include "chrome/browser/chromeos/cros/mock_touchpad_library.h"
15 #include "chrome/browser/profiles/profile_manager.h"
16 #include "chrome/browser/ui/browser.h"
17 #include "chrome/common/chrome_switches.h"
18 #include "chrome/test/in_process_browser_test.h"
19 #include "chrome/test/ui_test_utils.h"
20 #include "testing/gmock/include/gmock/gmock.h"
21 #include "testing/gtest/include/gtest/gtest.h"
22
23 namespace chromeos {
24 using ::testing::_;
25 using ::testing::AnyNumber;
26 using ::testing::InvokeWithoutArgs;
27 using ::testing::NiceMock;
28 using ::testing::Return;
29 using ::testing::ReturnRef;
30
31 class LoginTestBase : public CrosInProcessBrowserTest {
32 public:
LoginTestBase()33 LoginTestBase() : mock_cryptohome_library_(NULL),
34 mock_screen_lock_library_(NULL) {
35 }
36
37 protected:
SetUpInProcessBrowserTestFixture()38 virtual void SetUpInProcessBrowserTestFixture() {
39 cros_mock_->InitStatusAreaMocks();
40 cros_mock_->SetStatusAreaMocksExpectations();
41 cros_mock_->InitMockCryptohomeLibrary();
42 cros_mock_->InitMockScreenLockLibrary();
43 mock_cryptohome_library_ = cros_mock_->mock_cryptohome_library();
44 mock_screen_lock_library_ = cros_mock_->mock_screen_lock_library();
45 EXPECT_CALL(*mock_cryptohome_library_, IsMounted())
46 .WillRepeatedly(Return(true));
47 }
48
49 MockCryptohomeLibrary* mock_cryptohome_library_;
50 MockScreenLockLibrary* mock_screen_lock_library_;
51
52 private:
53 DISALLOW_COPY_AND_ASSIGN(LoginTestBase);
54 };
55
56 class LoginUserTest : public LoginTestBase {
57 protected:
SetUpInProcessBrowserTestFixture()58 virtual void SetUpInProcessBrowserTestFixture() {
59 LoginTestBase::SetUpInProcessBrowserTestFixture();
60 EXPECT_CALL(*mock_screen_lock_library_, AddObserver(_))
61 .WillOnce(Return());
62 }
63
SetUpCommandLine(CommandLine * command_line)64 virtual void SetUpCommandLine(CommandLine* command_line) {
65 command_line->AppendSwitchASCII(switches::kLoginUser, "TestUser@gmail.com");
66 command_line->AppendSwitchASCII(switches::kLoginProfile, "user");
67 command_line->AppendSwitch(switches::kNoFirstRun);
68 }
69 };
70
71 class LoginProfileTest : public LoginTestBase {
72 protected:
SetUpCommandLine(CommandLine * command_line)73 virtual void SetUpCommandLine(CommandLine* command_line) {
74 command_line->AppendSwitchASCII(switches::kLoginProfile, "user");
75 command_line->AppendSwitch(switches::kNoFirstRun);
76 }
77 };
78
79 // After a chrome crash, the session manager will restart chrome with
80 // the -login-user flag indicating that the user is already logged in.
81 // This profile should NOT be an OTR profile.
IN_PROC_BROWSER_TEST_F(LoginUserTest,UserPassed)82 IN_PROC_BROWSER_TEST_F(LoginUserTest, UserPassed) {
83 Profile* profile = browser()->profile();
84 EXPECT_EQ("user", profile->GetPath().BaseName().value());
85 EXPECT_FALSE(profile->IsOffTheRecord());
86 }
87
88 // On initial launch, we should get the OTR default profile.
IN_PROC_BROWSER_TEST_F(LoginProfileTest,UserNotPassed)89 IN_PROC_BROWSER_TEST_F(LoginProfileTest, UserNotPassed) {
90 Profile* profile = browser()->profile();
91 EXPECT_EQ("Default", profile->GetPath().BaseName().value());
92 EXPECT_TRUE(profile->IsOffTheRecord());
93 // Ensure there's extension service for this profile.
94 EXPECT_TRUE(profile->GetExtensionService());
95 }
96
97 } // namespace chromeos
98