OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/chromeos/login/mock_update_screen.h" | 5 #include "chrome/browser/chromeos/login/mock_update_screen.h" |
6 | 6 |
7 namespace chromeos { | 7 namespace chromeos { |
8 | 8 |
| 9 using ::testing::AtLeast; |
| 10 using ::testing::NotNull; |
| 11 |
9 MockUpdateScreen::MockUpdateScreen(ScreenObserver* screen_observer, | 12 MockUpdateScreen::MockUpdateScreen(ScreenObserver* screen_observer, |
10 UpdateScreenActor* actor) | 13 UpdateScreenActor* actor) |
11 : UpdateScreen(screen_observer, actor) { | 14 : UpdateScreen(screen_observer, actor) { |
12 } | 15 } |
13 | 16 |
14 MockUpdateScreen::~MockUpdateScreen() { | 17 MockUpdateScreen::~MockUpdateScreen() { |
15 } | 18 } |
16 | 19 |
17 MockUpdateScreenActor::MockUpdateScreenActor() { | 20 MockUpdateScreenActor::MockUpdateScreenActor() |
| 21 : screen_(NULL) { |
| 22 EXPECT_CALL(*this, MockSetDelegate(NotNull())).Times(AtLeast(1)); |
18 } | 23 } |
19 | 24 |
20 MockUpdateScreenActor::~MockUpdateScreenActor() { | 25 MockUpdateScreenActor::~MockUpdateScreenActor() { |
| 26 if (screen_) |
| 27 screen_->OnActorDestroyed(this); |
| 28 } |
| 29 |
| 30 void MockUpdateScreenActor::SetDelegate(UpdateScreenActor::Delegate* screen) { |
| 31 screen_ = screen; |
| 32 MockSetDelegate(screen); |
21 } | 33 } |
22 | 34 |
23 } // namespace chromeos | 35 } // namespace chromeos |
OLD | NEW |