Support EC certs in OriginBoundCertService and OriginBoundCertStore.

net/base/default_origin_bound_cert_store_unittest.cc

 // Copyright (c) 2011 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/base/default_origin_bound_cert_store.h"
 
 #include <map>
 #include <string>
 #include <vector>
 
@@ skipping 58 matching lines @@
 
 void MockPersistentStore::Flush(Task* completion_task) {
   NOTREACHED();
 }
 
 TEST(DefaultOriginBoundCertStoreTest, TestLoading) {
   scoped_refptr<MockPersistentStore> persistent_store(new MockPersistentStore);
 
   persistent_store->AddOriginBoundCert(
       DefaultOriginBoundCertStore::OriginBoundCert(
-          "https://encrypted.google.com/", "a", "b"));
+          "https://encrypted.google.com/", ORIGIN_BOUND_RSA_CERT, "a", "b"));
   persistent_store->AddOriginBoundCert(
       DefaultOriginBoundCertStore::OriginBoundCert(
-          "https://www.verisign.com/", "c", "d"));
+          "https://www.verisign.com/", ORIGIN_BOUND_EC_CERT, "c", "d"));
 
   // Make sure certs load properly.
   DefaultOriginBoundCertStore store(persistent_store.get());
   EXPECT_EQ(2, store.GetCertCount());
-  store.SetOriginBoundCert("https://www.verisign.com/", "e", "f");
+  store.SetOriginBoundCert(
+      "https://www.verisign.com/", ORIGIN_BOUND_RSA_CERT, "e", "f");
   EXPECT_EQ(2, store.GetCertCount());
-  store.SetOriginBoundCert("https://www.twitter.com/", "g", "h");
+  store.SetOriginBoundCert(
+      "https://www.twitter.com/", ORIGIN_BOUND_RSA_CERT, "g", "h");
   EXPECT_EQ(3, store.GetCertCount());
 }
 
 TEST(DefaultOriginBoundCertStoreTest, TestSettingAndGetting) {
   DefaultOriginBoundCertStore store(NULL);
+  OriginBoundCertType type;
   std::string private_key, cert;
   EXPECT_EQ(0, store.GetCertCount());
   EXPECT_FALSE(store.GetOriginBoundCert("https://www.verisign.com/",
-                                         &private_key,
-                                         &cert));
+                                        &type,
+                                        &private_key,
+                                        &cert));
   EXPECT_TRUE(private_key.empty());
   EXPECT_TRUE(cert.empty());
