1 // Copyright (c) 2012 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 "chrome/browser/chrome_notification_types.h"
6 #include "chrome/browser/extensions/crx_installer.h"
7 #include "chrome/browser/extensions/extension_browsertest.h"
8 #include "chrome/browser/extensions/extension_service.h"
9 #include "chrome/browser/extensions/extension_util.h"
10 #include "chrome/browser/profiles/profile.h"
11 #include "chrome/browser/ui/browser_commands.h"
12 #include "chrome/test/base/ui_test_utils.h"
13 #include "content/public/browser/notification_service.h"
14
15 namespace extensions {
16
17 class ExtensionFunctionalTest : public ExtensionBrowserTest {
18 public:
InstallExtensionSilently(ExtensionService * service,const char * filename)19 void InstallExtensionSilently(ExtensionService* service,
20 const char* filename) {
21 service->set_show_extensions_prompts(false);
22 size_t num_before = service->extensions()->size();
23
24 base::FilePath path = test_data_dir_.AppendASCII(filename);
25
26 content::WindowedNotificationObserver extension_loaded_observer(
27 chrome::NOTIFICATION_EXTENSION_LOADED_DEPRECATED,
28 content::NotificationService::AllSources());
29
30 scoped_refptr<extensions::CrxInstaller> installer(
31 extensions::CrxInstaller::CreateSilent(service));
32 installer->set_is_gallery_install(false);
33 installer->set_allow_silent_install(true);
34 installer->set_install_source(Manifest::INTERNAL);
35 installer->set_off_store_install_allow_reason(
36 extensions::CrxInstaller::OffStoreInstallAllowedInTest);
37
38 observer_->Watch(
39 chrome::NOTIFICATION_CRX_INSTALLER_DONE,
40 content::Source<extensions::CrxInstaller>(installer.get()));
41
42 installer->InstallCrx(path);
43 observer_->Wait();
44
45 size_t num_after = service->extensions()->size();
46 EXPECT_EQ(num_before + 1, num_after);
47
48 extension_loaded_observer.Wait();
49 const Extension* extension = service->GetExtensionById(
50 last_loaded_extension_id(), false);
51 EXPECT_TRUE(extension != NULL);
52 }
53 };
54
IN_PROC_BROWSER_TEST_F(ExtensionFunctionalTest,PRE_TestAdblockExtensionCrash)55 IN_PROC_BROWSER_TEST_F(ExtensionFunctionalTest,
56 PRE_TestAdblockExtensionCrash) {
57 ExtensionService* service = profile()->GetExtensionService();
58 InstallExtensionSilently(service, "adblock.crx");
59 }
60
IN_PROC_BROWSER_TEST_F(ExtensionFunctionalTest,TestAdblockExtensionCrash)61 IN_PROC_BROWSER_TEST_F(ExtensionFunctionalTest, TestAdblockExtensionCrash) {
62 ExtensionService* service = profile()->GetExtensionService();
63 // Verify that the extension is enabled and allowed in incognito
64 // is disabled.
65 EXPECT_TRUE(service->IsExtensionEnabled(last_loaded_extension_id()));
66 EXPECT_FALSE(util::IsIncognitoEnabled(last_loaded_extension_id(), profile()));
67 }
68
IN_PROC_BROWSER_TEST_F(ExtensionFunctionalTest,TestSetExtensionsState)69 IN_PROC_BROWSER_TEST_F(ExtensionFunctionalTest, TestSetExtensionsState) {
70 ExtensionService* service = profile()->GetExtensionService();
71 InstallExtensionSilently(service, "google_talk.crx");
72
73 // Disable the extension and verify.
74 util::SetIsIncognitoEnabled(last_loaded_extension_id(), profile(), false);
75 service->DisableExtension(last_loaded_extension_id(),
76 Extension::DISABLE_USER_ACTION);
77 EXPECT_FALSE(service->IsExtensionEnabled(last_loaded_extension_id()));
78
79 // Enable the extension and verify.
80 util::SetIsIncognitoEnabled(last_loaded_extension_id(), profile(), false);
81 service->EnableExtension(last_loaded_extension_id());
82 EXPECT_TRUE(service->IsExtensionEnabled(last_loaded_extension_id()));
83
84 // Allow extension in incognito mode and verify.
85 service->EnableExtension(last_loaded_extension_id());
86 util::SetIsIncognitoEnabled(last_loaded_extension_id(), profile(), true);
87 EXPECT_TRUE(util::IsIncognitoEnabled(last_loaded_extension_id(), profile()));
88
89 // Disallow extension in incognito mode and verify.
90 service->EnableExtension(last_loaded_extension_id());
91 util::SetIsIncognitoEnabled(last_loaded_extension_id(), profile(), false);
92 EXPECT_FALSE(util::IsIncognitoEnabled(last_loaded_extension_id(), profile()));
93 }
94 } // namespace extensions
95