Add functionality to OriginBoundCertStore interface and implementations.

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 91 matching lines @@
   EXPECT_TRUE(private_key.empty());
   EXPECT_TRUE(cert.empty());
   EXPECT_TRUE(store->SetOriginBoundCert("https://www.verisign.com/", "i", "j"));
   EXPECT_TRUE(store->GetOriginBoundCert("https://www.verisign.com/",
                                         &private_key,
                                         &cert));
   EXPECT_EQ("i", private_key);
   EXPECT_EQ("j", cert);
 }
 
+TEST(DefaultOriginBoundCertStoreTest, TestDuplicateCerts) {
+  scoped_refptr<MockPersistentStore> persistent_store(new MockPersistentStore);
+  scoped_ptr<DefaultOriginBoundCertStore> store(
+      new DefaultOriginBoundCertStore(persistent_store));

[wtc, 2011/08/10 21:35:59]

Consider using a local variable:
  DefaultOriginBoundCertStore store(persistent_store);

Can we do this?
  DefaultOriginBoundCertStore store(new MockPersistentStore);

I remember that's how you create the real origin-bound cert
stores in profile IO data, without using a scoped_refptr
for the persistent store.

[rkn, 2011/08/16 21:11:45]

I know this isn't exactly the change you were suggesting, but
DefaultOriginBoundCertStore doesn't delete it's SQLiteOriginBoundCertStore, so I
think it makes sense to store it in a scoped_refptr.

+
+  std::string private_key, cert;
+  EXPECT_EQ(0, store->GetCertCount());
+  EXPECT_TRUE(store->SetOriginBoundCert("https://www.verisign.com/", "a", "b"));
+  EXPECT_TRUE(store->SetOriginBoundCert("https://www.verisign.com/", "c", "d"));
+
+  EXPECT_EQ(1, store->GetCertCount());
+  EXPECT_TRUE(store->GetOriginBoundCert("https://www.verisign.com/",
+                                        &private_key,
+                                        &cert));
+  EXPECT_EQ("c", private_key);
+  EXPECT_EQ("d", cert);
+}
+
+TEST(DefaultOriginBoundCertStoreTest, TestDeleteAll) {
+  scoped_refptr<MockPersistentStore> persistent_store(new MockPersistentStore);
+  scoped_ptr<DefaultOriginBoundCertStore> store(
+      new DefaultOriginBoundCertStore(persistent_store));
+
+  EXPECT_EQ(0, store->GetCertCount());
+  EXPECT_TRUE(store->SetOriginBoundCert("https://www.verisign.com/", "a", "b"));
+  EXPECT_TRUE(store->SetOriginBoundCert("https://www.google.com/", "c", "d"));
+  EXPECT_TRUE(store->SetOriginBoundCert("https://www.harvard.com/", "e", "f"));
+
+  EXPECT_EQ(3, store->GetCertCount());
+  store->DeleteAll();
+  EXPECT_EQ(0, store->GetCertCount());
+}
+
+TEST(DefaultOriginBoundCertStoreTest, TestDelete) {
+  scoped_refptr<MockPersistentStore> persistent_store(new MockPersistentStore);
+  scoped_ptr<DefaultOriginBoundCertStore> store(
+      new DefaultOriginBoundCertStore(persistent_store));
+
+  std::string private_key, cert;
+  EXPECT_EQ(0, store->GetCertCount());
+  EXPECT_TRUE(store->SetOriginBoundCert("https://www.verisign.com/", "a", "b"));
+  EXPECT_TRUE(store->SetOriginBoundCert("https://www.google.com/", "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/",
+                                        &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));
+}
+
+TEST(DefaultOriginBoundCertStoreTest, TestGetAll) {
+  scoped_refptr<MockPersistentStore> persistent_store(new MockPersistentStore);
+  scoped_ptr<DefaultOriginBoundCertStore> store(
+      new DefaultOriginBoundCertStore(persistent_store));
+
+  EXPECT_EQ(0, store->GetCertCount());
+  EXPECT_TRUE(store->SetOriginBoundCert("https://www.verisign.com/", "a", "b"));
+  EXPECT_TRUE(store->SetOriginBoundCert("https://www.google.com/", "c", "d"));
+  EXPECT_TRUE(store->SetOriginBoundCert("https://www.harvard.com/", "e", "f"));
+  EXPECT_TRUE(store->SetOriginBoundCert("https://www.mit.com/", "g", "h"));
+
+  EXPECT_EQ(4, store->GetCertCount());
+  std::vector<OriginBoundCertStore::OriginBoundCertInfo> certs;
+  store->GetAllOriginBoundCerts(&certs);
+  EXPECT_EQ(4u, certs.size());
+}
+
 }  // namespace net