OLD | NEW |
1 // Copyright (c) 2014 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2014 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 #ifndef CONTENT_BROWSER_SITE_PER_PROCESS_BROWSERTEST_H_ | 5 #ifndef CONTENT_BROWSER_SITE_PER_PROCESS_BROWSERTEST_H_ |
6 #define CONTENT_BROWSER_SITE_PER_PROCESS_BROWSERTEST_H_ | 6 #define CONTENT_BROWSER_SITE_PER_PROCESS_BROWSERTEST_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 | 9 |
| 10 #include "content/browser/web_contents/web_contents_impl.h" |
10 #include "content/public/test/content_browser_test.h" | 11 #include "content/public/test/content_browser_test.h" |
| 12 #include "content/shell/browser/shell.h" |
11 #include "content/test/content_browser_test_utils_internal.h" | 13 #include "content/test/content_browser_test_utils_internal.h" |
12 #include "url/gurl.h" | 14 #include "url/gurl.h" |
13 | 15 |
14 namespace content { | 16 namespace content { |
15 | 17 |
16 class FrameTreeNode; | 18 class FrameTreeNode; |
17 class Shell; | |
18 | 19 |
19 class SitePerProcessBrowserTest : public ContentBrowserTest { | 20 class SitePerProcessBrowserTest : public ContentBrowserTest { |
20 public: | 21 public: |
21 SitePerProcessBrowserTest(); | 22 SitePerProcessBrowserTest(); |
22 | 23 |
23 protected: | 24 protected: |
24 std::string DepictFrameTree(FrameTreeNode* node); | 25 std::string DepictFrameTree(FrameTreeNode* node); |
25 | 26 |
26 void SetUpCommandLine(base::CommandLine* command_line) override; | 27 void SetUpCommandLine(base::CommandLine* command_line) override; |
27 void SetUpOnMainThread() override; | 28 void SetUpOnMainThread() override; |
28 | 29 |
| 30 WebContentsImpl* web_contents() { |
| 31 return static_cast<WebContentsImpl*>(shell()->web_contents()); |
| 32 } |
| 33 |
29 private: | 34 private: |
30 FrameTreeVisualizer visualizer_; | 35 FrameTreeVisualizer visualizer_; |
31 }; | 36 }; |
32 | 37 |
33 } // namespace content | 38 } // namespace content |
34 | 39 |
35 #endif // CONTENT_BROWSER_SITE_PER_PROCESS_BROWSERTEST_H_ | 40 #endif // CONTENT_BROWSER_SITE_PER_PROCESS_BROWSERTEST_H_ |
OLD | NEW |