Filter Web Store app from ChromeOS launcher according to policy.

chrome/browser/ui/app_list/apps_model_builder_unittest.cc

 // 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.
 
 #include "chrome/browser/ui/app_list/apps_model_builder.h"
 
 #include <string>
 
 #include "base/files/file_path.h"
 #include "base/memory/scoped_ptr.h"
+#include "base/prefs/pref_service.h"
 #include "base/values.h"
+#include "chrome/browser/extensions/extension_function_test_utils.h"
 #include "chrome/browser/extensions/extension_service_unittest.h"
 #include "chrome/browser/extensions/extension_sorting.h"
+#include "chrome/common/extensions/extension_constants.h"
+#include "chrome/common/extensions/manifest.h"
+#include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_profile.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/app_list/app_list_item_model.h"
 
 namespace {
 
 const char kHostedAppId[] = "dceacbkfkmllgmjmbhgkpjegnodmildf";
 const char kPackagedApp1Id[] = "emfkafnhnpcmabnnkckkchdilgeoekbo";
 const char kPackagedApp2Id[] = "jlklkagmeajbjiobondfhiekepofmljl";
 
 // Get a string of all apps in |model| joined with ','.
 std::string GetModelContent(app_list::AppListModel::Apps* model) {
   std::string content;
   for (size_t i = 0; i < model->item_count(); ++i) {
     if (i > 0)
       content += ',';
     content += model->GetItemAt(i)->title();
   }
   return content;
 }
 
+scoped_refptr<extensions::Extension> MakeApp(const std::string& name,
+                                             const std::string& version,
+                                             const std::string& url,
+                                             const std::string& id) {
+  std::string err;
+  DictionaryValue value;
+  value.SetString("name", name);
+  value.SetString("version", version);
+  value.SetString("app.launch.web_url", url);
+  scoped_refptr<extensions::Extension> app =
+      extensions::Extension::Create(
+          base::FilePath(),
+          extensions::Manifest::INTERNAL,
+          value,
+          extensions::Extension::WAS_INSTALLED_BY_DEFAULT,
+          id,
+          &err);
+  EXPECT_EQ(err, "");
+  return app;
+}
+
 }  // namespace
 
 class AppsModelBuilderTest : public ExtensionServiceTestBase {
  public:
   AppsModelBuilderTest() {}
   virtual ~AppsModelBuilderTest() {}
 
   virtual void SetUp() OVERRIDE {
     // Load "app_list" extensions test profile.
     // The test profile has 4 extensions:
@@ skipping 17 matching lines @@
   scoped_ptr<app_list::AppListModel::Apps> model(
       new app_list::AppListModel::Apps);
   AppsModelBuilder builder(profile_.get(), model.get(), NULL);
   builder.Build();
 
   // The apps list would have 3 extension apps in the profile.
   EXPECT_EQ(std::string("Packaged App 1,Packaged App 2,Hosted App"),
             GetModelContent(model.get()));
 }
 
+TEST_F(AppsModelBuilderTest, HideWebStore) {
+  // Install a "web store" app.
+  scoped_refptr<extensions::Extension> store =
+      MakeApp("webstore",
+              "0.0",
+              "http://google.com",
+              std::string(extension_misc::kWebStoreAppId));
+  service_->AddExtension(store);
+
+  // Install an "enterprise web store" app.
+  scoped_refptr<extensions::Extension> enterprise_store =
+      MakeApp("enterprise_webstore",
+              "0.0",
+              "http://google.com",
+              std::string(extension_misc::kEnterpriseWebStoreAppId));
+  service_->AddExtension(enterprise_store);
+
+  // Web stores should be present in the AppListModel.
+  app_list::AppListModel::Apps model1;
+  AppsModelBuilder builder1(profile_.get(), &model1, NULL);
+  builder1.Build();
+  std::string content = GetModelContent(&model1);
+  EXPECT_NE(std::string::npos, content.find("webstore"));
+  EXPECT_NE(std::string::npos, content.find("enterprise_webstore"));
+
+  // Activate the HideWebStoreIcon policy.
+  profile_->GetPrefs()->SetBoolean(prefs::kHideWebStoreIcon, true);
+
+  // Web stores should NOT be in the AppListModel.
+  app_list::AppListModel::Apps model2;
+  AppsModelBuilder builder2(profile_.get(), &model2, NULL);
+  builder2.Build();
+  content = GetModelContent(&model2);
+  EXPECT_EQ(std::string::npos, content.find("webstore"));
+  EXPECT_EQ(std::string::npos, content.find("enterprise_webstore"));
+}
+
 TEST_F(AppsModelBuilderTest, DisableAndEnable) {
   scoped_ptr<app_list::AppListModel::Apps> model(
       new app_list::AppListModel::Apps);
   AppsModelBuilder builder(profile_.get(), model.get(), NULL);
   builder.Build();
 
   service_->DisableExtension(kHostedAppId,
                              extensions::Extension::DISABLE_NONE);
   EXPECT_EQ(std::string("Packaged App 1,Packaged App 2,Hosted App"),
             GetModelContent(model.get()));
@@ skipping 76 matching lines @@
       "page_ordinal",
       base::Value::CreateStringValue("a corrupted ordinal"));
 
   scoped_ptr<app_list::AppListModel::Apps> model(
       new app_list::AppListModel::Apps);
   AppsModelBuilder builder(profile_.get(), model.get(), NULL);
 
   // This should not assert or crash.
   builder.Build();
 }