// 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 "components/sync_sessions/synced_tab_delegate.h"
#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 SessionID kTab1 = SessionID::FromSerializedValue(15);
const SessionID kTab2 = SessionID::FromSerializedValue(25);
}  // namespace

TEST(TaskTrackerTest, GetTabTasks) {
  TaskTracker task_tracker;
  TabTasks* tab_tasks =
      task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue());
  ASSERT_NE(tab_tasks, nullptr);
  EXPECT_EQ(task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue()),
            tab_tasks);
  EXPECT_NE(task_tracker.GetTabTasks(kTab2, SessionID::InvalidValue()),
            tab_tasks);
}

TEST(TaskTrackerTest, CleanTabTasks) {
  TaskTracker task_tracker;
  TabTasks* tab_tasks =
      task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue());
  ASSERT_NE(tab_tasks, nullptr);
  ASSERT_FALSE(task_tracker.local_tab_tasks_map_.empty());

  task_tracker.CleanTabTasks(kTab1);
  EXPECT_TRUE(task_tracker.local_tab_tasks_map_.empty());
}

TEST(TaskTrackerTest, UpdateTasksWithMultipleClicks) {
  TaskTracker task_tracker;
  TabTasks* tab_tasks =
      task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue());

  tab_tasks->UpdateWithNavigation(1, ui::PageTransition::PAGE_TRANSITION_TYPED,
                                  100);
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(1), ElementsAre(100));

  tab_tasks->UpdateWithNavigation(2, ui::PageTransition::PAGE_TRANSITION_LINK,
                                  200);
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(1), ElementsAre(100));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(2), ElementsAre(100, 200));

  tab_tasks->UpdateWithNavigation(3, ui::PageTransition::PAGE_TRANSITION_LINK,
                                  300);
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(1), ElementsAre(100));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(2), ElementsAre(100, 200));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(3),
              ElementsAre(100, 200, 300));
}

TEST(TaskTrackerTest, UpdateTasksWithMultipleClicksAndTypes) {
  TaskTracker task_tracker;
  TabTasks* tab_tasks =
      task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue());

  tab_tasks->UpdateWithNavigation(1, ui::PAGE_TRANSITION_LINK, 100);
  tab_tasks->UpdateWithNavigation(2, ui::PAGE_TRANSITION_LINK, 200);
  tab_tasks->UpdateWithNavigation(3, ui::PAGE_TRANSITION_LINK, 300);
  tab_tasks->UpdateWithNavigation(4, ui::PAGE_TRANSITION_TYPED, 400);
  tab_tasks->UpdateWithNavigation(5, ui::PAGE_TRANSITION_LINK, 500);
  tab_tasks->UpdateWithNavigation(6, ui::PAGE_TRANSITION_TYPED, 600);
  tab_tasks->UpdateWithNavigation(7, ui::PAGE_TRANSITION_LINK, 700);

  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(1), ElementsAre(100));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(2), ElementsAre(100, 200));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(3),
              ElementsAre(100, 200, 300));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(4), ElementsAre(400));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(5), ElementsAre(400, 500));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(6), ElementsAre(600));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(7), ElementsAre(600, 700));
}

TEST(TaskTrackerTest, UpdateTasksWithBackforwards) {
  TaskTracker task_tracker;
  TabTasks* tab_tasks =
      task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue());

  tab_tasks->UpdateWithNavigation(1, ui::PAGE_TRANSITION_TYPED, 100);
  tab_tasks->UpdateWithNavigation(2, ui::PAGE_TRANSITION_LINK, 200);
  tab_tasks->UpdateWithNavigation(3, ui::PAGE_TRANSITION_LINK, 300);

  tab_tasks->UpdateWithNavigation(
      1,
      ui::PageTransitionFromInt(ui::PAGE_TRANSITION_TYPED |
                                ui::PAGE_TRANSITION_FORWARD_BACK),
      400);
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(1), ElementsAre(100));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(2), ElementsAre(100, 200));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(3),
              ElementsAre(100, 200, 300));

  tab_tasks->UpdateWithNavigation(
      3,
      ui::PageTransitionFromInt(ui::PAGE_TRANSITION_LINK |
                                ui::PAGE_TRANSITION_FORWARD_BACK),
      500);
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(1), ElementsAre(100));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(2), ElementsAre(100, 200));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(3),
              ElementsAre(100, 200, 300));
}

