[TabManager] Unit tests to make sure of the number of times tabs can be discarded.

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>
 
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/strings/string16.h"
+#include "base/test/mock_entropy_provider.h"
 #include "base/test/simple_test_tick_clock.h"
 #include "base/time/time.h"
 #include "build/build_config.h"
 #include "chrome/browser/memory/tab_manager_web_contents_data.h"
 #include "chrome/browser/memory/tab_stats.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/browser/ui/tabs/test_tab_strip_model_delegate.h"
+#include "chrome/common/chrome_features.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
 #include "chrome/test/base/testing_profile.h"
+#include "components/variations/variations_associated_data.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/web_contents_tester.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 
 using content::WebContents;
 using content::WebContentsTester;
 
 namespace memory {
@@ skipping 392 matching lines @@
   test_contents->SetLastActiveTime(new_last_active_time);
   EXPECT_EQ(new_last_active_time, test_contents->GetLastActiveTime());
 
   WebContents* null_contents = tab_manager.DiscardWebContentsAt(1, &tabstrip);
   EXPECT_EQ(new_last_active_time, null_contents->GetLastActiveTime());
 
   tabstrip.CloseAllTabs();
   EXPECT_TRUE(tabstrip.empty());
 }
 
+// Test to see if a tab can only be discarded once. On Windows and Mac, this
+// defaults to true unless overridden through a variation parameter. On other
+// platforms, it's always false
+#if defined(OS_WIN) || defined(OS_MACOSX)
+TEST_F(TabManagerTest, CanOnlyDiscardOnce) {
+  TabManager tab_manager;
+  const std::string kTrialName = features::kAutomaticTabDiscarding.name;
+
+  // Not setting the variation parameter.
+  {
+    bool discard_once_value = tab_manager.CanOnlyDiscardOnce();
+    EXPECT_TRUE(discard_once_value);
+  }
+
+  // Setting the variation parameter to true.
+  {
+    std::unique_ptr<base::FieldTrialList> field_trial_list_;
+    field_trial_list_.reset(
+        new base::FieldTrialList(new base::MockEntropyProvider()));
+    variations::testing::ClearAllVariationParams();
+
+    std::map<std::string, std::string> params;
+    params["AllowMultipleDiscards"] = "true";
+    ASSERT_TRUE(variations::AssociateVariationParams(kTrialName, "A", params));
+    base::FieldTrialList::CreateFieldTrial(kTrialName, "A");
+
+    bool discard_once_value = tab_manager.CanOnlyDiscardOnce();
+    EXPECT_FALSE(discard_once_value);
+  }
+
+  // Setting the variation parameter to something else.
+  {
+    std::unique_ptr<base::FieldTrialList> field_trial_list_;
+    field_trial_list_.reset(
+        new base::FieldTrialList(new base::MockEntropyProvider()));
+    variations::testing::ClearAllVariationParams();
+
+    std::map<std::string, std::string> params;
+    params["AllowMultipleDiscards"] = "somethingElse";
+    ASSERT_TRUE(variations::AssociateVariationParams(kTrialName, "B", params));
+    base::FieldTrialList::CreateFieldTrial(kTrialName, "B");
+
+    bool discard_once_value = tab_manager.CanOnlyDiscardOnce();
+    EXPECT_TRUE(discard_once_value);
+  }
+}
+#else
+TEST_F(TabManagerTest, CanOnlyDiscardOnce) {
+  TabManager tab_manager;
+
+  bool discard_once_value = tab_manager.CanOnlyDiscardOnce();
+  EXPECT_FALSE(discard_once_value);
+}
+#endif  // defined(OS_WIN) || defined(OS_MACOSX)
+
 namespace {
 
 using MemoryPressureLevel = base::MemoryPressureListener::MemoryPressureLevel;
 
 // Function that always indicates the absence of memory pressure.
 MemoryPressureLevel ReturnNoPressure() {
   return base::MemoryPressureListener::MEMORY_PRESSURE_LEVEL_NONE;
 }
 
 // Function that simply parrots back an externally specified memory pressure
@@ skipping 146 matching lines @@
   EXPECT_EQ(0u, task_runner->size());
   EXPECT_EQ(0u, tm.notified_renderers_.size());
 
 
   // Clean up the tabstrip.
   tabstrip.CloseAllTabs();
   ASSERT_TRUE(tabstrip.empty());
 }
 
 }  // namespace memory