Index: content/browser/loader/buffered_resource_handler_unittest.cc |
diff --git a/content/browser/loader/buffered_resource_handler_unittest.cc b/content/browser/loader/buffered_resource_handler_unittest.cc |
index 97b9668c88c5cbfaba4bb1e6427e781733623761..3d9b219628854ec8b8b8a44c231aba42261f63ca 100644 |
--- a/content/browser/loader/buffered_resource_handler_unittest.cc |
+++ b/content/browser/loader/buffered_resource_handler_unittest.cc |
@@ -12,6 +12,7 @@ |
#include "content/public/browser/resource_dispatcher_host_delegate.h" |
#include "content/public/browser/resource_request_info.h" |
#include "content/public/common/resource_response.h" |
+#include "content/public/common/webplugininfo.h" |
#include "content/public/test/test_browser_thread_bundle.h" |
#include "content/public/test/test_utils.h" |
#include "content/test/fake_plugin_service.h" |
@@ -99,6 +100,7 @@ class TestResourceDispatcherHost : public ResourceDispatcherHostImpl { |
} |
scoped_ptr<ResourceHandler> MaybeInterceptAsStream( |
+ const base::FilePath& plugin_path, |
net::URLRequest* request, |
ResourceResponse* response, |
std::string* payload) override { |
@@ -151,14 +153,57 @@ class TestResourceController : public ResourceController { |
} |
}; |
+class TestFakePluginService : public FakePluginService { |
+ public: |
+ TestFakePluginService(bool fake_plugin_available) |
raymes
2015/03/24 03:01:49
nit: plugin_available is fine
jochen (gone - plz use gerrit)
2015/03/24 09:20:54
explicit
Deepak
2015/03/24 10:33:46
Done.
Deepak
2015/03/24 10:33:46
Done.
|
+ : fake_plugin_available_(fake_plugin_available) {} |
+ bool GetPluginInfo(int render_process_id, |
+ int render_frame_id, |
+ ResourceContext* context, |
+ const GURL& url, |
+ const GURL& page_url, |
+ const std::string& mime_type, |
+ bool allow_wildcard, |
+ bool* is_stale, |
+ WebPluginInfo* info, |
+ std::string* actual_mime_type) override; |
+ |
+ private: |
+ bool fake_plugin_available_; |
+}; |
jochen (gone - plz use gerrit)
2015/03/24 09:20:54
DISALLOW_COPY_AND_ASSIGN()
Deepak
2015/03/24 10:33:46
Done.
|
+ |
+bool TestFakePluginService::GetPluginInfo(int render_process_id, |
+ int render_frame_id, |
+ ResourceContext* context, |
+ const GURL& url, |
+ const GURL& page_url, |
+ const std::string& mime_type, |
+ bool allow_wildcard, |
+ bool* is_stale, |
+ WebPluginInfo* info, |
+ std::string* actual_mime_type) { |
+ *is_stale = false; |
+ if (!fake_plugin_available_) |
+ return false; |
+ info->type = WebPluginInfo::PLUGIN_TYPE_BROWSER_PLUGIN; |
+ info->path = base::FilePath( |
+ std::string("chrome-extension://mhjfbmdgcfjbbpaeojofohoefgiehjai/")); |
+ return true; |
+} |
+ |
class BufferedResourceHandlerTest : public testing::Test { |
public: |
- BufferedResourceHandlerTest() : stream_has_handler_(false) {} |
+ BufferedResourceHandlerTest() |
+ : stream_has_handler_(false), plugin_available_(false) {} |
void set_stream_has_handler(bool stream_has_handler) { |
stream_has_handler_ = stream_has_handler; |
} |
+ void set_plugin_available(bool plugin_available) { |
+ plugin_available_ = plugin_available; |
+ } |
+ |
bool TestStreamIsIntercepted(bool allow_download, |
bool must_download, |
ResourceType request_resource_type); |
@@ -166,6 +211,7 @@ class BufferedResourceHandlerTest : public testing::Test { |
private: |
// Whether the URL request should be intercepted as a stream. |
bool stream_has_handler_; |
+ bool plugin_available_; |
TestBrowserThreadBundle thread_bundle_; |
}; |
@@ -194,7 +240,7 @@ bool BufferedResourceHandlerTest::TestStreamIsIntercepted( |
TestResourceDispatcherHostDelegate host_delegate(must_download); |
host.SetDelegate(&host_delegate); |
- FakePluginService plugin_service; |
+ TestFakePluginService plugin_service(plugin_available_); |
scoped_ptr<ResourceHandler> buffered_handler( |
new BufferedResourceHandler( |
scoped_ptr<ResourceHandler>(new TestResourceHandler()).Pass(), |
@@ -227,6 +273,7 @@ TEST_F(BufferedResourceHandlerTest, StreamHandling) { |
// ResourceDispatcherHost. |
set_stream_has_handler(true); |
+ set_plugin_available(true); |
// Main frame request with no download allowed. Stream shouldn't be |
// intercepted. |
allow_download = false; |
@@ -279,6 +326,16 @@ TEST_F(BufferedResourceHandlerTest, StreamHandling) { |
resource_type = RESOURCE_TYPE_MAIN_FRAME; |
EXPECT_FALSE( |
TestStreamIsIntercepted(allow_download, must_download, resource_type)); |
+ |
+ // Test the cases where the stream handled by MaybeInterceptAsStream |
+ // with plugin not available. |
raymes
2015/03/24 03:01:49
nit: add to the comment: "This is the case when in
Deepak
2015/03/24 10:33:46
Done.
|
+ set_stream_has_handler(true); |
+ allow_download = false; |
+ must_download = false; |
+ resource_type = RESOURCE_TYPE_OBJECT; |
+ set_plugin_available(false); |
+ EXPECT_TRUE( |
+ TestStreamIsIntercepted(allow_download, must_download, resource_type)); |
} |
} // namespace |