Commit 5de873b8 authored by Matt Menke's avatar Matt Menke Committed by Commit Bot

Make trial token tests use two argument ResponseHeader manipulation calls

We want to get rid of the single argument AddHeader() method as the two
argument AddHeader() and SetHeader() methods are safer and more flexible.

Bug: 1068194
Change-Id: If46c344d0fbf3751041568aff529feef2ffb7d19
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2147652
Commit-Queue: Marijn Kruisselbrink <mek@chromium.org>
Auto-Submit: Matt Menke <mmenke@chromium.org>
Reviewed-by: default avatarMarijn Kruisselbrink <mek@chromium.org>
Cr-Commit-Position: refs/heads/master@{#758516}
parent e7c84a47
...@@ -341,15 +341,14 @@ TEST_F(TrialTokenValidatorTest, ValidatorRespectsDisabledTokens) { ...@@ -341,15 +341,14 @@ TEST_F(TrialTokenValidatorTest, ValidatorRespectsDisabledTokens) {
} }
TEST_F(TrialTokenValidatorTest, ValidateRequestInsecure) { TEST_F(TrialTokenValidatorTest, ValidateRequestInsecure) {
response_headers_->AddHeader(std::string("Origin-Trial: ") + response_headers_->AddHeader("Origin-Trial", kInsecureOriginToken);
kInsecureOriginToken);
EXPECT_FALSE(validator_.RequestEnablesFeature( EXPECT_FALSE(validator_.RequestEnablesFeature(
GURL(kInsecureOrigin), response_headers_.get(), kAppropriateFeatureName, GURL(kInsecureOrigin), response_headers_.get(), kAppropriateFeatureName,
Now())); Now()));
} }
TEST_F(TrialTokenValidatorTest, ValidateRequestValidToken) { TEST_F(TrialTokenValidatorTest, ValidateRequestValidToken) {
response_headers_->AddHeader(std::string("Origin-Trial: ") + kSampleToken); response_headers_->AddHeader("Origin-Trial", kSampleToken);
EXPECT_TRUE(validator_.RequestEnablesFeature(GURL(kAppropriateOrigin), EXPECT_TRUE(validator_.RequestEnablesFeature(GURL(kAppropriateOrigin),
response_headers_.get(), response_headers_.get(),
kAppropriateFeatureName, Now())); kAppropriateFeatureName, Now()));
...@@ -362,8 +361,8 @@ TEST_F(TrialTokenValidatorTest, ValidateRequestNoTokens) { ...@@ -362,8 +361,8 @@ TEST_F(TrialTokenValidatorTest, ValidateRequestNoTokens) {
} }
TEST_F(TrialTokenValidatorTest, ValidateRequestMultipleHeaders) { TEST_F(TrialTokenValidatorTest, ValidateRequestMultipleHeaders) {
response_headers_->AddHeader(std::string("Origin-Trial: ") + kSampleToken); response_headers_->AddHeader("Origin-Trial", kSampleToken);
response_headers_->AddHeader(std::string("Origin-Trial: ") + kExpiredToken); response_headers_->AddHeader("Origin-Trial", kExpiredToken);
EXPECT_TRUE(validator_.RequestEnablesFeature(GURL(kAppropriateOrigin), EXPECT_TRUE(validator_.RequestEnablesFeature(GURL(kAppropriateOrigin),
response_headers_.get(), response_headers_.get(),
kAppropriateFeatureName, Now())); kAppropriateFeatureName, Now()));
...@@ -376,8 +375,8 @@ TEST_F(TrialTokenValidatorTest, ValidateRequestMultipleHeaders) { ...@@ -376,8 +375,8 @@ TEST_F(TrialTokenValidatorTest, ValidateRequestMultipleHeaders) {
} }
TEST_F(TrialTokenValidatorTest, ValidateRequestMultipleHeaderValues) { TEST_F(TrialTokenValidatorTest, ValidateRequestMultipleHeaderValues) {
response_headers_->AddHeader(std::string("Origin-Trial: ") + kExpiredToken + response_headers_->AddHeader(
", " + kSampleToken); "Origin-Trial", std::string(kExpiredToken) + ", " + kSampleToken);
EXPECT_TRUE(validator_.RequestEnablesFeature(GURL(kAppropriateOrigin), EXPECT_TRUE(validator_.RequestEnablesFeature(GURL(kAppropriateOrigin),
response_headers_.get(), response_headers_.get(),
kAppropriateFeatureName, Now())); kAppropriateFeatureName, Now()));
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment