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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_UI_VIEWS_EXTENSIONS_EXTENSIONS_TOOLBAR_INTERACTIVE_UITEST_H_
#define CHROME_BROWSER_UI_VIEWS_EXTENSIONS_EXTENSIONS_TOOLBAR_INTERACTIVE_UITEST_H_
#include <string>
#include <vector>
#include "base/memory/raw_ptr.h"
#include "chrome/browser/ui/test/test_browser_dialog.h"
#include "chrome/browser/ui/views/extensions/extensions_menu_button.h"
#include "extensions/common/extension.h"
class ExtensionsToolbarContainer;
class ToolbarActionView;
class ExtensionsToolbarButton;
class ExtensionsMenuCoordinator;
namespace extensions {
class Extension;
}
namespace views {
class Button;
}
// Base class for interactive ui tests that use the toolbar area. This is used
// for browser test fixtures that are generally related to the
// ExtensionsToolbarContainer in the ToolbarView area. For example, this is used
// by ExtensionsToolbarContainer and ExtensionsMenuView separately to clarify
// what the suite is primarily trying to test.
class ExtensionsToolbarUITest : public DialogBrowserTest {
public:
ExtensionsToolbarUITest(const ExtensionsToolbarUITest&) = delete;
ExtensionsToolbarUITest& operator=(const ExtensionsToolbarUITest&) = delete;
protected:
ExtensionsToolbarUITest();
~ExtensionsToolbarUITest() override;
void SetUpOnMainThread() override;
Profile* profile();
Browser* incognito_browser() { return incognito_browser_; }
const std::vector<scoped_refptr<const extensions::Extension>>& extensions() {
return extensions_;
}
// Loads and returns a test extension from |chrome::DIR_TEST_DATA|.
// |allow_incognito| is used to declare whether the extension is allowed to
// run in incognito.
scoped_refptr<const extensions::Extension> LoadTestExtension(
const std::string& path,
bool allow_incognito = false);
scoped_refptr<const extensions::Extension> ForceInstallExtension(
const std::string& name);
// Loads and returns a extension given a `name`.
scoped_refptr<const extensions::Extension> InstallExtension(
const std::string& name);
// Loads and returns a extension given a `name`, `host_permission` and
// optional `content_script_run_location`.
scoped_refptr<const extensions::Extension>
InstallExtensionWithHostPermissions(
const std::string& name,
const std::string& host_permission,
const std::string& content_script_run_location = "");
// Adds |extension| to the back of |extensions_|.
void AppendExtension(scoped_refptr<const extensions::Extension> extension);
// Disables the extension of the given `extension_id`.
void DisableExtension(const extensions::ExtensionId& extension_id);
// Sets up |incognito_browser_|.
void SetUpIncognitoBrowser();
// Gets the extensions toolbar container from the browser() toolbar.
ExtensionsToolbarContainer* GetExtensionsToolbarContainer() const;
// Returns the extensions toolbar container for the given `browser`.
ExtensionsToolbarContainer* GetExtensionsToolbarContainerForBrowser(
Browser* browser) const;
// Gets the ToolbarActionView instances inside
// GetExtensionsToolbarContainer().
std::vector<ToolbarActionView*> GetToolbarActionViews() const;
// Returns the ToolbarActionView instances within the extensions toolbar for
// the given `browser`.
std::vector<ToolbarActionView*> GetToolbarActionViewsForBrowser(
Browser* browser) const;
// Gets only the visible ToolbarActionView instances from
// GetToolbarActionViews().
std::vector<ToolbarActionView*> GetVisibleToolbarActionViews() const;
// Returns the extensions button in the toolbar.
ExtensionsToolbarButton* extensions_button();
// Returns the extensions menu coordinator.
ExtensionsMenuCoordinator* menu_coordinator();
// Triggers the press and release event of the given `button`.
void ClickButton(views::Button* button) const;
// Returns whether the extension injected a script by checking the document
// title. Extension must use 'extensions/blocked_actions/content_scripts'.
bool DidInjectScript(content::WebContents* web_contents);
// Navigate to `url` in the currently active web contents.
void NavigateTo(const GURL& url);
// Waits for the extensions container to animate (on pin, unpin, pop-out,
// etc.)
void WaitForAnimation();
private:
raw_ptr<Browser, AcrossTasksDanglingUntriaged> incognito_browser_ = nullptr;
std::vector<scoped_refptr<const extensions::Extension>> extensions_;
};
#endif // CHROME_BROWSER_UI_VIEWS_EXTENSIONS_EXTENSIONS_TOOLBAR_INTERACTIVE_UITEST_H_
|
