Allows to manually change the stored permanent country in Variations from the translate-internals d…

components/variations/service/variations_service.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 "components/variations/service/variations_service.h"
 
 #include <stddef.h>
 #include <stdint.h>
 #include <utility>
 
@@ skipping 824 matching lines @@
     if (!is_pref_valid)
       local_state_->ClearPref(prefs::kVariationsPermanentConsistencyCountry);
     // If we've never received a country code from the server, use an empty
     // country so that it won't pass any filters that specifically include
     // countries, but so that it will pass any filters that specifically exclude
     // countries.
     return std::string();
   }
 
   // Otherwise, update the pref with the current Chrome version and country.
-  base::ListValue new_list_value;
-  new_list_value.AppendString(version.GetString());
-  new_list_value.AppendString(latest_country);
-  local_state_->Set(prefs::kVariationsPermanentConsistencyCountry,
-                    new_list_value);
+  SetStoredPermanentCountry(version, latest_country);
   return latest_country;
 }
 
+void VariationsService::SetStoredPermanentCountry(const base::Version& version,
+                                                  const std::string& country) {

[Alexei Svitkine (slow), 2016/04/20 21:46:44]

Nit: Make this a free-standing function in an anon namespace at the top of the
file.

[hamelphi, 2016/04/21 14:25:10]

It needs to be a class method since we need access to local_state_.

+  base::ListValue new_list_value;
+  new_list_value.AppendString(version.GetString());
+  new_list_value.AppendString(country);
+  local_state_->Set(prefs::kVariationsPermanentConsistencyCountry,
+                    new_list_value);
+}
+
 std::string VariationsService::GetStoredPermanentCountry() {
   const base::ListValue* list_value =
       local_state_->GetList(prefs::kVariationsPermanentConsistencyCountry);
   std::string stored_country;
 
   if (list_value->GetSize() == 2) {
     list_value->GetString(1, &stored_country);
   }
 
   return stored_country;
 }
 
+bool VariationsService::ForceSetStoredPermanentCountry(
+    const std::string& country_override) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
+  if (country_override.empty())
+    return false;
+
+  const base::ListValue* list_value =
+      local_state_->GetList(prefs::kVariationsPermanentConsistencyCountry);
+
+  std::string stored_country;
+  const bool got_stored_country =
+      list_value->GetSize() == 2 && list_value->GetString(1, &stored_country);
+
+  const bool needs_update =
+      !got_stored_country || stored_country != country_override;
+
+  if (!needs_update)
+    return false;
+
+  base::Version version(version_info::GetVersionNumber());
+  SetStoredPermanentCountry(version, country_override);
+  return true;
+}
+
 }  // namespace variations