Serialisation code for Certificate Transparency data

net/cert/ct_serialization_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/ct_serialization.h"
 
 #include <string>
 
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
@@ skipping 144 matching lines @@
 
   ASSERT_FALSE(
       ct::DecodeSignedCertificateTimestamp(&invalid_version_sct, &sct));
 
   // Valid version, invalid length (missing data)
   base::StringPiece invalid_length_sct("\x0\xa\xb\xc", 4);
   ASSERT_FALSE(
       ct::DecodeSignedCertificateTimestamp(&invalid_length_sct, &sct));
 }
 
-TEST_F(CtSerializationTest, EncodesValidSignedTreeHead) {
+TEST_F(CtSerializationTest, EncodesSignedCertificateTimestamp) {
+  scoped_refptr<ct::SignedCertificateTimestamp> sct;
+  ct::GetX509CertSCT(&sct);
+
+  std::string output;
+  EXPECT_TRUE(ct::EncodeSignedCertificateTimestamp(*sct, &output));
+  EXPECT_EQ(ct::GetTestSignedCertificateTimestamp(), output);
+}
+
+TEST_F(CtSerializationTest, EncodesValidTreeHeadForSignature) {
   ct::SignedTreeHead signed_tree_head;
   GetSampleSignedTreeHead(&signed_tree_head);
 
   std::string encoded;
   ct::EncodeTreeHeadSignature(signed_tree_head, &encoded);
   // Expected size is 50 bytes:
   // Byte 0 is version, byte 1 is signature type
   // Bytes 2-9 are timestamp
   // Bytes 10-17 are tree size
   // Bytes 18-49 are sha256 root hash
   ASSERT_EQ(50u, encoded.length());
   std::string expected_buffer(
       "\x0\x1\x0\x0\x1\x45\x3c\x5f\xb8\x35\x0\x0\x0\x0\x0\x0\x0\x15", 18);
   expected_buffer.append(ct::GetSampleSTHSHA256RootHash());
   ASSERT_EQ(expected_buffer, encoded);
 }
 
+TEST_F(CtSerializationTest, EncodesSignedTreeHead) {
+  ct::SignedTreeHead sth;
+  GetSampleSignedTreeHead(&sth);
+
+  std::string output;
+  EXPECT_TRUE(ct::EncodeSignedTreeHead(sth, &output));
+
+  EXPECT_EQ(ct::GetSampleSTH(), output);
+}
+
+TEST_F(CtSerializationTest, DecodesSignedTreeHead) {
+  ct::SignedTreeHead expected_sth;
+  GetSampleSignedTreeHead(&expected_sth);
+
+  std::string encoded_sth = ct::GetSampleSTH();
+  base::StringPiece encoded_sth_ptr(encoded_sth);
+  ct::SignedTreeHead sth;
+
+  EXPECT_TRUE(ct::DecodeSignedTreeHead(&encoded_sth_ptr, &sth));
+  EXPECT_EQ(expected_sth, sth);
+}
+
 }  // namespace net