Support an --install-from-webstore command line switch

chrome/browser/extensions/webstore_inline_install_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 "base/command_line.h"
 #include "base/stringprintf.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/extensions/extension_host.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/startup_helper.h"
 #include "chrome/browser/extensions/webstore_inline_installer.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_tabstrip.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
+#include "content/public/browser/notification_registrar.h"
+#include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/browser_test_utils.h"
 #include "googleurl/src/gurl.h"
 #include "net/base/host_port_pair.h"
 #include "net/base/mock_host_resolver.h"
 
 using content::WebContents;
+using extensions::Extension;
 
 const char kWebstoreDomain[] = "cws.com";
 const char kAppDomain[] = "app.com";
 const char kNonAppDomain[] = "nonapp.com";
+const char kTestExtensionId[] = "ecglahbcnmdpdciemllbhojghbkagdje";
 
 class WebstoreInlineInstallTest : public InProcessBrowserTest {
  public:
   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     // We start the test server now instead of in
     // SetUpInProcessBrowserTestFixture so that we can get its port number.
     ASSERT_TRUE(test_server()->Start());
 
     net::HostPortPair host_port = test_server()->host_port_pair();
     test_gallery_url_ = base::StringPrintf(
@@ skipping 44 matching lines @@
 IN_PROC_BROWSER_TEST_F(WebstoreInlineInstallTest, Install) {
   CommandLine::ForCurrentProcess()->AppendSwitchASCII(
       switches::kAppsGalleryInstallAutoConfirmForTests, "accept");
 
   ui_test_utils::NavigateToURL(
       browser(), GenerateTestServerUrl(kAppDomain, "install.html"));
 
   RunInlineInstallTest("runTest");
 
   const extensions::Extension* extension = browser()->profile()->
-      GetExtensionService()->GetExtensionById(
-          "ecglahbcnmdpdciemllbhojghbkagdje", false);
+      GetExtensionService()->GetExtensionById(kTestExtensionId, false);
   EXPECT_TRUE(extension);
 }
 
 IN_PROC_BROWSER_TEST_F(WebstoreInlineInstallTest,
     InstallNotAllowedFromNonVerifiedDomains) {
   CommandLine::ForCurrentProcess()->AppendSwitchASCII(
       switches::kAppsGalleryInstallAutoConfirmForTests, "cancel");
   ui_test_utils::NavigateToURL(
       browser(),
       GenerateTestServerUrl(kNonAppDomain, "install_non_verified_domain.html"));
@@ skipping 59 matching lines @@
 IN_PROC_BROWSER_TEST_F(WebstoreInlineInstallUnpackFailureTest,
     WebstoreInlineInstallUnpackFailureTest) {
   CommandLine::ForCurrentProcess()->AppendSwitchASCII(
       switches::kAppsGalleryInstallAutoConfirmForTests, "accept");
 
   ui_test_utils::NavigateToURL(browser(),
       GenerateTestServerUrl(kAppDomain, "install_unpack_failure.html"));
 
   RunInlineInstallTest("runTest");
 }
+
+class CommandLineWebstoreInstall : public WebstoreInlineInstallTest,
+                                   public content::NotificationObserver {
+ public:
+  CommandLineWebstoreInstall() : saw_install_(false) {}
+  virtual ~CommandLineWebstoreInstall() {}
+
+  virtual void SetUpOnMainThread() OVERRIDE {
+    WebstoreInlineInstallTest::SetUpOnMainThread();
+    registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_INSTALLED,
+                   content::NotificationService::AllSources());
+    CommandLine::ForCurrentProcess()->AppendSwitchASCII(
+        switches::kInstallFromWebstore, kTestExtensionId);
+  }
+
+  bool saw_install() { return saw_install_; }
+
+ protected:
+  // NotificationObserver interface.
+  virtual void Observe(int type,
+                       const content::NotificationSource& source,
+                       const content::NotificationDetails& details)  OVERRIDE {
+    ASSERT_EQ(chrome::NOTIFICATION_EXTENSION_INSTALLED, type);
+    const Extension* extension = content::Details<Extension>(details).ptr();
+    ASSERT_TRUE(extension != NULL);
+    EXPECT_EQ(extension->id(), kTestExtensionId);
+    saw_install_ = true;
+  }
+
+  content::NotificationRegistrar registrar_;
+
+  // Have we seen an installation notification for kTestExtensionId ?
+  bool saw_install_;
+};
+
+IN_PROC_BROWSER_TEST_F(CommandLineWebstoreInstall, Accept) {
+  CommandLine* command_line = CommandLine::ForCurrentProcess();
+  command_line->AppendSwitchASCII(
+      switches::kAppsGalleryInstallAutoConfirmForTests, "accept");
+  extensions::StartupHelper helper;
+  EXPECT_TRUE(helper.InstallFromWebstore(*command_line, browser()->profile()));
+  EXPECT_TRUE(saw_install());
+}
+
+IN_PROC_BROWSER_TEST_F(CommandLineWebstoreInstall, Cancel) {
+  CommandLine* command_line = CommandLine::ForCurrentProcess();
+  command_line->AppendSwitchASCII(
+      switches::kAppsGalleryInstallAutoConfirmForTests, "cancel");
+  extensions::StartupHelper helper;
+  EXPECT_FALSE(helper.InstallFromWebstore(*command_line, browser()->profile()));
+  EXPECT_FALSE(saw_install());
+}