Chromium Code Reviews| Index: content/browser/site_per_process_browsertest.h |
| diff --git a/content/browser/site_per_process_browsertest.h b/content/browser/site_per_process_browsertest.h |
| index 16db097ed49ecb0bee58f427169f791f31003ef5..9870cd92715eb50fdc0228e2bd257e9e1e29ae87 100644 |
| --- a/content/browser/site_per_process_browsertest.h |
| +++ b/content/browser/site_per_process_browsertest.h |
| @@ -7,7 +7,9 @@ |
| #include <string> |
| +#include "content/browser/web_contents/web_contents_impl.h" |
| #include "content/public/test/content_browser_test.h" |
| +#include "content/shell/browser/shell.h" |
|
EhsanK
2016/03/15 23:51:18
This was necessary for web_contents() to be define
Charlie Reis
2016/03/16 16:55:13
Yep, that's fine.
|
| #include "content/test/content_browser_test_utils_internal.h" |
| #include "url/gurl.h" |
| @@ -15,6 +17,7 @@ namespace content { |
| class FrameTreeNode; |
| class Shell; |
| +class WebContentsImpl; |
|
Charlie Reis
2016/03/16 16:55:13
nit: Don't need these 2 lines anymore.
EhsanK
2016/03/30 20:46:04
Done.
|
| class SitePerProcessBrowserTest : public ContentBrowserTest { |
| public: |
| @@ -26,6 +29,10 @@ class SitePerProcessBrowserTest : public ContentBrowserTest { |
| void SetUpCommandLine(base::CommandLine* command_line) override; |
| void SetUpOnMainThread() override; |
| + WebContentsImpl* web_contents() { |
| + return static_cast<WebContentsImpl*>(shell()->web_contents()); |
| + } |
| + |
| private: |
| FrameTreeVisualizer visualizer_; |
| }; |