src/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/chrome_switches.h"
  17 #include "chrome/common/extensions/extension.h"
  18 #include "chrome/common/extensions/features/feature_channel.h"
  19 #include "chrome/test/base/ui_test_utils.h"
  20 #include "content/public/browser/web_contents.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 SetUpCommandLine(CommandLine* command_line) OVERRIDE {
  34     ExtensionApiTest::SetUpCommandLine(command_line);
  35     command_line->AppendSwitchASCII(switches::kScriptBadges, "1");
  36   }
  37
  38  private:
  39   extensions::ScopedCurrentChannel trunk_;
  40 };
  41
  42 IN_PROC_BROWSER_TEST_F(ScriptBadgeApiTest, Basics) {
  43   ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady());
  44   ASSERT_TRUE(RunExtensionTest("script_badge/basics")) << message_;
  45   const Extension* extension = GetSingleLoadedExtension();
  46   ASSERT_TRUE(extension) << message_;
  47   ExtensionAction* script_badge =
  48       ExtensionActionManager::Get(browser()->profile())->
  49       GetScriptBadge(*extension);
  50   ASSERT_TRUE(script_badge);
  51   const extensions::LocationBarController* location_bar_controller =
  52       extensions::TabHelper::FromWebContents(
  53           browser()->tab_strip_model()->GetActiveWebContents())->
  54               location_bar_controller();
  55
  56   const int tab_id = SessionID::IdForTab(
  57       browser()->tab_strip_model()->GetActiveWebContents());
  58   EXPECT_EQ(GURL(extension->GetResourceURL("default_popup.html")),
  59             script_badge->GetPopupUrl(tab_id));
  60
  61   EXPECT_FALSE(script_badge->GetIsVisible(tab_id));
  62   EXPECT_THAT(location_bar_controller->GetCurrentActions(),
  63               testing::ElementsAre());
  64
  65   {
  66     ResultCatcher catcher;
  67     // Tell the extension to update the script badge state.
  68     ui_test_utils::NavigateToURL(
  69         browser(), GURL(extension->GetResourceURL("update.html")));
  70     ASSERT_TRUE(catcher.GetNextResult());
  71   }
  72
  73   // Test that we received the changes.
  74   EXPECT_EQ(GURL(extension->GetResourceURL("set_popup.html")),
  75             script_badge->GetPopupUrl(tab_id));
  76
  77   {
  78     // Simulate the script badge being clicked.
  79     ResultCatcher catcher;
  80     ExtensionActionAPI::ScriptBadgeExecuted(
  81         browser()->profile(), *script_badge, tab_id);
  82     EXPECT_TRUE(catcher.GetNextResult());
  83   }
  84
  85   {
  86     ResultCatcher catcher;
  87     // Visit a non-extension page so the extension can run a content script and
  88     // cause the script badge to be animated in.
  89     ui_test_utils::NavigateToURL(
  90         browser(), embedded_test_server()->GetURL("/title1.html"));
  91     ASSERT_TRUE(catcher.GetNextResult());
  92   }
  93   EXPECT_TRUE(script_badge->GetIsVisible(tab_id));
  94   EXPECT_THAT(location_bar_controller->GetCurrentActions(),
  95               testing::ElementsAre(script_badge));
  96 }
  97
  98 }  // namespace
  99 }  // namespace extensions