Index: components/rappor/rappor_service_unittest.cc |
diff --git a/components/rappor/rappor_service_unittest.cc b/components/rappor/rappor_service_unittest.cc |
index a91db8c03221d9f44194ade2fd50cc0a8421d4da..4430ee3eb9827ed160d02b6c0147efb40fb2d325 100644 |
--- a/components/rappor/rappor_service_unittest.cc |
+++ b/components/rappor/rappor_service_unittest.cc |
@@ -4,23 +4,44 @@ |
#include "components/rappor/rappor_service.h" |
+#include "base/base64.h" |
+#include "base/prefs/testing_pref_service.h" |
#include "components/rappor/byte_vector_utils.h" |
#include "components/rappor/proto/rappor_metric.pb.h" |
#include "components/rappor/rappor_parameters.h" |
+#include "components/rappor/rappor_pref_names.h" |
#include "testing/gtest/include/gtest/gtest.h" |
namespace rappor { |
class TestRapporService : public RapporService { |
public: |
+ TestRapporService() : RapporService(&prefs_) { |
+ RegisterPrefs(prefs_.registry()); |
+ prefs_.SetInteger(prefs::kRapporCohortSeed, 0); |
+ std::string secret = HmacByteVectorGenerator::GenerateEntropyInput(); |
+ std::string secret_base64; |
+ base::Base64Encode(secret, &secret_base64); |
+ prefs_.SetString(prefs::kRapporSecret, secret_base64); |
+ LoadCohort(); |
+ LoadSecret(); |
+ } |
+ |
void GetReports(RapporReports* reports) { |
ExportMetrics(reports); |
} |
+ |
void TestRecordSample(const std::string& metric_name, |
const RapporParameters& parameters, |
const std::string& sample) { |
RecordSampleInternal(metric_name, parameters, sample); |
} |
+ |
+ protected: |
+ TestingPrefServiceSimple prefs_; |
+ |
+ private: |
+ DISALLOW_COPY_AND_ASSIGN(TestRapporService); |
}; |
TEST(RapporServiceTest, RecordAndExportMetrics) { |
@@ -34,9 +55,6 @@ TEST(RapporServiceTest, RecordAndExportMetrics) { |
PROBABILITY_50 /* Zero coin probability */}; |
TestRapporService rappor_service; |
- rappor_service.SetCohortForTesting(0); |
- rappor_service.SetSecretForTesting( |
- HmacByteVectorGenerator::GenerateEntropyInput()); |
rappor_service.TestRecordSample("MyMetric", kTestRapporParameters, "foo"); |
rappor_service.TestRecordSample("MyMetric", kTestRapporParameters, "bar"); |