Move chrome-specific SerializedNavigation code to chrome/.

chrome/browser/sessions/session_common_utils_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/sessions/session_common_utils.h"
 
 #include <memory>
 
 #include "components/sessions/content/content_serialized_navigation_builder.h"
+#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 "components/sessions/core/session_types.h"
 #include "content/public/browser/navigation_entry.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 {
 
 void AppendNavigations(sessions::SessionTab* tab,
                        content::NavigationEntry* entry) {
   tab->navigations.push_back(
       sessions::ContentSerializedNavigationBuilder::FromNavigationEntry(
           0, *entry));
 }
 
 }  // namespace
+
 class SessionCommonUtilTest : public ::testing::Test {};
 
 TEST_F(SessionCommonUtilTest, GetSelectedIndex) {
   const GURL settings_page("chrome://settings");
   const GURL sign_out_page("chrome://settings/signOut");
   sessions::SessionTab tab;
 
   std::unique_ptr<content::NavigationEntry> entry1(
       content::NavigationEntry::Create());
   std::unique_ptr<content::NavigationEntry> entry2(
       content::NavigationEntry::Create());
   entry1->SetVirtualURL(settings_page);
   entry2->SetVirtualURL(sign_out_page);
 
   AppendNavigations(&tab, entry1.get());
   AppendNavigations(&tab, entry2.get());
 
   tab.current_navigation_index = 0;
   ASSERT_EQ(0, GetNavigationIndexToSelect(tab));
   tab.current_navigation_index = 1;
   ASSERT_EQ(0, GetNavigationIndexToSelect(tab));
 }
+
+// Tests that the input data is left unsanitized when the referrer policy is
+// Always.
+TEST_F(SessionCommonUtilTest, SanitizeWithReferrerPolicyAlways) {
+  sessions::SerializedNavigationDriver::SetSanitizeCallback(
+      base::Bind(&SanitizeNavigation));
+  sessions::ContentSerializedNavigationDriver* driver =
+      sessions::ContentSerializedNavigationDriver::GetInstance();
+  sessions::SerializedNavigationEntry navigation =
+      sessions::SerializedNavigationEntryTestHelper::CreateNavigationForTest();
+  sessions::SerializedNavigationEntryTestHelper::SetReferrerPolicy(
+      blink::WebReferrerPolicyAlways, &navigation);
+
+  content::PageState page_state =
+      content::PageState::CreateFromURL(sessions::test_data::kVirtualURL);
+  sessions::SerializedNavigationEntryTestHelper::SetEncodedPageState(
+      page_state.ToEncodedData(), &navigation);
+
+  driver->Sanitize(&navigation);
+  EXPECT_EQ(sessions::test_data::kIndex, navigation.index());
+  EXPECT_EQ(sessions::test_data::kUniqueID, navigation.unique_id());
+  EXPECT_EQ(sessions::test_data::kReferrerURL, navigation.referrer_url());
+  EXPECT_EQ(blink::WebReferrerPolicyAlways, navigation.referrer_policy());
+  EXPECT_EQ(sessions::test_data::kVirtualURL, navigation.virtual_url());
+  EXPECT_EQ(sessions::test_data::kTitle, navigation.title());
+  EXPECT_EQ(page_state.ToEncodedData(), navigation.encoded_page_state());
+  EXPECT_TRUE(ui::PageTransitionTypeIncludingQualifiersIs(
+      navigation.transition_type(), sessions::test_data::kTransitionType));
+  EXPECT_EQ(sessions::test_data::kHasPostData, navigation.has_post_data());
+  EXPECT_EQ(sessions::test_data::kPostID, navigation.post_id());
+  EXPECT_EQ(sessions::test_data::kOriginalRequestURL,
+            navigation.original_request_url());
+  EXPECT_EQ(sessions::test_data::kIsOverridingUserAgent,
+            navigation.is_overriding_user_agent());
+  EXPECT_EQ(sessions::test_data::kTimestamp, navigation.timestamp());
+  EXPECT_EQ(sessions::test_data::kSearchTerms, navigation.search_terms());
+  EXPECT_EQ(sessions::test_data::kFaviconURL, navigation.favicon_url());
+  EXPECT_EQ(sessions::test_data::kHttpStatusCode,
+            navigation.http_status_code());
+}
+
+// Tests that the input data is properly sanitized when the referrer policy is
+// Never.
+TEST_F(SessionCommonUtilTest, SanitizeWithReferrerPolicyNever) {
+  sessions::SerializedNavigationDriver::SetSanitizeCallback(
+      base::Bind(&SanitizeNavigation));
+  sessions::ContentSerializedNavigationDriver* driver =
+      sessions::ContentSerializedNavigationDriver::GetInstance();
+  sessions::SerializedNavigationEntry navigation =
+      sessions::SerializedNavigationEntryTestHelper::CreateNavigationForTest();
+  sessions::SerializedNavigationEntryTestHelper::SetReferrerPolicy(
+      blink::WebReferrerPolicyNever, &navigation);
+
+  content::PageState page_state =
+      content::PageState::CreateFromURL(sessions::test_data::kVirtualURL);
+  sessions::SerializedNavigationEntryTestHelper::SetEncodedPageState(
+      page_state.ToEncodedData(), &navigation);
+
+  driver->Sanitize(&navigation);
+
+  // Fields that should remain untouched.
+  EXPECT_EQ(sessions::test_data::kIndex, navigation.index());
+  EXPECT_EQ(sessions::test_data::kUniqueID, navigation.unique_id());
+  EXPECT_EQ(sessions::test_data::kVirtualURL, navigation.virtual_url());
+  EXPECT_EQ(sessions::test_data::kTitle, navigation.title());
+  EXPECT_TRUE(ui::PageTransitionTypeIncludingQualifiersIs(
+      navigation.transition_type(), sessions::test_data::kTransitionType));
+  EXPECT_EQ(sessions::test_data::kHasPostData, navigation.has_post_data());
+  EXPECT_EQ(sessions::test_data::kPostID, navigation.post_id());
+  EXPECT_EQ(sessions::test_data::kOriginalRequestURL,
+            navigation.original_request_url());
+  EXPECT_EQ(sessions::test_data::kIsOverridingUserAgent,
+            navigation.is_overriding_user_agent());
+  EXPECT_EQ(sessions::test_data::kTimestamp, navigation.timestamp());
+  EXPECT_EQ(sessions::test_data::kSearchTerms, navigation.search_terms());
+  EXPECT_EQ(sessions::test_data::kFaviconURL, navigation.favicon_url());
+  EXPECT_EQ(sessions::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());
+}