Refactor OriginTrialPolicy to be an abstract interface in content/public/common

content/common/origin_trials/trial_token_validator_unittest.cc

 // Copyright 2016 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 "content/common/origin_trials/trial_token_validator.h"
 
 #include <memory>
 
 #include "base/macros.h"
 #include "base/strings/string_util.h"
 #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"
 
 namespace content {
 
 namespace {
 
 // This is a sample public key for testing the API. The corresponding private
 // key (use this to generate new samples for this test file) is:
@@ skipping 53 matching lines @@
 // Generate this token with the command (in tools/origin_trials):
 // generate_token.py valid.example.com Frobulate --expire-timestamp=1000000000
 const char kExpiredToken[] =
     "AmHPUIXMaXe9jWW8kJeDFXolVjT93p4XMnK4+jMYd2pjqtFcYB1bUmdD8PunQKM+"
     "RrOtlAwa0gPqqn+A8GTD3AQAAABZeyJvcmlnaW4iOiAiaHR0cHM6Ly92YWxpZC5l"
     "eGFtcGxlLmNvbTo0NDMiLCAiZmVhdHVyZSI6ICJGcm9idWxhdGUiLCAiZXhwaXJ5"
     "IjogMTAwMDAwMDAwMH0=";
 
 const char kUnparsableToken[] = "abcde";
 
-class TestContentClient : public ContentClient {
+class TestOriginTrialPolicy : public OriginTrialPolicy {
  public:
-  base::StringPiece GetOriginTrialPublicKey() override {
+  base::StringPiece GetPublicKey() const override {
     return base::StringPiece(reinterpret_cast<const char*>(key_),
                              arraysize(kTestPublicKey));
   }
-  void SetOriginTrialPublicKey(const uint8_t* key) { key_ = key; }
+
+  // Test setup methods
+  void SetPublicKey(const uint8_t* key) { key_ = key; }
+
+ private:
   const uint8_t* key_ = nullptr;
 };
 
+class TestContentClient : public ContentClient {
+ public:
+  // ContentRendererClient methods
+  OriginTrialPolicy* GetOriginTrialPolicy() override {
+    return &origin_trial_policy_;
+  }
+  // Test setup methods
+  void SetOriginTrialPublicKey(const uint8_t* key) {
+    origin_trial_policy_.SetPublicKey(key);
+  }
+
+ private:
+  TestOriginTrialPolicy origin_trial_policy_;
+};
+
 }  // namespace
 
 class TrialTokenValidatorTest : public testing::Test {
  public:
   TrialTokenValidatorTest()
       : appropriate_origin_(GURL(kAppropriateOrigin)),
         inappropriate_origin_(GURL(kInappropriateOrigin)),
         insecure_origin_(GURL(kInsecureOrigin)) {
     SetPublicKey(kTestPublicKey);
     SetContentClient(&test_content_client_);
@@ skipping 55 matching lines @@
 }
 
 TEST_F(TrialTokenValidatorTest, ValidateValidTokenWithIncorrectKey) {
   SetPublicKey(kTestPublicKey2);
   EXPECT_EQ(blink::WebOriginTrialTokenStatus::InvalidSignature,
             TrialTokenValidator::ValidateToken(
                 kSampleToken, appropriate_origin_, kAppropriateFeatureName));
 }
 
 }  // namespace content