Index: chrome/browser/sessions/session_service_test_helper.cc |
diff --git a/chrome/browser/sessions/session_service_test_helper.cc b/chrome/browser/sessions/session_service_test_helper.cc |
index 4bafec1ce13c2aca3c9bb31d43e6d2b3943cce9a..bd1841e05aabcb0b5630a1f352b2050d3e8fcec0 100644 |
--- a/chrome/browser/sessions/session_service_test_helper.cc |
+++ b/chrome/browser/sessions/session_service_test_helper.cc |
@@ -43,6 +43,13 @@ void SessionServiceTestHelper::SetTabExtensionAppID( |
service()->SetTabExtensionAppID(window_id, tab_id, extension_app_id); |
} |
+void SessionServiceTestHelper::SetTabUserAgentOverride( |
+ const SessionID& window_id, |
+ const SessionID& tab_id, |
+ const std::string& user_agent_override) { |
+ service()->SetTabUserAgentOverride(window_id, tab_id, user_agent_override); |
+} |
+ |
// Be sure and null out service to force closing the file. |
void SessionServiceTestHelper::ReadWindows( |
std::vector<SessionWindow*>* windows) { |
@@ -84,6 +91,7 @@ void SessionServiceTestHelper::AssertNavigationEquals( |
EXPECT_EQ(expected.transition(), actual.transition()); |
EXPECT_EQ(expected.type_mask(), actual.type_mask()); |
EXPECT_TRUE(expected.original_request_url() == actual.original_request_url()); |
+ EXPECT_EQ(expected.override_user_agent(), actual.override_user_agent()); |
} |
void SessionServiceTestHelper::AssertSingleWindowWithSingleTab( |