Add info about user agent overrides to WebContents

chrome/browser/sessions/session_service_unittest.cc

 // Copyright (c) 2012 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 "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/file_util.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/scoped_vector.h"
 #include "base/path_service.h"
@@ skipping 67 matching lines @@
                         bool select) {
     scoped_ptr<NavigationEntry> entry(NavigationEntry::Create());
     entry->SetURL(navigation.virtual_url());
     entry->SetReferrer(navigation.referrer());
     entry->SetTitle(navigation.title());
     entry->SetContentState(navigation.state());
     entry->SetTransitionType(navigation.transition());
     entry->SetHasPostData(
         navigation.type_mask() & TabNavigation::HAS_POST_DATA);
     entry->SetOriginalRequestURL(navigation.original_request_url());
+    entry->SetIsOverridingUserAgent(navigation.is_overriding_user_agent());
     service()->UpdateTabNavigation(window_id, tab_id, index, *entry.get());
     if (select)
       service()->SetSelectedNavigationIndex(window_id, tab_id, index);
   }
 
   void ReadWindows(std::vector<SessionWindow*>* windows) {
     // Forces closing the file.
     helper_.set_service(NULL);
 
     SessionService* session_service = new SessionService(path_);
@@ skipping 560 matching lines @@
 // Don't set the pinned state and make sure the pinned value is false.
 TEST_F(SessionServiceTest, PinnedDefaultsToFalse) {
   EXPECT_FALSE(CreateAndWriteSessionWithOneTab(false, false));
 }
 
 // Explicitly set the pinned state to false and make sure we get back false.
 TEST_F(SessionServiceTest, PinnedFalseWhenSetToFalse) {
   EXPECT_FALSE(CreateAndWriteSessionWithOneTab(false, true));
 }
 
+// Explicitly set the pinned state to true and make sure we get back true.
+TEST_F(SessionServiceTest, PinnedTrue) {
+  EXPECT_TRUE(CreateAndWriteSessionWithOneTab(true, true));
+}
+
 // Make sure application extension ids are persisted.
 TEST_F(SessionServiceTest, PersistApplicationExtensionID) {
   SessionID tab_id;
   ASSERT_NE(window_id.id(), tab_id.id());
   std::string app_id("foo");
 
   TabNavigation nav1(0, GURL("http://google.com"), content::Referrer(),
                      ASCIIToUTF16("abc"), std::string(),
                      content::PAGE_TRANSITION_QUALIFIER_MASK);
 
   helper_.PrepareTabInWindow(window_id, tab_id, 0, true);
   UpdateNavigation(window_id, tab_id, nav1, 0, true);
   helper_.SetTabExtensionAppID(window_id, tab_id, app_id);
 
   ScopedVector<SessionWindow> windows;
   ReadWindows(&(windows.get()));
 
   helper_.AssertSingleWindowWithSingleTab(windows.get(), 1);
   EXPECT_TRUE(app_id == windows[0]->tabs[0]->extension_app_id);
 }
 
-// Explicitly set the pinned state to true and make sure we get back true.
-TEST_F(SessionServiceTest, PinnedTrue) {
-  EXPECT_TRUE(CreateAndWriteSessionWithOneTab(true, true));
+// Check that user agent overrides are persisted.
+TEST_F(SessionServiceTest, PersistUserAgentOverrides) {
+  SessionID tab_id;
+  ASSERT_NE(window_id.id(), tab_id.id());
+  std::string user_agent_override("overridden user agent");
+
+  TabNavigation nav1(0, GURL("http://google.com"), content::Referrer(),
+                     ASCIIToUTF16("abc"), std::string(),
+                     content::PAGE_TRANSITION_QUALIFIER_MASK);
+  nav1.set_is_overriding_user_agent(true);
+
+  helper_.PrepareTabInWindow(window_id, tab_id, 0, true);
+  UpdateNavigation(window_id, tab_id, nav1, 0, true);
+  helper_.SetTabUserAgentOverride(window_id, tab_id, user_agent_override);
+
+  ScopedVector<SessionWindow> windows;
+  ReadWindows(&(windows.get()));
+  helper_.AssertSingleWindowWithSingleTab(windows.get(), 1);
+
+  SessionTab* tab = windows[0]->tabs[0];
+  helper_.AssertTabEquals(window_id, tab_id, 0, 0, 1, *tab);
+  helper_.AssertNavigationEquals(nav1, tab->navigations[0]);
+  EXPECT_TRUE(user_agent_override == tab->user_agent_override);
 }
 
 // Test that the notification for SESSION_SERVICE_SAVED is working properly.
 TEST_F(SessionServiceTest, SavedSessionNotification) {
   content::NotificationRegistrar registrar_;
   registrar_.Add(this, chrome::NOTIFICATION_SESSION_SERVICE_SAVED,
                  content::NotificationService::AllSources());
   service()->Save();
   EXPECT_EQ(sync_save_count_, 1);
 }
@@ skipping 174 matching lines @@
   ASSERT_TRUE(file_util::CopyFile(v1_file_path, dest_file_path));
 
   SessionService* session_service = new SessionService(path_);
   helper_.set_service(session_service);
   ScopedVector<SessionWindow> windows;
   helper_.ReadWindows(&(windows.get()));
   ASSERT_EQ(1u, windows.size());
   EXPECT_EQ(1u, windows[0]->tabs.size());
 }
 #endif