Enable Chrome OS to load the user's nssdb later.

net/base/keygen_handler_unittest.cc

 // Copyright (c) 2010 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/keygen_handler.h"
 
 #include <string>
 
 #include "base/base64.h"
 #include "base/logging.h"
+#include "base/nss_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace net {
 
 namespace {
 
 KeygenHandler::KeyLocation ValidKeyLocation() {
   KeygenHandler::KeyLocation result;
 #if defined(OS_WIN)
   result.container_name = L"Unit tests";
   result.provider_name = L"Test Provider";
 #elif defined(OS_MACOSX)
   result.keychain_path = "/Users/tests/test.chain";
 #elif defined(USE_NSS)
   result.slot_name = "Sample slot";
 #endif
 
   return result;
 }
 
-TEST(KeygenHandlerTest, FLAKY_SmokeTest) {
+class KeygenHandlerTest : public ::testing::Test {
+ public:
+  KeygenHandlerTest() {}
+  virtual ~KeygenHandlerTest() {}
+
+  virtual void SetUp() {
+#if defined(OS_CHROMEOS)
+  base::OpenPersistentNSSDB();
+#endif
+  }
+};
+
+TEST_F(KeygenHandlerTest, FLAKY_SmokeTest) {
   KeygenHandler handler(2048, "some challenge");
   handler.set_stores_key(false);  // Don't leave the key-pair behind
   std::string result = handler.GenKeyAndSignChallenge();
   LOG(INFO) << "KeygenHandler produced: " << result;
   ASSERT_GT(result.length(), 0U);
 
   // Verify it's valid base64:
   std::string spkac;
   ASSERT_TRUE(base::Base64Decode(result, &spkac));
   // In lieu of actually parsing and validating the DER data,
@@ skipping 16 matching lines @@
   //     Exponent: 65537 (0x10001)
   //     Challenge String: some challenge
   //     Signature Algorithm: md5WithRSAEncryption
   //         92:f3:cc:ff:0b:d3:d0:4a:3a:4c:ba:ff:d6:38:7f:a5:4b:b5: .....
   //   Signature OK
   //
   // The value of |spkac| can be ASN.1-parsed with:
   //    openssl asn1parse -inform DER
 }
 
-TEST(KeygenHandlerTest, Cache) {
+TEST_F(KeygenHandlerTest, Cache) {
   KeygenHandler::Cache* cache = KeygenHandler::Cache::GetInstance();
   KeygenHandler::KeyLocation location1;
   KeygenHandler::KeyLocation location2;
 
   std::string key1("abcd");
   cache->Insert(key1, location1);
 
   // The cache should have stored location1 at key1.
   EXPECT_TRUE(cache->Find(key1, &location2));
 
@@ skipping 10 matching lines @@
   std::string key2("def");
   EXPECT_FALSE(cache->Find(key2, &location2));
 
   // A cache miss should leave the original location unmolested.
   EXPECT_TRUE(location2.Equals(location3));
 }
 
 }  // namespace
 
 }  // namespace net