Adding show profile switcher field trial.

chrome/browser/profiles/avatar_menu_model_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/profiles/avatar_menu_model.h"
 
+#include "base/metrics/field_trial.h"
 #include "base/string16.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/profiles/avatar_menu_model_observer.h"
 #include "chrome/browser/profiles/profile_info_cache.h"
+#include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_profile_manager.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 class MockObserver : public AvatarMenuModelObserver {
  public:
   MockObserver() : count_(0) {}
   virtual ~MockObserver() {}
@@ skipping 16 matching lines @@
   }
 
   virtual void SetUp() {
     ASSERT_TRUE(manager_.SetUp());
 #if defined(OS_CHROMEOS)
     // AvatarMenu and multiple profiles works after user logged in.
     manager_.SetLoggedIn(true);
 #endif
   }
 
+  virtual void TearDown() {
+    // Some tests force IsMultipleProfilesEnabled to return a particular
+    // value. We make sure it's always set back to a normal state.
+    ProfileManager::UnforceMultipleProfilesForTests();
+  }
+
   Browser* browser() { return NULL; }
 
   TestingProfileManager* manager() { return &manager_; }
 
  private:
   TestingProfileManager manager_;
 };
 
 TEST_F(AvatarMenuModelTest, InitialCreation) {
   string16 name1(ASCIIToUTF16("Test 1"));
@@ skipping 100 matching lines @@
 
   const AvatarMenuModel::Item& item2 = model.GetItemAt(1);
   EXPECT_EQ(1U, item2.model_index);
   EXPECT_EQ(name2, item2.name);
 
   const AvatarMenuModel::Item& item3 = model.GetItemAt(2);
   EXPECT_EQ(2U, item3.model_index);
   EXPECT_EQ(name3, item3.name);
 }
 
+// Tests for ShouldShowAvatarMenu are only applicable when multiple profiles is
+// enabled, so for these tests we force it to be enabled.
+TEST_F(AvatarMenuModelTest, ShowAvatarMenuInTrial) {
+  ProfileManager::ForceMultipleProfilesForTests(true);
+  base::FieldTrialList field_trial_list_(NULL);
+  base::FieldTrialList::CreateFieldTrial("ShowProfileSwitcher", "AlwaysShow");
+
+  EXPECT_TRUE(AvatarMenuModel::ShouldShowAvatarMenu());
+}
+
+TEST_F(AvatarMenuModelTest, DontShowAvatarMenu) {
+  ProfileManager::ForceMultipleProfilesForTests(true);
+  string16 name1(ASCIIToUTF16("Test 1"));
+  manager()->CreateTestingProfile("p1", name1, 0);
+
+  EXPECT_FALSE(AvatarMenuModel::ShouldShowAvatarMenu());
+
+  ProfileManager::ForceMultipleProfilesForTests(false);
+  string16 name2(ASCIIToUTF16("Test 2"));
+  manager()->CreateTestingProfile("p2", name2, 0);
+
+  EXPECT_FALSE(AvatarMenuModel::ShouldShowAvatarMenu());
+}
+
+TEST_F(AvatarMenuModelTest, ShowAvatarMenu) {
+  ProfileManager::ForceMultipleProfilesForTests(true);
+  string16 name1(ASCIIToUTF16("Test 1"));
+  string16 name2(ASCIIToUTF16("Test 2"));
+
+  manager()->CreateTestingProfile("p1", name1, 0);
+  manager()->CreateTestingProfile("p2", name2, 0);
+
+  EXPECT_TRUE(AvatarMenuModel::ShouldShowAvatarMenu());
+}
+
 }  // namespace