Add a policy to hide the Web Store on new tabs.

chrome/browser/policy/policy_browsertest.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 <algorithm>
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 1079 matching lines @@
   policies.Set(key::kDeveloperToolsDisabled, POLICY_LEVEL_MANDATORY,
                POLICY_SCOPE_USER, base::Value::CreateBooleanValue(true));
   UpdateProviderPolicy(policies);
   // The existing devtools window should have closed.
   EXPECT_FALSE(DevToolsWindow::GetDockedInstanceForInspectedTab(contents));
   // And it's not possible to open it again.
   EXPECT_FALSE(chrome::ExecuteCommand(browser(), IDC_DEV_TOOLS));
   EXPECT_FALSE(DevToolsWindow::GetDockedInstanceForInspectedTab(contents));
 }
 
+bool ContainsElement(content::WebContents* contents, const std::string id) {

[Joao da Silva, 2013/01/21 07:44:51]

Helper functions like this come before the tests, in the anonymous namespace.
Place this earlier in the file, right after FlushBlacklistPolicy().

Also, pass |id| by const &.

[dconnelly, 2013/01/21 09:39:03]

Done.

+  bool result;
+  EXPECT_TRUE(content::ExecuteScriptAndExtractBool(
+      contents,
+      "domAutomationController.send(!!document.getElementById(\"" + id + "\"))",

[Joao da Silva, 2013/01/21 07:44:51]

Single quotes could be used instead of \"; just a suggestion.

[dconnelly, 2013/01/21 09:39:03]

Done.

+      &result));
+  return result;
+}
+
+IN_PROC_BROWSER_TEST_F(PolicyTest, WebStoreIconHidden) {
+  // Verifies that the web store icons can be hidden from the new tab page.
+
+  // Open new tab page and look for the web store icons.
+  ui_test_utils::NavigateToURL(browser(), GURL(chrome::kChromeUINewTabURL));
+  content::WebContents* contents = chrome::GetActiveWebContents(browser());
+  EXPECT_TRUE(ContainsElement(contents, "ahfgeienlihckogmohjhadlkjgocpleb"));

[Joao da Silva, 2013/01/21 07:44:51]

What does that ID refer to? Please comment it (I'm guessing it's the
extension-ID of the webstore app?)

[dconnelly, 2013/01/21 09:39:03]

Done.

+  // The next NTP has no footer.
+  if (ContainsElement(contents, "footer"))
+      EXPECT_TRUE(ContainsElement(contents, "chrome-web-store-link"));

[Joao da Silva, 2013/01/21 07:44:51]

nit: indent (2 spaces instead of 4)

[dconnelly, 2013/01/21 09:39:03]

Done.

+
+  // Turn off the web store icons.
+  PolicyMap policies;
+  policies.Set(key::kHideWebStoreIcon, POLICY_LEVEL_MANDATORY,
+               POLICY_SCOPE_USER, base::Value::CreateBooleanValue(true));
+  UpdateProviderPolicy(policies);
+
+  // The web store icons should now be hidden.
+  ui_test_utils::NavigateToURL(browser(), GURL(chrome::kChromeUINewTabURL));
+  contents = chrome::GetActiveWebContents(browser());
+  EXPECT_FALSE(ContainsElement(contents, "ahfgeienlihckogmohjhadlkjgocpleb"));
+  EXPECT_FALSE(ContainsElement(contents, "chrome-web-store-link"));
+}
+
 // This policy isn't available on Chrome OS.
 #if !defined(OS_CHROMEOS)
 IN_PROC_BROWSER_TEST_F(PolicyTest, DownloadDirectory) {
   // Verifies that the download directory can be forced by policy.
 
   // Set the initial download directory.
   base::ScopedTempDir initial_dir;
   ASSERT_TRUE(initial_dir.CreateUniqueTempDir());
   browser()->profile()->GetPrefs()->SetFilePath(
       prefs::kDownloadDefaultDirectory, initial_dir.path());
@@ skipping 643 matching lines @@
   EXPECT_EQ(kExpectedLabel, text.substr(0, kExpectedLabel.size()));
   // HomepageLocation has policy ID 1.
   EXPECT_NE(std::string::npos, text.find("<br>1   ---"));
   // ShowHomeButton has policy ID 35.
   EXPECT_NE(std::string::npos, text.find("<br>35  ---"));
   // BookmarkBarEnabled has policy ID 82.
   EXPECT_NE(std::string::npos, text.find("<br>82  ---"));
 }
 
 }  // namespace policy