Index: chromecast/browser/test/chromecast_browser_test_helper_default.cc |
diff --git a/chromecast/browser/test/chromecast_browser_test_helper_default.cc b/chromecast/browser/test/chromecast_browser_test_helper_default.cc |
index f5aad5f660f31a2b745498d92444ba3d8713f411..8a50925444b7a05a8bbc9c06d16b52cbc59aa428 100644 |
--- a/chromecast/browser/test/chromecast_browser_test_helper_default.cc |
+++ b/chromecast/browser/test/chromecast_browser_test_helper_default.cc |
@@ -8,6 +8,7 @@ |
#include "chromecast/browser/cast_browser_context.h" |
#include "chromecast/browser/cast_browser_process.h" |
#include "chromecast/browser/cast_content_window.h" |
+#include "chromecast/browser/cast_media_blocker.h" |
#include "content/public/browser/web_contents.h" |
#include "content/public/test/browser_test_utils.h" |
#include "content/public/test/test_navigation_observer.h" |
@@ -26,6 +27,7 @@ class DefaultHelper : public ChromecastBrowserTestHelper { |
web_contents_ = window_->CreateWebContents( |
CastBrowserProcess::GetInstance()->browser_context()); |
window_->CreateWindowTree(web_contents_.get()); |
+ blocker_.reset(new CastMediaBlocker(web_contents_.get())); |
content::WaitForLoadStop(web_contents_.get()); |
content::TestNavigationObserver same_tab_observer(web_contents_.get(), 1); |
@@ -38,9 +40,14 @@ class DefaultHelper : public ChromecastBrowserTestHelper { |
return web_contents_.get(); |
} |
+ void BlockMediaLoading(bool block) override { |
+ blocker_->BlockMediaLoading(block); |
+ } |
+ |
private: |
std::unique_ptr<CastContentWindow> window_; |
std::unique_ptr<content::WebContents> web_contents_; |
+ std::unique_ptr<CastMediaBlocker> blocker_; |
}; |
} // namespace |