src/chrome/browser/extensions/webstore_reinstaller_browsertest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/bind.h"
   6 #include "base/run_loop.h"
   7 #include "chrome/browser/extensions/extension_service.h"
   8 #include "chrome/browser/extensions/webstore_installer_test.h"
   9 #include "chrome/browser/extensions/webstore_reinstaller.h"
  10 #include "chrome/browser/ui/browser.h"
  11 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  12 #include "chrome/common/extensions/webstore_install_result.h"
  13 #include "extensions/browser/extension_registry.h"
  14 #include "extensions/common/extension.h"
  15 #include "extensions/common/extension_builder.h"
  16 #include "extensions/common/value_builder.h"
  17
  18 namespace extensions {
  19
  20 namespace {
  21
  22 const char kWebstoreDomain[] = "cws.com";
  23 const char kAppDomain[] = "app.com";
  24 const char kNonAppDomain[] = "nonapp.com";
  25 const char kTestExtensionId[] = "ecglahbcnmdpdciemllbhojghbkagdje";
  26 const char kTestDataPath[] = "extensions/api_test/webstore_inline_install";
  27 const char kCrxFilename[] = "extension.crx";
  28
  29 }  // namespace
  30
  31 class WebstoreReinstallerBrowserTest : public WebstoreInstallerTest {
  32  public:
  33   WebstoreReinstallerBrowserTest()
  34       : WebstoreInstallerTest(
  35             kWebstoreDomain,
  36             kTestDataPath,
  37             kCrxFilename,
  38             kAppDomain,
  39             kNonAppDomain) {}
  40   virtual ~WebstoreReinstallerBrowserTest() {}
  41
  42   void OnInstallCompletion(const base::Closure& quit_closure,
  43                            bool success,
  44                            const std::string& error,
  45                            webstore_install::Result result);
  46
  47   bool last_install_result() const { return last_install_result_; }
  48
  49  private:
  50   bool last_install_result_;
  51 };
  52
  53 void WebstoreReinstallerBrowserTest::OnInstallCompletion(
  54     const base::Closure& quit_closure,
  55     bool success,
  56     const std::string& error,
  57     webstore_install::Result result) {
  58   last_install_result_ = success;
  59   quit_closure.Run();
  60 }
  61
  62 IN_PROC_BROWSER_TEST_F(WebstoreReinstallerBrowserTest, TestWebstoreReinstall) {
  63   // Build an extension with the same id as our test extension and add it.
  64   const std::string kExtensionName("ReinstallerExtension");
  65   scoped_refptr<const Extension> extension =
  66       ExtensionBuilder().SetLocation(Manifest::INTERNAL)
  67                       .SetID(kTestExtensionId)
  68                       .SetManifest(
  69                           DictionaryBuilder().Set("name", kExtensionName)
  70                                              .Set("description", "Foo")
  71                                              .Set("manifest_version", 2)
  72                                              .Set("version", "1.0")
  73                                              .Build())
  74                       .Build();
  75   extension_service()->AddExtension(extension.get());
  76   ExtensionRegistry* registry = ExtensionRegistry::Get(profile());
  77   ASSERT_TRUE(registry->enabled_extensions().GetByID(kTestExtensionId));
  78
  79   content::WebContents* active_web_contents =
  80       browser()->tab_strip_model()->GetActiveWebContents();
  81   ASSERT_TRUE(active_web_contents);
  82
  83   // Start by canceling the repair prompt.
  84   AutoCancelInstall();
  85
  86   // Create and run a WebstoreReinstaller.
  87   base::RunLoop run_loop;
  88   scoped_refptr<WebstoreReinstaller> reinstaller(
  89       new WebstoreReinstaller(
  90           active_web_contents,
  91           kTestExtensionId,
  92           base::Bind(&WebstoreReinstallerBrowserTest::OnInstallCompletion,
  93                      base::Unretained(this),
  94                      run_loop.QuitClosure())));
  95   reinstaller->BeginReinstall();
  96   run_loop.Run();
  97
  98   // We should have failed, and the old extension should still be present.
  99   EXPECT_FALSE(last_install_result());
 100   extension = registry->enabled_extensions().GetByID(kTestExtensionId);
 101   ASSERT_TRUE(extension.get());
 102   EXPECT_EQ(kExtensionName, extension->name());
 103
 104   // Now accept the repair prompt.
 105   AutoAcceptInstall();
 106   base::RunLoop run_loop2;
 107   reinstaller =
 108       new WebstoreReinstaller(
 109           active_web_contents,
 110           kTestExtensionId,
 111           base::Bind(&WebstoreReinstallerBrowserTest::OnInstallCompletion,
 112                      base::Unretained(this),
 113                      run_loop2.QuitClosure()));
 114   reinstaller->BeginReinstall();
 115   run_loop2.Run();
 116
 117   // The reinstall should have succeeded, and the extension should have been
 118   // "updated" (which in this case means that it should have been replaced with
 119   // the inline install test extension, since that's the id we used).
 120   EXPECT_TRUE(last_install_result());
 121   extension = registry->enabled_extensions().GetByID(kTestExtensionId);
 122   ASSERT_TRUE(extension.get());
 123   // The name should not match, since the extension changed.
 124   EXPECT_NE(kExtensionName, extension->name());
 125 }
 126
 127 }  // namespace extensions