[Net] Propagate priority changes from URLRequest to HttpTransaction

net/url_request/url_request_unittest.cc

 // Copyright (c) 2012 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 "build/build_config.h"
 
 #if defined(OS_WIN)
 #include <windows.h>
 #include <shlobj.h>
 #endif
@@ skipping 1386 matching lines @@
 // content is empty.
 TEST_F(URLRequestTest, RequestCompletionForEmptyResponse) {
   TestDelegate d;
   URLRequest req(GURL("data:,"), &d, &default_context_);
   req.Start();
   MessageLoop::current()->Run();
   EXPECT_EQ("", d.data_received());
   EXPECT_EQ(1, default_network_delegate_.completed_requests());
 }
 
+// Make sure that URLRequest passes on its priority updates to its
+// job.
+TEST_F(URLRequestTest, Priority) {
+  TestDelegate d;
+  URLRequest req(GURL("http://test_intercept/foo"), &d, &default_context_);
+  EXPECT_EQ(LOWEST, req.priority());
+
+  scoped_refptr<URLRequestTestJob> job =
+      new URLRequestTestJob(&req, &default_network_delegate_);
+  AddTestInterceptor()->set_main_intercept_job(job);
+  EXPECT_EQ(IDLE, job->priority());
+
+  req.SetPriority(LOW);
+  EXPECT_EQ(LOW, req.priority());
+  // |req| doesn't know about |job| yet.
+  EXPECT_EQ(IDLE, job->priority());
+
+  req.Start();
+  // |req| should set |job|'s initial priority.
+  EXPECT_EQ(LOW, job->priority());
+
+  req.SetPriority(HIGHEST);
+  EXPECT_EQ(HIGHEST, req.priority());
+  EXPECT_EQ(HIGHEST, job->priority());
+}
+
 // TODO(droger): Support TestServer on iOS (see http://crbug.com/148666).
 #if !defined(OS_IOS)
 // A subclass of TestServer that uses a statically-configured hostname. This is
 // to work around mysterious failures in chrome_frame_net_tests. See:
 // http://crbug.com/114369
 class LocalHttpTestServer : public TestServer {
  public:
   explicit LocalHttpTestServer(const base::FilePath& document_root)
       : TestServer(TestServer::TYPE_HTTP,
                    ScopedCustomUrlRequestTestHttpHost::value(),
@@ skipping 2706 matching lines @@
   for (size_t i = 0; i < ARRAYSIZE_UNSAFE(tests); i++) {
     TestDelegate d;
     URLRequest req(test_server_.GetURL(tests[i].request), &d, &context);
     req.Start();
     MessageLoop::current()->Run();
     EXPECT_EQ(tests[i].expected_response, d.data_received())
         << " Request = \"" << tests[i].request << "\"";
   }
 }
 
+// Make sure that URLRequest passes on its priority updates to its job
+// even on redirect.
+TEST_F(URLRequestTestHTTP, PriorityRedirect) {
+  ASSERT_TRUE(test_server_.Start());
+
+  TestDelegate d;
+  URLRequest req(test_server_.GetURL("empty.html"), &d, &default_context_);
+  EXPECT_EQ(LOWEST, req.priority());
+
+  scoped_refptr<URLRequestRedirectJob> redirect_job =
+      new URLRequestRedirectJob(
+          &req, &default_network_delegate_, test_server_.GetURL("echo"),
+          URLRequestRedirectJob::REDIRECT_302_FOUND);
+  scoped_refptr<URLRequestTestJob> job =
+      new URLRequestTestJob(&req, &default_network_delegate_);
+
+  AddTestInterceptor()->set_main_intercept_job(redirect_job);
+
+  EXPECT_EQ(IDLE, job->priority());
+
+  req.SetPriority(LOW);
+  EXPECT_EQ(LOW, req.priority());
+  EXPECT_EQ(IDLE, job->priority());
+
+  req.Start();
+  EXPECT_TRUE(req.is_pending());
+  EXPECT_EQ(IDLE, job->priority());
+
+  AddTestInterceptor()->set_main_intercept_job(job);
+  EXPECT_EQ(IDLE, job->priority());
+
+  MessageLoop::current()->Run();
+  EXPECT_EQ(LOW, job->priority());
+
+  req.SetPriority(HIGHEST);
+  EXPECT_EQ(HIGHEST, req.priority());
+  EXPECT_EQ(HIGHEST, job->priority());
+}
+
+
 class HTTPSRequestTest : public testing::Test {
  public:
   HTTPSRequestTest() : default_context_(true) {
     default_context_.set_network_delegate(&default_network_delegate_);
     default_context_.Init();
   }
   virtual ~HTTPSRequestTest() {}
 
  protected:
   TestNetworkDelegate default_network_delegate_;  // Must outlive URLRequest.
@@ skipping 1224 matching lines @@
 
     EXPECT_FALSE(r.is_pending());
     EXPECT_EQ(1, d->response_started_count());
     EXPECT_FALSE(d->received_data_before_response());
     EXPECT_EQ(d->bytes_received(), static_cast<int>(file_size));
   }
 }
 #endif  // !defined(DISABLE_FTP_SUPPORT)
 
 }  // namespace net