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.
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"
18 namespace extensions {
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";
31 class WebstoreReinstallerBrowserTest : public WebstoreInstallerTest {
33 WebstoreReinstallerBrowserTest()
34 : WebstoreInstallerTest(
40 ~WebstoreReinstallerBrowserTest() override {}
42 void OnInstallCompletion(const base::Closure& quit_closure,
44 const std::string& error,
45 webstore_install::Result result);
47 bool last_install_result() const { return last_install_result_; }
50 bool last_install_result_;
53 void WebstoreReinstallerBrowserTest::OnInstallCompletion(
54 const base::Closure& quit_closure,
56 const std::string& error,
57 webstore_install::Result result) {
58 last_install_result_ = success;
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)
69 DictionaryBuilder().Set("name", kExtensionName)
70 .Set("description", "Foo")
71 .Set("manifest_version", 2)
72 .Set("version", "1.0")
75 extension_service()->AddExtension(extension.get());
76 ExtensionRegistry* registry = ExtensionRegistry::Get(profile());
77 ASSERT_TRUE(registry->enabled_extensions().GetByID(kTestExtensionId));
79 content::WebContents* active_web_contents =
80 browser()->tab_strip_model()->GetActiveWebContents();
81 ASSERT_TRUE(active_web_contents);
83 // Start by canceling the repair prompt.
86 // Create and run a WebstoreReinstaller.
87 base::RunLoop run_loop;
88 scoped_refptr<WebstoreReinstaller> reinstaller(
89 new WebstoreReinstaller(
92 base::Bind(&WebstoreReinstallerBrowserTest::OnInstallCompletion,
93 base::Unretained(this),
94 run_loop.QuitClosure())));
95 reinstaller->BeginReinstall();
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());
104 // Now accept the repair prompt.
106 base::RunLoop run_loop2;
108 new WebstoreReinstaller(
111 base::Bind(&WebstoreReinstallerBrowserTest::OnInstallCompletion,
112 base::Unretained(this),
113 run_loop2.QuitClosure()));
114 reinstaller->BeginReinstall();
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());
127 } // namespace extensions