Add custom default theme support and create a managed user default theme.

chrome/browser/themes/theme_service_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/themes/theme_service.h"
 
 #include "base/json/json_reader.h"
 #include "chrome/browser/extensions/extension_service_unittest.h"
+#include "chrome/browser/themes/custom_theme_supplier.h"
 #include "chrome/browser/themes/theme_service_factory.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/extensions/extension_manifest_constants.h"
+#include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_profile.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
-namespace {
+namespace theme_service_internal {
 
 class ThemeServiceTest : public ExtensionServiceTestBase {
  public:
   ThemeServiceTest() {}
   virtual ~ThemeServiceTest() {}
 
   scoped_refptr<extensions::Extension> MakeThemeExtension(base::FilePath path) {
     DictionaryValue source;
     source.SetString(extension_manifest_keys::kName, "theme");
     source.Set(extension_manifest_keys::kTheme, new DictionaryValue());
     source.SetString(extension_manifest_keys::kUpdateURL, "http://foo.com");
     source.SetString(extension_manifest_keys::kVersion, "0.0.0.0");
     std::string error;
     scoped_refptr<extensions::Extension> extension =
         extensions::Extension::Create(
             path, extensions::Manifest::EXTERNAL_PREF_DOWNLOAD,
             source, extensions::Extension::NO_FLAGS, &error);
     EXPECT_TRUE(extension.get());
     EXPECT_EQ("", error);
     return extension;
   }
 
   virtual void SetUp() {
     ExtensionServiceTestBase::SetUp();
     InitializeEmptyExtensionService();
   }
+
+  const CustomThemeSupplier* get_theme_supplier(ThemeService* theme_service) {
+    return theme_service->get_theme_supplier();
+  }
 };
 
 // Installs then uninstalls a theme and makes sure that the ThemeService
 // reverts to the default theme after the uninstall.
 TEST_F(ThemeServiceTest, ThemeInstallUninstall) {
   base::ScopedTempDir temp_dir;
   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
   ThemeService* theme_service =
       ThemeServiceFactory::GetForProfile(profile_.get());
   theme_service->UseDefaultTheme();
@@ skipping 26 matching lines @@
   base::MessageLoop::current()->RunUntilIdle();
   EXPECT_FALSE(theme_service->UsingDefaultTheme());
   EXPECT_EQ(extension->id(), theme_service->GetThemeID());
 
   // Now unload the extension, should revert to the default theme.
   service_->UnloadExtension(extension->id(),
                             extension_misc::UNLOAD_REASON_UPDATE);
   EXPECT_FALSE(theme_service->UsingDefaultTheme());
 }
 
-}; // namespace
+// Checks that managed users have their own default theme.
+TEST_F(ThemeServiceTest, ManagedUserThemeReplacesDefaultTheme) {
+  profile_->GetPrefs()->SetBoolean(prefs::kProfileIsManaged, true);
+  ThemeService* theme_service =
+      ThemeServiceFactory::GetForProfile(profile_.get());
+  theme_service->UseDefaultTheme();
+  EXPECT_TRUE(theme_service->UsingDefaultTheme());
+  EXPECT_TRUE(get_theme_supplier(theme_service));
+  EXPECT_EQ(get_theme_supplier(theme_service)->get_theme_type(),
+            CustomThemeSupplier::MANAGED_USER_THEME);
+}
+
+#if defined(OS_LINUX) && !defined(OS_CHROMEOS)
+// Checks that managed users don't use the system theme even if it is the
+// default. The system theme is only available on Linux.
+TEST_F(ThemeServiceTest, ManagedUserThemeReplacesNativeTheme) {
+  profile_->GetPrefs()->SetBoolean(prefs::kProfileIsManaged, true);
+  profile_->GetPrefs()->SetBoolean(prefs::kUsesSystemTheme, true);
+  ThemeService* theme_service =
+      ThemeServiceFactory::GetForProfile(profile_.get());
+  theme_service->UseDefaultTheme();
+  EXPECT_TRUE(theme_service->UsingDefaultTheme());
+  EXPECT_TRUE(get_theme_supplier(theme_service));
+  EXPECT_EQ(get_theme_supplier(theme_service)->get_theme_type(),
+            CustomThemeSupplier::MANAGED_USER_THEME);
+}
+#endif
+
+}; // namespace theme_service_internal