blob: bc655e310cb5463d2efcbc46c367ad4858236a33 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/extensions/browser_event_router.h"
#include "chrome/browser/extensions/extension_apitest.h"
#include "chrome/browser/extensions/extension_service.h"
#include "chrome/browser/extensions/location_bar_controller.h"
#include "chrome/browser/extensions/tab_helper.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/browser_tabstrip.h"
#include "chrome/browser/ui/tab_contents/tab_contents.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/common/extensions/extension.h"
#include "chrome/common/extensions/extension_action.h"
#include "chrome/common/extensions/features/feature.h"
#include "chrome/test/base/ui_test_utils.h"
#include "content/public/browser/web_contents.h"
#include "net/base/mock_host_resolver.h"
#include "net/test/test_server.h"
#include "testing/gmock/include/gmock/gmock.h"
using extensions::Extension;
class ScriptBadgeApiTest : public ExtensionApiTest {
public:
ScriptBadgeApiTest() : trunk_(chrome::VersionInfo::CHANNEL_UNKNOWN) {}
protected:
virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
ExtensionApiTest::SetUpCommandLine(command_line);
command_line->AppendSwitch(switches::kEnableScriptBadges);
}
private:
extensions::Feature::ScopedCurrentChannel trunk_;
};
IN_PROC_BROWSER_TEST_F(ScriptBadgeApiTest, Basics) {
ASSERT_TRUE(test_server()->Start());
ASSERT_TRUE(RunExtensionTest("script_badge/basics")) << message_;
const Extension* extension = GetSingleLoadedExtension();
ASSERT_TRUE(extension) << message_;
ExtensionAction* script_badge = extension->script_badge();
ASSERT_TRUE(script_badge);
const extensions::LocationBarController* location_bar_controller =
extensions::TabHelper::FromWebContents(
chrome::GetActiveWebContents(browser()))->
location_bar_controller();
const int tab_id = SessionID::IdForTab(
chrome::GetActiveWebContents(browser()));
EXPECT_EQ(GURL(extension->GetResourceURL("default_popup.html")),
script_badge->GetPopupUrl(tab_id));
EXPECT_FALSE(script_badge->GetIsVisible(tab_id));
EXPECT_THAT(location_bar_controller->GetCurrentActions(),
testing::ElementsAre());
{
ResultCatcher catcher;
// Tell the extension to update the script badge state.
ui_test_utils::NavigateToURL(
browser(), GURL(extension->GetResourceURL("update.html")));
ASSERT_TRUE(catcher.GetNextResult());
}
// Test that we received the changes.
EXPECT_EQ(GURL(extension->GetResourceURL("set_popup.html")),
script_badge->GetPopupUrl(tab_id));
{
// Simulate the script badge being clicked.
ResultCatcher catcher;
ExtensionService* service = browser()->profile()->GetExtensionService();
service->browser_event_router()->ScriptBadgeExecuted(
browser()->profile(), *script_badge, tab_id);
EXPECT_TRUE(catcher.GetNextResult());
}
{
ResultCatcher catcher;
// Visit a non-extension page so the extension can run a content script and
// cause the script badge to be animated in.
ui_test_utils::NavigateToURL(browser(), test_server()->GetURL(""));
ASSERT_TRUE(catcher.GetNextResult());
}
EXPECT_TRUE(script_badge->GetIsVisible(tab_id));
EXPECT_THAT(location_bar_controller->GetCurrentActions(),
testing::ElementsAre(script_badge));
}
|
