Add "SHOW ALL SAVED PAGES" button to offline error page

components/error_page/renderer/net_error_helper_core_unittest.cc

Index: components/error_page/renderer/net_error_helper_core_unittest.cc
diff --git a/components/error_page/renderer/net_error_helper_core_unittest.cc b/components/error_page/renderer/net_error_helper_core_unittest.cc
index 6fa8caf15c7a2a94aa79bbce61e4ddf420a67b55..da56f36f709505d3f711ce702a406be9996c9194 100644
--- a/components/error_page/renderer/net_error_helper_core_unittest.cc
+++ b/components/error_page/renderer/net_error_helper_core_unittest.cc
@@ -156,8 +156,9 @@ class NetErrorHelperCoreTest : public testing::Test,
                              update_count_(0),
                              error_html_update_count_(0),
                              reload_count_(0),
-                             show_saved_count_(0),
+                             show_saved_copy_count_(0),
                              diagnose_error_count_(0),
+                             show_saved_pages_count_(0),
                              enable_page_helper_functions_count_(0),
                              default_url_(GURL(kFailedUrl)),
                              error_url_(GURL(content::kUnreachableWebDataURL)),
@@ -195,12 +196,12 @@ class NetErrorHelperCoreTest : public testing::Test,
     return reload_count_;
   }
 
-  int show_saved_count() const {
-    return show_saved_count_;
+  int show_saved_copy_count() const {
+    return show_saved_copy_count_;
   }
 
-  const GURL& show_saved_url() const {
-    return show_saved_url_;
+  const GURL& show_saved_copy_url() const {
+    return show_saved_copy_url_;
   }
 
   int diagnose_error_count() const {
@@ -211,6 +212,10 @@ class NetErrorHelperCoreTest : public testing::Test,
     return diagnose_error_url_;
   }
 
+  int show_saved_pages_count() const {
+    return show_saved_pages_count_;
+  }
+
   const GURL& default_url() const {
     return default_url_;
   }
@@ -233,6 +238,10 @@ class NetErrorHelperCoreTest : public testing::Test,
     return last_can_show_network_diagnostics_dialog_;
   }
 
+  bool last_has_offline_pages() const {
+    return last_has_offline_pages_;
+  }
+
   const ErrorPageParams* last_error_page_params() const {
     return last_error_page_params_.get();
   }
@@ -320,17 +329,21 @@ class NetErrorHelperCoreTest : public testing::Test,
   void GenerateLocalizedErrorPage(const WebURLError& error,
                                   bool is_failed_post,
                                   bool can_show_network_diagnostics_dialog,
+                                  bool has_offline_pages,
                                   scoped_ptr<ErrorPageParams> params,
                                   bool* reload_button_shown,
                                   bool* show_saved_copy_button_shown,
                                   bool* show_cached_copy_button_shown,
+                                  bool* show_saved_pages_button_shown,
                                   std::string* html) const override {
     last_can_show_network_diagnostics_dialog_ =
         can_show_network_diagnostics_dialog;
+    last_has_offline_pages_ = has_offline_pages;
     last_error_page_params_.reset(params.release());
     *reload_button_shown = false;
     *show_saved_copy_button_shown = false;
     *show_cached_copy_button_shown = false;
+    *show_saved_pages_button_shown = false;
     *html = ErrorToString(error, is_failed_post);
   }
 
@@ -344,10 +357,12 @@ class NetErrorHelperCoreTest : public testing::Test,
   }
 
   void UpdateErrorPage(const WebURLError& error, bool is_failed_post,
-                       bool can_show_network_diagnostics_dialog) override {
+                       bool can_show_network_diagnostics_dialog,
+                       bool has_offline_pages) override {
     update_count_++;
     last_can_show_network_diagnostics_dialog_ =
         can_show_network_diagnostics_dialog;
+    last_has_offline_pages_ = has_offline_pages;
     last_error_page_params_.reset(nullptr);
     last_error_html_ = ErrorToString(error, is_failed_post);
   }
@@ -385,8 +400,8 @@ class NetErrorHelperCoreTest : public testing::Test,
   void ReloadPage() override { reload_count_++; }
 
   void LoadPageFromCache(const GURL& page_url) override {
-    show_saved_count_++;
-    show_saved_url_ = page_url;
+    show_saved_copy_count_++;
+    show_saved_copy_url_ = page_url;
   }
 
   void DiagnoseError(const GURL& page_url) override {
@@ -394,6 +409,10 @@ class NetErrorHelperCoreTest : public testing::Test,
     diagnose_error_url_ = page_url;
   }
 
+  void ShowOfflinePages() override {
+    show_saved_pages_count_++;
+  }
+
   void SendTrackingRequest(const GURL& tracking_url,
                            const std::string& tracking_request_body) override {
     last_tracking_url_ = tracking_url;
@@ -436,14 +455,15 @@ class NetErrorHelperCoreTest : public testing::Test,
   // Values passed in to the last call of GenerateLocalizedErrorPage or
   // UpdateErrorPage.  Mutable because GenerateLocalizedErrorPage is const.
   mutable bool last_can_show_network_diagnostics_dialog_;
+  mutable bool last_has_offline_pages_;
   mutable scoped_ptr<ErrorPageParams> last_error_page_params_;
 
   int reload_count_;
-  int show_saved_count_;
-  GURL show_saved_url_;
+  int show_saved_copy_count_;
+  GURL show_saved_copy_url_;
   int diagnose_error_count_;
   GURL diagnose_error_url_;
-
+  int show_saved_pages_count_;
 
   int enable_page_helper_functions_count_;
 
@@ -2464,10 +2484,10 @@ TEST_F(NetErrorHelperCoreTest, ExplicitReloadSucceeds) {
 
 TEST_F(NetErrorHelperCoreTest, ExplicitShowSavedSucceeds) {
   DoErrorLoad(net::ERR_CONNECTION_RESET);
-  EXPECT_EQ(0, show_saved_count());
+  EXPECT_EQ(0, show_saved_copy_count());
   core()->ExecuteButtonPress(NetErrorHelperCore::SHOW_SAVED_COPY_BUTTON);
-  EXPECT_EQ(1, show_saved_count());
-  EXPECT_EQ(GURL(kFailedUrl), show_saved_url());
+  EXPECT_EQ(1, show_saved_copy_count());
+  EXPECT_EQ(GURL(kFailedUrl), show_saved_copy_url());
 }
 
 TEST_F(NetErrorHelperCoreTest, CanNotShowNetworkDiagnostics) {
@@ -2486,5 +2506,14 @@ TEST_F(NetErrorHelperCoreTest, CanShowNetworkDiagnostics) {
   EXPECT_EQ(GURL(kFailedUrl), diagnose_error_url());
 }
 
+TEST_F(NetErrorHelperCoreTest, ShowSavedPages) {
+  core()->OnSetHasOfflinePages(true);
+  DoErrorLoad(net::ERR_INTERNET_DISCONNECTED);
+  EXPECT_TRUE(last_has_offline_pages());
+  EXPECT_EQ(0, show_saved_pages_count());
+  core()->ExecuteButtonPress(NetErrorHelperCore::SHOW_SAVED_PAGES_BUTTON);
+  EXPECT_EQ(1, show_saved_pages_count());
+}
+
 }  // namespace
 }  // namespace error_page