| Index: chrome/browser/metrics/variations_service_unittest.cc
|
| ===================================================================
|
| --- chrome/browser/metrics/variations_service_unittest.cc (revision 151546)
|
| +++ chrome/browser/metrics/variations_service_unittest.cc (working copy)
|
| @@ -1,458 +0,0 @@
|
| -// 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 "base/base64.h"
|
| -#include "base/string_split.h"
|
| -#include "chrome/browser/metrics/proto/study.pb.h"
|
| -#include "chrome/browser/metrics/variations_service.h"
|
| -#include "chrome/common/chrome_version_info.h"
|
| -#include "chrome/common/pref_names.h"
|
| -#include "chrome/test/base/testing_pref_service.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -
|
| -namespace chrome_variations {
|
| -
|
| -namespace {
|
| -
|
| -// Converts |time| to Study proto format.
|
| -int64 TimeToProtoTime(const base::Time& time) {
|
| - return (time - base::Time::UnixEpoch()).InSeconds();
|
| -}
|
| -
|
| -// Populates |seed| with simple test data. The resulting seed will contain one
|
| -// study called "test", which contains one experiment called "abc" with
|
| -// probability weight 100. |seed|'s study field will be cleared before adding
|
| -// the new study.
|
| -TrialsSeed CreateTestSeed() {
|
| - TrialsSeed seed;
|
| - Study* study = seed.add_study();
|
| - study->set_name("test");
|
| - study->set_default_experiment_name("abc");
|
| - Study_Experiment* experiment = study->add_experiment();
|
| - experiment->set_name("abc");
|
| - experiment->set_probability_weight(100);
|
| - return seed;
|
| -}
|
| -
|
| -} // namespace
|
| -
|
| -TEST(VariationsServiceTest, CheckStudyChannel) {
|
| - const chrome::VersionInfo::Channel channels[] = {
|
| - chrome::VersionInfo::CHANNEL_CANARY,
|
| - chrome::VersionInfo::CHANNEL_DEV,
|
| - chrome::VersionInfo::CHANNEL_BETA,
|
| - chrome::VersionInfo::CHANNEL_STABLE,
|
| - };
|
| - const Study_Channel study_channels[] = {
|
| - Study_Channel_CANARY,
|
| - Study_Channel_DEV,
|
| - Study_Channel_BETA,
|
| - Study_Channel_STABLE,
|
| - };
|
| - ASSERT_EQ(arraysize(channels), arraysize(study_channels));
|
| - bool channel_added[arraysize(channels)] = { 0 };
|
| -
|
| - Study_Filter filter;
|
| -
|
| - // Check in the forwarded order. The loop cond is <= arraysize(study_channels)
|
| - // instead of < so that the result of adding the last channel gets checked.
|
| - for (size_t i = 0; i <= arraysize(study_channels); ++i) {
|
| - for (size_t j = 0; j < arraysize(channels); ++j) {
|
| - const bool expected = channel_added[j] || filter.channel_size() == 0;
|
| - const bool result = VariationsService::CheckStudyChannel(filter,
|
| - channels[j]);
|
| - EXPECT_EQ(expected, result) << "Case " << i << "," << j << " failed!";
|
| - }
|
| -
|
| - if (i < arraysize(study_channels)) {
|
| - filter.add_channel(study_channels[i]);
|
| - channel_added[i] = true;
|
| - }
|
| - }
|
| -
|
| - // Do the same check in the reverse order.
|
| - filter.clear_channel();
|
| - memset(&channel_added, 0, sizeof(channel_added));
|
| - for (size_t i = 0; i <= arraysize(study_channels); ++i) {
|
| - for (size_t j = 0; j < arraysize(channels); ++j) {
|
| - const bool expected = channel_added[j] || filter.channel_size() == 0;
|
| - const bool result = VariationsService::CheckStudyChannel(filter,
|
| - channels[j]);
|
| - EXPECT_EQ(expected, result) << "Case " << i << "," << j << " failed!";
|
| - }
|
| -
|
| - if (i < arraysize(study_channels)) {
|
| - const int index = arraysize(study_channels) - i - 1;
|
| - filter.add_channel(study_channels[index]);
|
| - channel_added[index] = true;
|
| - }
|
| - }
|
| -}
|
| -
|
| -TEST(VariationsServiceTest, CheckStudyLocale) {
|
| - struct {
|
| - const char* filter_locales;
|
| - bool en_us_result;
|
| - bool en_ca_result;
|
| - bool fr_result;
|
| - } test_cases[] = {
|
| - {"en-US", true, false, false},
|
| - {"en-US,en-CA,fr", true, true, true},
|
| - {"en-US,en-CA,en-GB", true, true, false},
|
| - {"en-GB,en-CA,en-US", true, true, false},
|
| - {"ja,kr,vi", false, false, false},
|
| - {"fr-CA", false, false, false},
|
| - {"", true, true, true},
|
| - };
|
| -
|
| - for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_cases); ++i) {
|
| - std::vector<std::string> filter_locales;
|
| - Study_Filter filter;
|
| - base::SplitString(test_cases[i].filter_locales, ',', &filter_locales);
|
| - for (size_t j = 0; j < filter_locales.size(); ++j)
|
| - filter.add_locale(filter_locales[j]);
|
| - EXPECT_EQ(test_cases[i].en_us_result,
|
| - VariationsService::CheckStudyLocale(filter, "en-US"));
|
| - EXPECT_EQ(test_cases[i].en_ca_result,
|
| - VariationsService::CheckStudyLocale(filter, "en-CA"));
|
| - EXPECT_EQ(test_cases[i].fr_result,
|
| - VariationsService::CheckStudyLocale(filter, "fr"));
|
| - }
|
| -}
|
| -
|
| -TEST(VariationsServiceTest, CheckStudyPlatform) {
|
| - const Study_Platform platforms[] = {
|
| - Study_Platform_PLATFORM_WINDOWS,
|
| - Study_Platform_PLATFORM_MAC,
|
| - Study_Platform_PLATFORM_LINUX,
|
| - Study_Platform_PLATFORM_CHROMEOS,
|
| - Study_Platform_PLATFORM_ANDROID,
|
| - Study_Platform_PLATFORM_IOS,
|
| - };
|
| - ASSERT_EQ(Study_Platform_Platform_ARRAYSIZE,
|
| - static_cast<int>(arraysize(platforms)));
|
| - bool platform_added[arraysize(platforms)] = { 0 };
|
| -
|
| - Study_Filter filter;
|
| -
|
| - // Check in the forwarded order. The loop cond is <= arraysize(platforms)
|
| - // instead of < so that the result of adding the last channel gets checked.
|
| - for (size_t i = 0; i <= arraysize(platforms); ++i) {
|
| - for (size_t j = 0; j < arraysize(platforms); ++j) {
|
| - const bool expected = platform_added[j] || filter.platform_size() == 0;
|
| - const bool result = VariationsService::CheckStudyPlatform(filter,
|
| - platforms[j]);
|
| - EXPECT_EQ(expected, result) << "Case " << i << "," << j << " failed!";
|
| - }
|
| -
|
| - if (i < arraysize(platforms)) {
|
| - filter.add_platform(platforms[i]);
|
| - platform_added[i] = true;
|
| - }
|
| - }
|
| -
|
| - // Do the same check in the reverse order.
|
| - filter.clear_platform();
|
| - memset(&platform_added, 0, sizeof(platform_added));
|
| - for (size_t i = 0; i <= arraysize(platforms); ++i) {
|
| - for (size_t j = 0; j < arraysize(platforms); ++j) {
|
| - const bool expected = platform_added[j] || filter.platform_size() == 0;
|
| - const bool result = VariationsService::CheckStudyPlatform(filter,
|
| - platforms[j]);
|
| - EXPECT_EQ(expected, result) << "Case " << i << "," << j << " failed!";
|
| - }
|
| -
|
| - if (i < arraysize(platforms)) {
|
| - const int index = arraysize(platforms) - i - 1;
|
| - filter.add_platform(platforms[index]);
|
| - platform_added[index] = true;
|
| - }
|
| - }
|
| -}
|
| -
|
| -TEST(VariationsServiceTest, CheckStudyVersion) {
|
| - const struct {
|
| - const char* min_version;
|
| - const char* version;
|
| - bool expected_result;
|
| - } min_test_cases[] = {
|
| - { "1.2.2", "1.2.3", true },
|
| - { "1.2.3", "1.2.3", true },
|
| - { "1.2.4", "1.2.3", false },
|
| - { "1.3.2", "1.2.3", false },
|
| - { "2.1.2", "1.2.3", false },
|
| - { "0.3.4", "1.2.3", true },
|
| - // Wildcards.
|
| - { "1.*", "1.2.3", true },
|
| - { "1.2.*", "1.2.3", true },
|
| - { "1.2.3.*", "1.2.3", true },
|
| - { "1.2.4.*", "1.2.3", false },
|
| - { "2.*", "1.2.3", false },
|
| - { "0.3.*", "1.2.3", true },
|
| - };
|
| -
|
| - const struct {
|
| - const char* max_version;
|
| - const char* version;
|
| - bool expected_result;
|
| - } max_test_cases[] = {
|
| - { "1.2.2", "1.2.3", false },
|
| - { "1.2.3", "1.2.3", true },
|
| - { "1.2.4", "1.2.3", true },
|
| - { "2.1.1", "1.2.3", true },
|
| - { "2.1.1", "2.3.4", false },
|
| - // Wildcards
|
| - { "2.1.*", "2.3.4", false },
|
| - { "2.*", "2.3.4", true },
|
| - { "2.3.*", "2.3.4", true },
|
| - { "2.3.4.*", "2.3.4", true },
|
| - { "2.3.4.0.*", "2.3.4", true },
|
| - { "2.4.*", "2.3.4", true },
|
| - { "1.3.*", "2.3.4", false },
|
| - { "1.*", "2.3.4", false },
|
| - };
|
| -
|
| - Study_Filter filter;
|
| -
|
| - // Min/max version not set should result in true.
|
| - EXPECT_TRUE(VariationsService::CheckStudyVersion(filter, "1.2.3"));
|
| -
|
| - for (size_t i = 0; i < ARRAYSIZE_UNSAFE(min_test_cases); ++i) {
|
| - filter.set_min_version(min_test_cases[i].min_version);
|
| - const bool result =
|
| - VariationsService::CheckStudyVersion(filter, min_test_cases[i].version);
|
| - EXPECT_EQ(min_test_cases[i].expected_result, result) <<
|
| - "Min. version case " << i << " failed!";
|
| - }
|
| - filter.clear_min_version();
|
| -
|
| - for (size_t i = 0; i < ARRAYSIZE_UNSAFE(max_test_cases); ++i) {
|
| - filter.set_max_version(max_test_cases[i].max_version);
|
| - const bool result =
|
| - VariationsService::CheckStudyVersion(filter, max_test_cases[i].version);
|
| - EXPECT_EQ(max_test_cases[i].expected_result, result) <<
|
| - "Max version case " << i << " failed!";
|
| - }
|
| -
|
| - // Check intersection semantics.
|
| - for (size_t i = 0; i < ARRAYSIZE_UNSAFE(min_test_cases); ++i) {
|
| - for (size_t j = 0; j < ARRAYSIZE_UNSAFE(max_test_cases); ++j) {
|
| - filter.set_min_version(min_test_cases[i].min_version);
|
| - filter.set_max_version(max_test_cases[j].max_version);
|
| -
|
| - if (!min_test_cases[i].expected_result) {
|
| - const bool result =
|
| - VariationsService::CheckStudyVersion(filter,
|
| - min_test_cases[i].version);
|
| - EXPECT_FALSE(result) << "Case " << i << "," << j << " failed!";
|
| - }
|
| -
|
| - if (!max_test_cases[j].expected_result) {
|
| - const bool result =
|
| - VariationsService::CheckStudyVersion(filter,
|
| - max_test_cases[j].version);
|
| - EXPECT_FALSE(result) << "Case " << i << "," << j << " failed!";
|
| - }
|
| - }
|
| - }
|
| -}
|
| -
|
| -TEST(VariationsServiceTest, CheckStudyStartDate) {
|
| - const base::Time now = base::Time::Now();
|
| - const base::TimeDelta delta = base::TimeDelta::FromHours(1);
|
| - const struct {
|
| - const base::Time start_date;
|
| - bool expected_result;
|
| - } start_test_cases[] = {
|
| - { now - delta, true },
|
| - { now, true },
|
| - { now + delta, false },
|
| - };
|
| -
|
| - Study_Filter filter;
|
| -
|
| - // Start date not set should result in true.
|
| - EXPECT_TRUE(VariationsService::CheckStudyStartDate(filter, now));
|
| -
|
| - for (size_t i = 0; i < ARRAYSIZE_UNSAFE(start_test_cases); ++i) {
|
| - filter.set_start_date(TimeToProtoTime(start_test_cases[i].start_date));
|
| - const bool result = VariationsService::CheckStudyStartDate(filter, now);
|
| - EXPECT_EQ(start_test_cases[i].expected_result, result)
|
| - << "Case " << i << " failed!";
|
| - }
|
| -}
|
| -
|
| -TEST(VariationsServiceTest, IsStudyExpired) {
|
| - const base::Time now = base::Time::Now();
|
| - const base::TimeDelta delta = base::TimeDelta::FromHours(1);
|
| - const struct {
|
| - const base::Time expiry_date;
|
| - bool expected_result;
|
| - } expiry_test_cases[] = {
|
| - { now - delta, true },
|
| - { now, true },
|
| - { now + delta, false },
|
| - };
|
| -
|
| - Study study;
|
| -
|
| - // Expiry date not set should result in false.
|
| - EXPECT_FALSE(VariationsService::IsStudyExpired(study, now));
|
| -
|
| - for (size_t i = 0; i < ARRAYSIZE_UNSAFE(expiry_test_cases); ++i) {
|
| - study.set_expiry_date(TimeToProtoTime(expiry_test_cases[i].expiry_date));
|
| - const bool result = VariationsService::IsStudyExpired(study, now);
|
| - EXPECT_EQ(expiry_test_cases[i].expected_result, result)
|
| - << "Case " << i << " failed!";
|
| - }
|
| -}
|
| -
|
| -TEST(VariationsServiceTest, LoadSeed) {
|
| - TestingPrefService pref_service;
|
| -
|
| - VariationsService::RegisterPrefs(&pref_service);
|
| -
|
| - // Store good seed data to test if loading from prefs works.
|
| - TrialsSeed seed = CreateTestSeed();
|
| -
|
| - std::string serialized_seed;
|
| - seed.SerializeToString(&serialized_seed);
|
| - std::string base64_serialized_seed;
|
| - ASSERT_TRUE(base::Base64Encode(serialized_seed, &base64_serialized_seed));
|
| - pref_service.SetString(prefs::kVariationsSeed, base64_serialized_seed);
|
| -
|
| - VariationsService variations_service;
|
| - TrialsSeed loaded_seed;
|
| - EXPECT_TRUE(
|
| - variations_service.LoadTrialsSeedFromPref(&pref_service, &loaded_seed));
|
| -
|
| - std::string serialized_loaded_seed;
|
| - loaded_seed.SerializeToString(&serialized_loaded_seed);
|
| - // Check that the loaded data is the same as the original.
|
| - EXPECT_EQ(serialized_seed, serialized_loaded_seed);
|
| - // Make sure the pref hasn't been changed.
|
| - EXPECT_FALSE(
|
| - pref_service.FindPreference(prefs::kVariationsSeed)->IsDefaultValue());
|
| - EXPECT_EQ(base64_serialized_seed,
|
| - pref_service.GetString(prefs::kVariationsSeed));
|
| -
|
| - // Check that loading a bad seed returns false and clears the pref.
|
| - pref_service.ClearPref(prefs::kVariationsSeed);
|
| - pref_service.SetString(prefs::kVariationsSeed, "this should fail");
|
| - EXPECT_FALSE(
|
| - pref_service.FindPreference(prefs::kVariationsSeed)->IsDefaultValue());
|
| - EXPECT_FALSE(
|
| - variations_service.LoadTrialsSeedFromPref(&pref_service, &loaded_seed));
|
| - EXPECT_TRUE(
|
| - pref_service.FindPreference(prefs::kVariationsSeed)->IsDefaultValue());
|
| -}
|
| -
|
| -TEST(VariationsServiceTest, StoreSeed) {
|
| - TestingPrefService pref_service;
|
| -
|
| - VariationsService::RegisterPrefs(&pref_service);
|
| - const base::Time now = base::Time::Now();
|
| -
|
| - TrialsSeed seed = CreateTestSeed();
|
| -
|
| - VariationsService variations_service;
|
| - std::string serialized_seed;
|
| - seed.SerializeToString(&serialized_seed);
|
| - EXPECT_TRUE(
|
| - variations_service.StoreSeedData(serialized_seed, now, &pref_service));
|
| - // Make sure the pref was actually set.
|
| - EXPECT_FALSE(
|
| - pref_service.FindPreference(prefs::kVariationsSeed)->IsDefaultValue());
|
| -
|
| - std::string loaded_serialized_seed =
|
| - pref_service.GetString(prefs::kVariationsSeed);
|
| - std::string decoded_serialized_seed;
|
| - ASSERT_TRUE(base::Base64Decode(loaded_serialized_seed,
|
| - &decoded_serialized_seed));
|
| - // Make sure the stored seed from pref is the same as the seed we created.
|
| - EXPECT_EQ(serialized_seed, decoded_serialized_seed);
|
| -
|
| - // Check if trying to store a bad seed leaves the pref unchanged.
|
| - pref_service.ClearPref(prefs::kVariationsSeed);
|
| - EXPECT_FALSE(
|
| - variations_service.StoreSeedData("this should fail", now, &pref_service));
|
| - EXPECT_TRUE(
|
| - pref_service.FindPreference(prefs::kVariationsSeed)->IsDefaultValue());
|
| -}
|
| -
|
| -TEST(VariationsServiceTest, ValidateStudy) {
|
| - Study study;
|
| - study.set_default_experiment_name("def");
|
| -
|
| - Study_Experiment* experiment = study.add_experiment();
|
| - experiment->set_name("abc");
|
| - experiment->set_probability_weight(100);
|
| -
|
| - Study_Experiment* default_group = study.add_experiment();
|
| - default_group->set_name("def");
|
| - default_group->set_probability_weight(200);
|
| -
|
| - base::FieldTrial::Probability total_probability = 0;
|
| - bool valid = VariationsService::ValidateStudyAndComputeTotalProbability(
|
| - study, &total_probability);
|
| - EXPECT_TRUE(valid);
|
| - EXPECT_EQ(300, total_probability);
|
| -
|
| - // Min version checks.
|
| - study.mutable_filter()->set_min_version("1.2.3.*");
|
| - valid = VariationsService::ValidateStudyAndComputeTotalProbability(
|
| - study, &total_probability);
|
| - EXPECT_TRUE(valid);
|
| - study.mutable_filter()->set_min_version("1.*.3");
|
| - valid = VariationsService::ValidateStudyAndComputeTotalProbability(
|
| - study, &total_probability);
|
| - EXPECT_FALSE(valid);
|
| - study.mutable_filter()->set_min_version("1.2.3");
|
| - valid = VariationsService::ValidateStudyAndComputeTotalProbability(
|
| - study, &total_probability);
|
| - EXPECT_TRUE(valid);
|
| -
|
| - // Max version checks.
|
| - study.mutable_filter()->set_max_version("2.3.4.*");
|
| - valid = VariationsService::ValidateStudyAndComputeTotalProbability(
|
| - study, &total_probability);
|
| - EXPECT_TRUE(valid);
|
| - study.mutable_filter()->set_max_version("*.3");
|
| - valid = VariationsService::ValidateStudyAndComputeTotalProbability(
|
| - study, &total_probability);
|
| - EXPECT_FALSE(valid);
|
| - study.mutable_filter()->set_max_version("2.3.4");
|
| - valid = VariationsService::ValidateStudyAndComputeTotalProbability(
|
| - study, &total_probability);
|
| - EXPECT_TRUE(valid);
|
| -
|
| - study.clear_default_experiment_name();
|
| - valid = VariationsService::ValidateStudyAndComputeTotalProbability(study,
|
| - &total_probability);
|
| - EXPECT_FALSE(valid);
|
| -
|
| - study.set_default_experiment_name("xyz");
|
| - valid = VariationsService::ValidateStudyAndComputeTotalProbability(study,
|
| - &total_probability);
|
| - EXPECT_FALSE(valid);
|
| -
|
| - study.set_default_experiment_name("def");
|
| - default_group->clear_name();
|
| - valid = VariationsService::ValidateStudyAndComputeTotalProbability(study,
|
| - &total_probability);
|
| - EXPECT_FALSE(valid);
|
| -
|
| - default_group->set_name("def");
|
| - valid = VariationsService::ValidateStudyAndComputeTotalProbability(study,
|
| - &total_probability);
|
| - ASSERT_TRUE(valid);
|
| - Study_Experiment* repeated_group = study.add_experiment();
|
| - repeated_group->set_name("abc");
|
| - repeated_group->set_probability_weight(1);
|
| - valid = VariationsService::ValidateStudyAndComputeTotalProbability(study,
|
| - &total_probability);
|
| - EXPECT_FALSE(valid);
|
| -}
|
| -
|
| -} // namespace chrome_variations
|
|
|