Chromium Code Reviews| Index: chrome/browser/extensions/extension_resource_request_policy_apitest.cc |
| diff --git a/chrome/browser/extensions/extension_resource_request_policy_apitest.cc b/chrome/browser/extensions/extension_resource_request_policy_apitest.cc |
| index db8a6065555358bd2dde5327d29a6f03180bf70c..c3b2a14a90fb0ed57974db50df775ec24b80ed7b 100644 |
| --- a/chrome/browser/extensions/extension_resource_request_policy_apitest.cc |
| +++ b/chrome/browser/extensions/extension_resource_request_policy_apitest.cc |
| @@ -12,6 +12,7 @@ |
| #include "chrome/test/base/test_switches.h" |
| #include "chrome/test/base/ui_test_utils.h" |
| #include "content/public/browser/navigation_handle.h" |
| +#include "content/public/browser/notification_service.h" |
| #include "content/public/browser/web_contents.h" |
| #include "content/public/browser/web_contents_observer.h" |
| #include "content/public/test/browser_test_utils.h" |
| @@ -261,27 +262,25 @@ IN_PROC_BROWSER_TEST_F(ExtensionResourceRequestPolicyTest, |
| &result)); |
| EXPECT_EQ("about:blank", result); |
| + // Redirects can sometimes occur before the load event, so use a |
| + // UrlLoadObserver instead of blocking waiting for two load events. |
| + ui_test_utils::UrlLoadObserver accessible_observer( |
| + GURL("chrome-extension://ggmldgjhdenlnjjjmehkomheglpmijnf/test.png"), |
|
Devlin
2016/08/10 20:20:08
nit: instead, use a reference to the extension (Lo
Charlie Harrison
2016/08/10 20:35:34
Done.
|
| + content::NotificationService::AllSources()); |
| GURL accessible_client_redirect_resource(embedded_test_server()->GetURL( |
| "/extensions/api_test/extension_resource_request_policy/" |
| "web_accessible/accessible_redirect_resource.html")); |
| - ui_test_utils::NavigateToURLBlockUntilNavigationsComplete(browser(), |
| - accessible_client_redirect_resource, 2); |
| - ASSERT_TRUE(content::ExecuteScriptAndExtractString( |
|
Devlin
2016/08/10 20:20:08
Using JS to get the url was probably overkill, but
Charlie Harrison
2016/08/10 20:35:34
Done.
|
| - browser()->tab_strip_model()->GetActiveWebContents(), |
| - "window.domAutomationController.send(document.URL)", |
| - &result)); |
| - EXPECT_NE("about:blank", result); |
| + ui_test_utils::NavigateToURL(browser(), accessible_client_redirect_resource); |
| + accessible_observer.Wait(); |
| + ui_test_utils::UrlLoadObserver nonaccessible_observer( |
| + GURL("about:blank"), content::NotificationService::AllSources()); |
| GURL nonaccessible_client_redirect_resource(embedded_test_server()->GetURL( |
| "/extensions/api_test/extension_resource_request_policy/" |
| "web_accessible/nonaccessible_redirect_resource.html")); |
| - ui_test_utils::NavigateToURLBlockUntilNavigationsComplete(browser(), |
| - nonaccessible_client_redirect_resource, 2); |
| - ASSERT_TRUE(content::ExecuteScriptAndExtractString( |
|
Devlin
2016/08/10 20:20:08
ditto
Charlie Harrison
2016/08/10 20:35:34
Done.
|
| - browser()->tab_strip_model()->GetActiveWebContents(), |
| - "window.domAutomationController.send(document.URL)", |
| - &result)); |
| - EXPECT_EQ("about:blank", result); |
| + ui_test_utils::NavigateToURL(browser(), |
| + nonaccessible_client_redirect_resource); |
| + nonaccessible_observer.Wait(); |
| } |
| IN_PROC_BROWSER_TEST_F(ExtensionResourceRequestPolicyTest, |