| // Copyright (c) 2010 The Chromium OS 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 "login_manager/nss_util.h" |
| |
| #include <base/crypto/rsa_private_key.h> |
| #include <base/nss_util.h> |
| #include <base/scoped_ptr.h> |
| #include <gtest/gtest.h> |
| |
| namespace login_manager { |
| class NssUtilTest : public ::testing::Test { |
| public: |
| NssUtilTest() : util_(NssUtil::Create()) {} |
| virtual ~NssUtilTest() {} |
| |
| virtual void SetUp() { |
| ASSERT_TRUE(util_->OpenUserDB()); |
| } |
| |
| scoped_ptr<NssUtil> util_; |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(NssUtilTest); |
| }; |
| |
| TEST_F(NssUtilTest, FindFromPublicKey) { |
| // Create a keypair, which will put the keys in the user's NSSDB. |
| scoped_ptr<base::RSAPrivateKey> key_pair(base::RSAPrivateKey::Create(256)); |
| ASSERT_NE(key_pair.get(), reinterpret_cast<base::RSAPrivateKey*>(NULL)); |
| |
| std::vector<uint8> public_key; |
| ASSERT_TRUE(key_pair->ExportPublicKey(&public_key)); |
| |
| EXPECT_NE(util_->GetPrivateKey(public_key), |
| reinterpret_cast<base::RSAPrivateKey*>(NULL)); |
| } |
| |
| } // namespace login_manager |