Add browser test for initial languages.

chrome/browser/chromeos/customization_document.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/chromeos/customization_document.h"
 
 #include <algorithm>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 91 matching lines @@
 // "ServicesCustomization.LoadResult" histogram.
 // It is append-only enum due to use in a histogram!
 enum HistogramServicesCustomizationLoadResult {
   HISTOGRAM_LOAD_RESULT_SUCCESS = 0,
   HISTOGRAM_LOAD_RESULT_FILE_NOT_FOUND = 1,
   HISTOGRAM_LOAD_RESULT_PARSING_ERROR = 2,
   HISTOGRAM_LOAD_RESULT_RETRIES_FAIL = 3,
   HISTOGRAM_LOAD_RESULT_MAX_VALUE = 4
 };
 
+// In tests we need to enforce initial settings.
+const char* enforce_initial_locale = NULL;
+const char* enforce_keyboard_layout = NULL;
+
 void LogManifestLoadResult(HistogramServicesCustomizationLoadResult result) {
   UMA_HISTOGRAM_ENUMERATION("ServicesCustomization.LoadResult",
                             result,
                             HISTOGRAM_LOAD_RESULT_MAX_VALUE);
 }
 
 std::string GetLocaleSpecificStringImpl(
     const base::DictionaryValue* root,
     const std::string& locale,
     const std::string& dictionary_name,
@@ skipping 157 matching lines @@
       DefaultSingletonTraits<StartupCustomizationDocument> >::get();
 }
 
 void StartupCustomizationDocument::Init(
     system::StatisticsProvider* statistics_provider) {
   if (IsReady()) {
     root_->GetString(kInitialLocaleAttr, &initial_locale_);
     root_->GetString(kInitialTimezoneAttr, &initial_timezone_);
     root_->GetString(kKeyboardLayoutAttr, &keyboard_layout_);
 
+    if (enforce_initial_locale)

[Dmitry Polukhin, 2014/09/30 09:29:54]

You override before and after checking machine statistics. I think you need it
only once and I prefer doing it before checking machine statistics, i.e. just
provide default value. But if ii doesn't work, please keep only last overriders.

+      initial_locale_ = enforce_initial_locale;
+
+    if (enforce_keyboard_layout)
+      keyboard_layout_ = enforce_keyboard_layout;
+
     std::string hwid;
     if (statistics_provider->GetMachineStatistic(
             system::kHardwareClassKey, &hwid)) {
       base::ListValue* hwid_list = NULL;
       if (root_->GetList(kHwidMapAttr, &hwid_list)) {
         for (size_t i = 0; i < hwid_list->GetSize(); ++i) {
           base::DictionaryValue* hwid_dictionary = NULL;
           std::string hwid_mask;
           if (hwid_list->GetDictionary(i, &hwid_dictionary) &&
               hwid_dictionary->GetString(kHwidMaskAttr, &hwid_mask)) {
@@ skipping 21 matching lines @@
     }
   }
 
   // If manifest doesn't exist still apply values from VPD.
   statistics_provider->GetMachineStatistic(kInitialLocaleAttr,
                                            &initial_locale_);
   statistics_provider->GetMachineStatistic(kInitialTimezoneAttr,
                                            &initial_timezone_);
   statistics_provider->GetMachineStatistic(kKeyboardLayoutAttr,
                                            &keyboard_layout_);
+  if (enforce_initial_locale)
+    initial_locale_ = enforce_initial_locale;
+
+  if (enforce_keyboard_layout)
+    keyboard_layout_ = enforce_keyboard_layout;
+
   configured_locales_.resize(0);
   base::SplitString(initial_locale_, ',', &configured_locales_);
 
   // Convert ICU locale to chrome ("en_US" to "en-US", etc.).
   std::for_each(configured_locales_.begin(),
                 configured_locales_.end(),
                 l10n_util::GetCanonicalLocale);
 
   // Let's always have configured_locales_.front() a valid entry.
   if (configured_locales_.size() == 0)
@@ skipping 617 matching lines @@
 
   apply_tasks_success_ += success;
 
   if (apply_tasks_started_ != apply_tasks_finished_)
     return;
 
   if (apply_tasks_success_ == apply_tasks_finished_)
     SetApplied(true);
 }
 
+void SetEnforceInitialLocaleForTesting(const char* locale) {
+  enforce_initial_locale = locale;
+}
+
+void SetEnforceInitialKeyboardForTesting(const char* keyboard) {
+  enforce_keyboard_layout = keyboard;
+}
+
 }  // namespace chromeos