diff --git a/tests/dhtrunnertester.cpp b/tests/dhtrunnertester.cpp index 01b8a347b9e0dccf3e9d770d17111fe0c294aa4d..8d6d23f8b1cbff7fd16ff5abe52818bfc7a6be24 100644 --- a/tests/dhtrunnertester.cpp +++ b/tests/dhtrunnertester.cpp @@ -90,6 +90,31 @@ DhtRunnerTester::testGetPut() { CPPUNIT_ASSERT(vals.front()->data == val_data); } +void +DhtRunnerTester::testPutDuplicate() { + auto key = dht::InfoHash::get("123"); + auto val = std::make_shared<dht::Value>("hey"); + val->id = 42; + auto val_data = val->data; + std::promise<bool> p1; + std::promise<bool> p2; + node2.put(key, val, [&](bool ok){ + p1.set_value(ok); + }); + node2.put(key, val, [&](bool ok){ + p2.set_value(ok); + }); + auto p1ret = p1.get_future().get(); + auto p2ret = p2.get_future().get(); + CPPUNIT_ASSERT(p1ret); + CPPUNIT_ASSERT(p2ret); + auto vals = node1.get(key).get(); + CPPUNIT_ASSERT(not vals.empty()); + CPPUNIT_ASSERT(vals.size() == 1); + CPPUNIT_ASSERT(vals.front()->data == val_data); +} + + void DhtRunnerTester::testListen() { std::mutex mutex; diff --git a/tests/dhtrunnertester.h b/tests/dhtrunnertester.h index 3f5afd3e486654ce2032fc2fdc38eed1f75f3e3e..536ffe23e3e8698eec485fa088ca48547c62377f 100644 --- a/tests/dhtrunnertester.h +++ b/tests/dhtrunnertester.h @@ -31,6 +31,7 @@ class DhtRunnerTester : public CppUnit::TestFixture { CPPUNIT_TEST_SUITE(DhtRunnerTester); CPPUNIT_TEST(testConstructors); CPPUNIT_TEST(testGetPut); + CPPUNIT_TEST(testPutDuplicate); CPPUNIT_TEST(testListen); CPPUNIT_TEST(testListenLotOfBytes); CPPUNIT_TEST(testIdOps); @@ -55,6 +56,10 @@ class DhtRunnerTester : public CppUnit::TestFixture { * Test get and put methods */ void testGetPut(); + /** + * Test get and multiple put + */ + void testPutDuplicate(); /** * Test listen method */