Track task ids for navigations cross multiple tabs.

components/sync_sessions/task_tracker_unittest.cc

 // Copyright 2017 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 "components/sync_sessions/task_tracker.h"
 
 #include <utility>
 
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::ElementsAre;
+using testing::ElementsAreArray;
 using testing::SizeIs;
 
 namespace sync_sessions {
 
 namespace {
 const int kTab1 = 15;
 const int kTab2 = 25;
 }  // namespace
 
 TEST(TaskTrackerTest, GetTabTasks) {
@@ skipping 165 matching lines @@
 
   tab_tasks->UpdateWithNavigation(1, ui::PAGE_TRANSITION_LINK,
                                   (kMaxNumTasksPerTab + 2) * 100);
   ASSERT_THAT(tab_tasks->task_ids_, SizeIs(1));
   ASSERT_EQ(tab_tasks->GetNavigationsCount(), 2);
   EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(0), ElementsAre());
   EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(1),
               ElementsAre((kMaxNumTasksPerTab + 2) * 100));
 }
 
+TEST(TaskTrackerTest, CreateTabTasksFromSourceTab) {
+  TaskTracker task_tracker;
+  TabTasks* source_tab_tasks = task_tracker.GetTabTasks(kTab1);
+  source_tab_tasks->UpdateWithNavigation(1, ui::PAGE_TRANSITION_LINK, 100);
+  source_tab_tasks->UpdateWithNavigation(2, ui::PAGE_TRANSITION_LINK, 200);
+  source_tab_tasks->UpdateWithNavigation(3, ui::PAGE_TRANSITION_TYPED, 300);
+  source_tab_tasks->UpdateWithNavigation(4, ui::PAGE_TRANSITION_LINK, 400);
+  source_tab_tasks->UpdateWithNavigation(5, ui::PAGE_TRANSITION_LINK, 500);
+
+  TabTasks* target_tab_tasks = task_tracker.GetTabTasks(kTab2, kTab1);
+  target_tab_tasks->UpdateWithNavigation(1, ui::PAGE_TRANSITION_LINK, 600);
+  target_tab_tasks->UpdateWithNavigation(2, ui::PAGE_TRANSITION_LINK, 700);
+  target_tab_tasks->UpdateWithNavigation(3, ui::PAGE_TRANSITION_TYPED, 800);
+
+  EXPECT_THAT(target_tab_tasks->GetTaskIdsForNavigation(1),
+              ElementsAre(300, 400, 500, 600));
+  EXPECT_THAT(target_tab_tasks->GetTaskIdsForNavigation(2),
+              ElementsAre(300, 400, 500, 600, 700));
+  EXPECT_THAT(target_tab_tasks->GetTaskIdsForNavigation(3), ElementsAre(800));
+}
+
+TEST(TaskTrackerTest, CreateTabTasksFromSourceTabAfterGoingBack) {
+  TaskTracker task_tracker;
+  TabTasks* source_tab_tasks = task_tracker.GetTabTasks(kTab1);
+  source_tab_tasks->UpdateWithNavigation(1, ui::PAGE_TRANSITION_LINK, 100);
+  source_tab_tasks->UpdateWithNavigation(2, ui::PAGE_TRANSITION_LINK, 200);
+  source_tab_tasks->UpdateWithNavigation(3, ui::PAGE_TRANSITION_TYPED, 300);
+  source_tab_tasks->UpdateWithNavigation(4, ui::PAGE_TRANSITION_LINK, 400);
+  source_tab_tasks->UpdateWithNavigation(5, ui::PAGE_TRANSITION_LINK, 500);
+  source_tab_tasks->UpdateWithNavigation(
+      2,
+      ui::PageTransitionFromInt(ui::PAGE_TRANSITION_LINK |
+                                ui::PAGE_TRANSITION_FORWARD_BACK),
+      600);
+  ASSERT_THAT(source_tab_tasks->GetTaskIdsForNavigation(2),
+              ElementsAre(100, 200));
+
+  TabTasks* target_tab_tasks = task_tracker.GetTabTasks(kTab2, kTab1);
+  target_tab_tasks->UpdateWithNavigation(1, ui::PAGE_TRANSITION_LINK, 700);
+
+  EXPECT_THAT(target_tab_tasks->GetTaskIdsForNavigation(1),
+              ElementsAre(100, 200, 700));
+}
+
+TEST(TaskTrackerTest, CreateTabTasksFromSourceTabWithLimitedTaskNum) {
+  const int kMaxNumTasksPerTab = 100;
+  TaskTracker task_tracker;
+
+  TabTasks* source_tab_tasks = task_tracker.GetTabTasks(kTab1);
+  // Adding max number of tasks to tab1.
+  for (int i = 0; i < kMaxNumTasksPerTab; i++) {
+    source_tab_tasks->UpdateWithNavigation(i, ui::PAGE_TRANSITION_LINK,
+                                           i * 100);
+  }
+
+  TabTasks* tab_tasks = task_tracker.GetTabTasks(kTab2, kTab1);
+  tab_tasks->UpdateWithNavigation(0, ui::PAGE_TRANSITION_LINK,
+                                  kMaxNumTasksPerTab * 100);
+
+  std::vector<int64_t> task_ids = tab_tasks->GetTaskIdsForNavigation(0);
+  // The first task from source tab is excluded because of max task number
+  // limit.
+  int expected_task_ids[kMaxNumTasksPerTab];
+  for (int i = 0; i < kMaxNumTasksPerTab; i++)
+    expected_task_ids[i] = (i + 1) * 100;
+  EXPECT_THAT(task_ids,
+              ElementsAreArray(expected_task_ids, kMaxNumTasksPerTab));
+}
+
 }  // namespace sync_sessions