mock_update_screen.cc revision c2e0dbddbe15c98d52c4786dac06cb8952a8ae6d 
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 "chrome/browser/chromeos/login/screens/mock_update_screen.h"
6
7namespace chromeos {
8
9using ::testing::AtLeast;
10using ::testing::NotNull;
11
12MockUpdateScreen::MockUpdateScreen(ScreenObserver* screen_observer,
13                                   UpdateScreenActor* actor)
14    : UpdateScreen(screen_observer, actor) {
15}
16
17MockUpdateScreen::~MockUpdateScreen() {
18}
19
20MockUpdateScreenActor::MockUpdateScreenActor()
21    : screen_(NULL) {
22  EXPECT_CALL(*this, MockSetDelegate(NotNull())).Times(AtLeast(1));
23}
24
25MockUpdateScreenActor::~MockUpdateScreenActor() {
26  if (screen_)
27    screen_->OnActorDestroyed(this);
28}
29
30void MockUpdateScreenActor::SetDelegate(UpdateScreenActor::Delegate* screen) {
31  screen_ = screen;
32  MockSetDelegate(screen);
33}
34
35}  // namespace chromeos
36