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) { |