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

components/variations/service/variations_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 "components/variations/service/variations_service.h"
 
 #include <stddef.h>
 #include <utility>
 #include <vector>
 
 #include "base/base64.h"
 #include "base/feature_list.h"
 #include "base/json/json_string_value_serializer.h"
 #include "base/macros.h"
 #include "base/message_loop/message_loop.h"
 #include "base/sha1.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_split.h"
 #include "base/strings/string_util.h"
 #include "base/test/histogram_tester.h"
 #include "base/version.h"
 #include "components/prefs/testing_pref_service.h"
 #include "components/variations/pref_names.h"
 #include "components/variations/proto/study.pb.h"
 #include "components/variations/proto/variations_seed.pb.h"
+#include "components/version_info/version_info.h"
 #include "components/web_resource/resource_request_allowed_notifier_test_util.h"
 #include "net/base/url_util.h"
 #include "net/http/http_response_headers.h"
 #include "net/http/http_status_code.h"
 #include "net/url_request/test_url_fetcher_factory.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace variations {
 
 namespace {
@@ skipping 685 matching lines @@
               ListValueToString(*pref_value))
         << test.pref_value_before << ", " << test.version << ", "
         << test.latest_country_code;
 
     histogram_tester.ExpectUniqueSample(
         "Variations.LoadPermanentConsistencyCountryResult",
         test.expected_result, 1);
   }
 }
 
+TEST_F(VariationsServiceTest, ForceSetStoredPermanentCountry) {
+  std::string kTestVersion = version_info::GetVersionNumber();
+  std::string kPrefCa = version_info::GetVersionNumber() + ",ca";
+  std::string kPrefUs = version_info::GetVersionNumber() + ",us";
+
+  struct {
+    // Comma separated list, empty string if the pref isn't set initially.
+    std::string pref_value_before;
+    std::string country_code_override;
+    // Comma separated list.
+    std::string expected_pref_value_after;
+    // Is the pref expected to be updated or not.
+    const bool has_updated;
+  } test_cases[] = {
+      {kPrefUs, "ca", kPrefCa, true},
+      {kPrefUs, "us", kPrefUs, false},
+      {kPrefUs, "", kPrefUs, false},
+      {"", "ca", kPrefCa, true},
+      {"", "", "", false},
+      {"19.0.0.0,us", "ca", kPrefCa, true},
+      {"19.0.0.0,us", "us", "19.0.0.0,us", false},
+  };
+
+  for (const auto& test : test_cases) {
+    TestingPrefServiceSimple prefs;
+    VariationsService::RegisterPrefs(prefs.registry());
+    TestVariationsService service(
+        make_scoped_ptr(new web_resource::TestRequestAllowedNotifier(&prefs)),
+        &prefs);
+
+    if (test.pref_value_before != "") {

[Alexei Svitkine (slow), 2016/04/20 20:30:24]

Nit: !.empty()

[hamelphi, 2016/04/20 21:39:47]

Done.

+      base::ListValue list_value;
+      for (const std::string& component :
+           base::SplitString(test.pref_value_before, ",", base::TRIM_WHITESPACE,
+                             base::SPLIT_WANT_ALL)) {
+        list_value.AppendString(component);
+      }
+      prefs.Set(prefs::kVariationsPermanentConsistencyCountry, list_value);
+    }
+
+    variations::VariationsSeed seed(CreateTestSeed());
+
+    EXPECT_EQ(test.has_updated, service.ForceSetStoredPermanentCountry(
+                                    test.country_code_override))
+        << test.pref_value_before << ", " << test.country_code_override;
+
+    base::ListValue expected_list_value;
+    for (const std::string& component :
+             base::SplitString(test.expected_pref_value_after, ",",
+                               base::TRIM_WHITESPACE, base::SPLIT_WANT_ALL)) {
+      expected_list_value.AppendString(component);
+    }
+    const base::ListValue* pref_value =
+        prefs.GetList(prefs::kVariationsPermanentConsistencyCountry);
+    EXPECT_EQ(ListValueToString(expected_list_value),
+              ListValueToString(*pref_value))
+        << test.pref_value_before << ", " << test.country_code_override;
+  }
+}
+
 }  // namespace variations