Add UploadProgressTracker unittest

content/browser/loader/async_resource_handler.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 "content/browser/loader/async_resource_handler.h"
 
 #include <algorithm>
 #include <vector>
 
 #include "base/command_line.h"
@@ skipping 319 matching lines @@
 }
 
 bool AsyncResourceHandler::OnWillStart(const GURL& url, bool* defer) {
   ResourceMessageFilter* filter = GetFilter();
   if (!filter)
     return false;
 
   if (GetRequestInfo()->is_upload_progress_enabled() &&
       request()->has_upload()) {
     upload_progress_tracker_ = base::MakeUnique<UploadProgressTracker>(
-        FROM_HERE,
-        base::BindRepeating(&AsyncResourceHandler::SendUploadProgress,
-                            base::Unretained(this)),
-        request());
+        FROM_HERE, this, base::ThreadTaskRunnerHandle::Get());
   }
   return true;
 }
 
 bool AsyncResourceHandler::OnWillRead(scoped_refptr<net::IOBuffer>* buf,
                                       int* buf_size,
                                       int min_size) {
   DCHECK_EQ(-1, min_size);
 
   if (!CheckForSufficientResource())
@@ skipping 123 matching lines @@
   request_complete_data.encoded_data_length =
       request()->GetTotalReceivedBytes();
   request_complete_data.encoded_body_length = request()->GetRawBodyBytes();
   filter->Send(
       new ResourceMsg_RequestComplete(GetRequestID(), request_complete_data));
 
   if (status.is_success())
     RecordHistogram();
 }
 
+net::UploadProgress AsyncResourceHandler::GetUploadProgress() {
+  return request()->GetUploadProgress();
+}
+
+void AsyncResourceHandler::ReportUploadProgress(int64_t current_position,
+                                                int64_t total_size) {
+  ResourceMessageFilter* filter = GetFilter();
+  if (!filter)
+    return;
+  filter->Send(new ResourceMsg_UploadProgress(GetRequestID(), current_position,
+                                              total_size));
+}
+
 bool AsyncResourceHandler::EnsureResourceBufferIsInitialized() {
   DCHECK(has_checked_for_sufficient_resources_);
 
   if (buffer_.get() && buffer_->IsInitialized())
     return true;
 
   buffer_ = new ResourceBuffer();
   return buffer_->Initialize(kBufferSize,
                              kMinAllocationSize,
                              kMaxAllocationSize);
@@ skipping 49 matching lines @@
         elapsed_time, 1, 100000, 100);
   } else {
     UMA_HISTOGRAM_CUSTOM_COUNTS(
         "Net.ResourceLoader.ResponseStartToEnd.Over_512kB",
         elapsed_time, 1, 100000, 100);
   }
 
   inlining_helper_->RecordHistogram(elapsed_time);
 }
 
-void AsyncResourceHandler::SendUploadProgress(int64_t current_position,
-                                              int64_t total_size) {
-  ResourceMessageFilter* filter = GetFilter();
-  if (!filter)
-    return;
-  filter->Send(new ResourceMsg_UploadProgress(
-      GetRequestID(), current_position, total_size));
-}
-
 }  // namespace content