[Net] Propagate priority changes from URLRequest to HttpTransaction

net/http/http_transaction_unittest.h

 // 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.
 
 #ifndef NET_HTTP_HTTP_TRANSACTION_UNITTEST_H_
 #define NET_HTTP_HTTP_TRANSACTION_UNITTEST_H_
 
 #include "net/http/http_transaction.h"
 
 #include <string>
 
 #include "base/callback.h"
 #include "base/compiler_specific.h"
+#include "base/memory/weak_ptr.h"
 #include "base/string16.h"
 #include "net/base/io_buffer.h"
 #include "net/base/load_flags.h"
 #include "net/base/net_errors.h"
 #include "net/base/request_priority.h"
 #include "net/base/test_completion_callback.h"
 #include "net/disk_cache/disk_cache.h"
 #include "net/http/http_cache.h"
 #include "net/http/http_request_info.h"
 #include "net/http/http_response_headers.h"
@@ skipping 122 matching lines @@
 
 //-----------------------------------------------------------------------------
 // mock network layer
 
 class MockNetworkLayer;
 
 // This transaction class inspects the available set of mock transactions to
 // find data for the request URL.  It supports IO operations that complete
 // synchronously or asynchronously to help exercise different code paths in the
 // HttpCache implementation.
-class MockNetworkTransaction : public net::HttpTransaction {
+class MockNetworkTransaction
+    : public net::HttpTransaction,
+      public base::SupportsWeakPtr<MockNetworkTransaction> {
  public:
   MockNetworkTransaction(net::RequestPriority priority,
                          MockNetworkLayer* factory);
   virtual ~MockNetworkTransaction();
 
   virtual int Start(const net::HttpRequestInfo* request,
                     const net::CompletionCallback& callback,
                     const net::BoundNetLog& net_log) OVERRIDE;
 
   virtual int RestartIgnoringLastError(
@@ skipping 18 matching lines @@
 
   virtual const net::HttpResponseInfo* GetResponseInfo() const OVERRIDE;
 
   virtual net::LoadState GetLoadState() const OVERRIDE;
 
   virtual net::UploadProgress GetUploadProgress() const OVERRIDE;
 
   virtual bool GetLoadTimingInfo(
       net::LoadTimingInfo* load_timing_info) const OVERRIDE;
 
+  virtual void SetPriority(net::RequestPriority priority) OVERRIDE;
+
+  net::RequestPriority priority() const { return priority_; }
+
  private:
   void CallbackLater(const net::CompletionCallback& callback, int result);
   void RunCallback(const net::CompletionCallback& callback, int result);
 
   base::WeakPtrFactory<MockNetworkTransaction> weak_factory_;
   net::HttpResponseInfo response_;
   std::string data_;
   int data_cursor_;
   int test_mode_;
+  net::RequestPriority priority_;
   base::WeakPtr<MockNetworkLayer> transaction_factory_;
 };
 
 class MockNetworkLayer : public net::HttpTransactionFactory,
                          public base::SupportsWeakPtr<MockNetworkLayer> {
  public:
   MockNetworkLayer();
   virtual ~MockNetworkLayer();
 
   int transaction_count() const { return transaction_count_; }
   bool done_reading_called() const { return done_reading_called_; }
   void TransactionDoneReading();
 
+  // Returns the last priority passed to CreateTransaction, or
+  // DEFAULT_PRIORITY if it hasn't been called yet.
+  net::RequestPriority last_create_transaction_priority() const {
+    return last_create_transaction_priority_;
+  }
+
+  // Returns the last transaction created by
+  // CreateTransaction. Returns a NULL WeakPtr if one has not been
+  // created yet, or the last transaction has been destroyed, or
+  // ClearLastTransaction() has been called and a new transaction
+  // hasn't been created yet.
+  base::WeakPtr<MockNetworkTransaction> last_transaction() {
+    return last_transaction_;
+  }
+
+  // Makes last_transaction() return NULL until the next transaction
+  // is created.
+  void ClearLastTransaction() {
+    last_transaction_.reset();
+  }
+
   // net::HttpTransactionFactory:
   virtual int CreateTransaction(
       net::RequestPriority priority,
       scoped_ptr<net::HttpTransaction>* trans,
       net::HttpTransactionDelegate* delegate) OVERRIDE;
   virtual net::HttpCache* GetCache() OVERRIDE;
   virtual net::HttpNetworkSession* GetSession() OVERRIDE;
 
  private:
   int transaction_count_;
   bool done_reading_called_;
+  net::RequestPriority last_create_transaction_priority_;
+  base::WeakPtr<MockNetworkTransaction> last_transaction_;
 };
 
 //-----------------------------------------------------------------------------
 // helpers
 
 // read the transaction completely
 int ReadTransaction(net::HttpTransaction* trans, std::string* result);
 
 #endif  // NET_HTTP_HTTP_TRANSACTION_UNITTEST_H_