Persist offline page info in a navigation entry if needed

components/sessions/content/content_serialized_navigation_driver_unittest.cc

 // Copyright 2014 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 "components/sessions/content/content_serialized_navigation_driver.h"
 
 #include "components/sessions/core/serialized_navigation_entry.h"
 #include "components/sessions/core/serialized_navigation_entry_test_helper.h"
 #include "content/public/common/page_state.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/WebKit/public/platform/WebReferrerPolicy.h"
 #include "ui/base/page_transition_types.h"
 
 namespace sessions {
 
+namespace {
+const char kExtendedInfoKey[] = "Key 1";
+const char kExtendedInfoValue[] = "Value 1";
+
+class TestExtendedInfoHandler :
+    public SerializedNavigationDriver::ExtendedInfoHandler {
+ public:
+  TestExtendedInfoHandler() {}
+  ~TestExtendedInfoHandler() override {}
+
+  std::string GetExtendedInfo(
+      const content::NavigationEntry& entry) const override {
+    return kExtendedInfoValue;
+  }
+
+  void RestoreExtendedInfo(
+      const std::string& info, content::NavigationEntry* entry) override {
+
+  }
+};
+
+}  // namespace
+
 // Tests that PageState data is properly sanitized when post data is present.
 TEST(ContentSerializedNavigationDriverTest, PickleSanitizationWithPostData) {
   ContentSerializedNavigationDriver* driver =
       ContentSerializedNavigationDriver::GetInstance();
   SerializedNavigationEntry navigation =
       SerializedNavigationEntryTestHelper::CreateNavigationForTest();
   ASSERT_TRUE(navigation.has_post_data());
 
   // When post data is present, the page state should be sanitized.
   std::string sanitized_page_state =
@@ skipping 84 matching lines @@
   EXPECT_EQ(test_data::kSearchTerms, navigation.search_terms());
   EXPECT_EQ(test_data::kFaviconURL, navigation.favicon_url());
   EXPECT_EQ(test_data::kHttpStatusCode, navigation.http_status_code());
 
   // Fields that were sanitized.
   EXPECT_EQ(GURL(), navigation.referrer_url());
   EXPECT_EQ(blink::WebReferrerPolicyDefault, navigation.referrer_policy());
   EXPECT_EQ(page_state.ToEncodedData(), navigation.encoded_page_state());
 }
 
+TEST(ContentSerializedNavigationDriverTest, ExtendedInfoHandler) {
+  ContentSerializedNavigationDriver* driver =
+      ContentSerializedNavigationDriver::GetInstance();
+
+  std::unique_ptr<SerializedNavigationDriver::ExtendedInfoHandler> handler(
+      new TestExtendedInfoHandler);
+  driver->RegisterExtendedInfoHandler(kExtendedInfoKey, std::move(handler));
+  EXPECT_EQ(1U, driver->GetAllExtendedInfoHandlers().size());
+}
+
 }  // namespace sessions