Certificate Transparency: Fetching of Signed Tree Heads (DRAFT)

net/cert/multi_log_ct_verifier_unittest.cc

 // Copyright 2013 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 "net/cert/multi_log_ct_verifier.h"
 
 #include <string>
 
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
@@ skipping 24 matching lines @@
 const char kLogDescription[] = "somelog";
 const char kSCTCountHistogram[] =
     "Net.CertificateTransparency.SCTsPerConnection";
 
 class MultiLogCTVerifierTest : public ::testing::Test {
  public:
   void SetUp() override {
     scoped_ptr<CTLogVerifier> log(CTLogVerifier::Create(
         ct::GetTestPublicKey(), kLogDescription, "https://ct.example.com"));
     ASSERT_TRUE(log);
+    log_verifiers_.clear();
+    log_verifiers_.push_back(linked_ptr<CTLogVerifier>(log.release()));
 
     verifier_.reset(new MultiLogCTVerifier());
-    verifier_->AddLog(log.Pass());
+    verifier_->AddLogs(log_verifiers_);
     std::string der_test_cert(ct::GetDerEncodedX509Cert());
     chain_ = X509Certificate::CreateFromBytes(
         der_test_cert.data(),
         der_test_cert.length());
     ASSERT_TRUE(chain_.get());
 
     embedded_sct_chain_ =
         CreateCertificateChainFromFile(GetTestCertsDirectory(),
                                        "ct-test-embedded-cert.pem",
                                        X509Certificate::FORMAT_AUTO);
@@ skipping 128 matching lines @@
 
   int NumValidSCTsInStatusHistogram() {
     return GetValueFromHistogram("Net.CertificateTransparency.SCTStatus",
                                  ct::SCT_STATUS_OK);
   }
 
  protected:
   scoped_ptr<MultiLogCTVerifier> verifier_;
   scoped_refptr<X509Certificate> chain_;
   scoped_refptr<X509Certificate> embedded_sct_chain_;
+  std::vector<linked_ptr<CTLogVerifier>> log_verifiers_;
 };
 
 TEST_F(MultiLogCTVerifierTest, VerifiesEmbeddedSCT) {
   ASSERT_TRUE(CheckPrecertificateVerification(embedded_sct_chain_));
 }
 
 TEST_F(MultiLogCTVerifierTest, VerifiesEmbeddedSCTWithPreCA) {
   scoped_refptr<X509Certificate> chain(
       CreateCertificateChainFromFile(GetTestCertsDirectory(),
                                      "ct-test-embedded-with-preca-chain.pem",
@@ skipping 89 matching lines @@
 TEST_F(MultiLogCTVerifierTest, CountsZeroSCTsCorrectly) {
   int connections_without_scts = GetValueFromHistogram(kSCTCountHistogram, 0);
   EXPECT_FALSE(VerifySinglePrecertificateChain(chain_));
   ASSERT_EQ(connections_without_scts + 1,
             GetValueFromHistogram(kSCTCountHistogram, 0));
 }
 
 }  // namespace
 
 }  // namespace net