[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());
 }
 
+#if defined(OS_WIN) || defined(OS_MACOSX)
+// Make sure of the number of times a tab is discarded.
+// Tests related to platforms that fully supports TabManager.

[Georges Khalil, 2016/05/25 17:15:09]

nit: Reword this comment and put it above the #ifdef.

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.

+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
+// Make sure of the number of times a tab is discarded.
+// Tests for platforms that doesn't fully support TabManager.

[Georges Khalil, 2016/05/25 17:15:09]

nit: remove this comment as the above one will take care of it.

+TEST_F(TabManagerTest, CanOnlyDiscardOnce) {
+  TabManager tab_manager;
+
+  bool discard_once_value = tab_manager.CanOnlyDiscardOnce();
+  EXPECT_FALSE(discard_once_value);
+}
+#endif
+

[Georges Khalil, 2016/05/25 17:15:09]

nit: add "  // defined(OS_WIN) || defined(OS_MACOSX)" (2 spaces between #endif
and //).

 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