OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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/enrollment/enrollment_screen.h" |
5 #include "chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper_
mock.h" | 6 #include "chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper_
mock.h" |
6 #include "chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper_
impl.h" | 7 #include "chrome/browser/chromeos/login/enrollment/enterprise_enrollment_helper_
impl.h" |
| 8 #include "chrome/browser/chromeos/policy/enrollment_status_chromeos.h" |
| 9 #include "testing/gmock/include/gmock/gmock.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" |
| 11 |
| 12 using testing::Invoke; |
7 | 13 |
8 namespace chromeos { | 14 namespace chromeos { |
9 | 15 |
10 EnterpriseEnrollmentHelperMock::EnterpriseEnrollmentHelperMock( | 16 EnterpriseEnrollmentHelperMock::EnterpriseEnrollmentHelperMock( |
11 EnrollmentStatusConsumer* status_consumer) | 17 EnrollmentStatusConsumer* status_consumer) |
12 : EnterpriseEnrollmentHelper(status_consumer) {} | 18 : EnterpriseEnrollmentHelper(status_consumer) { |
| 19 ON_CALL(*this, EnrollUsingAttestation()) |
| 20 .WillByDefault( |
| 21 Invoke(this, &EnterpriseEnrollmentHelperMock::TryEnrollment)); |
| 22 } |
13 | 23 |
14 EnterpriseEnrollmentHelperMock::~EnterpriseEnrollmentHelperMock() {} | 24 EnterpriseEnrollmentHelperMock::~EnterpriseEnrollmentHelperMock() {} |
15 | 25 |
16 EnterpriseEnrollmentHelper::EnrollmentStatusConsumer* | 26 EnterpriseEnrollmentHelper::EnrollmentStatusConsumer* |
17 EnterpriseEnrollmentHelperMock::status_consumer() { | 27 EnterpriseEnrollmentHelperMock::status_consumer() { |
18 return EnterpriseEnrollmentHelper::status_consumer(); | 28 return EnterpriseEnrollmentHelper::status_consumer(); |
19 } | 29 } |
20 | 30 |
| 31 void EnterpriseEnrollmentHelperMock::TryEnrollment() { |
| 32 if (should_enroll) { |
| 33 static_cast<EnrollmentScreen*>(status_consumer()) |
| 34 ->ShowEnrollmentStatusOnSuccess(); |
| 35 } else { |
| 36 status_consumer()->OnEnrollmentError(policy::EnrollmentStatus::ForStatus( |
| 37 policy::EnrollmentStatus::REGISTRATION_FAILED)); |
| 38 } |
| 39 } |
| 40 |
| 41 bool EnterpriseEnrollmentHelperMock::should_enroll = true; |
| 42 |
21 } // namespace chromeos | 43 } // namespace chromeos |
OLD | NEW |