Get enrollment token from DMServer when an Active Directory user uses ARC

chrome/browser/chromeos/arc/arc_util_unittest.cc

 // Copyright 2017 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/arc/arc_util.h"
 
 #include <memory>
 
 #include "base/command_line.h"
 #include "base/macros.h"
@@ skipping 20 matching lines @@
 constexpr char kTestGaiaId[] = "1234567890";
 
 class ScopedLogIn {
  public:
   ScopedLogIn(
       chromeos::FakeChromeUserManager* fake_user_manager,
       const AccountId& account_id,
       user_manager::UserType user_type = user_manager::USER_TYPE_REGULAR)
       : fake_user_manager_(fake_user_manager), account_id_(account_id) {
     switch (user_type) {
-      case user_manager::USER_TYPE_REGULAR:
+      case user_manager::USER_TYPE_REGULAR:  // fallthrough
+      case user_manager::USER_TYPE_ACTIVE_DIRECTORY:
         LogIn();
         break;
       case user_manager::USER_TYPE_ARC_KIOSK_APP:
         LogInArcKioskApp();
         break;
       default:
         NOTREACHED();
     }
   }
 
@@ skipping 95 matching lines @@
 TEST_F(ChromeArcUtilTest, IsArcAllowedForProfile_NonPrimaryProfile) {
   ScopedLogIn login2(
       GetFakeUserManager(),
       AccountId::FromUserEmailGaiaId("user2@gmail.com", "0123456789"));
   ScopedLogIn login(GetFakeUserManager(),
                     AccountId::FromUserEmailGaiaId(
                         profile()->GetProfileUserName(), kTestGaiaId));
   EXPECT_FALSE(IsArcAllowedForProfile(profile()));
 }
 
+TEST_F(ChromeArcUtilTest, IsArcAllowedForProfile_ActiveDirectoryEnabled) {
+  base::CommandLine::ForCurrentProcess()->InitFromArgv(
+      {"", "--arc-availability=officially-supported-with-active-directory"});
+  ScopedLogIn login(
+      GetFakeUserManager(),
+      AccountId::AdFromObjGuid("f04557de-5da2-40ce-ae9d-b8874d8da96e"),
+      user_manager::USER_TYPE_ACTIVE_DIRECTORY);
+  EXPECT_FALSE(chromeos::ProfileHelper::Get()
+                   ->GetUserByProfile(profile())
+                   ->HasGaiaAccount());
+  EXPECT_TRUE(IsArcAllowedForProfile(profile()));
+}
+
+TEST_F(ChromeArcUtilTest, IsArcAllowedForProfile_ActiveDirectoryDisabled) {
+  ScopedLogIn login(
+      GetFakeUserManager(),
+      AccountId::AdFromObjGuid("f04557de-5da2-40ce-ae9d-b8874d8da96e"),
+      user_manager::USER_TYPE_ACTIVE_DIRECTORY);
+  EXPECT_FALSE(chromeos::ProfileHelper::Get()
+                   ->GetUserByProfile(profile())
+                   ->HasGaiaAccount());
+  EXPECT_FALSE(IsArcAllowedForProfile(profile()));
+}
+
 TEST_F(ChromeArcUtilTest, IsArcAllowedForProfile_Kiosk) {
   ScopedLogIn login(GetFakeUserManager(),
                     AccountId::FromUserEmail(profile()->GetProfileUserName()),
                     user_manager::USER_TYPE_ARC_KIOSK_APP);
   EXPECT_FALSE(chromeos::ProfileHelper::Get()
                    ->GetUserByProfile(profile())
                    ->HasGaiaAccount());
   EXPECT_TRUE(IsArcAllowedForProfile(profile()));
 }
 
 TEST_F(ChromeArcUtilTest, IsArcAllowedForProfile_SupervisedUserFlow) {
   auto manager_id = AccountId::FromUserEmailGaiaId(
       profile()->GetProfileUserName(), kTestGaiaId);
   ScopedLogIn login(GetFakeUserManager(), manager_id);
   GetFakeUserManager()->SetUserFlow(
       manager_id, new chromeos::SupervisedUserCreationFlow(manager_id));
   EXPECT_FALSE(IsArcAllowedForProfile(profile()));
   GetFakeUserManager()->ResetUserFlow(manager_id);
 }
 
 // TODO(hidehiko): Add test for Ephemeral users. There seems no way to easily
 // simulate ephemeral user.
 
 }  // namespace util
 }  // namespace arc