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..faeebdea44716faf7001fad7efa960d1e94f2fd0 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" |
| @@ -235,9 +236,9 @@ IN_PROC_BROWSER_TEST_F(ExtensionResourceRequestPolicyTest, |
| LinkToWebAccessibleResources) { |
| std::string result; |
| ASSERT_TRUE(embedded_test_server()->Start()); |
| - ASSERT_TRUE(LoadExtension(test_data_dir_ |
| - .AppendASCII("extension_resource_request_policy") |
| - .AppendASCII("web_accessible"))); |
| + const extensions::Extension* extension = LoadExtension( |
| + test_data_dir_.AppendASCII("extension_resource_request_policy") |
| + .AppendASCII("web_accessible")); |
|
Devlin
2016/08/10 20:42:35
nitty nit: leave in an ASSERT_TRUE(extension) so t
Charlie Harrison
2016/08/10 20:45:35
Done. Sorry about that.
|
| GURL accessible_linked_resource(embedded_test_server()->GetURL( |
| "/extensions/api_test/extension_resource_request_policy/" |
| @@ -261,27 +262,36 @@ 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. |
| + GURL accessible_url = extension->GetResourceURL("/test.png"); |
| + ui_test_utils::UrlLoadObserver accessible_observer( |
| + accessible_url, 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( |
| - 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(); |
| + EXPECT_EQ(browser() |
|
Devlin
2016/08/10 20:42:35
nitty nit: expected, actual
so
EXPECT_EQ(accessibl
Charlie Harrison
2016/08/10 20:45:35
Done.
|
| + ->tab_strip_model() |
| + ->GetActiveWebContents() |
| + ->GetLastCommittedURL(), |
| + accessible_url); |
| + |
| + 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( |
| - 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(); |
| + EXPECT_EQ(browser() |
| + ->tab_strip_model() |
| + ->GetActiveWebContents() |
| + ->GetLastCommittedURL(), |
| + GURL("about:blank")); |
| } |
| IN_PROC_BROWSER_TEST_F(ExtensionResourceRequestPolicyTest, |