Use std::unique_ptr in DHTRegistry

1 files changed, 3 insertions, 3 deletions

diff --git a/test/DHTMessageTrackerTest.cc b/test/DHTMessageTrackerTest.cc
index 3c939f9d..1bf44cff 100644
--- a/test/DHTMessageTrackerTest.cc
+++ b/test/DHTMessageTrackerTest.cc
@@ -35,8 +35,8 @@ CPPUNIT_TEST_SUITE_REGISTRATION(DHTMessageTrackerTest);
 void DHTMessageTrackerTest::testMessageArrived()
 {
   auto localNode = std::make_shared<DHTNode>();
-  auto routingTable = std::make_shared<DHTRoutingTable>(localNode);
-  auto factory = std::make_shared<MockDHTMessageFactory>();
+  auto routingTable = make_unique<DHTRoutingTable>(localNode);
+  auto factory = make_unique<MockDHTMessageFactory>();
   factory->setLocalNode(localNode);
 
   auto r1 = std::make_shared<DHTNode>();
@@ -54,7 +54,7 @@ void DHTMessageTrackerTest::testMessageArrived()
   auto m3 = make_unique<MockDHTMessage>(localNode, r3);
 
   DHTMessageTracker tracker;
-  tracker.setRoutingTable(routingTable);
+  tracker.setRoutingTable(routingTable.get());
   tracker.setMessageFactory(factory.get());
   tracker.addMessage(m1.get(), DHT_MESSAGE_TIMEOUT);
   tracker.addMessage(m2.get(), DHT_MESSAGE_TIMEOUT);