Add a webstore API for installing bundles of extensions.

chrome/browser/extensions/extension_webstore_private_apitest.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 <vector>
 
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/stringprintf.h"
+#include "chrome/browser/extensions/api/webstore/bundle_installer.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/browser/extensions/extension_function_test_utils.h"
 #include "chrome/browser/extensions/extension_install_dialog.h"
 #include "chrome/browser/extensions/extension_install_ui.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_webstore_private_api.h"
 #include "chrome/browser/extensions/webstore_installer.h"
 #include "chrome/browser/gpu_blacklist.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
@@ skipping 10 matching lines @@
 
 using namespace extension_function_test_utils;
 
 namespace {
 
 class WebstoreInstallListener : public WebstoreInstaller::Delegate {
  public:
   WebstoreInstallListener()
       : received_failure_(false), received_success_(false), waiting_(false) {}
 
-  void OnExtensionInstallSuccess(const std::string& id) OVERRIDE;
+  void OnExtensionInstallSuccess(const std::string& id) OVERRIDE {
+    received_success_ = true;
+    id_ = id;
+
+    if (waiting_) {
+      waiting_ = false;
+      MessageLoopForUI::current()->Quit();
+    }
+  }
+
   void OnExtensionInstallFailure(const std::string& id,
-                                 const std::string& error) OVERRIDE;
-  void Wait();
+                                 const std::string& error) OVERRIDE {
+    received_failure_ = true;
+    id_ = id;
+    error_ = error;
+
+    if (waiting_) {
+      waiting_ = false;
+      MessageLoopForUI::current()->Quit();
+    }
+  }
+
+  void Wait() {
+    if (received_success_ || received_failure_)
+      return;
+
+    waiting_ = true;
+    ui_test_utils::RunMessageLoop();
+  }
 
   bool received_failure() const { return received_failure_; }
   bool received_success() const { return received_success_; }
   const std::string& id() const { return id_; }
   const std::string& error() const { return error_; }
 
  private:
   bool received_failure_;
   bool received_success_;
   bool waiting_;
   std::string id_;
   std::string error_;
 };
 
