components/variations/caching_permuted_entropy_provider_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "components/variations/caching_permuted_entropy_provider.h"
   6
   7 #include <string>
   8
   9 #include "base/basictypes.h"
  10 #include "base/prefs/testing_pref_service.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 namespace metrics {
  14
  15 // Size of the low entropy source to use for the permuted entropy provider
  16 // in tests.
  17 const size_t kMaxLowEntropySize = 8000;
  18
  19 // Field trial names used in unit tests.
  20 const char* const kTestTrialNames[] = {"TestTrial", "AnotherTestTrial",
  21                                        "NewTabButton"};
  22
  23 TEST(CachingPermutedEntropyProviderTest, HasConsistentResults) {
  24   TestingPrefServiceSimple prefs;
  25   CachingPermutedEntropyProvider::RegisterPrefs(prefs.registry());
  26   const int kEntropyValue = 1234;
  27
  28   // Check that the caching provider returns the same results as the non caching
  29   // one. Loop over the trial names twice, to test that caching returns the
  30   // expected results.
  31   PermutedEntropyProvider provider(kEntropyValue, kMaxLowEntropySize);
  32   for (size_t i = 0; i < 2 * arraysize(kTestTrialNames); ++i) {
  33     CachingPermutedEntropyProvider cached_provider(
  34         &prefs, kEntropyValue, kMaxLowEntropySize);
  35     const std::string trial_name =
  36         kTestTrialNames[i % arraysize(kTestTrialNames)];
  37     EXPECT_DOUBLE_EQ(provider.GetEntropyForTrial(trial_name, 0),
  38                      cached_provider.GetEntropyForTrial(trial_name, 0));
  39   }
  40
  41   // Now, do the same test re-using the same caching provider.
  42   CachingPermutedEntropyProvider cached_provider(
  43       &prefs, kEntropyValue, kMaxLowEntropySize);
  44   for (size_t i = 0; i < 2 * arraysize(kTestTrialNames); ++i) {
  45     const std::string trial_name =
  46         kTestTrialNames[i % arraysize(kTestTrialNames)];
  47     EXPECT_DOUBLE_EQ(provider.GetEntropyForTrial(trial_name, 0),
  48                      cached_provider.GetEntropyForTrial(trial_name, 0));
  49   }
  50 }
  51
  52 }  // namespace metrics