OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <chrome/browser/extensions/chrome_extension_test_notification_observer.
h> | 5 #include <chrome/browser/extensions/chrome_extension_test_notification_observer.
h> |
6 #include "base/scoped_observer.h" | 6 #include "base/scoped_observer.h" |
7 #include "chrome/browser/extensions/extension_action_test_util.h" | 7 #include "chrome/browser/extensions/extension_action_test_util.h" |
8 #include "chrome/browser/extensions/extension_util.h" | 8 #include "chrome/browser/extensions/extension_util.h" |
9 #include "chrome/browser/profiles/profile_manager.h" | 9 #include "chrome/browser/profiles/profile_manager.h" |
10 #include "chrome/browser/ui/browser.h" | 10 #include "chrome/browser/ui/browser.h" |
11 #include "chrome/browser/ui/tabs/tab_strip_model.h" | 11 #include "chrome/browser/ui/tabs/tab_strip_model.h" |
12 #include "content/public/browser/browser_context.h" | 12 #include "content/public/browser/browser_context.h" |
13 #include "content/public/browser/web_contents.h" | 13 #include "content/public/browser/web_contents.h" |
14 #include "extensions/browser/notification_types.h" | 14 #include "extensions/browser/notification_types.h" |
15 #include "extensions/browser/process_manager.h" | 15 #include "extensions/browser/process_manager.h" |
16 #include "extensions/common/extension.h" | 16 #include "extensions/common/extension.h" |
17 | 17 |
18 using extensions::Extension; | 18 namespace extensions { |
19 | 19 |
20 namespace { | 20 namespace { |
21 | 21 |
22 // A callback that returns true if the condition has been met and takes no | 22 // A callback that returns true if the condition has been met and takes no |
23 // arguments. | 23 // arguments. |
24 using ConditionCallback = base::Callback<bool(void)>; | 24 using ConditionCallback = base::Callback<bool(void)>; |
25 | 25 |
26 bool HasPageActionVisibilityReachedTarget( | 26 bool HasPageActionVisibilityReachedTarget( |
27 Browser* browser, | 27 Browser* browser, |
28 size_t target_visible_page_action_count) { | 28 size_t target_visible_page_action_count) { |
29 return extensions::extension_action_test_util::GetVisiblePageActionCount( | 29 return extension_action_test_util::GetVisiblePageActionCount( |
30 browser->tab_strip_model()->GetActiveWebContents()) == | 30 browser->tab_strip_model()->GetActiveWebContents()) == |
31 target_visible_page_action_count; | 31 target_visible_page_action_count; |
32 } | 32 } |
33 | 33 |
34 bool HaveAllExtensionRenderFrameHostsFinishedLoading( | 34 bool HaveAllExtensionRenderFrameHostsFinishedLoading(ProcessManager* manager) { |
35 extensions::ProcessManager* manager) { | 35 ProcessManager::FrameSet all_views = manager->GetAllFrames(); |
36 extensions::ProcessManager::FrameSet all_views = manager->GetAllFrames(); | |
37 for (content::RenderFrameHost* host : manager->GetAllFrames()) { | 36 for (content::RenderFrameHost* host : manager->GetAllFrames()) { |
38 if (content::WebContents::FromRenderFrameHost(host)->IsLoading()) | 37 if (content::WebContents::FromRenderFrameHost(host)->IsLoading()) |
39 return false; | 38 return false; |
40 } | 39 } |
41 return true; | 40 return true; |
42 } | 41 } |
43 | 42 |
44 } // namespace | 43 } // namespace |
45 | 44 |
46 //////////////////////////////////////////////////////////////////////////////// | 45 //////////////////////////////////////////////////////////////////////////////// |
(...skipping 18 matching lines...) Expand all Loading... |
65 context_ = browser_->profile(); | 64 context_ = browser_->profile(); |
66 else | 65 else |
67 context_ = ProfileManager::GetActiveUserProfile(); | 66 context_ = ProfileManager::GetActiveUserProfile(); |
68 } | 67 } |
69 return context_; | 68 return context_; |
70 } | 69 } |
71 | 70 |
72 bool ChromeExtensionTestNotificationObserver:: | 71 bool ChromeExtensionTestNotificationObserver:: |
73 WaitForPageActionVisibilityChangeTo(int count) { | 72 WaitForPageActionVisibilityChangeTo(int count) { |
74 DCHECK(browser_); | 73 DCHECK(browser_); |
75 ScopedObserver<extensions::ExtensionActionAPI, | 74 ScopedObserver<ExtensionActionAPI, ExtensionActionAPI::Observer> observer( |
76 extensions::ExtensionActionAPI::Observer> | 75 this); |
77 observer(this); | 76 observer.Add(ExtensionActionAPI::Get(GetBrowserContext())); |
78 observer.Add(extensions::ExtensionActionAPI::Get(GetBrowserContext())); | |
79 WaitForCondition( | 77 WaitForCondition( |
80 base::Bind(&HasPageActionVisibilityReachedTarget, browser_, count), NULL); | 78 base::Bind(&HasPageActionVisibilityReachedTarget, browser_, count), NULL); |
81 return true; | 79 return true; |
82 } | 80 } |
83 | 81 |
84 bool ChromeExtensionTestNotificationObserver::WaitForExtensionViewsToLoad() { | 82 bool ChromeExtensionTestNotificationObserver::WaitForExtensionViewsToLoad() { |
85 extensions::ProcessManager* manager = | 83 ProcessManager* manager = ProcessManager::Get(GetBrowserContext()); |
86 extensions::ProcessManager::Get(GetBrowserContext()); | |
87 NotificationSet notification_set; | 84 NotificationSet notification_set; |
88 notification_set.Add(content::NOTIFICATION_WEB_CONTENTS_DESTROYED); | 85 notification_set.Add(content::NOTIFICATION_WEB_CONTENTS_DESTROYED); |
89 notification_set.Add(content::NOTIFICATION_LOAD_STOP); | 86 notification_set.Add(content::NOTIFICATION_LOAD_STOP); |
90 notification_set.AddExtensionFrameUnregistration(manager); | 87 notification_set.AddExtensionFrameUnregistration(manager); |
91 WaitForCondition( | 88 WaitForCondition( |
92 base::Bind(&HaveAllExtensionRenderFrameHostsFinishedLoading, manager), | 89 base::Bind(&HaveAllExtensionRenderFrameHostsFinishedLoading, manager), |
93 ¬ification_set); | 90 ¬ification_set); |
94 return true; | 91 return true; |
95 } | 92 } |
96 | 93 |
97 bool ChromeExtensionTestNotificationObserver::WaitForExtensionIdle( | 94 bool ChromeExtensionTestNotificationObserver::WaitForExtensionIdle( |
98 const std::string& extension_id) { | 95 const std::string& extension_id) { |
99 NotificationSet notification_set; | 96 NotificationSet notification_set; |
100 notification_set.Add(content::NOTIFICATION_RENDERER_PROCESS_TERMINATED); | 97 notification_set.Add(content::NOTIFICATION_RENDERER_PROCESS_TERMINATED); |
101 WaitForCondition(base::Bind(&extensions::util::IsExtensionIdle, extension_id, | 98 WaitForCondition( |
102 GetBrowserContext()), | 99 base::Bind(&util::IsExtensionIdle, extension_id, GetBrowserContext()), |
103 ¬ification_set); | 100 ¬ification_set); |
104 return true; | 101 return true; |
105 } | 102 } |
106 | 103 |
107 bool ChromeExtensionTestNotificationObserver::WaitForExtensionNotIdle( | 104 bool ChromeExtensionTestNotificationObserver::WaitForExtensionNotIdle( |
108 const std::string& extension_id) { | 105 const std::string& extension_id) { |
109 NotificationSet notification_set; | 106 NotificationSet notification_set; |
110 notification_set.Add(content::NOTIFICATION_LOAD_STOP); | 107 notification_set.Add(content::NOTIFICATION_LOAD_STOP); |
111 WaitForCondition(base::Bind( | 108 WaitForCondition(base::Bind( |
112 [](const std::string& extension_id, | 109 [](const std::string& extension_id, |
113 content::BrowserContext* context) -> bool { | 110 content::BrowserContext* context) -> bool { |
114 return !extensions::util::IsExtensionIdle(extension_id, | 111 return !util::IsExtensionIdle(extension_id, context); |
115 context); | |
116 }, | 112 }, |
117 extension_id, GetBrowserContext()), | 113 extension_id, GetBrowserContext()), |
118 ¬ification_set); | 114 ¬ification_set); |
119 return true; | 115 return true; |
120 } | 116 } |
121 | 117 |
122 void ChromeExtensionTestNotificationObserver::WaitForExtensionAndViewLoad() { | 118 void ChromeExtensionTestNotificationObserver::WaitForExtensionAndViewLoad() { |
123 WaitForExtensionLoad(); | 119 WaitForExtensionLoad(); |
124 WaitForExtensionViewsToLoad(); | 120 WaitForExtensionViewsToLoad(); |
125 } | 121 } |
126 | 122 |
127 void ChromeExtensionTestNotificationObserver::OnPageActionsUpdated( | 123 void ChromeExtensionTestNotificationObserver::OnPageActionsUpdated( |
128 content::WebContents* web_contents) { | 124 content::WebContents* web_contents) { |
129 MaybeQuit(); | 125 MaybeQuit(); |
130 } | 126 } |
| 127 |
| 128 } // namespace extensions |
OLD | NEW |