AbstractSingleDiskAdaptor: Use std::unique_ptr for diskWriter_

1 files changed, 5 insertions, 4 deletions

diff --git a/test/PieceTest.cc b/test/PieceTest.cc
index 6f9b9cf3..7037548d 100644
--- a/test/PieceTest.cc
+++ b/test/PieceTest.cc
@@ -28,13 +28,14 @@ class PieceTest:public CppUnit::TestFixture {
   CPPUNIT_TEST_SUITE_END();
 private:
   std::shared_ptr<DirectDiskAdaptor> adaptor_;
-  std::shared_ptr<ByteArrayDiskWriter> writer_;
+  ByteArrayDiskWriter* writer_;
 public:
   void setUp()
   {
-    adaptor_.reset(new DirectDiskAdaptor());
-    writer_.reset(new ByteArrayDiskWriter());
-    adaptor_->setDiskWriter(writer_);
+    adaptor_ = std::make_shared<DirectDiskAdaptor>();
+    auto dw = make_unique<ByteArrayDiskWriter>();
+    writer_ = dw.get();
+    adaptor_->setDiskWriter(std::move(dw));
   }
 
   void testCompleteBlock();