tests/core/identity-fixture.hpp

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/**
 * Copyright (c) 2014,  Regents of the University of California
 *
 * This file is part of NSL (NDN Signature Logger).
 * See AUTHORS.md for complete list of NSL authors and contributors.
 *
 * NSL is free software: you can redistribute it and/or modify it under the terms
 * of the GNU General Public License as published by the Free Software Foundation,
 * either version 3 of the License, or (at your option) any later version.
 *
 * NSL is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
 * PURPOSE.  See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * NSL, e.g., in COPYING.md file.  If not, see <http://www.gnu.org/licenses/>.
 *
 * See AUTHORS.md for complete list of nsl authors and contributors.
 */

#ifndef NSL_TESTS_IDENTITY_FIXTURE_HPP
#define NSL_TESTS_IDENTITY_FIXTURE_HPP

#include "unit-test-time-fixture.hpp"
#include <ndn-cxx/security/key-chain.hpp>
#include <vector>

#include <boost/filesystem.hpp>

namespace nsl {
namespace tests {

class IdentityFixture : public UnitTestTimeFixture
{
public:
  IdentityFixture()
    : m_keyChainTmpPath(boost::filesystem::path(TEST_KEYCHAIN_PATH) / "IdentityFixture")
    , m_keyChain(std::string("pib-sqlite3:").append(m_keyChainTmpPath.string()),
                 std::string("tpm-file:").append(m_keyChainTmpPath.string()))
  {
  }

  ~IdentityFixture()
  {
    for (const auto& identity : m_identities) {
      m_keyChain.deleteIdentity(identity);
    }

    boost::filesystem::remove_all(m_keyChainTmpPath);
  }
  /// @brief add identity, return true if succeed.
  bool
  addIdentity(const Name& identity,
              const ndn::KeyParams& params = ndn::KeyChain::DEFAULT_KEY_PARAMS)
  {
    try {
      m_keyChain.createIdentity(identity, params);
      m_identities.push_back(identity);
      return true;
    }
    catch (std::runtime_error&) {
      return false;
    }
  }

protected:
  boost::filesystem::path m_keyChainTmpPath;
  ndn::KeyChain m_keyChain;
  std::vector<Name> m_identities;
};

} // namespace tests
} // namespace nsl

#endif // NSL_TESTS_IDENTITY_FIXTURE_HPP