1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #ifndef CHROME_BROWSER_UI_TOOLBAR_RECENT_TABS_BUILDER_TEST_HELPER_H_ 6 #define CHROME_BROWSER_UI_TOOLBAR_RECENT_TABS_BUILDER_TEST_HELPER_H_ 7 8 #include <vector> 9 10 #include "base/strings/string16.h" 11 #include "base/time/time.h" 12 #include "components/sessions/session_id.h" 13 14 namespace browser_sync { 15 class OpenTabsUIDelegate; 16 class SessionsSyncManager; 17 } 18 namespace sync_pb { 19 class SessionSpecifics; 20 } 21 22 // Utility class to help add recent tabs for testing. 23 class RecentTabsBuilderTestHelper { 24 public: 25 RecentTabsBuilderTestHelper(); 26 ~RecentTabsBuilderTestHelper(); 27 28 void AddSession(); 29 int GetSessionCount(); 30 SessionID::id_type GetSessionID(int session_index); 31 base::Time GetSessionTimestamp(int session_index); 32 33 void AddWindow(int session_index); 34 int GetWindowCount(int session_index); 35 SessionID::id_type GetWindowID(int session_index, int window_index); 36 37 void AddTab(int session_index, int window_index); 38 void AddTabWithInfo(int session_index, 39 int window_index, 40 base::Time timestamp, 41 const base::string16& title); 42 int GetTabCount(int session_index, int window_index); 43 SessionID::id_type GetTabID(int session_index, 44 int window_index, 45 int tab_index); 46 base::Time GetTabTimestamp(int session_index, 47 int window_index, 48 int tab_index); 49 base::string16 GetTabTitle(int session_index, 50 int window_index, 51 int tab_index); 52 53 void ExportToSessionsSyncManager( 54 browser_sync::SessionsSyncManager* manager); 55 56 std::vector<base::string16> GetTabTitlesSortedByRecency(); 57 58 private: 59 void BuildSessionSpecifics(int session_index, 60 sync_pb::SessionSpecifics* meta); 61 void BuildWindowSpecifics(int session_index, 62 int window_index, 63 sync_pb::SessionSpecifics* meta); 64 void BuildTabSpecifics(int session_index, 65 int window_index, 66 int tab_index, 67 sync_pb::SessionSpecifics* tab_base); 68 void VerifyExport(browser_sync::OpenTabsUIDelegate* delegate); 69 70 struct TabInfo; 71 struct WindowInfo; 72 struct SessionInfo; 73 74 std::vector<SessionInfo> sessions_; 75 base::Time start_time_; 76 77 int max_tab_node_id_; 78 79 DISALLOW_COPY_AND_ASSIGN(RecentTabsBuilderTestHelper); 80 }; 81 82 #endif // CHROME_BROWSER_UI_TOOLBAR_RECENT_TABS_BUILDER_TEST_HELPER_H_ 83