Track task ids for navigations cross multiple tabs.

components/sync_sessions/sessions_sync_manager_unittest.cc

 // Copyright 2014 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/sessions_sync_manager.h"
 
 #include <stdint.h>
 
 #include <utility>
 
@@ skipping 21 matching lines @@
 using sessions::SerializedNavigationEntryTestHelper;
 using syncer::DeviceInfo;
 using syncer::LocalDeviceInfoProvider;
 using syncer::LocalDeviceInfoProviderMock;
 using syncer::SyncChange;
 using syncer::SyncChangeList;
 using syncer::SyncData;
 using syncer::SyncDataList;
 using syncer::SyncDataLocal;
 using syncer::SyncError;
+using testing::ElementsAre;
 
 namespace sync_sessions {
 
 namespace {
 
 const char kTitle[] = "title";
 const char kFoo1[] = "http://foo1/";
 const char kFoo2[] = "http://foo2/";
 const char kBar1[] = "http://bar1/";
 const char kBar2[] = "http://bar2/";
@@ skipping 152 matching lines @@
     int http_count = 0;
     for (auto& entry : entries_) {
       if (!entry->virtual_url().is_valid())
         return false;
       if (entry->virtual_url().SchemeIsHTTPOrHTTPS())
         http_count++;
     }
     return http_count > 0;
   }
 
-  SessionID::id_type GetSourceTabID() const override { return kInvalidTabID; }
+  SessionID::id_type GetSourceTabID() const override { return source_tab_id_; }
+  void SetSourceTabID(SessionID::id_type source_tab_id) {
+    source_tab_id_ = source_tab_id;
+  }
 
   void AppendEntry(std::unique_ptr<sessions::SerializedNavigationEntry> entry) {
     entries_.push_back(std::move(entry));
   }
 
   void set_current_entry_index(int i) { current_entry_index_ = i; }
 
   void SetWindowId(SessionID::id_type window_id) {
     window_id_.set_id(window_id);
   }
@@ skipping 13 matching lines @@
     current_entry_index_ = 0;
     sync_id_ = TabNodePool::kInvalidTabNodeID;
     entries_.clear();
   }
 
  private:
   int current_entry_index_ = -1;
   bool is_supervised_ = false;
   int sync_id_ = kInvalidTabID;
   SessionID tab_id_;
+  SessionID::id_type source_tab_id_ = kInvalidTabID;
   SessionID window_id_;
   std::vector<std::unique_ptr<const sessions::SerializedNavigationEntry>>
       blocked_navigations_;
   std::vector<std::unique_ptr<const sessions::SerializedNavigationEntry>>
       entries_;
 };
 
 // A placeholder delegate. These delegates have no WebContents, simulating a tab
 // that has been restored without bringing its state fully into memory (for
 // example on Android), or where the tab's contents have been evicted from
@@ skipping 419 matching lines @@
 
   SyncData CreateRemoteData(const sync_pb::EntitySpecifics& entity,
                             base::Time mtime = base::Time()) const {
     // The server ID is never relevant to these tests, so just use 1.
     return SyncData::CreateRemoteData(
         1, entity, mtime, syncer::AttachmentIdList(),
         syncer::AttachmentServiceProxyForTest::Create(),
         SessionsSyncManager::TagHashFromSpecifics(entity.session()));
   }
 
