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