| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/common/origin_trials/trial_token_validator.h" | 5 #include "content/common/origin_trials/trial_token_validator.h" |
| 6 | 6 |
| 7 #include <memory> | 7 #include <memory> |
| 8 #include <set> | 8 #include <set> |
| 9 #include <string> | 9 #include <string> |
| 10 | 10 |
| (...skipping 139 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 150 | 150 |
| 151 const url::Origin appropriate_origin_; | 151 const url::Origin appropriate_origin_; |
| 152 const url::Origin inappropriate_origin_; | 152 const url::Origin inappropriate_origin_; |
| 153 const url::Origin insecure_origin_; | 153 const url::Origin insecure_origin_; |
| 154 | 154 |
| 155 private: | 155 private: |
| 156 TestContentClient test_content_client_; | 156 TestContentClient test_content_client_; |
| 157 }; | 157 }; |
| 158 | 158 |
| 159 TEST_F(TrialTokenValidatorTest, ValidateValidToken) { | 159 TEST_F(TrialTokenValidatorTest, ValidateValidToken) { |
| 160 std::string feature; |
| 160 EXPECT_EQ(blink::WebOriginTrialTokenStatus::Success, | 161 EXPECT_EQ(blink::WebOriginTrialTokenStatus::Success, |
| 161 TrialTokenValidator::ValidateToken( | 162 TrialTokenValidator::ValidateToken(kSampleToken, |
| 162 kSampleToken, appropriate_origin_, kAppropriateFeatureName)); | 163 appropriate_origin_, &feature)); |
| 164 EXPECT_EQ(kAppropriateFeatureName, feature); |
| 163 } | 165 } |
| 164 | 166 |
| 165 TEST_F(TrialTokenValidatorTest, ValidateInappropriateOrigin) { | 167 TEST_F(TrialTokenValidatorTest, ValidateInappropriateOrigin) { |
| 168 std::string feature; |
| 166 EXPECT_EQ(blink::WebOriginTrialTokenStatus::WrongOrigin, | 169 EXPECT_EQ(blink::WebOriginTrialTokenStatus::WrongOrigin, |
| 167 TrialTokenValidator::ValidateToken( | 170 TrialTokenValidator::ValidateToken( |
| 168 kSampleToken, inappropriate_origin_, kAppropriateFeatureName)); | 171 kSampleToken, inappropriate_origin_, &feature)); |
| 169 EXPECT_EQ(blink::WebOriginTrialTokenStatus::WrongOrigin, | 172 EXPECT_EQ(blink::WebOriginTrialTokenStatus::WrongOrigin, |
| 170 TrialTokenValidator::ValidateToken(kSampleToken, insecure_origin_, | 173 TrialTokenValidator::ValidateToken(kSampleToken, insecure_origin_, |
| 171 kAppropriateFeatureName)); | 174 &feature)); |
| 172 } | |
| 173 | |
| 174 TEST_F(TrialTokenValidatorTest, ValidateInappropriateFeature) { | |
| 175 EXPECT_EQ(blink::WebOriginTrialTokenStatus::WrongFeature, | |
| 176 TrialTokenValidator::ValidateToken( | |
| 177 kSampleToken, appropriate_origin_, kInappropriateFeatureName)); | |
| 178 } | 175 } |
| 179 | 176 |
| 180 TEST_F(TrialTokenValidatorTest, ValidateInvalidSignature) { | 177 TEST_F(TrialTokenValidatorTest, ValidateInvalidSignature) { |
| 178 std::string feature; |
| 181 EXPECT_EQ(blink::WebOriginTrialTokenStatus::InvalidSignature, | 179 EXPECT_EQ(blink::WebOriginTrialTokenStatus::InvalidSignature, |
| 182 TrialTokenValidator::ValidateToken(kInvalidSignatureToken, | 180 TrialTokenValidator::ValidateToken(kInvalidSignatureToken, |
| 183 appropriate_origin_, | 181 appropriate_origin_, &feature)); |
| 184 kAppropriateFeatureName)); | |
| 185 } | 182 } |
| 186 | 183 |
| 187 TEST_F(TrialTokenValidatorTest, ValidateUnparsableToken) { | 184 TEST_F(TrialTokenValidatorTest, ValidateUnparsableToken) { |
| 185 std::string feature; |
| 188 EXPECT_EQ(blink::WebOriginTrialTokenStatus::Malformed, | 186 EXPECT_EQ(blink::WebOriginTrialTokenStatus::Malformed, |
| 189 TrialTokenValidator::ValidateToken(kUnparsableToken, | 187 TrialTokenValidator::ValidateToken(kUnparsableToken, |
| 190 appropriate_origin_, | 188 appropriate_origin_, &feature)); |
| 191 kAppropriateFeatureName)); | |
| 192 } | 189 } |
| 193 | 190 |
| 194 TEST_F(TrialTokenValidatorTest, ValidateExpiredToken) { | 191 TEST_F(TrialTokenValidatorTest, ValidateExpiredToken) { |
| 192 std::string feature; |
| 195 EXPECT_EQ(blink::WebOriginTrialTokenStatus::Expired, | 193 EXPECT_EQ(blink::WebOriginTrialTokenStatus::Expired, |
| 196 TrialTokenValidator::ValidateToken( | 194 TrialTokenValidator::ValidateToken(kExpiredToken, |
| 197 kExpiredToken, appropriate_origin_, kAppropriateFeatureName)); | 195 appropriate_origin_, &feature)); |
| 198 } | 196 } |
| 199 | 197 |
| 200 TEST_F(TrialTokenValidatorTest, ValidateValidTokenWithIncorrectKey) { | 198 TEST_F(TrialTokenValidatorTest, ValidateValidTokenWithIncorrectKey) { |
| 199 std::string feature; |
| 201 SetPublicKey(kTestPublicKey2); | 200 SetPublicKey(kTestPublicKey2); |
| 202 EXPECT_EQ(blink::WebOriginTrialTokenStatus::InvalidSignature, | 201 EXPECT_EQ(blink::WebOriginTrialTokenStatus::InvalidSignature, |
| 203 TrialTokenValidator::ValidateToken( | 202 TrialTokenValidator::ValidateToken(kSampleToken, |
| 204 kSampleToken, appropriate_origin_, kAppropriateFeatureName)); | 203 appropriate_origin_, &feature)); |
| 205 } | 204 } |
| 206 | 205 |
| 207 TEST_F(TrialTokenValidatorTest, ValidatorRespectsDisabledFeatures) { | 206 TEST_F(TrialTokenValidatorTest, ValidatorRespectsDisabledFeatures) { |
| 207 std::string feature; |
| 208 // Disable an irrelevant feature; token should still validate | 208 // Disable an irrelevant feature; token should still validate |
| 209 DisableFeature(kInappropriateFeatureName); | 209 DisableFeature(kInappropriateFeatureName); |
| 210 EXPECT_EQ(blink::WebOriginTrialTokenStatus::Success, | 210 EXPECT_EQ(blink::WebOriginTrialTokenStatus::Success, |
| 211 TrialTokenValidator::ValidateToken( | 211 TrialTokenValidator::ValidateToken(kSampleToken, |
| 212 kSampleToken, appropriate_origin_, kAppropriateFeatureName)); | 212 appropriate_origin_, &feature)); |
| 213 EXPECT_EQ(kAppropriateFeatureName, feature); |
| 213 // Disable the token's feature; it should no longer be valid | 214 // Disable the token's feature; it should no longer be valid |
| 214 DisableFeature(kAppropriateFeatureName); | 215 DisableFeature(kAppropriateFeatureName); |
| 215 EXPECT_EQ(blink::WebOriginTrialTokenStatus::FeatureDisabled, | 216 EXPECT_EQ(blink::WebOriginTrialTokenStatus::FeatureDisabled, |
| 216 TrialTokenValidator::ValidateToken( | 217 TrialTokenValidator::ValidateToken(kSampleToken, |
| 217 kSampleToken, appropriate_origin_, kAppropriateFeatureName)); | 218 appropriate_origin_, &feature)); |
| 218 } | 219 } |
| 219 | 220 |
| 220 } // namespace content | 221 } // namespace content |
| OLD | NEW |