Index: net/cert/ct_policy_enforcer_unittest.cc |
diff --git a/net/cert/ct_policy_enforcer_unittest.cc b/net/cert/ct_policy_enforcer_unittest.cc |
index 435525293337ea7569e52ccf35e881580b1123b1..2e0b0260f99b8d85723b976abc529e24747ee674 100644 |
--- a/net/cert/ct_policy_enforcer_unittest.cc |
+++ b/net/cert/ct_policy_enforcer_unittest.cc |
@@ -123,15 +123,17 @@ class CTPolicyEnforcerTest : public ::testing::Test { |
for (size_t i = 0; i < required_scts - 1; ++i) { |
FillResultWithSCTsOfOrigin(ct::SignedCertificateTimestamp::SCT_EMBEDDED, |
1, std::vector<std::string>(), false, &result); |
- EXPECT_FALSE(policy_enforcer_->DoesConformToCTEVPolicy( |
- cert.get(), nullptr, result, BoundNetLog())) |
+ EXPECT_EQ(CTPolicyEnforcer::EV_POLICY_NOT_ENOUGH_SCTS, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ cert.get(), nullptr, result.verified_scts, BoundNetLog())) |
<< " for: " << (end - start).InDays() << " and " << required_scts |
<< " scts=" << result.verified_scts.size() << " i=" << i; |
} |
FillResultWithSCTsOfOrigin(ct::SignedCertificateTimestamp::SCT_EMBEDDED, 1, |
std::vector<std::string>(), false, &result); |
- EXPECT_TRUE(policy_enforcer_->DoesConformToCTEVPolicy( |
- cert.get(), nullptr, result, BoundNetLog())) |
+ EXPECT_EQ(CTPolicyEnforcer::EV_POLICY_COMPLIES_VIA_SCTS, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ cert.get(), nullptr, result.verified_scts, BoundNetLog())) |
<< " for: " << (end - start).InDays() << " and " << required_scts |
<< " scts=" << result.verified_scts.size(); |
} |
@@ -148,8 +150,9 @@ TEST_F(CTPolicyEnforcerTest, |
ct::CTVerifyResult result; |
FillResultWithRepeatedLogID(google_log_id_, 2, true, &result); |
- EXPECT_FALSE(policy_enforcer_->DoesConformToCTEVPolicy( |
- chain_.get(), nullptr, result, BoundNetLog())); |
+ EXPECT_EQ(CTPolicyEnforcer::EV_POLICY_NOT_DIVERSE_SCTS, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ chain_.get(), nullptr, result.verified_scts, BoundNetLog())); |
} |
TEST_F(CTPolicyEnforcerTest, |
@@ -157,16 +160,18 @@ TEST_F(CTPolicyEnforcerTest, |
ct::CTVerifyResult result; |
FillResultWithRepeatedLogID(non_google_log_id_, 2, true, &result); |
- EXPECT_FALSE(policy_enforcer_->DoesConformToCTEVPolicy( |
- chain_.get(), nullptr, result, BoundNetLog())); |
+ EXPECT_EQ(CTPolicyEnforcer::EV_POLICY_NOT_DIVERSE_SCTS, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ chain_.get(), nullptr, result.verified_scts, BoundNetLog())); |
} |
TEST_F(CTPolicyEnforcerTest, ConformsToCTEVPolicyIfSCTBeforeEnforcementDate) { |
ct::CTVerifyResult result; |
FillResultWithRepeatedLogID(non_google_log_id_, 2, false, &result); |
- EXPECT_TRUE(policy_enforcer_->DoesConformToCTEVPolicy(chain_.get(), nullptr, |
- result, BoundNetLog())); |
+ EXPECT_EQ(CTPolicyEnforcer::EV_POLICY_COMPLIES_VIA_SCTS, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ chain_.get(), nullptr, result.verified_scts, BoundNetLog())); |
} |
TEST_F(CTPolicyEnforcerTest, ConformsToCTEVPolicyWithNonEmbeddedSCTs) { |
@@ -174,8 +179,9 @@ TEST_F(CTPolicyEnforcerTest, ConformsToCTEVPolicyWithNonEmbeddedSCTs) { |
FillResultWithSCTsOfOrigin( |
ct::SignedCertificateTimestamp::SCT_FROM_TLS_EXTENSION, 2, &result); |
- EXPECT_TRUE(policy_enforcer_->DoesConformToCTEVPolicy(chain_.get(), nullptr, |
- result, BoundNetLog())); |
+ EXPECT_EQ(CTPolicyEnforcer::EV_POLICY_COMPLIES_VIA_SCTS, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ chain_.get(), nullptr, result.verified_scts, BoundNetLog())); |
} |
TEST_F(CTPolicyEnforcerTest, ConformsToCTEVPolicyWithEmbeddedSCTs) { |
@@ -184,8 +190,9 @@ TEST_F(CTPolicyEnforcerTest, ConformsToCTEVPolicyWithEmbeddedSCTs) { |
FillResultWithSCTsOfOrigin(ct::SignedCertificateTimestamp::SCT_EMBEDDED, 5, |
&result); |
- EXPECT_TRUE(policy_enforcer_->DoesConformToCTEVPolicy(chain_.get(), nullptr, |
- result, BoundNetLog())); |
+ EXPECT_EQ(CTPolicyEnforcer::EV_POLICY_COMPLIES_VIA_SCTS, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ chain_.get(), nullptr, result.verified_scts, BoundNetLog())); |
} |
TEST_F(CTPolicyEnforcerTest, DoesNotConformToCTEVPolicyNotEnoughSCTs) { |
@@ -198,14 +205,18 @@ TEST_F(CTPolicyEnforcerTest, DoesNotConformToCTEVPolicyNotEnoughSCTs) { |
FillResultWithSCTsOfOrigin(ct::SignedCertificateTimestamp::SCT_EMBEDDED, 1, |
&result); |
- EXPECT_FALSE(policy_enforcer_->DoesConformToCTEVPolicy( |
- chain_.get(), non_including_whitelist.get(), result, BoundNetLog())); |
+ EXPECT_EQ(CTPolicyEnforcer::EV_POLICY_NOT_ENOUGH_SCTS, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ chain_.get(), non_including_whitelist.get(), |
+ result.verified_scts, BoundNetLog())); |
// ... but should be OK if whitelisted. |
scoped_refptr<ct::EVCertsWhitelist> whitelist( |
new DummyEVCertsWhitelist(true, true)); |
- EXPECT_TRUE(policy_enforcer_->DoesConformToCTEVPolicy( |
- chain_.get(), whitelist.get(), result, BoundNetLog())); |
+ EXPECT_EQ( |
+ CTPolicyEnforcer::EV_POLICY_COMPLIES_VIA_WHITELIST, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ chain_.get(), whitelist.get(), result.verified_scts, BoundNetLog())); |
} |
TEST_F(CTPolicyEnforcerTest, DoesNotConformToPolicyInvalidDates) { |
@@ -214,13 +225,17 @@ TEST_F(CTPolicyEnforcerTest, DoesNotConformToPolicyInvalidDates) { |
ct::CTVerifyResult result; |
FillResultWithSCTsOfOrigin(ct::SignedCertificateTimestamp::SCT_EMBEDDED, 5, |
&result); |
- EXPECT_FALSE(policy_enforcer_->DoesConformToCTEVPolicy( |
- no_valid_dates_cert.get(), nullptr, result, BoundNetLog())); |
+ EXPECT_EQ(CTPolicyEnforcer::EV_POLICY_NOT_ENOUGH_SCTS, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ no_valid_dates_cert.get(), nullptr, result.verified_scts, |
+ BoundNetLog())); |
// ... but should be OK if whitelisted. |
scoped_refptr<ct::EVCertsWhitelist> whitelist( |
new DummyEVCertsWhitelist(true, true)); |
- EXPECT_TRUE(policy_enforcer_->DoesConformToCTEVPolicy( |
- chain_.get(), whitelist.get(), result, BoundNetLog())); |
+ EXPECT_EQ( |
+ CTPolicyEnforcer::EV_POLICY_COMPLIES_VIA_WHITELIST, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ chain_.get(), whitelist.get(), result.verified_scts, BoundNetLog())); |
} |
TEST_F(CTPolicyEnforcerTest, |
@@ -274,8 +289,10 @@ TEST_F(CTPolicyEnforcerTest, ConformsToPolicyByEVWhitelistPresence) { |
ct::CTVerifyResult result; |
FillResultWithSCTsOfOrigin(ct::SignedCertificateTimestamp::SCT_EMBEDDED, 1, |
&result); |
- EXPECT_TRUE(policy_enforcer_->DoesConformToCTEVPolicy( |
- chain_.get(), whitelist.get(), result, BoundNetLog())); |
+ EXPECT_EQ( |
+ CTPolicyEnforcer::EV_POLICY_COMPLIES_VIA_WHITELIST, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ chain_.get(), whitelist.get(), result.verified_scts, BoundNetLog())); |
} |
TEST_F(CTPolicyEnforcerTest, IgnoresInvalidEVWhitelist) { |
@@ -285,16 +302,19 @@ TEST_F(CTPolicyEnforcerTest, IgnoresInvalidEVWhitelist) { |
ct::CTVerifyResult result; |
FillResultWithSCTsOfOrigin(ct::SignedCertificateTimestamp::SCT_EMBEDDED, 1, |
&result); |
- EXPECT_FALSE(policy_enforcer_->DoesConformToCTEVPolicy( |
- chain_.get(), whitelist.get(), result, BoundNetLog())); |
+ EXPECT_EQ( |
+ CTPolicyEnforcer::EV_POLICY_NOT_ENOUGH_SCTS, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ chain_.get(), whitelist.get(), result.verified_scts, BoundNetLog())); |
} |
TEST_F(CTPolicyEnforcerTest, IgnoresNullEVWhitelist) { |
ct::CTVerifyResult result; |
FillResultWithSCTsOfOrigin(ct::SignedCertificateTimestamp::SCT_EMBEDDED, 1, |
&result); |
- EXPECT_FALSE(policy_enforcer_->DoesConformToCTEVPolicy( |
- chain_.get(), nullptr, result, BoundNetLog())); |
+ EXPECT_EQ(CTPolicyEnforcer::EV_POLICY_NOT_ENOUGH_SCTS, |
+ policy_enforcer_->DoesConformToCTEVPolicy( |
+ chain_.get(), nullptr, result.verified_scts, BoundNetLog())); |
} |
} // namespace |