[Mojo-Loading] Implement URLLoader::SetPriority

content/browser/loader/mojo_async_resource_handler_unittest.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/browser/loader/mojo_async_resource_handler.h"
 
 #include <string.h>
 
 #include <utility>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/files/file_path.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/memory/weak_ptr.h"
 #include "base/run_loop.h"
 #include "base/test/test_simple_task_runner.h"
 #include "content/browser/loader/mock_resource_loader.h"
 #include "content/browser/loader/resource_controller.h"
 #include "content/browser/loader/resource_dispatcher_host_impl.h"
 #include "content/browser/loader/resource_request_info_impl.h"
+#include "content/browser/loader/resource_scheduler.h"
 #include "content/browser/loader/test_url_loader_client.h"
 #include "content/common/resource_request_completion_status.h"
 #include "content/common/url_loader.mojom.h"
 #include "content/public/browser/appcache_service.h"
 #include "content/public/browser/navigation_data.h"
 #include "content/public/browser/resource_context.h"
 #include "content/public/browser/resource_dispatcher_host_delegate.h"
 #include "content/public/browser/resource_throttle.h"
 #include "content/public/browser/stream_info.h"
 #include "content/public/common/previews_state.h"
@@ skipping 312 matching lines @@
     // just need something non-NULL.
     net::URLRequestContext* request_context =
         browser_context_->GetResourceContext()->GetRequestContext();
     request_ = request_context->CreateRequest(
         GURL("http://foo/"), net::DEFAULT_PRIORITY, &url_request_delegate_);
     request_->set_upload(std::move(upload_stream));
     ResourceRequestInfo::AllocateForTesting(
         request_.get(),                          // request
         RESOURCE_TYPE_XHR,                       // resource_type
         browser_context_->GetResourceContext(),  // context
-        2,                                       // render_process_id
-        0,                                       // render_view_id
+        kChildId,                                // render_process_id
+        kRouteId,                                // render_view_id
         0,                                       // render_frame_id
         true,                                    // is_main_frame
         false,                                   // parent_is_main_frame
         false,                                   // allow_download
         true,                                    // is_async
         PREVIEWS_OFF                             // previews_state
         );
 
     ResourceRequest request;
     base::WeakPtr<mojo::StrongBinding<mojom::URLLoaderFactory>> weak_binding =
         mojo::MakeStrongBinding(base::MakeUnique<TestURLLoaderFactory>(),
                                 mojo::MakeRequest(&url_loader_factory_));
 
     url_loader_factory_->CreateLoaderAndStart(
         mojo::MakeRequest(&url_loader_proxy_,
                           url_loader_factory_.associated_group()),
-        0, 0, request, url_loader_client_.CreateRemoteAssociatedPtrInfo(
-                           url_loader_factory_.associated_group()));
+        kRouteId, kRequestId, request,
+        url_loader_client_.CreateRemoteAssociatedPtrInfo(
+            url_loader_factory_.associated_group()));
 
     url_loader_factory_.FlushForTesting();
     DCHECK(weak_binding);
     TestURLLoaderFactory* factory_impl =
         static_cast<TestURLLoaderFactory*>(weak_binding->impl());
 
     mojom::URLLoaderClientAssociatedPtr client_ptr;
     client_ptr.Bind(factory_impl->PassClientPtrInfo());
     handler_.reset(new MojoAsyncResourceHandlerWithStubOperations(
         request_.get(), &rdh_, factory_impl->PassLoaderRequest(),
@@ skipping 49 matching lines @@
   ResourceDispatcherHostImpl rdh_;
   mojom::URLLoaderFactoryPtr url_loader_factory_;
   mojom::URLLoaderAssociatedPtr url_loader_proxy_;
   TestURLLoaderClient url_loader_client_;
   std::unique_ptr<TestBrowserContext> browser_context_;
   net::TestDelegate url_request_delegate_;
   std::unique_ptr<net::URLRequest> request_;
   std::unique_ptr<MojoAsyncResourceHandlerWithStubOperations> handler_;
   std::unique_ptr<MockResourceLoader> mock_loader_;
 
+  static constexpr int kChildId = 25;
+  static constexpr int kRouteId = 12;
+  static constexpr int kRequestId = 41;
+
   DISALLOW_COPY_AND_ASSIGN(MojoAsyncResourceHandlerTestBase);
 };
 
 class MojoAsyncResourceHandlerTest : public MojoAsyncResourceHandlerTestBase,
                                      public ::testing::Test {
  protected:
   MojoAsyncResourceHandlerTest() : MojoAsyncResourceHandlerTestBase(nullptr) {}
 };
 
 // This test class is parameterized with MojoAsyncResourceHandler's allocation
@@ skipping 482 matching lines @@
 
   // Expect a upload progress report for the upload completion.
   url_loader_client_.reset_has_received_upload_progress();
   set_upload_progress(net::UploadProgress(1000, 1000));
   ASSERT_TRUE(CallOnResponseStarted());
   EXPECT_TRUE(url_loader_client_.has_received_upload_progress());
   EXPECT_EQ(1000, url_loader_client_.current_upload_position());
   EXPECT_EQ(1000, url_loader_client_.total_upload_size());
 }
 
+TEST_F(MojoAsyncResourceHandlerTest, SetPriority) {
+  constexpr int kIntraPriority = 5;
+  ASSERT_TRUE(CallOnWillStartAndOnResponseStarted());
+  std::unique_ptr<ResourceThrottle> throttle =
+      ResourceDispatcherHostImpl::Get()->scheduler()->ScheduleRequest(
+          kChildId, kRouteId, false, request_.get());
+
+  EXPECT_EQ(net::LOWEST, request_->priority());
+
+  handler_->SetPriority(net::RequestPriority::HIGHEST, kIntraPriority);
+
+  EXPECT_EQ(net::HIGHEST, request_->priority());
+}
+
 TEST_P(MojoAsyncResourceHandlerWithAllocationSizeTest,
        OnWillReadWithLongContents) {
   ASSERT_TRUE(CallOnWillStartAndOnResponseStarted());
   ASSERT_EQ(MockResourceLoader::Status::IDLE, mock_loader_->OnWillRead(-1));
   std::string expected;
   for (int i = 0; i < 3 * mock_loader_->io_buffer_size() + 2; ++i)
     expected += ('A' + i % 26);
 
   ASSERT_EQ(MockResourceLoader::Status::IDLE, mock_loader_->OnReadCompleted(0));
 
@@ skipping 304 matching lines @@
     }
   }
   EXPECT_EQ("B", body);
 }
 
 INSTANTIATE_TEST_CASE_P(MojoAsyncResourceHandlerWithAllocationSizeTest,
                         MojoAsyncResourceHandlerWithAllocationSizeTest,
                         ::testing::Values(8, 32 * 2014));
 }  // namespace
 }  // namespace content