Mixed Licenses Enrollment

chrome/browser/chromeos/login/enrollment/enrollment_screen.h

 // Copyright (c) 2012 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.
 
 #ifndef CHROME_BROWSER_CHROMEOS_LOGIN_ENROLLMENT_ENROLLMENT_SCREEN_H_
 #define CHROME_BROWSER_CHROMEOS_LOGIN_ENROLLMENT_ENROLLMENT_SCREEN_H_
 
 #include <memory>
 #include <string>
 
@@ skipping 56 matching lines @@
   void OnLicenseTypeSelected(const std::string& license_type) override;
   void OnRetry() override;
   void OnCancel() override;
   void OnConfirmationClosed() override;
   void OnAdJoined(const std::string& realm) override;
   void OnDeviceAttributeProvided(const std::string& asset_id,
                                  const std::string& location) override;
 
   // EnterpriseEnrollmentHelper::EnrollmentStatusConsumer implementation:
   void OnAuthError(const GoogleServiceAuthError& error) override;
+  void OnMultipleLicensesAvailable(
+      const EnrollmentLicenseMap& licenses) override;
   void OnEnrollmentError(policy::EnrollmentStatus status) override;
   void OnOtherError(EnterpriseEnrollmentHelper::OtherError error) override;
   void OnDeviceEnrolled(const std::string& additional_token) override;
   void OnDeviceAttributeUploadCompleted(bool success) override;
   void OnDeviceAttributeUpdatePermission(bool granted) override;
 
   // ActiveDirectoryJoinDelegate implementation:
   void JoinDomain(OnDomainJoinedCallback on_joined_callback) override;
 
   // Used for testing.
   EnrollmentScreenView* GetView() { return view_; }
 
  private:
-  friend class EnrollmentScreenUnitTest;
-  FRIEND_TEST_ALL_PREFIXES(EnrollmentScreenTest, TestSuccess);
+  friend class MultiLicenseEnrollmentScreenUnitTest;
+  friend class ZeroTouchEnrollmentScreenUnitTest;
   FRIEND_TEST_ALL_PREFIXES(AttestationAuthEnrollmentScreenTest, TestCancel);
   FRIEND_TEST_ALL_PREFIXES(ForcedAttestationAuthEnrollmentScreenTest,
                            TestCancel);
   FRIEND_TEST_ALL_PREFIXES(MultiAuthEnrollmentScreenTest, TestCancel);
   FRIEND_TEST_ALL_PREFIXES(EnterpriseEnrollmentTest,
                            TestProperPageGetsLoadedOnEnrollmentSuccess);
   FRIEND_TEST_ALL_PREFIXES(EnterpriseEnrollmentTest,
                            TestAttributePromptPageGetsLoaded);
   FRIEND_TEST_ALL_PREFIXES(EnterpriseEnrollmentTest,
                            TestAuthCodeGetsProperlyReceivedFromGaia);
   FRIEND_TEST_ALL_PREFIXES(EnterpriseEnrollmentTest,
                            TestActiveDirectoryEnrollment_Success);
   FRIEND_TEST_ALL_PREFIXES(EnterpriseEnrollmentTest,
                            TestActiveDirectoryEnrollment_UIErrors);
   FRIEND_TEST_ALL_PREFIXES(HandsOffNetworkScreenTest, RequiresNoInput);
   FRIEND_TEST_ALL_PREFIXES(HandsOffNetworkScreenTest, ContinueClickedOnlyOnce);
-  FRIEND_TEST_ALL_PREFIXES(EnrollmentScreenUnitTest, Retries);
-  FRIEND_TEST_ALL_PREFIXES(EnrollmentScreenUnitTest, DoesNotRetryOnTopOfUser);
-  FRIEND_TEST_ALL_PREFIXES(EnrollmentScreenUnitTest, DoesNotRetryAfterSuccess);
+  FRIEND_TEST_ALL_PREFIXES(ZeroTouchEnrollmentScreenUnitTest, Retries);
+  FRIEND_TEST_ALL_PREFIXES(ZeroTouchEnrollmentScreenUnitTest, TestSuccess);
+  FRIEND_TEST_ALL_PREFIXES(ZeroTouchEnrollmentScreenUnitTest,
+                           DoesNotRetryOnTopOfUser);
+  FRIEND_TEST_ALL_PREFIXES(ZeroTouchEnrollmentScreenUnitTest,
+                           DoesNotRetryAfterSuccess);
 
   // The authentication mechanisms that this class can use.
   enum Auth {
     AUTH_ATTESTATION,
     AUTH_OAUTH,
   };
 
   // Sets the current config to use for enrollment.
   void SetConfig();
 
@@ skipping 63 matching lines @@
   std::unique_ptr<EnterpriseEnrollmentHelper> enrollment_helper_;
   OnDomainJoinedCallback on_joined_callback_;
   base::WeakPtrFactory<EnrollmentScreen> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(EnrollmentScreen);
 };
 
 }  // namespace chromeos
 
 #endif  // CHROME_BROWSER_CHROMEOS_LOGIN_ENROLLMENT_ENROLLMENT_SCREEN_H_