blob: 016ad90c0e0c3febd74bf4d2f733dba3c50ab39b [file] [log] [blame]
Yingdi Yub8f8b342015-04-27 11:06:42 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/**
3 * Copyright (c) 2013-2015 Regents of the University of California.
4 *
5 * This file is part of ndn-cxx library (NDN C++ library with eXperimental eXtensions).
6 *
7 * ndn-cxx library is free software: you can redistribute it and/or modify it under the
8 * terms of the GNU Lesser General Public License as published by the Free Software
9 * Foundation, either version 3 of the License, or (at your option) any later version.
10 *
11 * ndn-cxx library is distributed in the hope that it will be useful, but WITHOUT ANY
12 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
13 * PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
14 *
15 * You should have received copies of the GNU General Public License and GNU Lesser
16 * General Public License along with ndn-cxx, e.g., in COPYING.md file. If not, see
17 * <http://www.gnu.org/licenses/>.
18 *
19 * See AUTHORS.md for complete list of ndn-cxx authors and contributors.
20 */
21
22#include "security/key-container.hpp"
23#include "security/pib.hpp"
Yingdi Yu3bf91f52015-06-12 19:39:40 -070024#include "security/pib-memory.hpp"
Yingdi Yub8f8b342015-04-27 11:06:42 -070025#include "pib-data-fixture.hpp"
26
27#include "boost-test.hpp"
28
29namespace ndn {
30namespace security {
31namespace tests {
32
33BOOST_AUTO_TEST_SUITE(SecurityKeyContainer)
34
35BOOST_FIXTURE_TEST_CASE(TestKeyContainer, PibDataFixture)
36{
Yingdi Yu3bf91f52015-06-12 19:39:40 -070037 auto pibImpl = make_shared<PibMemory>();
Yingdi Yub8f8b342015-04-27 11:06:42 -070038 Pib pib("pib-memory", "", pibImpl);
39
40 Identity identity1 = pib.addIdentity(id1);
41
42 Key key11 = identity1.addKey(id1Key1, id1Key1Name.get(-1));
43 Key key12 = identity1.addKey(id1Key2, id1Key2Name.get(-1));
44
45 KeyContainer container = identity1.getKeys();
46 BOOST_CHECK_EQUAL(container.size(), 2);
47 BOOST_CHECK(container.find(id1Key1Name.get(-1)) != container.end());
48 BOOST_CHECK(container.find(id1Key2Name.get(-1)) != container.end());
49
50 std::set<name::Component> keyNames;
51 keyNames.insert(id1Key1Name.get(-1));
52 keyNames.insert(id1Key2Name.get(-1));
53
54 KeyContainer::const_iterator it = container.begin();
55 std::set<name::Component>::const_iterator testIt = keyNames.begin();
56 BOOST_CHECK_EQUAL((*it).getKeyId(), *testIt);
57 it++;
58 testIt++;
59 BOOST_CHECK_EQUAL((*it).getKeyId(), *testIt);
60 ++it;
61 testIt++;
62 BOOST_CHECK(it == container.end());
63
64 size_t count = 0;
65 testIt = keyNames.begin();
66 for (const auto& key : container) {
67 BOOST_CHECK_EQUAL(key.getIdentity(), id1);
68 BOOST_CHECK_EQUAL(key.getKeyId(), *testIt);
69 testIt++;
70 count++;
71 }
72 BOOST_CHECK_EQUAL(count, 2);
73}
74
75BOOST_AUTO_TEST_SUITE_END()
76
77} // namespace tests
78} // namespace security
79} // namespace ndn