-  store.SetOriginBoundCert("https://www.verisign.com/", "i", "j");
+  store.SetOriginBoundCert(
+      "https://www.verisign.com/", ORIGIN_BOUND_RSA_CERT, "i", "j");
   EXPECT_TRUE(store.GetOriginBoundCert("https://www.verisign.com/",
-                                        &private_key,
-                                        &cert));
+                                       &type,
+                                       &private_key,
+                                       &cert));
+  EXPECT_EQ(ORIGIN_BOUND_RSA_CERT, type);
   EXPECT_EQ("i", private_key);
   EXPECT_EQ("j", cert);
 }
 
 TEST(DefaultOriginBoundCertStoreTest, TestDuplicateCerts) {
   scoped_refptr<MockPersistentStore> persistent_store(new MockPersistentStore);
   DefaultOriginBoundCertStore store(persistent_store.get());
 
+  OriginBoundCertType type;
   std::string private_key, cert;
   EXPECT_EQ(0, store.GetCertCount());
-  store.SetOriginBoundCert("https://www.verisign.com/", "a", "b");
-  store.SetOriginBoundCert("https://www.verisign.com/", "c", "d");
+  store.SetOriginBoundCert(
+      "https://www.verisign.com/", ORIGIN_BOUND_RSA_CERT, "a", "b");
+  store.SetOriginBoundCert(
+      "https://www.verisign.com/", ORIGIN_BOUND_EC_CERT, "c", "d");
 
   EXPECT_EQ(1, store.GetCertCount());
   EXPECT_TRUE(store.GetOriginBoundCert("https://www.verisign.com/",
-                                        &private_key,
-                                        &cert));
+                                       &type,
+                                       &private_key,
+                                       &cert));
+  EXPECT_EQ(ORIGIN_BOUND_EC_CERT, type);
   EXPECT_EQ("c", private_key);
   EXPECT_EQ("d", cert);
 }
 
 TEST(DefaultOriginBoundCertStoreTest, TestDeleteAll) {
   scoped_refptr<MockPersistentStore> persistent_store(new MockPersistentStore);
   DefaultOriginBoundCertStore store(persistent_store.get());
 
   EXPECT_EQ(0, store.GetCertCount());
-  store.SetOriginBoundCert("https://www.verisign.com/", "a", "b");
-  store.SetOriginBoundCert("https://www.google.com/", "c", "d");
-  store.SetOriginBoundCert("https://www.harvard.com/", "e", "f");
+  store.SetOriginBoundCert(
+      "https://www.verisign.com/", ORIGIN_BOUND_RSA_CERT, "a", "b");
+  store.SetOriginBoundCert(
+      "https://www.google.com/", ORIGIN_BOUND_RSA_CERT, "c", "d");
+  store.SetOriginBoundCert(
+      "https://www.harvard.com/", ORIGIN_BOUND_RSA_CERT, "e", "f");
 
   EXPECT_EQ(3, store.GetCertCount());
   store.DeleteAll();
   EXPECT_EQ(0, store.GetCertCount());
 }
 
 TEST(DefaultOriginBoundCertStoreTest, TestDelete) {
   scoped_refptr<MockPersistentStore> persistent_store(new MockPersistentStore);
   DefaultOriginBoundCertStore store(persistent_store.get());
 
+  OriginBoundCertType type;
   std::string private_key, cert;
   EXPECT_EQ(0, store.GetCertCount());
-  store.SetOriginBoundCert("https://www.verisign.com/", "a", "b");
-  store.SetOriginBoundCert("https://www.google.com/", "c", "d");
+  store.SetOriginBoundCert(
+      "https://www.verisign.com/", ORIGIN_BOUND_RSA_CERT, "a", "b");
+  store.SetOriginBoundCert(
+      "https://www.google.com/", ORIGIN_BOUND_EC_CERT, "c", "d");
 
   EXPECT_EQ(2, store.GetCertCount());
   store.DeleteOriginBoundCert("https://www.verisign.com/");
   EXPECT_EQ(1, store.GetCertCount());
   EXPECT_FALSE(store.GetOriginBoundCert("https://www.verisign.com/",
-                                         &private_key,
-                                         &cert));
-  EXPECT_TRUE(store.GetOriginBoundCert("https://www.google.com/",
+                                        &type,
                                         &private_key,
                                         &cert));
+  EXPECT_TRUE(store.GetOriginBoundCert("https://www.google.com/",
+                                       &type,
+                                       &private_key,
+                                       &cert));
   store.DeleteOriginBoundCert("https://www.google.com/");
   EXPECT_EQ(0, store.GetCertCount());
   EXPECT_FALSE(store.GetOriginBoundCert("https://www.google.com/",
-                                         &private_key,
-                                         &cert));
+                                        &type,
+                                        &private_key,
+                                        &cert));
 }
 
 TEST(DefaultOriginBoundCertStoreTest, TestGetAll) {
   scoped_refptr<MockPersistentStore> persistent_store(new MockPersistentStore);
   DefaultOriginBoundCertStore store(persistent_store.get());
 
   EXPECT_EQ(0, store.GetCertCount());
-  store.SetOriginBoundCert("https://www.verisign.com/", "a", "b");
-  store.SetOriginBoundCert("https://www.google.com/", "c", "d");
-  store.SetOriginBoundCert("https://www.harvard.com/", "e", "f");
-  store.SetOriginBoundCert("https://www.mit.com/", "g", "h");
+  store.SetOriginBoundCert(
+      "https://www.verisign.com/", ORIGIN_BOUND_RSA_CERT, "a", "b");
+  store.SetOriginBoundCert(
+      "https://www.google.com/", ORIGIN_BOUND_EC_CERT, "c", "d");
+  store.SetOriginBoundCert(
+      "https://www.harvard.com/", ORIGIN_BOUND_RSA_CERT, "e", "f");
+  store.SetOriginBoundCert(
+      "https://www.mit.com/", ORIGIN_BOUND_RSA_CERT, "g", "h");
 
   EXPECT_EQ(4, store.GetCertCount());
   std::vector<OriginBoundCertStore::OriginBoundCertInfo> certs;
   store.GetAllOriginBoundCerts(&certs);
   EXPECT_EQ(4u, certs.size());
 }
 
 }  // namespace net