Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(164)

Side by Side Diff: components/sync_sessions/synced_session_tracker_unittest.cc

Issue 2494533002: [Sync] Put session tracker in charge of maintaining local state. (Closed)
Patch Set: Rebase and update comment Created 4 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright 2012 The Chromium Authors. All rights reserved. 1 // Copyright 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "components/sync_sessions/synced_session_tracker.h" 5 #include "components/sync_sessions/synced_session_tracker.h"
6 6
7 #include "base/rand_util.h" 7 #include "base/rand_util.h"
8 #include "base/strings/stringprintf.h" 8 #include "base/strings/stringprintf.h"
9 #include "base/strings/utf_string_conversions.h" 9 #include "base/strings/utf_string_conversions.h"
10 #include "components/sessions/core/serialized_navigation_entry_test_helper.h" 10 #include "components/sessions/core/serialized_navigation_entry_test_helper.h"
11 #include "components/sync_sessions/fake_sync_sessions_client.h" 11 #include "components/sync_sessions/fake_sync_sessions_client.h"
12 #include "testing/gtest/include/gtest/gtest.h" 12 #include "testing/gtest/include/gtest/gtest.h"
13 13
14 namespace sync_sessions { 14 namespace sync_sessions {
15 15
16 namespace { 16 namespace {
17 17
18 const char kValidUrl[] = "http://www.example.com"; 18 const char kValidUrl[] = "http://www.example.com";
19 const char kInvalidUrl[] = "invalid.url"; 19 const char kInvalidUrl[] = "invalid.url";
20 const char kTag[] = "tag";
21 const char kTag2[] = "tag2";
22 const char kTag3[] = "tag3";
23 const char kTitle[] = "title";
20 24
21 } // namespace 25 } // namespace
22 26
23 class SyncedSessionTrackerTest : public testing::Test { 27 class SyncedSessionTrackerTest : public testing::Test {
24 public: 28 public:
25 SyncedSessionTrackerTest() : tracker_(&sessions_client_) {} 29 SyncedSessionTrackerTest() : tracker_(&sessions_client_) {}
26 ~SyncedSessionTrackerTest() override {} 30 ~SyncedSessionTrackerTest() override {}
27 31
28 SyncedSessionTracker* GetTracker() { return &tracker_; } 32 SyncedSessionTracker* GetTracker() { return &tracker_; }
29 33
30 private: 34 private:
31 FakeSyncSessionsClient sessions_client_; 35 FakeSyncSessionsClient sessions_client_;
32 SyncedSessionTracker tracker_; 36 SyncedSessionTracker tracker_;
33 }; 37 };
34 38
35 TEST_F(SyncedSessionTrackerTest, GetSession) { 39 TEST_F(SyncedSessionTrackerTest, GetSession) {
36 SyncedSession* session1 = GetTracker()->GetSession("tag"); 40 SyncedSession* session1 = GetTracker()->GetSession(kTag);
37 SyncedSession* session2 = GetTracker()->GetSession("tag2"); 41 SyncedSession* session2 = GetTracker()->GetSession(kTag2);
38 ASSERT_EQ(session1, GetTracker()->GetSession("tag")); 42 ASSERT_EQ(session1, GetTracker()->GetSession(kTag));
39 ASSERT_NE(session1, session2); 43 ASSERT_NE(session1, session2);
40 // Should clean up memory on its own. 44 // Should clean up memory on its own.
41 } 45 }
42 46
43 TEST_F(SyncedSessionTrackerTest, GetTabUnmapped) { 47 TEST_F(SyncedSessionTrackerTest, GetTabUnmapped) {
44 sessions::SessionTab* tab = GetTracker()->GetTab("tag", 0, 0); 48 sessions::SessionTab* tab = GetTracker()->GetTab(kTag, 0, 0);
45 ASSERT_EQ(tab, GetTracker()->GetTab("tag", 0, 0)); 49 ASSERT_EQ(tab, GetTracker()->GetTab(kTag, 0, 0));
46 // Should clean up memory on its own. 50 // Should clean up memory on its own.
47 } 51 }
48 52
49 TEST_F(SyncedSessionTrackerTest, PutWindowInSession) { 53 TEST_F(SyncedSessionTrackerTest, PutWindowInSession) {
50 GetTracker()->PutWindowInSession("tag", 0); 54 GetTracker()->PutWindowInSession(kTag, 0);
51 SyncedSession* session = GetTracker()->GetSession("tag"); 55 SyncedSession* session = GetTracker()->GetSession(kTag);
52 ASSERT_EQ(1U, session->windows.size()); 56 ASSERT_EQ(1U, session->windows.size());
53 // Should clean up memory on its own. 57 // Should clean up memory on its own.
54 } 58 }
55 59
56 TEST_F(SyncedSessionTrackerTest, PutTabInWindow) { 60 TEST_F(SyncedSessionTrackerTest, PutTabInWindow) {
57 GetTracker()->PutWindowInSession("tag", 10); 61 GetTracker()->PutWindowInSession(kTag, 10);
58 GetTracker()->PutTabInWindow("tag", 10, 15, 62 GetTracker()->PutTabInWindow(kTag, 10, 15,
59 0); // win id 10, tab id 15, tab ind 0. 63 0); // win id 10, tab id 15, tab ind 0.
60 SyncedSession* session = GetTracker()->GetSession("tag"); 64 SyncedSession* session = GetTracker()->GetSession(kTag);
61 ASSERT_EQ(1U, session->windows.size()); 65 ASSERT_EQ(1U, session->windows.size());
62 ASSERT_EQ(1U, session->windows[10]->tabs.size()); 66 ASSERT_EQ(1U, session->windows[10]->tabs.size());
63 ASSERT_EQ(GetTracker()->GetTab("tag", 15, 1), 67 ASSERT_EQ(GetTracker()->GetTab(kTag, 15, 1),
64 session->windows[10]->tabs[0].get()); 68 session->windows[10]->tabs[0].get());
65 // Should clean up memory on its own. 69 // Should clean up memory on its own.
66 } 70 }
67 71
68 TEST_F(SyncedSessionTrackerTest, LookupAllForeignSessions) { 72 TEST_F(SyncedSessionTrackerTest, LookupAllForeignSessions) {
69 std::vector<const SyncedSession*> sessions; 73 std::vector<const SyncedSession*> sessions;
70 ASSERT_FALSE(GetTracker()->LookupAllForeignSessions( 74 ASSERT_FALSE(GetTracker()->LookupAllForeignSessions(
71 &sessions, SyncedSessionTracker::PRESENTABLE)); 75 &sessions, SyncedSessionTracker::PRESENTABLE));
72 GetTracker()->GetSession("tag1"); 76 GetTracker()->GetSession(kTag);
73 GetTracker()->PutWindowInSession("tag1", 0); 77 GetTracker()->PutWindowInSession(kTag, 0);
74 GetTracker()->PutTabInWindow("tag1", 0, 15, 0); 78 GetTracker()->PutTabInWindow(kTag, 0, 15, 0);
75 sessions::SessionTab* tab = GetTracker()->GetTab("tag1", 15, 1); 79 sessions::SessionTab* tab = GetTracker()->GetTab(kTag, 15, 1);
76 ASSERT_TRUE(tab); 80 ASSERT_TRUE(tab);
77 tab->navigations.push_back( 81 tab->navigations.push_back(
78 sessions::SerializedNavigationEntryTestHelper::CreateNavigation(kValidUrl, 82 sessions::SerializedNavigationEntryTestHelper::CreateNavigation(kValidUrl,
79 "title")); 83 kTitle));
80 GetTracker()->GetSession("tag2"); 84 GetTracker()->GetSession(kTag2);
81 GetTracker()->GetSession("tag3"); 85 GetTracker()->GetSession(kTag3);
82 GetTracker()->PutWindowInSession("tag3", 0); 86 GetTracker()->PutWindowInSession(kTag3, 0);
83 GetTracker()->PutTabInWindow("tag3", 0, 15, 0); 87 GetTracker()->PutTabInWindow(kTag3, 0, 15, 0);
84 tab = GetTracker()->GetTab("tag3", 15, 1); 88 tab = GetTracker()->GetTab(kTag3, 15, 1);
85 ASSERT_TRUE(tab); 89 ASSERT_TRUE(tab);
86 tab->navigations.push_back( 90 tab->navigations.push_back(
87 sessions::SerializedNavigationEntryTestHelper::CreateNavigation( 91 sessions::SerializedNavigationEntryTestHelper::CreateNavigation(
88 kInvalidUrl, "title")); 92 kInvalidUrl, kTitle));
89 ASSERT_TRUE(GetTracker()->LookupAllForeignSessions( 93 ASSERT_TRUE(GetTracker()->LookupAllForeignSessions(
90 &sessions, SyncedSessionTracker::PRESENTABLE)); 94 &sessions, SyncedSessionTracker::PRESENTABLE));
91 // Only the session with a valid window and tab gets returned. 95 // Only the session with a valid window and tab gets returned.
92 ASSERT_EQ(1U, sessions.size()); 96 ASSERT_EQ(1U, sessions.size());
93 ASSERT_EQ("tag1", sessions[0]->session_tag); 97 ASSERT_EQ(kTag, sessions[0]->session_tag);
94 98
95 ASSERT_TRUE(GetTracker()->LookupAllForeignSessions( 99 ASSERT_TRUE(GetTracker()->LookupAllForeignSessions(
96 &sessions, SyncedSessionTracker::RAW)); 100 &sessions, SyncedSessionTracker::RAW));
97 ASSERT_EQ(3U, sessions.size()); 101 ASSERT_EQ(3U, sessions.size());
98 } 102 }
99 103
100 TEST_F(SyncedSessionTrackerTest, LookupSessionWindows) { 104 TEST_F(SyncedSessionTrackerTest, LookupSessionWindows) {
101 std::vector<const sessions::SessionWindow*> windows; 105 std::vector<const sessions::SessionWindow*> windows;
102 ASSERT_FALSE(GetTracker()->LookupSessionWindows("tag1", &windows)); 106 ASSERT_FALSE(GetTracker()->LookupSessionWindows(kTag, &windows));
103 GetTracker()->GetSession("tag1"); 107 GetTracker()->GetSession(kTag);
104 GetTracker()->PutWindowInSession("tag1", 0); 108 GetTracker()->PutWindowInSession(kTag, 0);
105 GetTracker()->PutWindowInSession("tag1", 2); 109 GetTracker()->PutWindowInSession(kTag, 2);
106 GetTracker()->GetSession("tag2"); 110 GetTracker()->GetSession(kTag2);
107 GetTracker()->PutWindowInSession("tag2", 0); 111 GetTracker()->PutWindowInSession(kTag2, 0);
108 GetTracker()->PutWindowInSession("tag2", 2); 112 GetTracker()->PutWindowInSession(kTag2, 2);
109 ASSERT_TRUE(GetTracker()->LookupSessionWindows("tag1", &windows)); 113 ASSERT_TRUE(GetTracker()->LookupSessionWindows(kTag, &windows));
110 ASSERT_EQ(2U, windows.size()); // Only windows from tag1 session. 114 ASSERT_EQ(2U, windows.size()); // Only windows from kTag session.
111 ASSERT_NE((sessions::SessionWindow*)nullptr, windows[0]); 115 ASSERT_NE((sessions::SessionWindow*)nullptr, windows[0]);
112 ASSERT_NE((sessions::SessionWindow*)nullptr, windows[1]); 116 ASSERT_NE((sessions::SessionWindow*)nullptr, windows[1]);
113 ASSERT_NE(windows[1], windows[0]); 117 ASSERT_NE(windows[1], windows[0]);
114 } 118 }
115 119
116 TEST_F(SyncedSessionTrackerTest, LookupSessionTab) { 120 TEST_F(SyncedSessionTrackerTest, LookupSessionTab) {
117 const sessions::SessionTab* tab; 121 const sessions::SessionTab* tab;
118 ASSERT_FALSE(GetTracker()->LookupSessionTab("tag1", 5, &tab)); 122 ASSERT_FALSE(GetTracker()->LookupSessionTab(kTag, 5, &tab));
119 GetTracker()->GetSession("tag1"); 123 GetTracker()->GetSession(kTag);
120 GetTracker()->PutWindowInSession("tag1", 0); 124 GetTracker()->PutWindowInSession(kTag, 0);
121 GetTracker()->PutTabInWindow("tag1", 0, 5, 0); 125 GetTracker()->PutTabInWindow(kTag, 0, 5, 0);
122 ASSERT_TRUE(GetTracker()->LookupSessionTab("tag1", 5, &tab)); 126 ASSERT_TRUE(GetTracker()->LookupSessionTab(kTag, 5, &tab));
123 ASSERT_NE((sessions::SessionTab*)nullptr, tab); 127 ASSERT_NE((sessions::SessionTab*)nullptr, tab);
124 } 128 }
125 129
126 TEST_F(SyncedSessionTrackerTest, Complex) { 130 TEST_F(SyncedSessionTrackerTest, Complex) {
127 const std::string tag1 = "tag";
128 const std::string tag2 = "tag2";
129 const std::string tag3 = "tag3";
130 std::vector<sessions::SessionTab *> tabs1, tabs2; 131 std::vector<sessions::SessionTab *> tabs1, tabs2;
131 sessions::SessionTab* temp_tab; 132 sessions::SessionTab* temp_tab;
132 ASSERT_TRUE(GetTracker()->Empty()); 133 ASSERT_TRUE(GetTracker()->Empty());
133 ASSERT_EQ(0U, GetTracker()->num_synced_sessions()); 134 ASSERT_EQ(0U, GetTracker()->num_synced_sessions());
134 ASSERT_EQ(0U, GetTracker()->num_synced_tabs(tag1)); 135 ASSERT_EQ(0U, GetTracker()->num_synced_tabs(kTag));
135 tabs1.push_back(GetTracker()->GetTab(tag1, 0, 0)); 136 tabs1.push_back(GetTracker()->GetTab(kTag, 0, 0));
136 tabs1.push_back(GetTracker()->GetTab(tag1, 1, 1)); 137 tabs1.push_back(GetTracker()->GetTab(kTag, 1, 1));
137 tabs1.push_back(GetTracker()->GetTab(tag1, 2, 2)); 138 tabs1.push_back(GetTracker()->GetTab(kTag, 2, 2));
138 ASSERT_EQ(3U, GetTracker()->num_synced_tabs(tag1)); 139 ASSERT_EQ(3U, GetTracker()->num_synced_tabs(kTag));
139 ASSERT_EQ(1U, GetTracker()->num_synced_sessions()); 140 ASSERT_EQ(1U, GetTracker()->num_synced_sessions());
140 temp_tab = GetTracker()->GetTab(tag1, 0, 0); // Already created. 141 temp_tab = GetTracker()->GetTab(kTag, 0, 0); // Already created.
141 ASSERT_EQ(3U, GetTracker()->num_synced_tabs(tag1)); 142 ASSERT_EQ(3U, GetTracker()->num_synced_tabs(kTag));
142 ASSERT_EQ(1U, GetTracker()->num_synced_sessions()); 143 ASSERT_EQ(1U, GetTracker()->num_synced_sessions());
143 ASSERT_EQ(tabs1[0], temp_tab); 144 ASSERT_EQ(tabs1[0], temp_tab);
144 tabs2.push_back(GetTracker()->GetTab(tag2, 0, 0)); 145 tabs2.push_back(GetTracker()->GetTab(kTag2, 0, 0));
145 ASSERT_EQ(1U, GetTracker()->num_synced_tabs(tag2)); 146 ASSERT_EQ(1U, GetTracker()->num_synced_tabs(kTag2));
146 ASSERT_EQ(2U, GetTracker()->num_synced_sessions()); 147 ASSERT_EQ(2U, GetTracker()->num_synced_sessions());
147 ASSERT_FALSE(GetTracker()->DeleteSession(tag3)); 148 ASSERT_FALSE(GetTracker()->DeleteSession(kTag3));
148 149
149 SyncedSession* session = GetTracker()->GetSession(tag1); 150 SyncedSession* session = GetTracker()->GetSession(kTag);
150 SyncedSession* session2 = GetTracker()->GetSession(tag2); 151 SyncedSession* session2 = GetTracker()->GetSession(kTag2);
151 SyncedSession* session3 = GetTracker()->GetSession(tag3); 152 SyncedSession* session3 = GetTracker()->GetSession(kTag3);
152 session3->device_type = SyncedSession::TYPE_OTHER; 153 session3->device_type = SyncedSession::TYPE_OTHER;
153 ASSERT_EQ(3U, GetTracker()->num_synced_sessions()); 154 ASSERT_EQ(3U, GetTracker()->num_synced_sessions());
154 155
155 ASSERT_TRUE(session); 156 ASSERT_TRUE(session);
156 ASSERT_TRUE(session2); 157 ASSERT_TRUE(session2);
157 ASSERT_TRUE(session3); 158 ASSERT_TRUE(session3);
158 ASSERT_NE(session, session2); 159 ASSERT_NE(session, session2);
159 ASSERT_NE(session2, session3); 160 ASSERT_NE(session2, session3);
160 ASSERT_TRUE(GetTracker()->DeleteSession(tag3)); 161 ASSERT_TRUE(GetTracker()->DeleteSession(kTag3));
161 ASSERT_EQ(2U, GetTracker()->num_synced_sessions()); 162 ASSERT_EQ(2U, GetTracker()->num_synced_sessions());
162 163
163 GetTracker()->PutWindowInSession(tag1, 0); // Create a window. 164 GetTracker()->PutWindowInSession(kTag, 0); // Create a window.
164 GetTracker()->PutTabInWindow(tag1, 0, 2, 0); // No longer unmapped. 165 GetTracker()->PutTabInWindow(kTag, 0, 2, 0); // No longer unmapped.
165 ASSERT_EQ(3U, GetTracker()->num_synced_tabs(tag1)); // Has not changed. 166 ASSERT_EQ(3U, GetTracker()->num_synced_tabs(kTag)); // Has not changed.
166 167
167 const sessions::SessionTab* tab_ptr; 168 const sessions::SessionTab* tab_ptr;
168 ASSERT_TRUE(GetTracker()->LookupSessionTab(tag1, 0, &tab_ptr)); 169 ASSERT_TRUE(GetTracker()->LookupSessionTab(kTag, 0, &tab_ptr));
169 ASSERT_EQ(tab_ptr, tabs1[0]); 170 ASSERT_EQ(tab_ptr, tabs1[0]);
170 ASSERT_TRUE(GetTracker()->LookupSessionTab(tag1, 2, &tab_ptr)); 171 ASSERT_TRUE(GetTracker()->LookupSessionTab(kTag, 2, &tab_ptr));
171 ASSERT_EQ(tab_ptr, tabs1[2]); 172 ASSERT_EQ(tab_ptr, tabs1[2]);
172 ASSERT_FALSE(GetTracker()->LookupSessionTab(tag1, 3, &tab_ptr)); 173 ASSERT_FALSE(GetTracker()->LookupSessionTab(kTag, 3, &tab_ptr));
173 ASSERT_FALSE(tab_ptr); 174 ASSERT_FALSE(tab_ptr);
174 175
175 std::vector<const sessions::SessionWindow*> windows; 176 std::vector<const sessions::SessionWindow*> windows;
176 ASSERT_TRUE(GetTracker()->LookupSessionWindows(tag1, &windows)); 177 ASSERT_TRUE(GetTracker()->LookupSessionWindows(kTag, &windows));
177 ASSERT_EQ(1U, windows.size()); 178 ASSERT_EQ(1U, windows.size());
178 ASSERT_TRUE(GetTracker()->LookupSessionWindows(tag2, &windows)); 179 ASSERT_TRUE(GetTracker()->LookupSessionWindows(kTag2, &windows));
179 ASSERT_EQ(0U, windows.size()); 180 ASSERT_EQ(0U, windows.size());
180 181
181 // The sessions don't have valid tabs, lookup should not succeed. 182 // The sessions don't have valid tabs, lookup should not succeed.
182 std::vector<const SyncedSession*> sessions; 183 std::vector<const SyncedSession*> sessions;
183 ASSERT_FALSE(GetTracker()->LookupAllForeignSessions( 184 ASSERT_FALSE(GetTracker()->LookupAllForeignSessions(
184 &sessions, SyncedSessionTracker::PRESENTABLE)); 185 &sessions, SyncedSessionTracker::PRESENTABLE));
185 ASSERT_TRUE(GetTracker()->LookupAllForeignSessions( 186 ASSERT_TRUE(GetTracker()->LookupAllForeignSessions(
186 &sessions, SyncedSessionTracker::RAW)); 187 &sessions, SyncedSessionTracker::RAW));
187 ASSERT_EQ(2U, sessions.size()); 188 ASSERT_EQ(2U, sessions.size());
188 189
189 GetTracker()->Clear(); 190 GetTracker()->Clear();
190 ASSERT_EQ(0U, GetTracker()->num_synced_tabs(tag1)); 191 ASSERT_EQ(0U, GetTracker()->num_synced_tabs(kTag));
191 ASSERT_EQ(0U, GetTracker()->num_synced_tabs(tag2)); 192 ASSERT_EQ(0U, GetTracker()->num_synced_tabs(kTag2));
192 ASSERT_EQ(0U, GetTracker()->num_synced_sessions()); 193 ASSERT_EQ(0U, GetTracker()->num_synced_sessions());
193 } 194 }
194 195
195 TEST_F(SyncedSessionTrackerTest, ManyGetTabs) { 196 TEST_F(SyncedSessionTrackerTest, ManyGetTabs) {
196 ASSERT_TRUE(GetTracker()->Empty()); 197 ASSERT_TRUE(GetTracker()->Empty());
197 const int kMaxSessions = 10; 198 const int kMaxSessions = 10;
198 const int kMaxTabs = 1000; 199 const int kMaxTabs = 1000;
199 const int kMaxAttempts = 10000; 200 const int kMaxAttempts = 10000;
200 for (int j = 0; j < kMaxSessions; ++j) { 201 for (int j = 0; j < kMaxSessions; ++j) {
201 std::string tag = base::StringPrintf("tag%d", j); 202 std::string tag = base::StringPrintf("tag%d", j);
202 for (int i = 0; i < kMaxAttempts; ++i) { 203 for (int i = 0; i < kMaxAttempts; ++i) {
203 // More attempts than tabs means we'll sometimes get the same tabs, 204 // More attempts than tabs means we'll sometimes get the same tabs,
204 // sometimes have to allocate new tabs. 205 // sometimes have to allocate new tabs.
205 int rand_tab_num = base::RandInt(0, kMaxTabs); 206 int rand_tab_num = base::RandInt(0, kMaxTabs);
206 sessions::SessionTab* tab = 207 sessions::SessionTab* tab =
207 GetTracker()->GetTab(tag, rand_tab_num, rand_tab_num + 1); 208 GetTracker()->GetTab(tag, rand_tab_num, rand_tab_num + 1);
208 ASSERT_TRUE(tab); 209 ASSERT_TRUE(tab);
209 } 210 }
210 } 211 }
211 } 212 }
212 213
213 TEST_F(SyncedSessionTrackerTest, LookupTabNodeIds) { 214 TEST_F(SyncedSessionTrackerTest, LookupTabNodeIds) {
214 std::set<int> result; 215 std::set<int> result;
215 std::string tag1 = "session1";
216 std::string tag2 = "session2";
217 std::string tag3 = "session3";
218 216
219 GetTracker()->GetTab(tag1, 1, 1); 217 GetTracker()->GetTab(kTag, 1, 1);
220 GetTracker()->GetTab(tag1, 2, 2); 218 GetTracker()->GetTab(kTag, 2, 2);
221 GetTracker()->LookupTabNodeIds(tag1, &result); 219 GetTracker()->LookupTabNodeIds(kTag, &result);
222 EXPECT_EQ(2U, result.size()); 220 EXPECT_EQ(2U, result.size());
223 EXPECT_FALSE(result.end() == result.find(1)); 221 EXPECT_FALSE(result.end() == result.find(1));
224 EXPECT_FALSE(result.end() == result.find(2)); 222 EXPECT_FALSE(result.end() == result.find(2));
225 GetTracker()->LookupTabNodeIds(tag2, &result); 223 GetTracker()->LookupTabNodeIds(kTag2, &result);
226 EXPECT_TRUE(result.empty()); 224 EXPECT_TRUE(result.empty());
227 225
228 GetTracker()->PutWindowInSession(tag1, 0); 226 GetTracker()->PutWindowInSession(kTag, 0);
229 GetTracker()->PutTabInWindow(tag1, 0, 3, 0); 227 GetTracker()->PutTabInWindow(kTag, 0, 3, 0);
230 GetTracker()->LookupTabNodeIds(tag1, &result); 228 GetTracker()->LookupTabNodeIds(kTag, &result);
231 EXPECT_EQ(2U, result.size()); 229 EXPECT_EQ(2U, result.size());
232 230
233 GetTracker()->GetTab(tag1, 3, 3); 231 GetTracker()->GetTab(kTag, 3, 3);
234 GetTracker()->LookupTabNodeIds(tag1, &result); 232 GetTracker()->LookupTabNodeIds(kTag, &result);
235 EXPECT_EQ(3U, result.size()); 233 EXPECT_EQ(3U, result.size());
236 EXPECT_FALSE(result.end() == result.find(3)); 234 EXPECT_FALSE(result.end() == result.find(3));
237 235
238 GetTracker()->GetTab(tag2, 1, 21); 236 GetTracker()->GetTab(kTag2, 1, 21);
239 GetTracker()->GetTab(tag2, 2, 22); 237 GetTracker()->GetTab(kTag2, 2, 22);
240 GetTracker()->LookupTabNodeIds(tag2, &result); 238 GetTracker()->LookupTabNodeIds(kTag2, &result);
241 EXPECT_EQ(2U, result.size()); 239 EXPECT_EQ(2U, result.size());
242 EXPECT_FALSE(result.end() == result.find(21)); 240 EXPECT_FALSE(result.end() == result.find(21));
243 EXPECT_FALSE(result.end() == result.find(22)); 241 EXPECT_FALSE(result.end() == result.find(22));
244 GetTracker()->LookupTabNodeIds(tag1, &result); 242 GetTracker()->LookupTabNodeIds(kTag, &result);
245 EXPECT_EQ(3U, result.size()); 243 EXPECT_EQ(3U, result.size());
246 EXPECT_FALSE(result.end() == result.find(1)); 244 EXPECT_FALSE(result.end() == result.find(1));
247 EXPECT_FALSE(result.end() == result.find(2)); 245 EXPECT_FALSE(result.end() == result.find(2));
248 246
249 GetTracker()->LookupTabNodeIds(tag3, &result); 247 GetTracker()->LookupTabNodeIds(kTag3, &result);
250 EXPECT_TRUE(result.empty()); 248 EXPECT_TRUE(result.empty());
251 GetTracker()->PutWindowInSession(tag3, 1); 249 GetTracker()->PutWindowInSession(kTag3, 1);
252 GetTracker()->PutTabInWindow(tag3, 1, 5, 0); 250 GetTracker()->PutTabInWindow(kTag3, 1, 5, 0);
253 GetTracker()->LookupTabNodeIds(tag3, &result); 251 GetTracker()->LookupTabNodeIds(kTag3, &result);
254 EXPECT_TRUE(result.empty()); 252 EXPECT_TRUE(result.empty());
255 EXPECT_FALSE(GetTracker()->DeleteSession(tag3)); 253 EXPECT_FALSE(GetTracker()->DeleteSession(kTag3));
256 GetTracker()->LookupTabNodeIds(tag3, &result); 254 GetTracker()->LookupTabNodeIds(kTag3, &result);
257 EXPECT_TRUE(result.empty()); 255 EXPECT_TRUE(result.empty());
258 256
259 EXPECT_FALSE(GetTracker()->DeleteSession(tag1)); 257 EXPECT_FALSE(GetTracker()->DeleteSession(kTag));
260 GetTracker()->LookupTabNodeIds(tag1, &result); 258 GetTracker()->LookupTabNodeIds(kTag, &result);
261 EXPECT_TRUE(result.empty()); 259 EXPECT_TRUE(result.empty());
262 GetTracker()->LookupTabNodeIds(tag2, &result); 260 GetTracker()->LookupTabNodeIds(kTag2, &result);
263 EXPECT_EQ(2U, result.size()); 261 EXPECT_EQ(2U, result.size());
264 EXPECT_FALSE(result.end() == result.find(21)); 262 EXPECT_FALSE(result.end() == result.find(21));
265 EXPECT_FALSE(result.end() == result.find(22)); 263 EXPECT_FALSE(result.end() == result.find(22));
266 264
267 GetTracker()->GetTab(tag2, 1, 21); 265 GetTracker()->GetTab(kTag2, 1, 21);
268 GetTracker()->GetTab(tag2, 2, 23); 266 GetTracker()->GetTab(kTag2, 2, 23);
269 GetTracker()->LookupTabNodeIds(tag2, &result); 267 GetTracker()->LookupTabNodeIds(kTag2, &result);
270 EXPECT_EQ(3U, result.size()); 268 EXPECT_EQ(3U, result.size());
271 EXPECT_FALSE(result.end() == result.find(21)); 269 EXPECT_FALSE(result.end() == result.find(21));
272 EXPECT_FALSE(result.end() == result.find(22)); 270 EXPECT_FALSE(result.end() == result.find(22));
273 EXPECT_FALSE(result.end() == result.find(23)); 271 EXPECT_FALSE(result.end() == result.find(23));
274 272
275 EXPECT_FALSE(GetTracker()->DeleteSession(tag2)); 273 EXPECT_FALSE(GetTracker()->DeleteSession(kTag2));
276 GetTracker()->LookupTabNodeIds(tag2, &result); 274 GetTracker()->LookupTabNodeIds(kTag2, &result);
277 EXPECT_TRUE(result.empty()); 275 EXPECT_TRUE(result.empty());
278 } 276 }
279 277
280 TEST_F(SyncedSessionTrackerTest, SessionTracking) { 278 TEST_F(SyncedSessionTrackerTest, SessionTracking) {
281 ASSERT_TRUE(GetTracker()->Empty()); 279 ASSERT_TRUE(GetTracker()->Empty());
282 std::string tag1 = "tag1";
283 std::string tag2 = "tag2";
284 280
285 // Create some session information that is stale. 281 // Create some session information that is stale.
286 SyncedSession* session1 = GetTracker()->GetSession(tag1); 282 SyncedSession* session1 = GetTracker()->GetSession(kTag);
287 GetTracker()->PutWindowInSession(tag1, 0); 283 GetTracker()->PutWindowInSession(kTag, 0);
288 GetTracker()->PutTabInWindow(tag1, 0, 0, 0); 284 GetTracker()->PutTabInWindow(kTag, 0, 0, 0);
289 GetTracker()->PutTabInWindow(tag1, 0, 1, 1); 285 GetTracker()->PutTabInWindow(kTag, 0, 1, 1);
290 GetTracker()->GetTab(tag1, 2, 3U)->window_id.set_id(0); // Will be unmapped. 286 GetTracker()->GetTab(kTag, 2, 3U)->window_id.set_id(0); // Will be unmapped.
291 GetTracker()->GetTab(tag1, 3, 4U)->window_id.set_id(0); // Will be unmapped. 287 GetTracker()->GetTab(kTag, 3, 4U)->window_id.set_id(0); // Will be unmapped.
292 GetTracker()->PutWindowInSession(tag1, 1); 288 GetTracker()->PutWindowInSession(kTag, 1);
293 GetTracker()->PutTabInWindow(tag1, 1, 4, 0); 289 GetTracker()->PutTabInWindow(kTag, 1, 4, 0);
294 GetTracker()->PutTabInWindow(tag1, 1, 5, 1); 290 GetTracker()->PutTabInWindow(kTag, 1, 5, 1);
295 ASSERT_EQ(2U, session1->windows.size()); 291 ASSERT_EQ(2U, session1->windows.size());
296 ASSERT_EQ(2U, session1->windows[0]->tabs.size()); 292 ASSERT_EQ(2U, session1->windows[0]->tabs.size());
297 ASSERT_EQ(2U, session1->windows[1]->tabs.size()); 293 ASSERT_EQ(2U, session1->windows[1]->tabs.size());
298 ASSERT_EQ(6U, GetTracker()->num_synced_tabs(tag1)); 294 ASSERT_EQ(6U, GetTracker()->num_synced_tabs(kTag));
299 295
300 // Create a session that should not be affected. 296 // Create a session that should not be affected.
301 SyncedSession* session2 = GetTracker()->GetSession(tag2); 297 SyncedSession* session2 = GetTracker()->GetSession(kTag2);
302 GetTracker()->PutWindowInSession(tag2, 2); 298 GetTracker()->PutWindowInSession(kTag2, 2);
303 GetTracker()->PutTabInWindow(tag2, 2, 1, 0); 299 GetTracker()->PutTabInWindow(kTag2, 2, 1, 0);
304 ASSERT_EQ(1U, session2->windows.size()); 300 ASSERT_EQ(1U, session2->windows.size());
305 ASSERT_EQ(1U, session2->windows[2]->tabs.size()); 301 ASSERT_EQ(1U, session2->windows[2]->tabs.size());
306 ASSERT_EQ(1U, GetTracker()->num_synced_tabs(tag2)); 302 ASSERT_EQ(1U, GetTracker()->num_synced_tabs(kTag2));
307 303
308 // Reset tracking and get the current windows/tabs. 304 // Reset tracking and get the current windows/tabs.
309 // We simulate moving a tab from one window to another, then closing the 305 // We simulate moving a tab from one window to another, then closing the
310 // first window (including its one remaining tab), and opening a new tab 306 // first window (including its one remaining tab), and opening a new tab
311 // on the remaining window. 307 // on the remaining window.
312 308
313 // New tab, arrived before meta node so unmapped. 309 // New tab, arrived before meta node so unmapped.
314 GetTracker()->GetTab(tag1, 6, 7U); 310 GetTracker()->GetTab(kTag, 6, 7U);
315 GetTracker()->ResetSessionTracking(tag1); 311 GetTracker()->ResetSessionTracking(kTag);
316 GetTracker()->PutWindowInSession(tag1, 0); 312 GetTracker()->PutWindowInSession(kTag, 0);
317 GetTracker()->PutTabInWindow(tag1, 0, 0, 0); 313 GetTracker()->PutTabInWindow(kTag, 0, 0, 0);
318 // Tab 1 is closed. 314 // Tab 1 is closed.
319 GetTracker()->PutTabInWindow(tag1, 0, 2, 1); // No longer unmapped. 315 GetTracker()->PutTabInWindow(kTag, 0, 2, 1); // No longer unmapped.
320 // Tab 3 was unmapped and does not get used. 316 // Tab 3 was unmapped and does not get used.
321 GetTracker()->PutTabInWindow(tag1, 0, 4, 2); // Moved from window 1. 317 GetTracker()->PutTabInWindow(kTag, 0, 4, 2); // Moved from window 1.
322 // Window 1 was closed, along with tab 5. 318 // Window 1 was closed, along with tab 5.
323 GetTracker()->PutTabInWindow(tag1, 0, 6, 3); // No longer unmapped. 319 GetTracker()->PutTabInWindow(kTag, 0, 6, 3); // No longer unmapped.
324 // Session 2 should not be affected. 320 // Session 2 should not be affected.
325 GetTracker()->CleanupSession(tag1); 321 GetTracker()->CleanupSession(kTag);
326 322
327 // Verify that only those parts of the session not owned have been removed. 323 // Verify that only those parts of the session not owned have been removed.
328 ASSERT_EQ(1U, session1->windows.size()); 324 ASSERT_EQ(1U, session1->windows.size());
329 ASSERT_EQ(4U, session1->windows[0]->tabs.size()); 325 ASSERT_EQ(4U, session1->windows[0]->tabs.size());
330 ASSERT_EQ(1U, session2->windows.size()); 326 ASSERT_EQ(1U, session2->windows.size());
331 ASSERT_EQ(1U, session2->windows[2]->tabs.size()); 327 ASSERT_EQ(1U, session2->windows[2]->tabs.size());
332 ASSERT_EQ(2U, GetTracker()->num_synced_sessions()); 328 ASSERT_EQ(2U, GetTracker()->num_synced_sessions());
333 ASSERT_EQ(4U, GetTracker()->num_synced_tabs(tag1)); 329 ASSERT_EQ(4U, GetTracker()->num_synced_tabs(kTag));
334 ASSERT_EQ(1U, GetTracker()->num_synced_tabs(tag2)); 330 ASSERT_EQ(1U, GetTracker()->num_synced_tabs(kTag2));
335 331
336 // All memory should be properly deallocated by destructor for the 332 // All memory should be properly deallocated by destructor for the
337 // SyncedSessionTracker. 333 // SyncedSessionTracker.
338 } 334 }
339 335
340 TEST_F(SyncedSessionTrackerTest, DeleteForeignTab) { 336 TEST_F(SyncedSessionTrackerTest, DeleteForeignTab) {
341 std::string session_tag = "session_tag";
342 int tab_id_1 = 1; 337 int tab_id_1 = 1;
343 int tab_id_2 = 2; 338 int tab_id_2 = 2;
344 int tab_node_id_3 = 3; 339 int tab_node_id_3 = 3;
345 int tab_node_id_4 = 4; 340 int tab_node_id_4 = 4;
346 std::set<int> result; 341 std::set<int> result;
347 342
348 GetTracker()->GetTab(session_tag, tab_id_1, tab_node_id_3); 343 GetTracker()->GetTab(kTag, tab_id_1, tab_node_id_3);
349 GetTracker()->GetTab(session_tag, tab_id_1, tab_node_id_4); 344 GetTracker()->GetTab(kTag, tab_id_1, tab_node_id_4);
350 GetTracker()->GetTab(session_tag, tab_id_2, tab_node_id_3); 345 GetTracker()->GetTab(kTag, tab_id_2, tab_node_id_3);
351 GetTracker()->GetTab(session_tag, tab_id_2, tab_node_id_4); 346 GetTracker()->GetTab(kTag, tab_id_2, tab_node_id_4);
352 347
353 GetTracker()->LookupTabNodeIds(session_tag, &result); 348 GetTracker()->LookupTabNodeIds(kTag, &result);
354 EXPECT_EQ(2U, result.size()); 349 EXPECT_EQ(2U, result.size());
355 EXPECT_TRUE(result.find(tab_node_id_3) != result.end()); 350 EXPECT_TRUE(result.find(tab_node_id_3) != result.end());
356 EXPECT_TRUE(result.find(tab_node_id_4) != result.end()); 351 EXPECT_TRUE(result.find(tab_node_id_4) != result.end());
357 352
358 GetTracker()->DeleteForeignTab(session_tag, tab_node_id_3); 353 GetTracker()->DeleteForeignTab(kTag, tab_node_id_3);
359 GetTracker()->LookupTabNodeIds(session_tag, &result); 354 GetTracker()->LookupTabNodeIds(kTag, &result);
360 EXPECT_EQ(1U, result.size()); 355 EXPECT_EQ(1U, result.size());
361 EXPECT_TRUE(result.find(tab_node_id_4) != result.end()); 356 EXPECT_TRUE(result.find(tab_node_id_4) != result.end());
362 357
363 GetTracker()->DeleteForeignTab(session_tag, tab_node_id_4); 358 GetTracker()->DeleteForeignTab(kTag, tab_node_id_4);
364 GetTracker()->LookupTabNodeIds(session_tag, &result); 359 GetTracker()->LookupTabNodeIds(kTag, &result);
365 EXPECT_TRUE(result.empty()); 360 EXPECT_TRUE(result.empty());
366 } 361 }
367 362
363 TEST_F(SyncedSessionTrackerTest, ReassociateTabMapped) {
364 const int kWindow1 = 1;
365 const int kTab1 = 15;
366 const int kTab2 = 25;
367 const int kTabNodeId = 1;
368
369 // First create the tab normally.
370 GetTracker()->PutWindowInSession(kTag, kWindow1);
371 GetTracker()->PutTabInWindow(kTag, kWindow1, kTab1, 0);
372 SyncedSession* session = GetTracker()->GetSession(kTag);
373 sessions::SessionTab* tab = session->windows[kWindow1]->tabs[0].get();
374 ASSERT_EQ(1U, session->windows.size());
375 ASSERT_EQ(1U, session->windows[kWindow1]->tabs.size());
376 ASSERT_EQ(GetTracker()->GetTab(kTag, kTab1, 1),
377 session->windows[kWindow1]->tabs[0].get());
378
379 // Then reassociate with a new tab id.
380 ASSERT_TRUE(GetTracker()->ReassociateTab(kTag, kTab1, kTab2));
381 ASSERT_EQ(tab, GetTracker()->GetTab(kTag, kTab2, kTabNodeId));
382
383 // Reset tracking, and put the new tab id into the window.
384 GetTracker()->ResetSessionTracking(kTag);
385 GetTracker()->PutWindowInSession(kTag, kWindow1);
386 GetTracker()->PutTabInWindow(kTag, kWindow1, kTab2, 0);
387 GetTracker()->CleanupSession(kTag);
388 ASSERT_EQ(GetTracker()->GetTab(kTag, kTab2, kTabNodeId),
389 session->windows[kWindow1]->tabs[0].get());
390 }
391
392 TEST_F(SyncedSessionTrackerTest, ReassociateTabUnmapped) {
393 const int kWindow1 = 1;
394 const int kTab1 = 15;
395 const int kTab2 = 25;
396 const int kTabNodeId = 1;
397
398 // First create the old tab in an unmapped state.
399 ASSERT_TRUE(GetTracker()->GetTab(kTag, kTab1, kTabNodeId));
400
401 // Then handle loading the window and reassociating the tab.
402 GetTracker()->ResetSessionTracking(kTag);
403 ASSERT_TRUE(GetTracker()->ReassociateTab(kTag, kTab1, kTab2));
404 GetTracker()->PutWindowInSession(kTag, kWindow1);
405 GetTracker()->PutTabInWindow(kTag, kWindow1, kTab2, 0);
406 GetTracker()->CleanupSession(kTag);
407
408 // Then reassociate with a new tab id. It should be accessible both via the
409 // GetSession as well as GetTab.
410 SyncedSession* session = GetTracker()->GetSession(kTag);
411 ASSERT_EQ(GetTracker()->GetTab(kTag, kTab2, 1),
412 session->windows[kWindow1]->tabs[0].get());
413 }
414
368 } // namespace sync_sessions 415 } // namespace sync_sessions
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698