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
23namespace chromeos {
24using ::testing::_;
25using ::testing::AnyNumber;
26using ::testing::InvokeWithoutArgs;
27using ::testing::NiceMock;
28using ::testing::Return;
29using ::testing::ReturnRef;
30
31class LoginTestBase : public CrosInProcessBrowserTest {
32 public:
33  LoginTestBase() : mock_cryptohome_library_(NULL),
34                    mock_screen_lock_library_(NULL) {
35  }
36
37 protected:
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
56class LoginUserTest : public LoginTestBase {
57 protected:
58  virtual void SetUpInProcessBrowserTestFixture() {
59    LoginTestBase::SetUpInProcessBrowserTestFixture();
60    EXPECT_CALL(*mock_screen_lock_library_, AddObserver(_))
61        .WillOnce(Return());
62  }
63
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
71class LoginProfileTest : public LoginTestBase {
72 protected:
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.
82IN_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.
89IN_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