Add more management policy checking after extension installed

chrome/browser/extensions/chrome_app_sorting_unittest.cc

 // Copyright 2013 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/extensions/chrome_app_sorting.h"
 
 #include <map>
 
 #include "chrome/browser/extensions/./extension_prefs_unittest.h"
 #include "components/crx_file/id_util.h"
@@ skipping 13 matching lines @@
   }
 };
 
 class ChromeAppSortingAppLocation : public ChromeAppSortingTest {
  public:
   void Initialize() override {
     extension_ = prefs_.AddExtension("not_an_app");
     // Non-apps should not have any app launch ordinal or page ordinal.
     prefs()->OnExtensionInstalled(extension_.get(),
                                   Extension::ENABLED,
+                                  Extension::DISABLE_NONE,
                                   syncer::StringOrdinal(),
                                   std::string());
   }
 
   void Verify() override {
     EXPECT_FALSE(
         app_sorting()->GetAppLaunchOrdinal(extension_->id()).IsValid());
     EXPECT_FALSE(
         app_sorting()->GetPageOrdinal(extension_->id()).IsValid());
   }
 
  private:
   scoped_refptr<Extension> extension_;
 };
 TEST_F(ChromeAppSortingAppLocation, ChromeAppSortingAppLocation) {}
 
 class ChromeAppSortingAppLaunchOrdinal : public ChromeAppSortingTest {
  public:
   void Initialize() override {
     // No extensions yet.
     syncer::StringOrdinal page = syncer::StringOrdinal::CreateInitialOrdinal();
     EXPECT_TRUE(syncer::StringOrdinal::CreateInitialOrdinal().Equals(
         app_sorting()->CreateNextAppLaunchOrdinal(page)));
 
     extension_ = prefs_.AddApp("on_extension_installed");
     EXPECT_FALSE(prefs()->IsExtensionDisabled(extension_->id()));
     prefs()->OnExtensionInstalled(extension_.get(),
                                   Extension::ENABLED,
+                                  Extension::DISABLE_NONE,
                                   syncer::StringOrdinal(),
                                   std::string());
   }
 
   void Verify() override {
     syncer::StringOrdinal launch_ordinal =
         app_sorting()->GetAppLaunchOrdinal(extension_->id());
     syncer::StringOrdinal page_ordinal =
         syncer::StringOrdinal::CreateInitialOrdinal();
 
@@ skipping 32 matching lines @@
 TEST_F(ChromeAppSortingAppLaunchOrdinal, ChromeAppSortingAppLaunchOrdinal) {}
 
 class ChromeAppSortingPageOrdinal : public ChromeAppSortingTest {
  public:
   void Initialize() override {
     extension_ = prefs_.AddApp("page_ordinal");
     // Install with a page preference.
     first_page_ = syncer::StringOrdinal::CreateInitialOrdinal();
     prefs()->OnExtensionInstalled(extension_.get(),
                                   Extension::ENABLED,
+                                  Extension::DISABLE_NONE,
                                   first_page_,
                                   std::string());
     EXPECT_TRUE(first_page_.Equals(
         app_sorting()->GetPageOrdinal(extension_->id())));
     EXPECT_EQ(0, app_sorting()->PageStringOrdinalAsInteger(first_page_));
 
     scoped_refptr<Extension> extension2 = prefs_.AddApp("page_ordinal_2");
     // Install without any page preference.
     prefs()->OnExtensionInstalled(extension2.get(),
                                   Extension::ENABLED,
+                                  Extension::DISABLE_NONE,
                                   syncer::StringOrdinal(),
                                   std::string());
     EXPECT_TRUE(first_page_.Equals(
         app_sorting()->GetPageOrdinal(extension2->id())));
   }
   void Verify() override {
     // Set the page ordinal.
     syncer::StringOrdinal new_page = first_page_.CreateAfter();
     app_sorting()->SetPageOrdinal(extension_->id(), new_page);
     // Verify the page ordinal.
@@ skipping 495 matching lines @@
     simple_dict.SetString(keys::kName, "unused");
     simple_dict.SetString(keys::kApp, "true");
     simple_dict.SetString(keys::kLaunchLocalPath, "fake.html");
 
     std::string error;
     app1_scoped_ = Extension::Create(
         prefs_.temp_dir().AppendASCII("app1_"), Manifest::EXTERNAL_PREF,
         simple_dict, Extension::NO_FLAGS, &error);
     prefs()->OnExtensionInstalled(app1_scoped_.get(),
                                   Extension::ENABLED,
+                                  Extension::DISABLE_NONE,
                                   syncer::StringOrdinal(),
                                   std::string());
 
     app2_scoped_ = Extension::Create(
         prefs_.temp_dir().AppendASCII("app2_"), Manifest::EXTERNAL_PREF,
         simple_dict, Extension::NO_FLAGS, &error);
     prefs()->OnExtensionInstalled(app2_scoped_.get(),
                                   Extension::ENABLED,
+                                  Extension::DISABLE_NONE,
                                   syncer::StringOrdinal(),
                                   std::string());
 
     app1_ = app1_scoped_.get();
     app2_ = app2_scoped_.get();
   }
   ~ChromeAppSortingPreinstalledAppsBase() override {}
 
  protected:
   // Weak references, for convenience.
@@ skipping 166 matching lines @@
         syncer::StringOrdinal::CreateInitialOrdinal().CreateAfter();
     default_app_launch_ordinal_ =
         syncer::StringOrdinal::CreateInitialOrdinal().CreateBefore();
   }
 
   virtual void SetupUserOrdinals() {}
 
   virtual void InstallApps() {
     prefs()->OnExtensionInstalled(app_.get(),
                                   Extension::ENABLED,
+                                  Extension::DISABLE_NONE,
                                   syncer::StringOrdinal(),
                                   std::string());
   }
 
   scoped_refptr<Extension> app_;
   syncer::StringOrdinal default_page_ordinal_;
   syncer::StringOrdinal default_app_launch_ordinal_;
 };
 
 // Tests that the app gets its default ordinals.
@@ skipping 27 matching lines @@
     EXPECT_FALSE(app_sorting->GetPageOrdinal(app_->id()).Equals(
         default_page_ordinal_));
     EXPECT_TRUE(app_sorting->GetPageOrdinal(app_->id()).Equals(install_page_));
   }
 
  protected:
   void InstallApps() override {
     install_page_ = default_page_ordinal_.CreateAfter();
     prefs()->OnExtensionInstalled(app_.get(),
                                   Extension::ENABLED,
+                                  Extension::DISABLE_NONE,
                                   install_page_,
                                   std::string());
   }
 
  private:
   syncer::StringOrdinal install_page_;
 };
 TEST_F(ChromeAppSortingDefaultOrdinalOverriddenByInstallPage,
        ChromeAppSortingDefaultOrdinalOverriddenByInstallPage) {}
 
@@ skipping 102 matching lines @@
 
  private:
   scoped_refptr<Extension> first_app_;
   scoped_refptr<Extension> second_app_;
 };
 TEST_F(ChromeAppSortingSetExtensionVisible,
        ChromeAppSortingSetExtensionVisible) {
 }
 
 }  // namespace extensions