TEST(TaskTrackerTest, UpdateWithNavigationsWithBackAndForkedNavigation) {
  TaskTracker task_tracker;
  TabTasks* tab_tasks =
      task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue());
  tab_tasks->UpdateWithNavigation(1, ui::PAGE_TRANSITION_LINK, 100);
  tab_tasks->UpdateWithNavigation(2, ui::PAGE_TRANSITION_LINK, 200);
  tab_tasks->UpdateWithNavigation(3, ui::PAGE_TRANSITION_LINK, 300);
  tab_tasks->UpdateWithNavigation(
      1,
      ui::PageTransitionFromInt(ui::PAGE_TRANSITION_LINK |
                                ui::PAGE_TRANSITION_FORWARD_BACK),
      400);
  tab_tasks->UpdateWithNavigation(4, ui::PAGE_TRANSITION_LINK, 500);
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(1), ElementsAre(100));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(4), ElementsAre(100, 500));
}

TEST(TaskTrackerTest, LimitMaxNumberOfTasksPerTab) {
  TaskTracker task_tracker;
  TabTasks* tab_tasks =
      task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue());

  // Reaching max number of tasks for a tab.
  for (int i = 1; i <= kMaxNumTasksPerTab; i++) {
    tab_tasks->UpdateWithNavigation(i, ui::PAGE_TRANSITION_LINK, i * 100);
  }

  tab_tasks->UpdateWithNavigation(kMaxNumTasksPerTab, ui::PAGE_TRANSITION_LINK,
                                  kMaxNumTasksPerTab * 100);

  ASSERT_THAT(tab_tasks->nav_to_task_id_map_, SizeIs(kMaxNumTasksPerTab));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(0), ElementsAre());
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(1), ElementsAre(100));
  std::vector<int64_t> task_ids =
      tab_tasks->GetTaskIdsForNavigation(kMaxNumTasksPerTab);
  EXPECT_THAT(task_ids, SizeIs(kMaxNumTasksPerTab));
  EXPECT_EQ(task_ids[0], 100);
  EXPECT_EQ(task_ids[kMaxNumTasksPerTab - 1], kMaxNumTasksPerTab * 100);

  tab_tasks->UpdateWithNavigation(kMaxNumTasksPerTab + 1,
                                  ui::PAGE_TRANSITION_LINK,
                                  (kMaxNumTasksPerTab + 1) * 100);

  ASSERT_THAT(tab_tasks->nav_to_task_id_map_, SizeIs(kMaxNumTasksPerTab));
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(0), ElementsAre());
  // Navigation 1 is the root navigation. It is preserved.
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(1), ElementsAre(100));
  // Navigation 2 is oldest non-root naviation. It is evicted.
  EXPECT_THAT(tab_tasks->GetTaskIdsForNavigation(2), ElementsAre());
  task_ids = tab_tasks->GetTaskIdsForNavigation(kMaxNumTasksPerTab + 1);
  // Navigation chain starts at navigation 3 since navigation 2 was evicted.
  EXPECT_THAT(task_ids, SizeIs(kMaxNumTasksPerTab - 1));
  EXPECT_EQ(task_ids.front(), 300);
  EXPECT_EQ(task_ids.back(), (kMaxNumTasksPerTab + 1) * 100);
}

// Regression test for crbug/766963. Tests that evicting and reinserting root
// navigation doesn't create loop in navigation chain.
TEST(TaskTrackerTest, LoopInNavigationChain) {
  TaskTracker task_tracker;
  TabTasks* tasks = task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue());
  // Create a few navigations that would later be evicted.
  tasks->UpdateWithNavigation(1, ui::PageTransition::PAGE_TRANSITION_LINK, 1);
  tasks->UpdateWithNavigation(2, ui::PageTransition::PAGE_TRANSITION_LINK, 2);
  tasks->UpdateWithNavigation(3, ui::PageTransition::PAGE_TRANSITION_LINK, 3);
  // Continue task from navigation 2.
  tasks->UpdateWithNavigation(2, ui::PageTransition::PAGE_TRANSITION_LINK, 4);
  // Create large number of navigations to get navigation 1 evicted.
  for (int i = 0; i < kMaxNumTasksPerTab - 2; i++) {
    tasks->UpdateWithNavigation(
        i + 10, ui::PageTransition::PAGE_TRANSITION_LINK, i + 10);
  }
  // Revisit navigation 1 to get it recreated.
  tasks->UpdateWithNavigation(1, ui::PageTransition::PAGE_TRANSITION_LINK,
                              1000);

  // If recreation of navigation 1 caused loop then the next call would either
  // DCHECK or OOM.
  tasks->GetTaskIdsForNavigation(1);
}

