| // Copyright 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 "chrome/browser/sessions/session_service_test_helper.h" |
| |
| #include "chrome/browser/sessions/session_service.h" |
| #include "components/sessions/core/base_session_service_test_helper.h" |
| #include "components/sessions/core/serialized_navigation_entry_test_helper.h" |
| #include "components/sessions/core/session_backend.h" |
| #include "components/sessions/core/session_id.h" |
| #include "components/sessions/core/session_types.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/test/test_utils.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| using base::Time; |
| |
| SessionServiceTestHelper::SessionServiceTestHelper() {} |
| |
| SessionServiceTestHelper::SessionServiceTestHelper(SessionService* service) |
| : service_(service) {} |
| |
| SessionServiceTestHelper::~SessionServiceTestHelper() {} |
| |
| void SessionServiceTestHelper::PrepareTabInWindow(const SessionID& window_id, |
| const SessionID& tab_id, |
| int visual_index, |
| bool select) { |
| service()->SetTabWindow(window_id, tab_id); |
| service()->SetTabIndexInWindow(window_id, tab_id, visual_index); |
| if (select) |
| service()->SetSelectedTabInWindow(window_id, visual_index); |
| } |
| |
| void SessionServiceTestHelper::SetTabExtensionAppID( |
| const SessionID& window_id, |
| const SessionID& tab_id, |
| const std::string& extension_app_id) { |
| 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); |
| } |
| |
| void SessionServiceTestHelper::SetForceBrowserNotAliveWithNoWindows( |
| bool force_browser_not_alive_with_no_windows) { |
| service()->force_browser_not_alive_with_no_windows_ = |
| force_browser_not_alive_with_no_windows; |
| } |
| |
| // Be sure and null out service to force closing the file. |
| void SessionServiceTestHelper::ReadWindows( |
| std::vector<std::unique_ptr<sessions::SessionWindow>>* windows, |
| SessionID* active_window_id) { |
| std::vector<std::unique_ptr<sessions::SessionCommand>> read_commands; |
| sessions::BaseSessionServiceTestHelper test_helper( |
| service_->GetBaseSessionServiceForTest()); |
| test_helper.ReadLastSessionCommands(&read_commands); |
| RestoreSessionFromCommands(read_commands, windows, active_window_id); |
| service()->RemoveUnusedRestoreWindows(windows); |
| } |
| |
| void SessionServiceTestHelper::AssertTabEquals( |
| const SessionID& window_id, |
| const SessionID& tab_id, |
| int visual_index, |
| int nav_index, |
| size_t nav_count, |
| const sessions::SessionTab& session_tab) { |
| EXPECT_EQ(window_id.id(), session_tab.window_id.id()); |
| EXPECT_EQ(tab_id.id(), session_tab.tab_id.id()); |
| AssertTabEquals(visual_index, nav_index, nav_count, session_tab); |
| } |
| |
| void SessionServiceTestHelper::AssertTabEquals( |
| int visual_index, |
| int nav_index, |
| size_t nav_count, |
| const sessions::SessionTab& session_tab) { |
| EXPECT_EQ(visual_index, session_tab.tab_visual_index); |
| EXPECT_EQ(nav_index, session_tab.current_navigation_index); |
| ASSERT_EQ(nav_count, session_tab.navigations.size()); |
| } |
| |
| // TODO(sky): nuke this and change to call directly into |
| // SerializedNavigationEntryTestHelper. |
| void SessionServiceTestHelper::AssertNavigationEquals( |
| const sessions::SerializedNavigationEntry& expected, |
| const sessions::SerializedNavigationEntry& actual) { |
| sessions::SerializedNavigationEntryTestHelper::ExpectNavigationEquals( |
| expected, actual); |
| } |
| |
| void SessionServiceTestHelper::AssertSingleWindowWithSingleTab( |
| const std::vector<std::unique_ptr<sessions::SessionWindow>>& windows, |
| size_t nav_count) { |
| ASSERT_EQ(1U, windows.size()); |
| EXPECT_EQ(1U, windows[0]->tabs.size()); |
| EXPECT_EQ(nav_count, windows[0]->tabs[0]->navigations.size()); |
| } |
| |
| void SessionServiceTestHelper::SetService(SessionService* service) { |
| service_.reset(service); |
| // Execute IO tasks posted by the SessionService. |
| content::RunAllTasksUntilIdle(); |
| } |
| |
| SessionService* SessionServiceTestHelper::ReleaseService() { |
| return service_.release(); |
| } |
| |
| void SessionServiceTestHelper::RunTaskOnBackendThread( |
| const base::Location& from_here, |
| const base::Closure& task) { |
| sessions::BaseSessionServiceTestHelper test_helper( |
| service_->GetBaseSessionServiceForTest()); |
| test_helper.RunTaskOnBackendThread(from_here, task); |
| } |
| |
| void SessionServiceTestHelper::SetAvailableRange( |
| const SessionID& tab_id, |
| const std::pair<int, int>& range) { |
| service_->SetAvailableRangeForTest(tab_id, range); |
| } |
| |
| bool SessionServiceTestHelper::GetAvailableRange(const SessionID& tab_id, |
| std::pair<int, int>* range) { |
| return service_->GetAvailableRangeForTest(tab_id, range); |
| } |