chrome/browser/extensions/api/extension_action/script_badge_apitest.cc

   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/extensions/api/extension_action/extension_action_api.h"
   6 #include "chrome/browser/extensions/extension_action.h"
   7 #include "chrome/browser/extensions/extension_action_manager.h"
   8 #include "chrome/browser/extensions/extension_apitest.h"
   9 #include "chrome/browser/extensions/extension_service.h"
  10 #include "chrome/browser/extensions/extension_system.h"
  11 #include "chrome/browser/extensions/location_bar_controller.h"
  12 #include "chrome/browser/extensions/tab_helper.h"
  13 #include "chrome/browser/profiles/profile.h"
  14 #include "chrome/browser/ui/browser.h"
  15 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  16 #include "chrome/common/extensions/features/feature_channel.h"
  17 #include "chrome/test/base/ui_test_utils.h"
  18 #include "content/public/browser/web_contents.h"
  19 #include "extensions/common/extension.h"
  20 #include "extensions/common/feature_switch.h"
  21 #include "net/dns/mock_host_resolver.h"
  22 #include "net/test/embedded_test_server/embedded_test_server.h"
  23 #include "testing/gmock/include/gmock/gmock.h"
  24
  25 namespace extensions {
  26 namespace {
  27
  28 class ScriptBadgeApiTest : public ExtensionApiTest {
  29  public:
  30   ScriptBadgeApiTest() : trunk_(chrome::VersionInfo::CHANNEL_UNKNOWN) {}
  31
  32  protected:
  33   virtual void SetUpInProcessBrowserTestFixture() OVERRIDE {
  34     ExtensionApiTest::SetUpInProcessBrowserTestFixture();
  35     FeatureSwitch::script_badges()->SetOverrideValue(
  36         FeatureSwitch::OVERRIDE_ENABLED);
  37   }
  38
  39  private:
  40   extensions::ScopedCurrentChannel trunk_;
  41 };
  42
  43 IN_PROC_BROWSER_TEST_F(ScriptBadgeApiTest, Basics) {
  44   ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady());
  45   ASSERT_TRUE(RunExtensionTest("script_badge/basics")) << message_;
  46   const Extension* extension = GetSingleLoadedExtension();
  47   ASSERT_TRUE(extension) << message_;
  48   ExtensionAction* script_badge =
  49       ExtensionActionManager::Get(browser()->profile())->
  50       GetScriptBadge(*extension);
  51   ASSERT_TRUE(script_badge);
  52   const extensions::LocationBarController* location_bar_controller =
  53       extensions::TabHelper::FromWebContents(
  54           browser()->tab_strip_model()->GetActiveWebContents())->
  55               location_bar_controller();
  56
  57   const int tab_id = SessionID::IdForTab(
  58       browser()->tab_strip_model()->GetActiveWebContents());
  59   EXPECT_EQ(GURL(extension->GetResourceURL("default_popup.html")),
  60             script_badge->GetPopupUrl(tab_id));
  61
  62   EXPECT_FALSE(script_badge->GetIsVisible(tab_id));
  63   EXPECT_THAT(location_bar_controller->GetCurrentActions(),
  64               testing::ElementsAre());
  65
  66   {
  67     ResultCatcher catcher;
  68     // Tell the extension to update the script badge state.
  69     ui_test_utils::NavigateToURL(
  70         browser(), GURL(extension->GetResourceURL("update.html")));
  71     ASSERT_TRUE(catcher.GetNextResult());
  72   }
  73
  74   // Test that we received the changes.
  75   EXPECT_EQ(GURL(extension->GetResourceURL("set_popup.html")),
  76             script_badge->GetPopupUrl(tab_id));
  77
  78   {
  79     // Simulate the script badge being clicked.
  80     ResultCatcher catcher;
  81     ExtensionActionAPI::ScriptBadgeExecuted(
  82         browser()->profile(), *script_badge, tab_id);
  83     EXPECT_TRUE(catcher.GetNextResult());
  84   }
  85
  86   {
  87     ResultCatcher catcher;
  88     // Visit a non-extension page so the extension can run a content script and
  89     // cause the script badge to be animated in.
  90     ui_test_utils::NavigateToURL(
  91         browser(), embedded_test_server()->GetURL("/title1.html"));
  92     ASSERT_TRUE(catcher.GetNextResult());
  93   }
  94   EXPECT_TRUE(script_badge->GetIsVisible(tab_id));
  95   EXPECT_THAT(location_bar_controller->GetCurrentActions(),
  96               testing::ElementsAre(script_badge));
  97 }
  98
  99 }  // namespace
 100 }  // namespace extensions