Index: content/common/origin_trials/trial_token_validator_unittest.cc |
diff --git a/content/common/origin_trials/trial_token_validator_unittest.cc b/content/common/origin_trials/trial_token_validator_unittest.cc |
index 363e3d8dd5c8c10150f9562282e2fe7181fe9da0..e3ac8fcadb9e06f26b6ec18b75b3f95d39b950a9 100644 |
--- a/content/common/origin_trials/trial_token_validator_unittest.cc |
+++ b/content/common/origin_trials/trial_token_validator_unittest.cc |
@@ -13,6 +13,7 @@ |
#include "base/test/simple_test_clock.h" |
#include "base/time/time.h" |
#include "content/public/common/content_client.h" |
+#include "content/public/common/origin_trial_policy.h" |
#include "testing/gtest/include/gtest/gtest.h" |
#include "third_party/WebKit/public/platform/WebOriginTrialTokenStatus.h" |
#include "url/gurl.h" |
@@ -86,19 +87,18 @@ const char kExpiredToken[] = |
const char kUnparsableToken[] = "abcde"; |
-class TestContentClient : public ContentClient { |
+class TestOriginTrialPolicy : public OriginTrialPolicy { |
public: |
- // ContentRendererClient methods |
- base::StringPiece GetOriginTrialPublicKey() const override { |
+ base::StringPiece GetPublicKey() const override { |
return base::StringPiece(reinterpret_cast<const char*>(key_), |
arraysize(kTestPublicKey)); |
} |
- bool IsOriginTrialFeatureDisabled(base::StringPiece feature) const override { |
+ bool IsFeatureDisabled(base::StringPiece feature) const override { |
return disabled_features_.count(feature.as_string()) > 0; |
} |
// Test setup methods |
- void SetOriginTrialPublicKey(const uint8_t* key) { key_ = key; } |
+ void SetPublicKey(const uint8_t* key) { key_ = key; } |
void DisableFeature(const std::string& feature) { |
disabled_features_.insert(feature); |
} |
@@ -108,6 +108,24 @@ class TestContentClient : public ContentClient { |
std::set<std::string> disabled_features_; |
}; |
+class TestContentClient : public ContentClient { |
+ public: |
+ // ContentRendererClient methods |
+ const OriginTrialPolicy* GetOriginTrialPolicy() const override { |
+ return &origin_trial_policy_; |
+ } |
+ // Test setup methods |
+ void SetOriginTrialPublicKey(const uint8_t* key) { |
+ origin_trial_policy_.SetPublicKey(key); |
+ } |
+ void DisableFeature(const std::string& feature) { |
+ origin_trial_policy_.DisableFeature(feature); |
+ } |
+ |
+ private: |
+ TestOriginTrialPolicy origin_trial_policy_; |
+}; |
+ |
} // namespace |
class TrialTokenValidatorTest : public testing::Test { |