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_; |
}; |