-void WebstoreInstallListener::OnExtensionInstallSuccess(const std::string& id) {
-  received_success_ = true;
-  id_ = id;
-
-  if (waiting_) {
-    waiting_ = false;
-    MessageLoopForUI::current()->Quit();
-  }
-}
-
-void WebstoreInstallListener::OnExtensionInstallFailure(
-    const std::string& id, const std::string& error) {
-  received_failure_ = true;
-  id_ = id;
-  error_ = error;
-
-  if (waiting_) {
-    waiting_ = false;
-    MessageLoopForUI::current()->Quit();
-  }
-}
-
-void WebstoreInstallListener::Wait() {
-  if (received_success_ || received_failure_)
-    return;
-
-  waiting_ = true;
-  ui_test_utils::RunMessageLoop();
-}
-
 }  // namespace
 
 // A base class for tests below.
 class ExtensionWebstorePrivateApiTest : public ExtensionApiTest {
  public:
   void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     ExtensionApiTest::SetUpCommandLine(command_line);
     command_line->AppendSwitchASCII(
         switches::kAppsGalleryURL, "http://www.example.com");
     command_line->AppendSwitchASCII(
@@ skipping 44 matching lines @@
     : public ExtensionWebstorePrivateApiTest {
  public:
   void SetUpInProcessBrowserTestFixture() OVERRIDE {
     ExtensionWebstorePrivateApiTest::SetUpInProcessBrowserTestFixture();
 
     // The test server needs to have already started, so setup the switch here
     // rather than in SetUpCommandLine.
     CommandLine::ForCurrentProcess()->AppendSwitchASCII(
         switches::kAppsGalleryDownloadURL,
         GetTestServerURL("bundle/%s.crx").spec());
-
-    PackCRX("begfmnajjkbjdgmffnjaojchoncnmngg");
-    PackCRX("bmfoocgfinpmkmlbjhcbofejhkhlbchk");
-    PackCRX("mpneghmdnmaolkljkipbhaienajcflfe");
   }
 
   void TearDownInProcessBrowserTestFixture() OVERRIDE {
     ExtensionWebstorePrivateApiTest::TearDownInProcessBrowserTestFixture();
     for (size_t i = 0; i < test_crx_.size(); ++i)
       ASSERT_TRUE(file_util::Delete(test_crx_[i], false));
   }
 
- private:
+ protected:
   void PackCRX(const std::string& id) {
+    FilePath dir_path = test_data_dir_
+        .AppendASCII("webstore_private/bundle")
+        .AppendASCII(id);
+
+    PackCRX(id, dir_path);
+  }
+
+  // Packs the |manifest| file into a CRX using |id|'s PEM key.
+  void PackCRX(const std::string& id, const std::string& manifest) {
+    // Move the extension to a temporary directory.
+    ScopedTempDir tmp;
+    ASSERT_TRUE(tmp.CreateUniqueTempDir());
+    ASSERT_TRUE(file_util::CreateDirectory(tmp.path()));
+
+    FilePath tmp_manifest = tmp.path().AppendASCII("manifest.json");
     FilePath data_path = test_data_dir_.AppendASCII("webstore_private/bundle");
-    FilePath dir_path = data_path.AppendASCII(id);
+    FilePath manifest_path = data_path.AppendASCII(manifest);
+
+    ASSERT_TRUE(file_util::PathExists(manifest_path));
+    ASSERT_TRUE(file_util::CopyFile(manifest_path, tmp_manifest));
+
+    PackCRX(id, tmp.path());
+  }
+
+  // Packs the extension at |ext_path| using |id|'s PEM key.
+  void PackCRX(const std::string& id, const FilePath& ext_path) {
+    FilePath data_path = test_data_dir_.AppendASCII("webstore_private/bundle");
     FilePath pem_path = data_path.AppendASCII(id + ".pem");
     FilePath crx_path = data_path.AppendASCII(id + ".crx");
     FilePath destination = PackExtensionWithOptions(
-        dir_path, crx_path, pem_path, FilePath());
+        ext_path, crx_path, pem_path, FilePath());
 
     ASSERT_FALSE(destination.empty());
     ASSERT_EQ(destination, crx_path);
 
     test_crx_.push_back(destination);
   }
 
+  // Creates an invalid CRX.
+  void PackInvalidCRX(const std::string& id) {
+    FilePath contents = test_data_dir_
+        .AppendASCII("webstore_private")
+        .AppendASCII("install_bundle_invalid.html");
+    FilePath crx_path = test_data_dir_
+        .AppendASCII("webstore_private/bundle")
+        .AppendASCII(id + ".crx");
+
+    ASSERT_TRUE(file_util::CopyFile(contents, crx_path));
+
+    test_crx_.push_back(crx_path);
+  }
+
+ private:
   std::vector<FilePath> test_crx_;
 };
 
 class ExtensionWebstoreGetWebGLStatusTest : public InProcessBrowserTest {
  public:
   void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     // In linux, we need to launch GPU process to decide if WebGL is allowed.
     // Run it on top of osmesa to avoid bot driver issues.
 #if defined(OS_LINUX)
     CHECK(test_launcher_utils::OverrideGLImplementation(
@@ skipping 85 matching lines @@
 
 // Tests using the iconUrl parameter to the install function.
 IN_PROC_BROWSER_TEST_F(ExtensionWebstorePrivateApiTest,
                        IconUrl) {
   ASSERT_TRUE(RunInstallTest("icon_url.html", "extension.crx"));
 }
 
 // Tests using silentlyInstall to install extensions.
 IN_PROC_BROWSER_TEST_F(ExtensionWebstorePrivateBundleTest, SilentlyInstall) {
   WebstorePrivateApi::SetTrustTestIDsForTesting(true);
+
+  PackCRX("bmfoocgfinpmkmlbjhcbofejhkhlbchk", "extension1.json");
+  PackCRX("mpneghmdnmaolkljkipbhaienajcflfe", "extension2.json");
+  PackCRX("begfmnajjkbjdgmffnjaojchoncnmngg", "app2.json");
+
   ASSERT_TRUE(RunPageTest(GetTestServerURL("silently_install.html").spec()));
 }
 
+// Tests successfully installing a bundle of 2 apps and 2 extensions.
+IN_PROC_BROWSER_TEST_F(ExtensionWebstorePrivateBundleTest, InstallBundle) {
+  webstore::BundleInstaller::SetAutoApproveForTesting(true);
+
+  PackCRX("bmfoocgfinpmkmlbjhcbofejhkhlbchk", "extension1.json");
+  PackCRX("pkapffpjmiilhlhbibjhamlmdhfneidj", "extension2.json");
+  PackCRX("begfmnajjkbjdgmffnjaojchoncnmngg", "app1.json");
+  PackCRX("mpneghmdnmaolkljkipbhaienajcflfe", "app2.json");
+
+  ASSERT_TRUE(RunPageTest(GetTestServerURL("install_bundle.html").spec()));
+}
+
+// Tests the user canceling the bundle install prompt.
+IN_PROC_BROWSER_TEST_F(ExtensionWebstorePrivateBundleTest,
+                       InstallBundleCancel) {
+  // We don't need to create the CRX files since we are aborting the install.
+  webstore::BundleInstaller::SetAutoApproveForTesting(false);
+  ASSERT_TRUE(RunPageTest(GetTestServerURL(
+      "install_bundle_cancel.html").spec()));
+}
+
+// Tests partially installing a bundle (2 succeed, 1 fails due to an invalid
+// CRX, and 1 fails due to the manifests not matching).
+IN_PROC_BROWSER_TEST_F(ExtensionWebstorePrivateBundleTest,
+                       InstallBundleInvalid) {
+  webstore::BundleInstaller::SetAutoApproveForTesting(true);
+
+  PackInvalidCRX("aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa");
+  PackCRX("bmfoocgfinpmkmlbjhcbofejhkhlbchk", "extension1.json");
+  PackCRX("pkapffpjmiilhlhbibjhamlmdhfneidj", "extension2.json");
+  PackCRX("begfmnajjkbjdgmffnjaojchoncnmngg", "app1.json");
+
+  ASSERT_TRUE(RunPageTest(GetTestServerURL(
+      "install_bundle_invalid.html").spec()));
+
+  ASSERT_TRUE(service()->GetExtensionById(
+      "begfmnajjkbjdgmffnjaojchoncnmngg", false));
+  ASSERT_TRUE(service()->GetExtensionById(
+      "pkapffpjmiilhlhbibjhamlmdhfneidj", false));
+  ASSERT_FALSE(service()->GetExtensionById(
+      "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa", true));
+  ASSERT_FALSE(service()->GetExtensionById(
+      "bmfoocgfinpmkmlbjhcbofejhkhlbchk", true));
+}
+
 // Tests getWebGLStatus function when WebGL is allowed.
 IN_PROC_BROWSER_TEST_F(ExtensionWebstoreGetWebGLStatusTest, Allowed) {
   bool webgl_allowed = true;
   RunTest(webgl_allowed);
 }
 
 // Tests getWebGLStatus function when WebGL is blacklisted.
 IN_PROC_BROWSER_TEST_F(ExtensionWebstoreGetWebGLStatusTest, Blocked) {
   static const std::string json_blacklist =
       "{\n"
@@ skipping 13 matching lines @@
 
   ASSERT_TRUE(blacklist->LoadGpuBlacklist(
       "1.0", json_blacklist, GpuBlacklist::kAllOs));
   blacklist->UpdateGpuDataManager();
   GpuFeatureType type = GpuDataManager::GetInstance()->GetGpuFeatureType();
   EXPECT_EQ(type, content::GPU_FEATURE_TYPE_WEBGL);
 
   bool webgl_allowed = false;
   RunTest(webgl_allowed);
 }