Home | History | Annotate | Download | only in extensions 
      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:
     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 
     55 IN_PROC_BROWSER_TEST_F(ExtensionFunctionalTest,
     56                        PRE_TestAdblockExtensionCrash) {
     57   ExtensionService* service = profile()->GetExtensionService();
     58   InstallExtensionSilently(service, "adblock.crx");
     59 }
     60 
     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 
     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