Introduced new Filesystem API.

1 files changed, 5 insertions, 4 deletions

diff --git a/src/builder_test.cc b/src/builder_test.cc
index 0980585..4f077c4 100644
--- a/src/builder_test.cc
+++ b/src/builder_test.cc
@@ -14,6 +14,7 @@
 
 #include "builder.h"
 #include "common.h"
+#include "filesystem.h"
 #include "flags.h"
 #include "normalizer.h"
 #include "sentencepiece_trainer.h"
@@ -167,10 +168,10 @@ TEST(BuilderTest, LoadCharsMapWithEmptyeTest) {
   test::ScopedTempFile test_tsv("test.tsv");
   test::ScopedTempFile test_out_tsv("test_out.tsv");
   {
-    io::OutputBuffer output(test_tsv.filename());
-    output.WriteLine("0061\t0041");
-    output.WriteLine("0062");
-    output.WriteLine("0063\t\t#foo=>bar");
+    auto output = filesystem::NewWritableFile(test_tsv.filename());
+    output->WriteLine("0061\t0041");
+    output->WriteLine("0062");
+    output->WriteLine("0063\t\t#foo=>bar");
   }
 
   Builder::CharsMap chars_map;