TEST(TaskTrackerTest, CreateTabTasksFromSourceTab) {
  TaskTracker task_tracker;
  TabTasks* source_tab_tasks =
      task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue());
  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);
  EXPECT_EQ(kTab1, target_tab_tasks->parent_tab_id());
  target_tab_tasks->UpdateWithNavigation(6, ui::PAGE_TRANSITION_LINK, 600);
  target_tab_tasks->UpdateWithNavigation(7, ui::PAGE_TRANSITION_LINK, 700);
  target_tab_tasks->UpdateWithNavigation(8, ui::PAGE_TRANSITION_TYPED, 800);

  EXPECT_THAT(target_tab_tasks->GetTaskIdsForNavigation(6),
              ElementsAre(300, 400, 500, 600));
  EXPECT_THAT(target_tab_tasks->GetTaskIdsForNavigation(7),
              ElementsAre(300, 400, 500, 600, 700));
  EXPECT_THAT(target_tab_tasks->GetTaskIdsForNavigation(8), ElementsAre(800));
}

TEST(TaskTrackerTest, CreateTabTasksFromSourceTabAfterGoingBack) {
  TaskTracker task_tracker;
  TabTasks* source_tab_tasks =
      task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue());
  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);
  EXPECT_EQ(kTab1, target_tab_tasks->parent_tab_id());
  target_tab_tasks->UpdateWithNavigation(7, ui::PAGE_TRANSITION_LINK, 700);

  EXPECT_THAT(target_tab_tasks->GetTaskIdsForNavigation(7),
              ElementsAre(100, 200, 700));
}

TEST(TaskTrackerTest, CreateTabTasksFromSourceTabWithLimitedTaskNum) {
  TaskTracker task_tracker;

  TabTasks* source_tab_tasks =
      task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue());
  // Adding max number of tasks to tab1.
  int nav_id = 1;
  for (; nav_id <= kMaxNumTasksPerTab; nav_id++) {
    source_tab_tasks->UpdateWithNavigation(nav_id, ui::PAGE_TRANSITION_LINK,
                                           nav_id * 100);
  }

  TabTasks* tab_tasks = task_tracker.GetTabTasks(kTab2, kTab1);
  EXPECT_EQ(kTab1, tab_tasks->parent_tab_id());
  tab_tasks->UpdateWithNavigation(nav_id, ui::PAGE_TRANSITION_LINK,
                                  nav_id * 100);

  std::vector<int64_t> task_ids = tab_tasks->GetTaskIdsForNavigation(nav_id);
  // The first task from source tab (id 100) is excluded because of max task
  // number limit, so start at id 200.
  int expected_task_ids[kMaxNumTasksPerTab];
  for (int i = 0; i < kMaxNumTasksPerTab; i++)
    expected_task_ids[i] = (i + 2) * 100;
  EXPECT_THAT(task_ids,
              ElementsAreArray(expected_task_ids, kMaxNumTasksPerTab));
}

TEST(TaskTrackerTest, GetTabTasksWithNewSource) {
  TaskTracker task_tracker;
  TabTasks* target_tab_tasks =
      task_tracker.GetTabTasks(kTab2, SessionID::InvalidValue());
  EXPECT_EQ(SessionID::InvalidValue(), target_tab_tasks->parent_tab_id());

  TabTasks* source_tab_tasks =
      task_tracker.GetTabTasks(kTab1, SessionID::InvalidValue());
  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);

  target_tab_tasks = task_tracker.GetTabTasks(kTab2, kTab1);
  EXPECT_EQ(kTab1, target_tab_tasks->parent_tab_id());
  target_tab_tasks->UpdateWithNavigation(6, ui::PAGE_TRANSITION_LINK, 600);
  target_tab_tasks->UpdateWithNavigation(7, ui::PAGE_TRANSITION_LINK, 700);
  target_tab_tasks->UpdateWithNavigation(8, ui::PAGE_TRANSITION_TYPED, 800);

  EXPECT_THAT(target_tab_tasks->GetTaskIdsForNavigation(6),
              ElementsAre(300, 400, 500, 600));
  EXPECT_THAT(target_tab_tasks->GetTaskIdsForNavigation(7),
              ElementsAre(300, 400, 500, 600, 700));
  EXPECT_THAT(target_tab_tasks->GetTaskIdsForNavigation(8), ElementsAre(800));
}

}  // namespace sync_sessions