chrome/browser/chromeos/login/enrollment/mock_auto_enrollment_check_screen.cc

// Copyright 2014 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "chrome/browser/chromeos/login/enrollment/mock_auto_enrollment_check_screen.h"

using ::testing::AtLeast;
using ::testing::NotNull;

namespace chromeos {

MockAutoEnrollmentCheckScreen::MockAutoEnrollmentCheckScreen(
    BaseScreenDelegate* base_screen_delegate,
    AutoEnrollmentCheckScreenActor* actor)
    : AutoEnrollmentCheckScreen(base_screen_delegate, actor) {
}

MockAutoEnrollmentCheckScreen::~MockAutoEnrollmentCheckScreen() { }

MockAutoEnrollmentCheckScreenActor::MockAutoEnrollmentCheckScreenActor()
    : screen_(NULL) {
  EXPECT_CALL(*this, MockSetDelegate(NotNull())).Times(AtLeast(1));
}


MockAutoEnrollmentCheckScreenActor::~MockAutoEnrollmentCheckScreenActor() {
  if (screen_)
    screen_->OnActorDestroyed(this);
}

void MockAutoEnrollmentCheckScreenActor::SetDelegate(Delegate* screen) {
  screen_ = screen;
  MockSetDelegate(screen);
}

}  // namespace chromeos