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 "base/basictypes.h" 6#include "base/memory/scoped_ptr.h" 7#include "chrome/browser/chromeos/login/enterprise_enrollment_screen.h" 8#include "chrome/browser/chromeos/login/mock_screen_observer.h" 9#include "chrome/browser/chromeos/login/wizard_controller.h" 10#include "chrome/browser/chromeos/login/wizard_in_process_browser_test.h" 11#include "testing/gmock/include/gmock/gmock.h" 12#include "testing/gtest/include/gtest/gtest.h" 13 14using testing::Mock; 15 16namespace chromeos { 17 18class EnterpriseEnrollmentScreenTest : public WizardInProcessBrowserTest { 19 public: 20 EnterpriseEnrollmentScreenTest() 21 : WizardInProcessBrowserTest( 22 WizardController::kEnterpriseEnrollmentScreenName) {} 23 24 private: 25 DISALLOW_COPY_AND_ASSIGN(EnterpriseEnrollmentScreenTest); 26}; 27 28IN_PROC_BROWSER_TEST_F(EnterpriseEnrollmentScreenTest, TestCancel) { 29 ASSERT_TRUE(controller() != NULL); 30 MockScreenObserver mock_screen_observer; 31 controller()->set_observer(&mock_screen_observer); 32 EnterpriseEnrollmentScreen* enterprise_enrollment_screen = 33 controller()->GetEnterpriseEnrollmentScreen(); 34 ASSERT_TRUE(enterprise_enrollment_screen != NULL); 35 ASSERT_EQ(controller()->current_screen(), enterprise_enrollment_screen); 36 37 EXPECT_CALL(mock_screen_observer, 38 OnExit(ScreenObserver::ENTERPRISE_ENROLLMENT_CANCELLED)); 39 enterprise_enrollment_screen->CancelEnrollment(); 40 Mock::VerifyAndClearExpectations(&mock_screen_observer); 41 42 controller()->set_observer(NULL); 43} 44 45} // namespace chromeos 46