-  // Creates a new tab within the window specified by |window_id|, and points it
-  // at |url|. Returns the newly created TestSyncedTabDelegate (not owned).
+  // Creates a new tab within the window specified by |window_id| and from the
+  // source tab |source_tab_id|, and points it at |url|. Returns the newly
+  // created TestSyncedTabDelegate (not owned).
   TestSyncedTabDelegate* AddTab(SessionID::id_type window_id,
                                 const std::string& url,
+                                SessionID::id_type source_tab_id,
                                 base::Time time) {
     tabs_.push_back(base::MakeUnique<TestSyncedTabDelegate>());
     for (auto& window : windows_) {
       if (window->GetSessionId() == window_id) {
         int tab_index = window->GetTabCount();
         window->OverrideTabAt(tab_index, tabs_.back().get());
       }
     }
+    tabs_.back()->SetSourceTabID(source_tab_id);
 
     // Simulate the browser firing a tab parented notification, ahead of the
     // actual navigation.
     router_->NotifyNav(tabs_.back().get());
 
     // Now do the actual navigation.
     NavigateTab(tabs_.back().get(), url, time);
     return tabs_.back().get();
   }
   TestSyncedTabDelegate* AddTab(SessionID::id_type window_id,
+                                const std::string& url,
+                                base::Time time) {
+    return AddTab(window_id, url, kInvalidTabID, time);
+  }
+  TestSyncedTabDelegate* AddTab(SessionID::id_type window_id,
+                                const std::string& url,
+                                SessionID::id_type source_tab_id) {
+    return AddTab(window_id, url, source_tab_id, base::Time::Now());
+  }
+  TestSyncedTabDelegate* AddTab(SessionID::id_type window_id,
                                 const std::string& url) {
-    return AddTab(window_id, url, base::Time::Now());
+    return AddTab(window_id, url, kInvalidTabID);
   }
 
   void NavigateTab(TestSyncedTabDelegate* delegate,
                    const std::string& url,
                    base::Time time,
                    ui::PageTransition transition) {
     auto entry = base::MakeUnique<sessions::SerializedNavigationEntry>(
         SerializedNavigationEntryTestHelper::CreateNavigation(url, kTitle));
     SerializedNavigationEntryTestHelper::SetTimestamp(time, entry.get());
     SerializedNavigationEntryTestHelper::SetTransitionType(transition,
@@ skipping 1866 matching lines @@
   tab = SyncDataLocal(changes[3].sync_data()).GetSpecifics().session().tab();
   EXPECT_EQ(tab.navigation_size(), 2);
   EXPECT_EQ(tab.navigation(0).global_id(), tab.navigation(0).task_id());
   EXPECT_TRUE(tab.navigation(0).ancestor_task_id().empty());
   EXPECT_EQ(tab.navigation(1).global_id(), tab.navigation(1).task_id());
   // navigation(1) is a subtask of navigation(0).
   EXPECT_EQ(tab.navigation(1).ancestor_task_id_size(), 1);
   EXPECT_EQ(tab.navigation(1).ancestor_task_id(0), tab.navigation(0).task_id());
 }
 
+// Tests that task ids are generated for navigations on multiple local tabs.
+TEST_F(SessionsSyncManagerTest, TrackTasksOnMultipleTabs) {
+  SyncChangeList changes;
+  TestSyncedWindowDelegate* window = AddWindow();
+  InitWithSyncDataTakeOutput(SyncDataList(), &changes);
+  SessionID::id_type window_id = window->GetSessionId();
+  ASSERT_FALSE(manager()->current_machine_tag().empty());
+  changes.clear();
+
+  // Tab 1
+  TestSyncedTabDelegate* tab1 = AddTab(window_id, kFoo1);
+  NavigateTab(tab1, kFoo2, ui::PAGE_TRANSITION_LINK);
+  // Tab 2 from Tab 1.
+  TestSyncedTabDelegate* tab2 = AddTab(window_id, kBar1, tab1->GetSessionId());
+  NavigateTab(tab2, kBar2, ui::PAGE_TRANSITION_LINK);
+  // Tab 3 from Tab 2.
+  TestSyncedTabDelegate* tab3 = AddTab(window_id, kBaz1, tab2->GetSessionId());
+  NavigateTab(tab3, kBaz2, ui::PAGE_TRANSITION_LINK);
+
+  // We only test changes for tab add and tab update, and ignore header updates.
+  FilterOutLocalHeaderChanges(&changes);
+  // Sync data of updating Tab 1 change
+  sync_pb::SessionTab sync_data_tab1 =
+      SyncDataLocal(changes[1].sync_data()).GetSpecifics().session().tab();
+  ASSERT_EQ(2, sync_data_tab1.navigation_size());
+
+  // Sync data of updating Tab 2 change
+  sync_pb::SessionTab sync_data_tab2 =
+      SyncDataLocal(changes[3].sync_data()).GetSpecifics().session().tab();
+  ASSERT_EQ(2, sync_data_tab1.navigation_size());
+
+  // Sync data of adding Tab 3 change
+  sync_pb::SessionTab sync_data_tab3 =
+      SyncDataLocal(changes[4].sync_data()).GetSpecifics().session().tab();
+  ASSERT_EQ(1, sync_data_tab3.navigation_size());
+  EXPECT_EQ(sync_data_tab3.navigation(0).global_id(),
+            sync_data_tab3.navigation(0).task_id());
+  EXPECT_THAT(sync_data_tab3.navigation(0).ancestor_task_id(),
+              ElementsAre(sync_data_tab1.navigation(0).task_id(),
+                          sync_data_tab1.navigation(1).task_id(),
+                          sync_data_tab2.navigation(0).task_id(),
+                          sync_data_tab2.navigation(1).task_id()));
+
+  // Sync data of adding Tab 3 change
+  sync_data_tab3 =
+      SyncDataLocal(changes[5].sync_data()).GetSpecifics().session().tab();
+  ASSERT_EQ(2, sync_data_tab3.navigation_size());
+  EXPECT_EQ(sync_data_tab3.navigation(0).global_id(),
+            sync_data_tab3.navigation(0).task_id());
+  EXPECT_THAT(sync_data_tab3.navigation(0).ancestor_task_id(),
+              ElementsAre(sync_data_tab1.navigation(0).task_id(),
+                          sync_data_tab1.navigation(1).task_id(),
+                          sync_data_tab2.navigation(0).task_id(),
+                          sync_data_tab2.navigation(1).task_id()));
+  EXPECT_EQ(sync_data_tab3.navigation(1).global_id(),
+            sync_data_tab3.navigation(1).task_id());
+  EXPECT_THAT(sync_data_tab3.navigation(1).ancestor_task_id(),
+              ElementsAre(sync_data_tab1.navigation(0).task_id(),
+                          sync_data_tab1.navigation(1).task_id(),
+                          sync_data_tab2.navigation(0).task_id(),
+                          sync_data_tab2.navigation(1).task_id(),
+                          sync_data_tab3.navigation(0).task_id()));
+}
+
 }  // namespace sync_sessions