| Index: chrome/browser/chromeos/login/login_browsertest.cc
|
| diff --git a/chrome/browser/chromeos/login/login_browsertest.cc b/chrome/browser/chromeos/login/login_browsertest.cc
|
| index c4ee33d5eef533807282a0ffc44b516027344d51..ebbf3e85b755c03937ec33753ce5e17a55b8f60b 100644
|
| --- a/chrome/browser/chromeos/login/login_browsertest.cc
|
| +++ b/chrome/browser/chromeos/login/login_browsertest.cc
|
| @@ -8,6 +8,7 @@
|
| #include "chrome/browser/chromeos/cros/mock_cryptohome_library.h"
|
| #include "chrome/browser/chromeos/cros/mock_library_loader.h"
|
| #include "chrome/browser/chromeos/cros/mock_network_library.h"
|
| +#include "chrome/browser/chromeos/cros/mock_screen_lock_library.h"
|
| #include "chrome/browser/profiles/profile_manager.h"
|
| #include "chrome/browser/ui/browser.h"
|
| #include "chrome/common/chrome_switches.h"
|
| @@ -23,7 +24,8 @@ using ::testing::Return;
|
|
|
| class LoginTestBase : public CrosInProcessBrowserTest {
|
| public:
|
| - LoginTestBase() : mock_cryptohome_library_(NULL) {
|
| + LoginTestBase() : mock_cryptohome_library_(NULL),
|
| + mock_screen_lock_library_(NULL) {
|
| }
|
|
|
| protected:
|
| @@ -31,12 +33,15 @@ class LoginTestBase : public CrosInProcessBrowserTest {
|
| cros_mock_->InitStatusAreaMocks();
|
| cros_mock_->SetStatusAreaMocksExpectations();
|
| cros_mock_->InitMockCryptohomeLibrary();
|
| + cros_mock_->InitMockScreenLockLibrary();
|
| mock_cryptohome_library_ = cros_mock_->mock_cryptohome_library();
|
| + mock_screen_lock_library_ = cros_mock_->mock_screen_lock_library();
|
| EXPECT_CALL(*mock_cryptohome_library_, IsMounted())
|
| .WillRepeatedly(Return(true));
|
| }
|
|
|
| MockCryptohomeLibrary* mock_cryptohome_library_;
|
| + MockScreenLockLibrary* mock_screen_lock_library_;
|
|
|
| private:
|
| DISALLOW_COPY_AND_ASSIGN(LoginTestBase);
|
| @@ -46,6 +51,12 @@ class LoginUserTest : public LoginTestBase {
|
| protected:
|
| virtual void SetUpInProcessBrowserTestFixture() {
|
| LoginTestBase::SetUpInProcessBrowserTestFixture();
|
| + EXPECT_CALL(*mock_screen_lock_library_, AddObserver(_))
|
| + .Times(AtLeast(1))
|
| + .WillRepeatedly(Return());
|
| + EXPECT_CALL(*mock_screen_lock_library_, RemoveObserver(_))
|
| + .Times(AtLeast(1))
|
| + .WillRepeatedly(Return());
|
| }
|
|
|
| virtual void SetUpCommandLine(CommandLine* command_line) {
|
|
|