Alexander Afanasyev | fa2f662 | 2016-12-25 12:28:00 -0800 | [diff] [blame] | 1 | /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */ |
| 2 | /** |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 3 | * Copyright (c) 2013-2017, Regents of the University of California. |
Alexander Afanasyev | fa2f662 | 2016-12-25 12:28:00 -0800 | [diff] [blame] | 4 | * |
| 5 | * This file is part of ChronoShare, a decentralized file sharing application over NDN. |
| 6 | * |
| 7 | * ChronoShare is free software: you can redistribute it and/or modify it under the terms |
| 8 | * of the GNU General Public License as published by the Free Software Foundation, either |
| 9 | * version 3 of the License, or (at your option) any later version. |
| 10 | * |
| 11 | * ChronoShare 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 General Public License for more details. |
| 14 | * |
| 15 | * You should have received copies of the GNU General Public License along with |
| 16 | * ChronoShare, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>. |
| 17 | * |
| 18 | * See AUTHORS.md for complete list of ChronoShare authors and contributors. |
| 19 | */ |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 20 | |
Alexander Afanasyev | eda3b7a | 2016-12-25 11:26:40 -0800 | [diff] [blame] | 21 | #include "sync-core.hpp" |
Alexander Afanasyev | abe952a | 2013-01-17 17:06:32 -0800 | [diff] [blame] | 22 | |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 23 | #include "test-common.hpp" |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 24 | |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 25 | namespace ndn { |
| 26 | namespace chronoshare { |
| 27 | namespace tests { |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 28 | |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 29 | namespace fs = boost::filesystem; |
Alexander Afanasyev | c507ac2 | 2013-01-21 16:01:58 -0800 | [diff] [blame] | 30 | |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 31 | INIT_LOGGER("Test.SyncCore") |
| 32 | |
| 33 | class TestSyncCoreFixture : public IdentityManagementTimeFixture |
| 34 | { |
| 35 | public: |
| 36 | void |
| 37 | advanceClocks() |
| 38 | { |
| 39 | for (int i = 0; i < 100; ++i) { |
| 40 | usleep(10000); |
| 41 | IdentityManagementTimeFixture::advanceClocks(time::milliseconds(10), 1); |
| 42 | } |
| 43 | } |
| 44 | }; |
| 45 | |
| 46 | BOOST_FIXTURE_TEST_SUITE(TestSyncCore, TestSyncCoreFixture) |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 47 | |
Alexander Afanasyev | eda3b7a | 2016-12-25 11:26:40 -0800 | [diff] [blame] | 48 | void |
| 49 | callback(const SyncStateMsgPtr& msg) |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 50 | { |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 51 | _LOG_DEBUG("Callback I'm called!!!!"); |
Zhenkai Zhu | 9501b8b | 2013-01-17 12:37:00 -0800 | [diff] [blame] | 52 | BOOST_CHECK(msg->state_size() > 0); |
Zhenkai Zhu | 085aae7 | 2013-01-17 21:09:01 -0800 | [diff] [blame] | 53 | int size = msg->state_size(); |
| 54 | int index = 0; |
Alexander Afanasyev | eda3b7a | 2016-12-25 11:26:40 -0800 | [diff] [blame] | 55 | while (index < size) { |
Zhenkai Zhu | 085aae7 | 2013-01-17 21:09:01 -0800 | [diff] [blame] | 56 | SyncState state = msg->state(index); |
| 57 | BOOST_CHECK(state.has_old_seq()); |
Alexander Afanasyev | eda3b7a | 2016-12-25 11:26:40 -0800 | [diff] [blame] | 58 | if (state.seq() != 0) { |
Alexander Afanasyev | dac8492 | 2013-01-20 23:32:17 -0800 | [diff] [blame] | 59 | BOOST_CHECK(state.old_seq() != state.seq()); |
Zhenkai Zhu | 085aae7 | 2013-01-17 21:09:01 -0800 | [diff] [blame] | 60 | } |
| 61 | index++; |
| 62 | } |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 63 | } |
| 64 | |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 65 | BOOST_AUTO_TEST_CASE(TwoNodes) |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 66 | { |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 67 | fs::path tmpdir = fs::unique_path(UNIT_TEST_CONFIG_PATH) / "SyncCoreTest"; |
| 68 | if (exists(tmpdir)) { |
| 69 | remove_all(tmpdir); |
Zhenkai Zhu | 4e1c1d9 | 2013-01-17 14:12:46 -0800 | [diff] [blame] | 70 | } |
| 71 | |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 72 | std::string dir1 = (tmpdir / "1").string(); |
| 73 | std::string dir2 = (tmpdir / "2").string(); |
| 74 | Name user1("/shuai"); |
| 75 | Name loc1("/locator1"); |
| 76 | Name user2("/loli"); |
| 77 | Name loc2("/locator2"); |
| 78 | Name syncPrefix("/broadcast/arslan"); |
Zhenkai Zhu | b330aed | 2013-01-17 13:29:37 -0800 | [diff] [blame] | 79 | |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 80 | shared_ptr<Face> c1 = make_shared<Face>(this->m_io); |
| 81 | auto log1 = make_shared<SyncLog>(dir1, user1); |
| 82 | auto core1 = make_shared<SyncCore>(*c1, log1, user1, loc1, syncPrefix, bind(&callback, _1)); |
Alexander Afanasyev | fc72036 | 2013-01-24 21:49:48 -0800 | [diff] [blame] | 83 | |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 84 | shared_ptr<Face> c2 = make_shared<Face>(this->m_io); |
| 85 | auto log2 = make_shared<SyncLog>(dir2, user2); |
| 86 | auto core2 = make_shared<SyncCore>(*c2, log2, user2, loc2, syncPrefix, bind(&callback, _1)); |
Zhenkai Zhu | 6e7d4d2 | 2013-01-15 18:18:18 -0800 | [diff] [blame] | 87 | |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 88 | // @TODO: Implement test using the dummy forwarder and disable dependency on real time |
| 89 | |
| 90 | this->advanceClocks(); |
| 91 | |
| 92 | BOOST_CHECK_EQUAL(toHex(*core1->root()), toHex(*core2->root())); |
| 93 | |
| 94 | // _LOG_TRACE("\n\n\n\n\n\n----------\n"); |
Alexander Afanasyev | c507ac2 | 2013-01-21 16:01:58 -0800 | [diff] [blame] | 95 | |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 96 | core1->updateLocalState(1); |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 97 | this->advanceClocks(); |
| 98 | BOOST_CHECK_EQUAL(toHex(*core1->root()), toHex(*core2->root())); |
Zhenkai Zhu | 9501b8b | 2013-01-17 12:37:00 -0800 | [diff] [blame] | 99 | BOOST_CHECK_EQUAL(core2->seq(user1), 1); |
Alexander Afanasyev | eda3b7a | 2016-12-25 11:26:40 -0800 | [diff] [blame] | 100 | BOOST_CHECK_EQUAL(log2->LookupLocator(user1), loc1); |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 101 | |
| 102 | core1->updateLocalState(5); |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 103 | this->advanceClocks(); |
| 104 | BOOST_CHECK_EQUAL(toHex(*core1->root()), toHex(*core2->root())); |
Zhenkai Zhu | 9501b8b | 2013-01-17 12:37:00 -0800 | [diff] [blame] | 105 | BOOST_CHECK_EQUAL(core2->seq(user1), 5); |
Alexander Afanasyev | eda3b7a | 2016-12-25 11:26:40 -0800 | [diff] [blame] | 106 | BOOST_CHECK_EQUAL(log2->LookupLocator(user1), loc1); |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 107 | |
| 108 | core2->updateLocalState(10); |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 109 | this->advanceClocks(); |
| 110 | BOOST_CHECK_EQUAL(toHex(*core1->root()), toHex(*core2->root())); |
Zhenkai Zhu | 9501b8b | 2013-01-17 12:37:00 -0800 | [diff] [blame] | 111 | BOOST_CHECK_EQUAL(core1->seq(user2), 10); |
Alexander Afanasyev | eda3b7a | 2016-12-25 11:26:40 -0800 | [diff] [blame] | 112 | BOOST_CHECK_EQUAL(log1->LookupLocator(user2), loc2); |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 113 | |
| 114 | // simple simultaneous data generation |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 115 | _LOG_TRACE("\n\n\n\n\n\n----------Simultaneous\n"); |
Alexander Afanasyev | c507ac2 | 2013-01-21 16:01:58 -0800 | [diff] [blame] | 116 | |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 117 | core1->updateLocalState(11); |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 118 | this->advanceClocks(); |
Zhenkai Zhu | 9501b8b | 2013-01-17 12:37:00 -0800 | [diff] [blame] | 119 | core2->updateLocalState(15); |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 120 | this->advanceClocks(); |
| 121 | BOOST_CHECK_EQUAL(toHex(*core1->root()), toHex(*core2->root())); |
Zhenkai Zhu | 9501b8b | 2013-01-17 12:37:00 -0800 | [diff] [blame] | 122 | BOOST_CHECK_EQUAL(core1->seq(user2), 15); |
| 123 | BOOST_CHECK_EQUAL(core2->seq(user1), 11); |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 124 | |
Alexander Afanasyev | eda3b7a | 2016-12-25 11:26:40 -0800 | [diff] [blame] | 125 | BOOST_CHECK_EQUAL(log1->LookupLocator(user1), loc1); |
| 126 | BOOST_CHECK_EQUAL(log1->LookupLocator(user2), loc2); |
| 127 | BOOST_CHECK_EQUAL(log2->LookupLocator(user1), loc1); |
| 128 | BOOST_CHECK_EQUAL(log2->LookupLocator(user2), loc2); |
Zhenkai Zhu | e29616f | 2013-01-14 15:40:57 -0800 | [diff] [blame] | 129 | } |
| 130 | |
| 131 | BOOST_AUTO_TEST_SUITE_END() |
Yukai Tu | ccdcdba | 2016-10-24 13:48:01 -0700 | [diff] [blame^] | 132 | |
| 133 | } // namespace tests |
| 134 | } // namespace chronoshare |
| 135 | } // namespace ndn |