Chromium Code Reviews| Index: content/browser/loader/resource_loader_unittest.cc |
| diff --git a/content/browser/loader/resource_loader_unittest.cc b/content/browser/loader/resource_loader_unittest.cc |
| index 92241b68d0beab40f4d03c6512289017c9578e5b..757fe414c94b964222e759178512ebf50c6297ee 100644 |
| --- a/content/browser/loader/resource_loader_unittest.cc |
| +++ b/content/browser/loader/resource_loader_unittest.cc |
| @@ -11,11 +11,15 @@ |
| #include "content/browser/browser_thread_impl.h" |
| #include "content/browser/loader/redirect_to_file_resource_handler.h" |
| #include "content/browser/loader/resource_loader_delegate.h" |
| +#include "content/public/browser/client_certificate_delegate.h" |
| #include "content/public/browser/resource_request_info.h" |
| #include "content/public/common/resource_response.h" |
| #include "content/public/test/mock_resource_context.h" |
| +#include "content/public/test/test_browser_context.h" |
| #include "content/public/test/test_browser_thread_bundle.h" |
| +#include "content/public/test/test_renderer_host.h" |
| #include "content/test/test_content_browser_client.h" |
| +#include "content/test/test_web_contents.h" |
| #include "ipc/ipc_message.h" |
| #include "net/base/io_buffer.h" |
| #include "net/base/mock_file_stream.h" |
| @@ -239,10 +243,9 @@ class SelectCertificateBrowserClient : public TestContentBrowserClient { |
| SelectCertificateBrowserClient() : call_count_(0) {} |
| void SelectClientCertificate( |
| - int render_process_id, |
| - int render_view_id, |
| + WebContents* web_contents, |
| net::SSLCertRequestInfo* cert_request_info, |
| - const base::Callback<void(net::X509Certificate*)>& callback) override { |
| + scoped_ptr<ClientCertificateDelegate> delegate) override { |
| ++call_count_; |
| passed_certs_ = cert_request_info->client_certs; |
| } |
| @@ -317,8 +320,12 @@ class ResourceLoaderTest : public testing::Test, |
| } |
| void SetUp() override { |
| - const int kRenderProcessId = 1; |
| - const int kRenderViewId = 2; |
| + browser_context_.reset(new TestBrowserContext()); |
| + scoped_refptr<SiteInstance> site_instance = |
| + SiteInstance::Create(browser_context_.get()); |
| + web_contents_.reset( |
| + TestWebContents::Create(browser_context_.get(), site_instance.get())); |
|
mmenke
2015/02/10 17:19:10
I'm confused. Why are these needed? Aren't we ju
davidben
2015/02/10 20:28:49
There's a handful of tests that install a custom o
mmenke
2015/02/11 17:09:16
Ah, somehow I was missing SelectCertificateBrowser
davidben
2015/02/12 22:55:15
Done.
|
| + RenderFrameHost* rfh = web_contents_->GetMainFrame(); |
| scoped_ptr<net::URLRequest> request( |
| resource_context_.GetRequestContext()->CreateRequest( |
| @@ -330,9 +337,9 @@ class ResourceLoaderTest : public testing::Test, |
| ResourceRequestInfo::AllocateForTesting(request.get(), |
| RESOURCE_TYPE_MAIN_FRAME, |
| &resource_context_, |
| - kRenderProcessId, |
| - kRenderViewId, |
| + rfh->GetProcess()->GetID(), |
| MSG_ROUTING_NONE, |
| + rfh->GetRoutingID(), |
|
mmenke
2015/02/10 17:19:10
Think you messed up the order here
davidben
2015/02/10 20:28:49
I'm passing in RFH ID, and not RVH ID. Getting the
|
| true, // is_main_frame |
| false, // parent_is_main_frame |
| true, // allow_download |
| @@ -346,6 +353,11 @@ class ResourceLoaderTest : public testing::Test, |
| this)); |
| } |
| + void TearDown() override { |
| + web_contents_.reset(); |
| + base::RunLoop().RunUntilIdle(); |
| + } |
| + |
| // ResourceLoaderDelegate: |
| ResourceDispatcherHostLoginDelegate* CreateLoginDelegate( |
| ResourceLoader* loader, |
| @@ -362,11 +374,14 @@ class ResourceLoaderTest : public testing::Test, |
| void DidReceiveResponse(ResourceLoader* loader) override {} |
| void DidFinishLoading(ResourceLoader* loader) override {} |
| - content::TestBrowserThreadBundle thread_bundle_; |
| + TestBrowserThreadBundle thread_bundle_; |
| + RenderViewHostTestEnabler rvh_test_enabler_; |
| net::URLRequestJobFactoryImpl job_factory_; |
| net::TestURLRequestContext test_url_request_context_; |
| ResourceContextStub resource_context_; |
| + scoped_ptr<TestBrowserContext> browser_context_; |
| + scoped_ptr<TestWebContents> web_contents_; |
| // The ResourceLoader owns the URLRequest and the ResourceHandler. |
| ResourceHandlerStub* raw_ptr_resource_handler_; |