Add Resume logic of Purge+Suspend to TabManager.

chrome/browser/memory/tab_manager_unittest.cc

 // Copyright (c) 2012 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 "chrome/browser/memory/tab_manager.h"
 
 #include <algorithm>
 #include <map>
 #include <vector>
 
@@ skipping 653 matching lines @@
   EXPECT_FALSE(tm.under_memory_pressure_);
   EXPECT_EQ(0u, task_runner->size());
   EXPECT_EQ(0u, tm.notified_renderers_.size());
 
 
   // Clean up the tabstrip.
   tabstrip.CloseAllTabs();
   ASSERT_TRUE(tabstrip.empty());
 }
 
+TEST_F(TabManagerTest, NextPurgeAndSuspendState) {
+  TabManager tab_manager;
+  TabStripDummyDelegate delegate;
+  TabStripModel tabstrip(&delegate, profile());
+  tabstrip.AddObserver(&tab_manager);
+
+  WebContents* test_contents = CreateWebContents();
+  tabstrip.AppendWebContents(test_contents, false);
+
+  base::TimeDelta threshold = base::TimeDelta::FromSeconds(180);
+  base::SimpleTestTickClock test_clock;
+
+  tab_manager.GetWebContentsData(test_contents)
+      ->SetPurgeAndSuspendState(TabManager::RUNNING);
+  tab_manager.GetWebContentsData(test_contents)
+      ->SetLastPurgeAndSuspendModifiedTimeForTesting(test_clock.NowTicks());
+
+  test_clock.Advance(base::TimeDelta::FromSeconds(180));
+  EXPECT_EQ(TabManager::RUNNING,
+            tab_manager.GetNextPurgeAndSuspendState(
+                test_contents, test_clock.NowTicks(), threshold));
+
+  test_clock.Advance(base::TimeDelta::FromSeconds(1));
+  EXPECT_EQ(TabManager::SUSPENDED,
+            tab_manager.GetNextPurgeAndSuspendState(
+                test_contents, test_clock.NowTicks(), threshold));
+
+  tab_manager.GetWebContentsData(test_contents)
+      ->SetPurgeAndSuspendState(TabManager::SUSPENDED);
+  tab_manager.GetWebContentsData(test_contents)
+      ->SetLastPurgeAndSuspendModifiedTimeForTesting(test_clock.NowTicks());
+
+  test_clock.Advance(base::TimeDelta::FromSeconds(120));
+  EXPECT_EQ(TabManager::SUSPENDED,
+            tab_manager.GetNextPurgeAndSuspendState(
+                test_contents, test_clock.NowTicks(), threshold));
+
+  test_clock.Advance(base::TimeDelta::FromSeconds(1));
+  EXPECT_EQ(TabManager::RESUMED,
+            tab_manager.GetNextPurgeAndSuspendState(
+                test_contents, test_clock.NowTicks(), threshold));
+
+  tab_manager.GetWebContentsData(test_contents)
+      ->SetPurgeAndSuspendState(TabManager::RESUMED);
+  tab_manager.GetWebContentsData(test_contents)
+      ->SetLastPurgeAndSuspendModifiedTimeForTesting(test_clock.NowTicks());
+
+  test_clock.Advance(base::TimeDelta::FromSeconds(10));
+  EXPECT_EQ(TabManager::RESUMED,
+            tab_manager.GetNextPurgeAndSuspendState(
+                test_contents, test_clock.NowTicks(), threshold));
+
+  test_clock.Advance(base::TimeDelta::FromSeconds(1));
+  EXPECT_EQ(TabManager::SUSPENDED,
+            tab_manager.GetNextPurgeAndSuspendState(
+                test_contents, test_clock.NowTicks(), threshold));
+
+  // Clean up the tabstrip.
+  tabstrip.CloseAllTabs();
+  EXPECT_TRUE(tabstrip.empty());
+}
+
+TEST_F(TabManagerTest, ActivateTabResetPurgeAndSuspendState) {
+  TabManager tab_manager;
+  TabStripDummyDelegate delegate;
+  TabStripModel tabstrip(&delegate, profile());
+  tabstrip.AddObserver(&tab_manager);
+
+  WebContents* tab1 = CreateWebContents();
+  WebContents* tab2 = CreateWebContents();
+  tabstrip.AppendWebContents(tab1, true);
+  tabstrip.AppendWebContents(tab2, false);
+
+  base::SimpleTestTickClock test_clock;
+
+  // Initially PurgeAndSuspend state should be RUNNING.
+  EXPECT_EQ(TabManager::RUNNING,
+            tab_manager.GetWebContentsData(tab2)->GetPurgeAndSuspendState());
+
+  tab_manager.GetWebContentsData(tab2)->SetPurgeAndSuspendState(
+      TabManager::SUSPENDED);
+  tab_manager.GetWebContentsData(tab2)
+      ->SetLastPurgeAndSuspendModifiedTimeForTesting(test_clock.NowTicks());
+
+  // Activate tab2. Tab2's PurgeAndSuspend state should be RUNNING.
+  tabstrip.ActivateTabAt(1, true /* user_gesture */);
+  EXPECT_EQ(TabManager::RUNNING,
+            tab_manager.GetWebContentsData(tab2)->GetPurgeAndSuspendState());
+
+  tab_manager.GetWebContentsData(tab1)->SetPurgeAndSuspendState(
+      TabManager::RESUMED);
+  tab_manager.GetWebContentsData(tab1)
+      ->SetLastPurgeAndSuspendModifiedTimeForTesting(test_clock.NowTicks());
+
+  // Activate tab1. Tab1's PurgeAndSuspend state should be RUNNING.
+  tabstrip.ActivateTabAt(0, true /* user_gesture */);
+  EXPECT_EQ(TabManager::RUNNING,
+            tab_manager.GetWebContentsData(tab1)->GetPurgeAndSuspendState());
+
+  // Clean up the tabstrip.
+  tabstrip.CloseAllTabs();
+  EXPECT_TRUE(tabstrip.empty());
+}
+
 }  // namespace memory