Index: components/sessions/content/content_serialized_navigation_driver_unittest.cc |
diff --git a/components/sessions/content/content_serialized_navigation_driver_unittest.cc b/components/sessions/content/content_serialized_navigation_driver_unittest.cc |
index 67a34512d8cd3cf2e86a7e711ae4b3d5fcfeffa2..98f1eaf559ab327f8b0d7d76137bdcdd880d7df6 100644 |
--- a/components/sessions/content/content_serialized_navigation_driver_unittest.cc |
+++ b/components/sessions/content/content_serialized_navigation_driver_unittest.cc |
@@ -13,6 +13,29 @@ |
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 = |
@@ -117,4 +140,14 @@ TEST(ContentSerializedNavigationDriverTest, SanitizeWithReferrerPolicyNever) { |
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 |