Make PackagesWriter streaming - you don't need to give count when creating it

1 files changed, 3 insertions, 3 deletions

diff --git a/test/test-packages.cc b/test/test-packages.cc
index d0c526c..fe56077 100644
--- a/test/test-packages.cc
+++ b/test/test-packages.cc
@@ -77,9 +77,9 @@ bool test_sanity() {
 
   std::stringstream stream;
   auto writer = std::unique_ptr<PackagesWriter>(
-      PackagesWriter::create(1, &stream));
+      PackagesWriter::create(&stream));
   writer->write(pkg1, "Hello World!");
-  writer.reset();
+  writer->flush();
 
   Delegate delegate;
   ASSERT_EQ(PackagesReader::GOOD, PackagesReader::read(stream, &delegate));
@@ -93,7 +93,7 @@ bool test_sanity() {
 
 bool test_empty() {
   std::stringstream stream;
-  delete PackagesWriter::create(0, &stream);
+  delete PackagesWriter::create(&stream);
   ASSERT_TRUE(!stream.str().empty());
   Delegate delegate;
   ASSERT_EQ(PackagesReader::GOOD, PackagesReader::read(stream, &delegate));