[Offline Pages] Make prerenderer notify us of network progress.

chrome/browser/android/offline_pages/prerender_adapter_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 "chrome/browser/android/offline_pages/prerender_adapter.h"
 
 #include "base/sys_info.h"
 #include "chrome/browser/prerender/prerender_manager_factory.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/browser/browser_thread.h"
@@ skipping 95 matching lines @@
 class PrerenderAdapterTest : public testing::Test,
                              public PrerenderAdapter::Observer {
  public:
   PrerenderAdapterTest();
   ~PrerenderAdapterTest() override;
 
   // PrerenderAdapter::Observer implementation:
   void OnPrerenderStopLoading() override;
   void OnPrerenderDomContentLoaded() override;
   void OnPrerenderStop() override;
+  void OnPrerenderNetworkBytesChanged(int64_t bytes) override;
 
   void SetUp() override;
 
   // Returns the PrerenderLoader to test.
   PrerenderAdapter* adapter() const { return adapter_.get(); }
   StubPrerenderContentsFactory* prerender_contents_factory() {
     return prerender_contents_factory_;
   }
   Profile* profile() { return &profile_; }
   PrerenderManager* prerender_manager() { return prerender_manager_; }
   bool observer_stop_loading_called() const {
     return observer_stop_loading_called_;
   }
   bool observer_dom_content_loaded_called() const {
     return observer_dom_content_loaded_called_;
   }
   bool observer_stop_called() const { return observer_stop_called_; }
+  int64_t observer_network_bytes_changed() const {
+    return observer_network_bytes_changed_;
+  }
 
  private:
   content::TestBrowserThreadBundle thread_bundle_;
   TestingProfile profile_;
   std::unique_ptr<PrerenderAdapter> adapter_;
   StubPrerenderContentsFactory* prerender_contents_factory_;
   PrerenderManager* prerender_manager_;
   bool observer_stop_loading_called_;
   bool observer_dom_content_loaded_called_;
   bool observer_stop_called_;
+  int64_t observer_network_bytes_changed_;
 
   DISALLOW_COPY_AND_ASSIGN(PrerenderAdapterTest);
 };
 
 PrerenderAdapterTest::PrerenderAdapterTest()
     : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP),
       prerender_manager_(nullptr),
       observer_stop_loading_called_(false),
       observer_dom_content_loaded_called_(false),
-      observer_stop_called_(false) {}
+      observer_stop_called_(false),
+      observer_network_bytes_changed_(0) {}
 
 PrerenderAdapterTest::~PrerenderAdapterTest() {
   if (prerender_manager_)
     prerender_manager_->Shutdown();
 }
 
 void PrerenderAdapterTest::OnPrerenderStopLoading() {
   observer_stop_loading_called_ = true;
 }
 
 void PrerenderAdapterTest::OnPrerenderDomContentLoaded() {
   observer_dom_content_loaded_called_ = true;
 }
 
 void PrerenderAdapterTest::OnPrerenderStop() {
   observer_stop_called_ = true;
 }
 
+void PrerenderAdapterTest::OnPrerenderNetworkBytesChanged(int64_t bytes) {
+  observer_network_bytes_changed_ = bytes;
+}
+
 void PrerenderAdapterTest::SetUp() {
   if (base::SysInfo::IsLowEndDevice())
     return;
   adapter_.reset(new PrerenderAdapter(this));
   prerender_contents_factory_ = new StubPrerenderContentsFactory();
   prerender_manager_ = PrerenderManagerFactory::GetForBrowserContext(profile());
   if (prerender_manager_) {
     prerender_manager_->SetPrerenderContentsFactoryForTest(
         prerender_contents_factory_);
     prerender_manager_->SetMode(PrerenderManager::PRERENDER_MODE_ENABLED);
@@ skipping 41 matching lines @@
   EXPECT_TRUE(adapter()->IsActive());
   EXPECT_FALSE(observer_stop_loading_called());
   EXPECT_FALSE(observer_dom_content_loaded_called());
   EXPECT_FALSE(observer_stop_called());
 
   // Exercise observer event call paths.
   prerender_contents_factory()
       ->last_prerender_contents()
       ->ReportDomContentEvent();
   EXPECT_TRUE(observer_dom_content_loaded_called());
+
+  // Expect byte count reported to Observer.
+  prerender_contents_factory()->last_prerender_contents()->AddNetworkBytes(153);
+  EXPECT_EQ(153LL, observer_network_bytes_changed());
+
   prerender_contents_factory()->last_prerender_contents()->ReportOnLoadEvent();
   EXPECT_TRUE(observer_stop_loading_called());
   prerender_contents_factory()->last_prerender_contents()->StopWithStatus(
       FinalStatus::FINAL_STATUS_CANCELLED);
   EXPECT_TRUE(observer_stop_called());
   EXPECT_EQ(FinalStatus::FINAL_STATUS_CANCELLED, adapter()->GetFinalStatus());
 
   // Exercise access methods even though no interesting values set beneath.
   EXPECT_EQ(nullptr, adapter()->GetWebContents());
 
   adapter()->DestroyActive();
   EXPECT_FALSE(adapter()->IsActive());
 }
 
 }  // namespace offline_pages