Chromium Code Reviews| Index: net/cert/merkle_audit_proof_unittest.cc |
| diff --git a/net/cert/merkle_audit_proof_unittest.cc b/net/cert/merkle_audit_proof_unittest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..1b924956ec46935ce031116a2a8dccf5259426be |
| --- /dev/null |
| +++ b/net/cert/merkle_audit_proof_unittest.cc |
| @@ -0,0 +1,53 @@ |
| +// 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 "net/cert/merkle_audit_proof.h" |
| + |
| +#include "base/logging.h" |
| +#include "testing/gmock/include/gmock/gmock.h" |
| +#include "testing/gtest/include/gtest/gtest.h" |
| + |
| +namespace net { |
| +namespace ct { |
| +namespace { |
| + |
| +TEST(MerkleAuditProofTest, CalculatesAuditPathLengthCorrectly) { |
| + // Test all leaves up to a tree size of 4: |
| + EXPECT_EQ(0u, CalculateAuditPathLength(0, 1)); |
| + EXPECT_EQ(1u, CalculateAuditPathLength(0, 2)); |
| + EXPECT_EQ(1u, CalculateAuditPathLength(1, 2)); |
| + EXPECT_EQ(2u, CalculateAuditPathLength(0, 3)); |
| + EXPECT_EQ(2u, CalculateAuditPathLength(1, 3)); |
| + EXPECT_EQ(1u, CalculateAuditPathLength(2, 3)); |
| + EXPECT_EQ(2u, CalculateAuditPathLength(0, 4)); |
| + EXPECT_EQ(2u, CalculateAuditPathLength(1, 4)); |
| + EXPECT_EQ(2u, CalculateAuditPathLength(2, 4)); |
| + EXPECT_EQ(2u, CalculateAuditPathLength(3, 4)); |
| + // Boundary cases for a larger tree size: |
| + EXPECT_EQ(9u, CalculateAuditPathLength(0, 257)); |
| + EXPECT_EQ(9u, CalculateAuditPathLength(255, 257)); |
| + EXPECT_EQ(1u, CalculateAuditPathLength(256, 257)); |
| + // Example from CT over DNS draft RFC: |
| + EXPECT_EQ(20u, CalculateAuditPathLength(123456, 999999)); |
| + // Test data from |
| + // https://github.com/google/certificate-transparency/blob/af98904302724c29aa6659ca372d41c9687de2b7/python/ct/crypto/merkle_test.py: |
| + EXPECT_EQ(22u, CalculateAuditPathLength(848049, 3630887)); |
| +} |
| + |
| +#if DCHECK_IS_ON() |
| + |
| +TEST(MerkleAuditProofDeathTest, DiesIfLeafIndexIsGreaterThanOrEqualToTreeSize) { |
|
eroman
2016/05/26 00:39:43
This test sounds fishy.
(a) if it is reachable co
Rob Percival
2016/05/26 03:17:13
The tree size will come from a signed, verified da
Rob Percival
2016/05/26 16:00:57
Done.
|
| + EXPECT_DEATH_IF_SUPPORTED(CalculateAuditPathLength(0, 0), |
| + "leaf_index < tree_size"); |
| + EXPECT_DEATH_IF_SUPPORTED(CalculateAuditPathLength(10, 10), |
| + "leaf_index < tree_size"); |
| + EXPECT_DEATH_IF_SUPPORTED(CalculateAuditPathLength(11, 10), |
| + "leaf_index < tree_size"); |
| +} |
| + |
| +#endif |
| + |
| +} // namespace |
| +} // namespace ct |
| +} // namespace net |