Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(635)

Side by Side Diff: chrome/browser/extensions/extension_unload_browsertest.cc

Issue 2847313002: Update some host_resolver()->AddRules in chrome/browser. (Closed)
Patch Set: Created 3 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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/extension_browsertest.h" 5 #include "chrome/browser/extensions/extension_browsertest.h"
6 #include "chrome/browser/extensions/extension_service.h" 6 #include "chrome/browser/extensions/extension_service.h"
7 #include "chrome/browser/ui/tabs/tab_strip_model.h" 7 #include "chrome/browser/ui/tabs/tab_strip_model.h"
8 #include "chrome/test/base/ui_test_utils.h" 8 #include "chrome/test/base/ui_test_utils.h"
9 #include "content/public/browser/render_frame_host.h" 9 #include "content/public/browser/render_frame_host.h"
10 #include "content/public/test/browser_test_utils.h" 10 #include "content/public/test/browser_test_utils.h"
11 #include "net/dns/mock_host_resolver.h" 11 #include "net/dns/mock_host_resolver.h"
12 #include "net/test/embedded_test_server/embedded_test_server.h" 12 #include "net/test/embedded_test_server/embedded_test_server.h"
13 #include "ui/base/window_open_disposition.h" 13 #include "ui/base/window_open_disposition.h"
14 14
15 namespace extensions { 15 namespace extensions {
16 16
17 using ExtensionUnloadBrowserTest = ExtensionBrowserTest; 17 class ExtensionUnloadBrowserTest : public ExtensionBrowserTest {
18 public:
19 void SetUpOnMainThread() override {
20 ExtensionBrowserTest::SetUpOnMainThread();
21 host_resolver()->AddRule("maps.google.com", "127.0.0.1");
22 }
23 };
18 24
19 IN_PROC_BROWSER_TEST_F(ExtensionUnloadBrowserTest, TestUnload) { 25 IN_PROC_BROWSER_TEST_F(ExtensionUnloadBrowserTest, TestUnload) {
20 // Load an extension that installs unload and beforeunload listeners. 26 // Load an extension that installs unload and beforeunload listeners.
21 const Extension* extension = 27 const Extension* extension =
22 LoadExtension(test_data_dir_.AppendASCII("unload_listener")); 28 LoadExtension(test_data_dir_.AppendASCII("unload_listener"));
23 ASSERT_TRUE(extension); 29 ASSERT_TRUE(extension);
24 std::string id = extension->id(); 30 std::string id = extension->id();
25 ASSERT_EQ(1, browser()->tab_strip_model()->count()); 31 ASSERT_EQ(1, browser()->tab_strip_model()->count());
26 GURL initial_tab_url = 32 GURL initial_tab_url =
27 browser()->tab_strip_model()->GetWebContentsAt(0)->GetLastCommittedURL(); 33 browser()->tab_strip_model()->GetWebContentsAt(0)->GetLastCommittedURL();
28 ui_test_utils::NavigateToURLWithDisposition( 34 ui_test_utils::NavigateToURLWithDisposition(
29 browser(), extension->GetResourceURL("page.html"), 35 browser(), extension->GetResourceURL("page.html"),
30 WindowOpenDisposition::NEW_FOREGROUND_TAB, 36 WindowOpenDisposition::NEW_FOREGROUND_TAB,
31 ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); 37 ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION);
32 EXPECT_EQ(2, browser()->tab_strip_model()->count()); 38 EXPECT_EQ(2, browser()->tab_strip_model()->count());
33 DisableExtension(id); 39 DisableExtension(id);
34 // There should only be one remaining web contents - the initial one. 40 // There should only be one remaining web contents - the initial one.
35 ASSERT_EQ(1, browser()->tab_strip_model()->count()); 41 ASSERT_EQ(1, browser()->tab_strip_model()->count());
36 EXPECT_EQ( 42 EXPECT_EQ(
37 initial_tab_url, 43 initial_tab_url,
38 browser()->tab_strip_model()->GetWebContentsAt(0)->GetLastCommittedURL()); 44 browser()->tab_strip_model()->GetWebContentsAt(0)->GetLastCommittedURL());
39 } 45 }
40 46
41 // After an extension is uninstalled, network requests from its content scripts 47 // After an extension is uninstalled, network requests from its content scripts
42 // should fail but not kill the renderer process. 48 // should fail but not kill the renderer process.
43 IN_PROC_BROWSER_TEST_F(ExtensionUnloadBrowserTest, UnloadWithContentScripts) { 49 IN_PROC_BROWSER_TEST_F(ExtensionUnloadBrowserTest, UnloadWithContentScripts) {
44 host_resolver()->AddRule("maps.google.com", "127.0.0.1");
45 ASSERT_TRUE(embedded_test_server()->Start()); 50 ASSERT_TRUE(embedded_test_server()->Start());
46 51
47 // Load an extension with a content script that has a button to send XHRs. 52 // Load an extension with a content script that has a button to send XHRs.
48 const Extension* extension = 53 const Extension* extension =
49 LoadExtension(test_data_dir_.AppendASCII("xhr_from_content_script")); 54 LoadExtension(test_data_dir_.AppendASCII("xhr_from_content_script"));
50 ASSERT_TRUE(extension); 55 ASSERT_TRUE(extension);
51 std::string id = extension->id(); 56 std::string id = extension->id();
52 ASSERT_EQ(1, browser()->tab_strip_model()->count()); 57 ASSERT_EQ(1, browser()->tab_strip_model()->count());
53 GURL test_url = embedded_test_server()->GetURL("/title1.html"); 58 GURL test_url = embedded_test_server()->GetURL("/title1.html");
54 ui_test_utils::NavigateToURL(browser(), test_url); 59 ui_test_utils::NavigateToURL(browser(), test_url);
(...skipping 26 matching lines...) Expand all
81 EXPECT_TRUE(browser() 86 EXPECT_TRUE(browser()
82 ->tab_strip_model() 87 ->tab_strip_model()
83 ->GetActiveWebContents() 88 ->GetActiveWebContents()
84 ->GetMainFrame() 89 ->GetMainFrame()
85 ->IsRenderFrameLive()); 90 ->IsRenderFrameLive());
86 } 91 }
87 92
88 // TODO(devlin): Investigate what to do for embedded iframes. 93 // TODO(devlin): Investigate what to do for embedded iframes.
89 94
90 } // namespace extensions 95 } // namespace extensions
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698