fix: Fix text writer abort test

facebookincubator · Jan 2, 2025 · 9e97b92 · 9e97b92
1 parent 98932e5
commit 9e97b92
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/velox/dwio/text/tests/writer/TextWriterTest.cpp b/velox/dwio/text/tests/writer/TextWriterTest.cpp
@@ -167,6 +167,12 @@ TEST_F(TextWriterTest, abort) {
   writer->abort();
 
   std::string result = readFile(filePath);
-  EXPECT_EQ(result.size(), 8);
+  // With defaultFlushCount as 10, it will trigger two times of flushes before
+  // abort, and abort will discard the remaining 5 characters in buffer. The
+  // written file would have:
+  // 1^Atrue
+  // 2^Atrue
+  // 3^A
+  EXPECT_EQ(result.size(), 14);
 }
 } // namespace facebook::velox::text