[Merge M58] arc: Reactivate OptIn flow on clicking Play Store.

chrome/browser/chromeos/arc/arc_session_manager_unittest.cc

 // Copyright 2016 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 <memory>
 #include <string>
 #include <tuple>
 #include <vector>
 
 #include "base/bind.h"
@@ skipping 491 matching lines @@
   EXPECT_EQ(ArcSessionManager::State::ACTIVE, arc_session_manager()->state());
   EXPECT_FALSE(prefs->GetBoolean(prefs::kArcBackupRestoreEnabled));
   EXPECT_FALSE(prefs->GetBoolean(prefs::kArcLocationServiceEnabled));
 
   // Stop ARC and shutdown the service.
   prefs->RemoveManagedPref(prefs::kArcEnabled);
   WaitForDataRemoved(ArcSessionManager::State::STOPPED);
   arc_session_manager()->Shutdown();
 }
 
+TEST_P(ArcSessionManagerPolicyTest, ReenableManagedArc) {
+  sync_preferences::TestingPrefServiceSyncable* const prefs =
+      profile()->GetTestingPrefService();
+
+  // Set ARC to be managed.
+  prefs->SetManagedPref(prefs::kArcEnabled, new base::Value(true));
+  EXPECT_TRUE(arc::IsArcPlayStoreEnabledForProfile(profile()));
+
+  arc_session_manager()->SetProfile(profile());
+  arc_session_manager()->RequestEnable();
+  EXPECT_TRUE(arc_session_manager()->enable_requested());
+
+  // Simulate close OptIn. Session manager should stop.
+  SetArcPlayStoreEnabledForProfile(profile(), false);
+  EXPECT_TRUE(arc::IsArcPlayStoreEnabledForProfile(profile()));
+  EXPECT_FALSE(arc_session_manager()->enable_requested());
+
+  // Restart ARC again
+  SetArcPlayStoreEnabledForProfile(profile(), true);
+  EXPECT_TRUE(arc::IsArcPlayStoreEnabledForProfile(profile()));
+  EXPECT_TRUE(arc_session_manager()->enable_requested());
+
+  arc_session_manager()->Shutdown();
+}
+
 INSTANTIATE_TEST_CASE_P(
     ArcSessionManagerPolicyTest,
     ArcSessionManagerPolicyTest,
     testing::Combine(
         testing::Values(base::Value(), base::Value(false), base::Value(true)),
         testing::Values(base::Value(), base::Value(false), base::Value(true))));
 
 class ArcSessionManagerKioskTest : public ArcSessionManagerTestBase {
  public:
   ArcSessionManagerKioskTest() = default;
@@ skipping 171 matching lines @@
             arc_session_manager()->state());
   ReportResult(true);
   EXPECT_EQ(ArcSessionManager::State::ACTIVE, arc_session_manager()->state());
 }
 
 TEST_P(ArcSessionOobeOptInNegotiatorTest, OobeTermsRejected) {
   view()->Show();
   EXPECT_EQ(ArcSessionManager::State::SHOWING_TERMS_OF_SERVICE,
             arc_session_manager()->state());
   ReportResult(false);
-  EXPECT_EQ(ArcSessionManager::State::STOPPED, arc_session_manager()->state());
+  if (!IsManagedUser()) {
+    // ArcPlayStoreEnabledPreferenceHandler is not running, so the state should
+    // be kept as is
+    EXPECT_EQ(/*ArcSessionManager::State::SHOWING_TERMS_OF_SERVICE,*/
+              ArcSessionManager::State::STOPPED,  // M58 Only, different logic
+              arc_session_manager()->state());
+    EXPECT_FALSE(IsArcPlayStoreEnabledForProfile(profile()));
+  } else {
+    // For managed case we handle closing outside of
+    // ArcPlayStoreEnabledPreferenceHandler. So it session turns to STOPPED.
+    EXPECT_EQ(ArcSessionManager::State::STOPPED,
+              arc_session_manager()->state());
+    // Managed user's preference should not be overwritten.
+    EXPECT_TRUE(IsArcPlayStoreEnabledForProfile(profile()));
+  }
 }
 
 TEST_P(ArcSessionOobeOptInNegotiatorTest, OobeTermsViewDestroyed) {
   view()->Show();
   EXPECT_EQ(ArcSessionManager::State::SHOWING_TERMS_OF_SERVICE,
             arc_session_manager()->state());
   CloseLoginDisplayHost();
   ReportViewDestroyed();
-  EXPECT_EQ(ArcSessionManager::State::STOPPED, arc_session_manager()->state());
+  if (!IsManagedUser()) {
+    // ArcPlayStoreEnabledPreferenceHandler is not running, so the state should
+    // be kept as is.
+    EXPECT_EQ(/*ArcSessionManager::State::SHOWING_TERMS_OF_SERVICE,*/
+              ArcSessionManager::State::STOPPED,  // M58 Only, different logic
+              arc_session_manager()->state());
+    EXPECT_FALSE(IsArcPlayStoreEnabledForProfile(profile()));
+  } else {
+    // For managed case we handle closing outside of
+    // ArcPlayStoreEnabledPreferenceHandler. So it session turns to STOPPED.
+    EXPECT_EQ(ArcSessionManager::State::STOPPED,
+              arc_session_manager()->state());
+    // Managed user's preference should not be overwritten.
+    EXPECT_TRUE(IsArcPlayStoreEnabledForProfile(profile()));
+  }
 }
 
 }  // namespace arc